From 7bba5237dc34d0aa8b6046423cd36ef62ee46ea2 Mon Sep 17 00:00:00 2001 From: Kuroshini Date: Wed, 16 Oct 2019 14:11:07 +0300 Subject: [PATCH] Closed_at fill command --- .../management/commands/add_closed_at_timetable.py | 14 ++++++++++++++ apps/timetable/transfer.py | 1 + 2 files changed, 15 insertions(+) create mode 100644 apps/establishment/management/commands/add_closed_at_timetable.py diff --git a/apps/establishment/management/commands/add_closed_at_timetable.py b/apps/establishment/management/commands/add_closed_at_timetable.py new file mode 100644 index 00000000..d7b80d61 --- /dev/null +++ b/apps/establishment/management/commands/add_closed_at_timetable.py @@ -0,0 +1,14 @@ +from django.core.management.base import BaseCommand +from timetable.models import Timetable + +class Command(BaseCommand): + help = '''Add closed_at, opening_at Timetable fields''' + + def handle(self, *args, **options): + for tt in Timetable.objects.all(): + end = tt.dinner_end or tt.lunch_end + start = tt.lunch_start or tt.dinner_start + if end or start: + tt.closed_at = end + tt.opening_at = start + tt.save() \ No newline at end of file diff --git a/apps/timetable/transfer.py b/apps/timetable/transfer.py index 915b9ad5..2b8fa45f 100644 --- a/apps/timetable/transfer.py +++ b/apps/timetable/transfer.py @@ -66,6 +66,7 @@ card = { "fields": { "Schedules": { # нет аналогов для weekday, opening_at, closed_at + # upd: запустить команду add_closed_at_timetable. она заполнит opening_at, closed_at "lunch_start": "lunch_start", "lunch_end": "lunch_end", "diner_start": "diner_start",