loko/loko/home/migrations/0006_create_exceptional_transport_widget.py

44 lines
1.4 KiB
Python

from django.db import migrations
def create_exceptional_transport_widget(apps, schema_editor):
HomeWidget = apps.get_model('home', 'HomeWidget')
HomeWidgetConditionGroup = apps.get_model('home', 'HomeWidgetConditionGroup')
HomeWidgetCondition = apps.get_model('home', 'HomeWidgetCondition')
widget, created = HomeWidget.objects.get_or_create(
widget_code='exceptional_transport',
defaults={'is_active': True}
)
if not widget.condition_groups.exists():
group = HomeWidgetConditionGroup.objects.create(
widget=widget,
label="Profil Transport Exceptionnel",
order=25,
col_width=12,
priority=0,
is_active=True
)
HomeWidgetCondition.objects.create(
group=group,
condition_type='exceptional_transport',
is_active=True,
is_negated=False
)
def remove_exceptional_transport_widget(apps, schema_editor):
HomeWidget = apps.get_model('home', 'HomeWidget')
HomeWidget.objects.filter(widget_code='exceptional_transport').delete()
class Migration(migrations.Migration):
dependencies = [
('home', '0005_alter_homewidget_widget_code_and_more'),
]
operations = [
migrations.RunPython(create_exceptional_transport_widget, remove_exceptional_transport_widget),
]