Переопределить проверку Django Unique_Together путем добавления дополнительного поля

Question

Мне нужна помощь в сохранении некоторых записей в БД.Работайте для приложения для парковки, и я хочу расширить способ, как unique_together запрашивает БД, имея дополнительный параметр.Просто чтобы понять, какова моя ситуация.

Например, я даю пользователю возможность зарезервировать участок на (стоянке) 25-го из 9-18 местоположения P1.Поскольку оба фильтра уникальны для БД, кто-то другой не может заказать один и тот же участок P1 на тот же день 25-го, но с 19-24 ==>, поэтому мне нужна некоторая проверка здесь или вручную изменить способвалидация unique_together работает.

Может кто-нибудь помочь, пожалуйста?

Ниже приведены мои модели и администратор

from django.db import models
from django.conf import settings
from django.core.exceptions import ValidationError
from django.utils.translation import gettext_lazy as _
from datetime import datetime, timedelta, time
from django.core.exceptions import NON_FIELD_ERRORS
today = datetime.now().date()
tomorrow = today + timedelta(1)
now = datetime.now()
l = now.hour
m = int(now.strftime("%H"))


class ParcareManager(models.Manager):
    def active(self, *args, **kwargs):
        return super(ParcareManager, self).filter(draft=False).filter(parking_on__lte=datetime.now())


class Parcare(models.Model):
    PARKING_PLOT = (('P1', 'Parking #1'),
                    ('P2', 'Parking #2'), ('P3', 'Parking #3'))
    user = models.ForeignKey(settings.AUTH_USER_MODEL,
                             blank=True, null=True, default=1, on_delete=True)
    email = models.EmailField(blank=True, null=True)
    parking_on = models.DateField(auto_now=False, auto_now_add=False, blank=True, null=True,
                                  help_text='Alege data cand doresti sa vii in office',)
    parking_off = models.DateField(auto_now=False, auto_now_add=False, blank=True, null=True,
                                   help_text='Alege Data Plecarii')
    numar_masina = models.CharField(max_length=8, default="IF77WXV", blank=True, null=True,
                                    help_text='Introdu Numarul Masinii')
    location = models.CharField(max_length=3, blank=True, default="P1", null=True, choices=PARKING_PLOT,
                                help_text='Alege Locul de Parcare Dorit')
    updated = models.DateTimeField(
        auto_now=True, auto_now_add=False, blank=True, null=True)
    timestamp = models.DateTimeField(
        auto_now=False, auto_now_add=True, blank=True, null=True)
    venire = models.TimeField(default=time(
        9, 00), auto_now=False, auto_now_add=False, help_text='Alege Ora Venirii')
    plecare = models.TimeField(default=time(
        18, 00), auto_now=False, auto_now_add=False, help_text='Alege Ora Plecarii')
    objects = ParcareManager()

    def __str__(self):
        return self.location + " | " + str(self.parking_on) + " | " + str(self.parking_off)

    class Meta:
        verbose_name_plural = "parcare"
        ordering = ["-parking_on"]
        unique_together = ("parking_on", "location")

    def clean(self):
        if self.parking_on == today:  # merge--vedem dak parcam azi
            raise ValidationError(
                {'parking_on': _('Please book for a date starting tomorrow')})
        if self.parking_off < self.parking_on:  # merge-vedem daca bookam in trecut
            raise ValidationError(
                {'parking_off': _('You cant book for a past date!')})



    def save(self):
        list = []
        d = self.parking_on
        while d <= self.parking_off:
            list.append(
                Parcare(user=self.user,
                        email=self.email,
                        parking_on=d,
                        parking_off=d,
                        location=self.location
                        )
            )
            d = d + timedelta(days=1)
        Parcare.objects.bulk_create(list)


from django.contrib import admin
from .models import Parcare


class ParcareModelAdmin(admin.ModelAdmin):
    list_display = ["user", "location",
                    "parking_on", "parking_off",  "venire", "plecare", "timestamp"]
    list_display_links = ["user", "location"]
    list_editable = ["parking_off", "parking_on", "venire", "plecare"]
    list_filter = ["parking_on", "location", "email"]
    search_fields = ["location", "parking_on"]
    date_hierarchy = 'parking_on'

    class Meta: 
        model = Parcare
    def get_form(self, request, obj=None, **kwargs):
        form = super().get_form(request, obj, **kwargs)
        if not obj:
            user = request.user
            form.base_fields['user'].initial = user
            form.base_fields['email'].initial = user.email
        return form 

admin.site.register(Parcare, ParcareModelAdmin)

Заранее спасибо!

Answer 1

У меня просто есть мысли по поводу этой проблемы.Я думаю, что лучшее решение - заменить DateField на DateTimeField.Затем вы можете использовать в методе clean что-то вроде этого (как в примере с плохой производительностью):

def clean(self):
    qs = Parcare.objects.filter(location=self.location)
                        .filter(parking_on__gt=self.parking_on)
                        .filter(parking_off__lt=self.parking_on)
    qs2 = Parcare.objects.filter(location=self.location)
                         .filter(parking_on__gt=self.parking_off)
                         .filter(parking_off__lt=self.parking_off)
    if len(qs) > 0 or len(qs2) > 0:
        raise ValidationError('Time is booked already')
    if self.parking_on < datetime.now():  # merge--vedem dak parcam azi
        raise ValidationError(
            {'parking_on': _('Please book for a time more than now')})
    if self.parking_off < self.parking_on:  # merge-vedem daca bookam in trecut
        raise ValidationError(
            {'parking_off': _('You cant book for a past date!')})