Django,@ property,在models.py内部具有两个功能

时间:2019-05-10 20:26:54

标签: python django django-models

我不知道如何正确命名该主题,因此如果该主题包含错误,我深表歉意。

我的模型包含以下内容:

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

如果该员工当前正在工作,则在我的views.py中返回“ True”。

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

以上我写的内容不起作用。我如何才能以最快的速度从@property获取“真”值。

对于我的发言不是很容易理解,我深表歉意,但我不知道如何更轻松地描述它。在我看来,我的方法相当复杂,因此,如果该论坛的任何用户对如何更快地提出建议,我将不胜感激。任何帮助将不胜感激。

2 个答案:

答案 0 :(得分:1)

由于要从Model类外部导入这些辅助函数,请尝试执行以下操作:

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

答案 1 :(得分:1)

通过稍微简化代码,这可能会更容易解决,并使其更加有效。

例如,我们可以将支票写为:

Employee(models.Model):

    # ...

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

现在,我们可以简单地定义另外两个@propertyday_st,分别是:

day_end

因此,通过在 @property def day_st(self): 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] @property def day_end(self): 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模型中为Employeemonday_st等使用了 integers 。 ,则应使用IntegerField fields [Django-doc]。通过使用monday_end,您的数据库可以帮助维护模型的类型完整性,此外,您不必自己进行很多手动转换。