transfer formulas menu

apps/establishment/management/commands/add_menus.py

@@ -1,4 +1,5 @@
 from django.core.management.base import BaseCommand
+from tqdm import tqdm
 
 from establishment.models import Establishment, Menu, Plate
 from transfer.models import Menus
@@ -10,14 +11,17 @@ class Command(BaseCommand):
     def handle(self, *args, **kwargs):
         count = 0
         menus = Menus.objects.filter(name__isnull=False).exclude(name='')
-        for old_menu in menus:
+        for old_menu in tqdm(menus, desc='Add formulas menu'):
             est = Establishment.objects.filter(
                 old_id=old_menu.establishment_id).first()
             if est:
 
                 menu, _ = Menu.objects.get_or_create(
                     category={'en-GB': 'formulas'},
-                    establishment=est
+                    establishment=est,
+                    old_id=old_menu.id,
+                    is_drinks_included=True if old_menu.drinks == 'included' else False,
+                    created=old_menu.created_at,
                 )
                 plate, created = Plate.objects.get_or_create(
                     name={"en-GB": old_menu.name},

make_data_migration.sh

@@ -37,4 +37,8 @@
 ./manage.py add_employee
 ./manage.py add_position
 ./manage.py add_empl_position
-./manage.py update_employee
+./manage.py update_employee
+
+# меню из Dishes(dessert, main_course, starter) и Menus(formulas)
+./manage.py transfer --menu
+./manage.py add_menus