nk20/apps/member/migrations/0003_create_bde_and_kfet.py

from django.db import migrations


def create_bde_and_kfet(apps, schema_editor):
    """
    The clubs BDE and Kfet are pre-injected.
    """
    Club = apps.get_model("member", "club")
    NoteClub = apps.get_model("note", "noteclub")
    Alias = apps.get_model("note", "alias")
    ContentType = apps.get_model('contenttypes', 'ContentType')
    polymorphic_ctype_id = ContentType.objects.get_for_model(NoteClub).id

    Club.objects.get_or_create(
        id=1,
        name="BDE",
        email="tresorerie.bde@example.com",
        require_memberships=True,
        membership_fee_paid=500,
        membership_fee_unpaid=500,
        membership_duration=396,
        membership_start="2020-08-01",
        membership_end="2021-09-30",
    )
    Club.objects.get_or_create(
        id=2,
        name="Kfet",
        parent_club_id=1,
        email="tresorerie.bde@example.com",
        require_memberships=True,
        membership_fee_paid=3500,
        membership_fee_unpaid=3500,
        membership_duration=396,
        membership_start="2020-08-01",
        membership_end="2021-09-30",
    )

    NoteClub.objects.get_or_create(
        id=5,
        club_id=1,
        polymorphic_ctype_id=polymorphic_ctype_id,
    )
    NoteClub.objects.get_or_create(
        id=6,
        club_id=2,
        polymorphic_ctype_id=polymorphic_ctype_id,
    )

    Alias.objects.get_or_create(
        id=5,
        note_id=5,
        name="BDE",
        normalized_name="bde",
    )
    Alias.objects.get_or_create(
        id=6,
        note_id=6,
        name="Kfet",
        normalized_name="kfet",
    )


class Migration(migrations.Migration):
    dependencies = [
        ('member', '0002_auto_20200904_2341'),
        ('note', '0002_create_special_notes'),
    ]

    operations = [
        migrations.RunPython(create_bde_and_kfet),
    ]