mirror of
https://github.com/ae-utbm/sith.git
synced 2024-11-18 04:03:22 +00:00
41 lines
1.3 KiB
Python
41 lines
1.3 KiB
Python
# -*- coding: utf-8 -*-
|
|
from __future__ import unicode_literals
|
|
|
|
from django.db import migrations, models
|
|
|
|
from club.models import Club
|
|
from core.operations import PsqlRunOnly
|
|
|
|
|
|
def generate_club_pages(apps, schema_editor):
|
|
def recursive_generate_club_page(club):
|
|
club.make_page()
|
|
for child in Club.objects.filter(parent=club).all():
|
|
recursive_generate_club_page(child)
|
|
for club in Club.objects.filter(parent=None).all():
|
|
recursive_generate_club_page(club)
|
|
|
|
|
|
class Migration(migrations.Migration):
|
|
|
|
dependencies = [
|
|
('core', '0024_auto_20170906_1317'),
|
|
('club', '0009_auto_20170822_2232'),
|
|
]
|
|
|
|
operations = [
|
|
migrations.AddField(
|
|
model_name='club',
|
|
name='is_active',
|
|
field=models.BooleanField(default=True, verbose_name='is active'),
|
|
),
|
|
migrations.AddField(
|
|
model_name='club',
|
|
name='page',
|
|
field=models.OneToOneField(related_name='club', blank=True, null=True, to='core.Page'),
|
|
),
|
|
PsqlRunOnly('SET CONSTRAINTS ALL IMMEDIATE', reverse_sql=migrations.RunSQL.noop),
|
|
migrations.RunPython(generate_club_pages),
|
|
PsqlRunOnly(migrations.RunSQL.noop, reverse_sql='SET CONSTRAINTS ALL IMMEDIATE'),
|
|
]
|