Как рассчитать количество дней между двумя датами и вычесть выходные дни 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
Спасибо вам всем.