31 lines
880 B
Python
31 lines
880 B
Python
# Generated by Django 3.0.7 on 2020-06-11 13:09
|
|
|
|
from django.db import migrations
|
|
from django.db.models import Q
|
|
|
|
|
|
def migrate_meal_types(apps, schema_editor):
|
|
MealPlan = apps.get_model('cookbook', 'MealPlan')
|
|
MealType = apps.get_model('cookbook', 'MealType')
|
|
User = apps.get_model('auth', 'User')
|
|
|
|
for u in User.objects.all():
|
|
for t in MealType.objects.filter(created_by=None).all():
|
|
user_type = MealType.objects.create(
|
|
name=t.name,
|
|
created_by=u,
|
|
)
|
|
MealPlan.objects.filter(Q(created_by=u) and Q(meal_type=t)).update(meal_type=user_type)
|
|
|
|
MealType.objects.filter(created_by=None).delete()
|
|
|
|
|
|
class Migration(migrations.Migration):
|
|
dependencies = [
|
|
('cookbook', '0049_mealtype_created_by'),
|
|
]
|
|
|
|
operations = [
|
|
migrations.RunPython(migrate_meal_types),
|
|
]
|