transfer formulas menu

This commit is contained in:
alex 2020-01-24 09:20:22 +03:00
parent 9b6918b3b0
commit 792c11e128
2 changed files with 11 additions and 3 deletions

View File

@ -1,4 +1,5 @@
from django.core.management.base import BaseCommand from django.core.management.base import BaseCommand
from tqdm import tqdm
from establishment.models import Establishment, Menu, Plate from establishment.models import Establishment, Menu, Plate
from transfer.models import Menus from transfer.models import Menus
@ -10,14 +11,17 @@ class Command(BaseCommand):
def handle(self, *args, **kwargs): def handle(self, *args, **kwargs):
count = 0 count = 0
menus = Menus.objects.filter(name__isnull=False).exclude(name='') 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( est = Establishment.objects.filter(
old_id=old_menu.establishment_id).first() old_id=old_menu.establishment_id).first()
if est: if est:
menu, _ = Menu.objects.get_or_create( menu, _ = Menu.objects.get_or_create(
category={'en-GB': 'formulas'}, 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( plate, created = Plate.objects.get_or_create(
name={"en-GB": old_menu.name}, name={"en-GB": old_menu.name},

View File

@ -37,4 +37,8 @@
./manage.py add_employee ./manage.py add_employee
./manage.py add_position ./manage.py add_position
./manage.py add_empl_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