Как рассчитать количество дней между двумя датами и вычесть выходные дни DJANGO MODELS

надеюсь, у вас все хорошо!

У меня есть модель под названием Vacation и я борюсь с одним полем: days_requested, это поле - количество дней из vacation_start и vacation_end, оно работает и дает мне целое число в качестве результата. Проблема, с которой я столкнулся сейчас, заключается в том, что мне нужно вычесть выходные дни (или не считать их).

Что у меня есть:

vacation_start = '2022-05-20'
vacation_end = '2022-05-24'
days_requested = 5

Что я пытаюсь получить:

vacation_start = '2022-05-20'
vacation_end = '2022-05-24'
days_requested = 3

#21/05 and 22/05 are weekends

Модели для отдыха:

class Vacation(models.Model):
    department = models.ForeignKey(
        'departments.Department', on_delete=models.SET_NULL, null=True)
    responsible = models.ForeignKey(
        User, on_delete=models.SET_NULL, null=True, related_name='responsible_vacation')
    status = models.CharField(
        max_length=20, choices=STATUS_CHOICES_VACATIONS, default='open')
    created_at = models.DateTimeField(auto_now_add=True)
    updated_at = models.DateTimeField(auto_now=True)
    created_by = models.ForeignKey(
        User, on_delete=models.SET_NULL, null=True, related_name='created_by_vacation')
    vacation_start = models.DateField(blank=False)
    vacation_end = models.DateField(blank=False)
    days_requested = property(
        lambda self: (self.vacation_end - self.vacation_start).days + 1
    )
    def __str__(self):
        return str(self.created_by.first_name + ' ' + self.created_by.last_name)

Я пробовал:

    days_requested = property(
         lambda self: [(self.vacation_start + datetime.timedelta(days=i)).date()
                      for i in range(0, (self.vacation_end - self.vacation_start)) if (self.vacation_start + datetime.timedelta(days=i)).weekday() not in [5, 6].days()])

Но я получаю следующую ошибку:

'datetime.timedelta' object cannot be interpreted as an integer

Поскольку я выполняю операции с заданным количеством дней, мне нужно иметь возможность получить целое число.

Заранее благодарю вас всех.

UPDATE


class Model:
    days_requested = models.IntegerField(blank=True,null=True)
    
    
      def save(self, *args, **kwargs):
        excluded = (6, 7)
        days = 0
        start_date =self.vacation_start
        while start_date < self.vacation_end:
            if start_date.isoweekday() not in excluded: #if you want to get only weekdays
                days += 1
                start_date+= timedelta(days=1)
         
        self.days_requested=days

        super(YourModel, self).save(*args, **kwargs)


После ответа Эльвина я переместил логику дыры в представление и установил логику внутри функции form_valid:

        start = form.instance.vacation_start
        end = form.instance.vacation_end
        delta = end - start
        excluded = (6, 7)
        days = 0
        for i in range(delta.days + 1):
            day = start + datetime.timedelta(days=i)
            if day.isoweekday() not in excluded:
                days += 1
        form.instance.days_requested = days

Спасибо вам всем.

Вернуться на верх