Создание объектов на основе даты и времени

У меня есть модель Event, и каждое событие будет иметь разные шоу.

class Event(models.Model):
    title = models.CharField(max_length=200)

class Show(models.Model):
    event = models.ForeignKey(Event, on_delete=models.CASCADE)
    date_time = models.DateTimeField(unique=True)

У меня есть еще одна модель билетов. Каждый билет должен быть уникальным. Значение каждого билета будет уникальным и относится к шоу и сидению.

class Ticket(models.Model):
    show = models.ForeignKey(Show)
    seat = models.ForeignKey(Seat)

    class Meta:
        unique_together = ('show', 'seat')

Мне нужно создавать шоу, основанные на дате начала и дате окончания, предоставляемой пользователем. Предположим, что это сообщение JSON:

{
    "event_id": 1,
    "start_date": "2018-02-16",
    "end_date": "2018-02-20",
    "time_list": ["11:00 AM", "8:00 PM"]
}

Из приведенного выше примера JSON мне нужно создать Показать запуск следующим образом:

# Start with the start_date as the date, and for each time from the time_list
Show.objects.create(
    event = 1,
    date_time = datetime.strptime('2018-02-16 11:00 AM', "%Y-%m-%d %I:%M %p")
)
Show.objects.create(
    event = 1,
    date_time = datetime.strptime('2018-02-16 8:00 PM', "%Y-%m-%d %I:%M %p")
)
# Next date after the start_date, i.e., 16+1 = 17
Show.objects.create(
    event = 1,
    date_time = datetime.strptime('2018-02-17 8:00 PM', "%Y-%m-%d %I:%M %p")
)
.
.
.
# Create Show objects till the end_date and for each time from the time_list
Show.objects.create(
    event = 1,
    date_time = datetime.strptime('2018-02-20 8:00 PM', "%Y-%m-%d %I:%M %p")
)

Прямо сейчас я создаю объекты Show:

def create_show_by_datetime(self, request):
    event_id = request.data['event_id']
    try:
        event = Event.objects.get(id=event_id)
    except Event.DoesNotExist:
        return Response(
            {'error': 'event with id: %s does not exist.' % event_id},
            status=status.HTTP_400_BAD_REQUEST
        )

    start_date = request.data['start_date']
    end_date = request.data['end_date']
    time_list = request.data['time_list']
    date_format = '%Y-%m-%d'
    time_format = "%I:%M %p"
    try:
        datetime.strptime(start_date, date_format)
        datetime.strptime(end_date, date_format)
        for i in range(len(time_list)):
            time = datetime.strptime(time_list[i], time_format)
    except ValueError as e:
        return Response(
            {'error': 'Time was not in a supported format. %s' % e},
            status=status.HTTP_400_BAD_REQUEST
        )

    delta_days = datetime.strptime(end_date, date_format).date() - datetime.strptime(start_date, date_format).date()
    delta_days = delta_days.days + 1
    dt = None
    try:
        with transaction.atomic():
            for i in range(delta_days):
                day = datetime.strptime(start_date, date_format) + timedelta(days=i)
                for i in range(len(time_list)):
                    hrs = datetime.strptime(time_list[i], time_format).hour
                    mins = datetime.strptime(time_list[i], time_format).minute
                    dt = day + timedelta(hours=hrs, minutes=mins)
                    show = Show.objects.create(
                        event=event,
                        date_time=dt
                    )
            return Response({"data": 'Post succesfull'}, status=status.HTTP_201_CREATED)
    except IntegrityError as e:
        return Response(
            {
                'error': "event with date and time already exsits. %s-%s-%s at %s:%s" % (
                    dt.day, dt.month, dt.year, dt.hour, dt.minute),
                'detail': str(e)
            }, status=status.HTTP_400_BAD_REQUEST

Но я надеюсь на гораздо более элегантный способ, чем на то, как я это делаю. Я использую python 3, django 2 и django restoforork. Как я могу создать Показать с событием и date_time на основе event_id, start_date, end_date и time_list?

Ответ 1

Мой подход немного отличается. Вы сказали в вопросительном теге, что используете django-rest-framework. Итак, где же сериализаторы?:)

Позволяет создать два сериализатора: один для проверки пользовательских данных (потому что мы не доверяем ПОЛЬЗОВАТЕЛЮ!) и один для вставки нескольких данных.

Я не проверял код! Но вы можете использовать это был пример...

class ShowEventSerializer(serializers.Serializer):
    event_id = serializers.IntegerField()
    start_date = serializers.DateField(required=True)
    end_date = serializers.DateField(required=True)
    time_list = serializers.ListField(
        child=serializers.TimeField()
    )

    class Meta:
        fields = ('event_id', 'start_date', 'end_date', 'time_list')

class ShowSerializer(serializers.Serializer):
    date_time = serializers.DateTimeField()

    class Meta:
        model = Show
        fields = ('event', 'date_time')

Теперь, с помощью сериализаторов, мы собираемся проверить данные пользователя, а затем создать объект данных json:

def create_show_by_datetime(self, request):

    show_event_serializer = ShowEventSerializer(data=request.data)
    if not show_event_serializer.is_valid():
        return Response({'error': show_event_serializer.errors},status=status.HTTP_400_BAD_REQUEST)

    event_id = show_event_serializer.data['event_id']
    try:
        event = Event.objects.get(id=event_id)
    except Event.DoesNotExist:
        return Response({'error': 'event with id: %s does not exist.' % event_id},status=status.HTTP_400_BAD_REQUEST)

    start_date = show_event_serializer.data['start_date']
    end_date = show_event_serializer.data['end_date']
    time_list = show_event_serializer.data['time_list']
    date_format = '%Y-%m-%d'
    time_format = "%I:%M %p"

    try:
        datetime.strptime(start_date, date_format)
        datetime.strptime(end_date, date_format)
        for i in range(len(time_list)):
            time = datetime.strptime(time_list[i], time_format)
    except ValueError as e:
        return Response(
            {'error': 'Time was not in a supported format. %s' % e},
            status=status.HTTP_400_BAD_REQUEST
        )

    delta_days = datetime.strptime(end_date, date_format).date() - datetime.strptime(start_date, date_format).date()
    delta_days = delta_days.days + 1
    dt = None
    show_data = []
    for i in range(delta_days):
        day = datetime.strptime(start_date, date_format) + timedelta(days=i)
        for i in range(len(time_list)):
            hrs = datetime.strptime(time_list[i], time_format).hour
            mins = datetime.strptime(time_list[i], time_format).minute
            dt = day + timedelta(hours=hrs, minutes=mins)
            show_data.append({
                "event": event,
                "date_time": dt
            })

    try:
        with transaction.atomic():
            show_serializer = ShowSerializer(data=show_data, many=True)
            if show_serializer.is_valid():
                show_serializer.save()

            return Response({"data": 'Post succesfull'}, status=status.HTTP_201_CREATED)
    except IntegrityError as e:
        return Response(
            {
                'error': "event with date and time already exsits. %s-%s-%s at %s:%s" % (
                    dt.day, dt.month, dt.year, dt.hour, dt.minute),
                'detail': str(e)
            }, status=status.HTTP_400_BAD_REQUEST

Итак, этот код в основном такой же, как ваш, с разницей в том, как объекты сохраняются с помощью DRF. Посмотрите переменную show_data.

Это решение - это просто другой способ взглянуть на вопрос.

УДАЧА!

http://www.django-rest-framework.org/api-guide/serializers/

http://www.django-rest-framework.org/api-guide/fields/

Ответ 2

Тогда ваша модель Show должна выглядеть как

from django.contrib.postgres.fields import ArrayField

class Show(models.Model):
    event = models.ForeignKey(Event, on_delete=models.CASCADE)
    start_date = models.DateField(blank=True, null=True)
    end_date = models.DateField(blank=True, null=True)
    board = ArrayField(
        models.TimeField(blank=True, null=True),
        size=10, # specify max array size
    )

поэтому у вас будет модель Show с указанным DateFields и массивом TimeFields.

django docs: Array Field, Поле времени, Поле даты

Ответ 3

Я покидаю часть проверки и фокусируюсь только на создании объектов Show из данных:

data = request.data

date_format = '%Y-%m-%d'
time_format = "%I:%M %p"
show_time_format = f"{date_format} {time_format}"

# get the total number of days by parsing start and end dates
start_date = datetime.strptime(data['start_date'], date_format)
end_date = datetime.strptime(data['end_date'], date_format)
total_days = (end_date - start_date).days + 1

# Get the timings for the first day.
# We will use this to generate the timings for the rest of the days.
first_day_timings = [
    datetime.strptime(f"{data['start_date']} {show_time}", show_time_format)
    for show_time in data['time_list']
]

# generate all show objects using list comprehension and bulk create later
show_objects = [
    Show(event=event, date_time=first_day_timing + timedelta(days=day_cnt))
    for day_cnt in range(total_days)
    for first_day_timing in first_day_timings
]

Show.objects.bulk_create(show_objects)

Улучшения, сделанные в существующем коде:

  • Уменьшено количество экземпляров. Дата/время получает синтаксический анализ и вычисляются часы/минуты.
  • Использовать список для генерации объектов show и массового создания вместо создания одного объекта за раз в транзакции.

Ответ 4

В библиотеке datetime есть несколько инструментов, которые могут дать вам более рациональный подход к генерации вашего времени. Вы можете использовать toordinal, чтобы превратить дату в целое число и fromordinal, чтобы превратить целое число в дату; это дает хороший способ создать диапазон дат. И вы можете использовать combine для объединения объекта date и объекта time в datetime. Я бы создал следующую функцию:

from datetime import datetime, date

def get_showtimes(post):
    start = datetime.strptime(post['start_date'], '%Y-%m-%d')
    end = datetime.strptime(post['end_date'], '%Y-%m-%d')
    times = [datetime.strptime(t, '%I:%M %p').time() for t in post['time_list']]
    for ordinal in range(start.toordinal(), end.toordinal() + 1):
        date = date.fromordinal(date)
        for time in times:
            yield datetime.combine(date, time)

то в вашем коде замените второй блок try: except: и что следует за ним:

try:
    showtimes = list(get_showtimes(post))
except ValueError as e:
    return Response(
        {'error': 'Time was not in a supported format. %s' % e},
        status=status.HTTP_400_BAD_REQUEST
    )

try:
    with transaction.atomic():
        for showtime in showtimes:
            show = Show.objects.create(event=event, date_time=showtime)
except IntegrityError as e:
    # etc.