Django, @property с двумя функциями внутри models.py - PullRequest
Новая
       16

Django, @property с двумя функциями внутри models.py

1 голос
/ 10 мая 2019

Я не знал, как правильно назвать тему, поэтому прошу прощения, если она содержит ошибки.

Моя модель состоит из чего-то вроде: 

class Employee(models.Model):
    #time work
    monday_st = models.CharField(max_length=2, choices=CHOICES_DAY_WORK, default='8')
    monday_end = models.CharField(max_length=2, choices=CHOICES_DAY_WORK, default='20')
    tuesday_st = models.CharField(max_length=2, choices=CHOICES_DAY_WORK, default='8')
    tuesday_end = models.CharField(max_length=2, choices=CHOICES_DAY_WORK, default='20')
    wednesday_st = models.CharField(max_length=2, choices=CHOICES_DAY_WORK, default='8')
    wednesday_end = models.CharField(max_length=2, choices=CHOICES_DAY_WORK, default='20')
    thursday_st = models.CharField(max_length=2, choices=CHOICES_DAY_WORK, default='8')
    thursday_end = models.CharField(max_length=2, choices=CHOICES_DAY_WORK, default='20')
    friday_st = models.CharField(max_length=2, choices=CHOICES_DAY_WORK, default='8')
    friday_end = models.CharField(max_length=2, choices=CHOICES_DAY_WORK, default='20')
    saturday_st = models.CharField(max_length=2, choices=CHOICES_DAY_WORK, default='0')
    saturday_end = models.CharField(max_length=2, choices=CHOICES_DAY_WORK, default='0')
    sunday_st = models.CharField(max_length=2, choices=CHOICES_DAY_WORK, default='0')
    sunday_end = models.CharField(max_length=2, choices=CHOICES_DAY_WORK, default='0')

Я бы хотел отфильтровать сотрудников, которые в настоящее время работают. Так что это отражение их работы в модели, которая выглядит следующим образом. 

@property
def is_expired(self):
    actual_hour = int(datetime.datetime.now().strftime('%H'))
    if actual_hour in list(range(int(self.monday_st), int(self.monday_end))):
        return True
    return False

Возвращает значение True в my views.py, если сотрудник в данный момент работает. 

if instance.is_expired == True:
    'the employee is working'
else:
    'no working'

Но ему нужна информация, если сотрудник работает только в определенный день, поэтому я создал файл со вспомогательными функциями, который выглядит следующим образом. 

def day_st():
    actual_day_number = datetime.datetime.today().weekday() + 1
    if actual_day_number == 1:
        return monday_st
    if actual_day_number == 2:
        return tuesday_st
    if actual_day_number == 3:
        return wednesday_st
    if actual_day_number == 4:
        return thursday_st
    if actual_day_number == 5:
        return friday_st
    if actual_day_number == 6:
        return saturday_st
    if actual_day_number == 7:
        return sunday_st

def day_end():
    actual_day_number = datetime.datetime.today().weekday() + 1
    if actual_day_number == 1:
        return monday_end
    if actual_day_number == 2:
        return tuesday_end
    if actual_day_number == 3:
        return wednesday_end
    if actual_day_number == 4:
        return thursday_end
    if actual_day_number == 5:
        return friday_st
    if actual_day_number == 6:
        return saturday_end
    if actual_day_number == 7:
        return sunday_end

Но я не могу использовать его в моей модели. Я пытаюсь что-то подобное. 

from .my_helped_function import day_st, day_end
    @property
    def is_expired(self):
        actual_hour = int(datetime.datetime.now().strftime('%H'))
        if actual_hour in list(range(int(self.day_st), int(self.day_end))):
            return True
        return False

Это выше того, что я написал, не работает. Как я могу получить значение 'True' из @property самое быстрое только за текущий день.

Я прошу прощения, если мое утверждение не очень понятно, но я не знаю, как его описать проще. Мой путь кажется мне довольно сложным, поэтому, если у кого-то из пользователей форума есть предложения о том, как это сделать быстрее, я буду очень признателен. Любая помощь будет оценена.

Ответы [ 2 ]

1 голос
/ 10 мая 2019

Если немного упростить код, это, вероятно, станет намного проще, а также сделает его более эффективным.

Например, мы можем написать чек как: 

Employee(models.Model):

    # ...

    @property
    def is_expired(self):
        hour = datetime.datetime.now()<b>.hour</b>
        return hour in range(int(<b>self.day_st</b>), int(<b>self.day_end</b>))

Теперь мы можем просто определить два дополнительных @property s day_st и day_end с помощью: 

    <b>@property</b>
    def day_st(<b>self</b>):
        day = datetime.datetime.today().weekday()
        return (
            self.monday_st,
            self.tuesday_st,
            self.wednesday_st,
            self.thursday_st,
            self.friday_st,
            self.saturday_st,
            self.sunday_st)[day]

    <b>@property</b>
    def day_end(<b>self</b>):
        day = datetime.datetime.today().weekday()
        return (
            self.monday_end,
            self.tuesday_end,
            self.wednesday_end,
            self.thursday_end,
            self.friday_end,
            self.saturday_end,
            self.sunday_end)[day]

Итак, объявив три @property с в модели Employee, мы закончили.

Примечание : поскольку кажется, что вы используете целые числа в вашей Employee модели для monday_st, monday_end и т. Д., Вам следует использовать IntegerField поля [Джанго-док] . Используя IntegerField, ваша база данных может помочь защитить целостность типа ваших моделей, и, кроме того, вам не нужно выполнять много ручной разборки самостоятельно.

1 голос
/ 10 мая 2019

Поскольку вы импортируете эти вспомогательные функции из-за пределов класса Model, попробуйте сделать это вместо этого: 

# . . .
def is_expired(self):
    # . . .
    if actual_hour in list(range(int(day_st()), int(day_end()))):
    # . . .
...