From 792c11e1286adfefdab58905f893dd88283f524d Mon Sep 17 00:00:00 2001 From: alex Date: Fri, 24 Jan 2020 09:20:22 +0300 Subject: [PATCH] transfer formulas menu --- apps/establishment/management/commands/add_menus.py | 8 ++++++-- make_data_migration.sh | 6 +++++- 2 files changed, 11 insertions(+), 3 deletions(-) diff --git a/apps/establishment/management/commands/add_menus.py b/apps/establishment/management/commands/add_menus.py index 3875b8ca..8692094b 100644 --- a/apps/establishment/management/commands/add_menus.py +++ b/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}, diff --git a/make_data_migration.sh b/make_data_migration.sh index d5edc793..6de3a713 100755 --- a/make_data_migration.sh +++ b/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 \ No newline at end of file +./manage.py update_employee + +# меню из Dishes(dessert, main_course, starter) и Menus(formulas) +./manage.py transfer --menu +./manage.py add_menus \ No newline at end of file