Kumi
893c13a0b4
Add fields for weekdays and days of month Update requirements to include dateutil, polymorphic, multiselectfield
45 lines
1.4 KiB
Python
45 lines
1.4 KiB
Python
from django.db import models
|
|
from django.utils import timezone
|
|
|
|
from colorfield.fields import ColorField
|
|
from polymorphic.models import PolymorphicModel
|
|
from dateutil.relativedelta import relativedelta
|
|
|
|
from common.fields import WeekdayField, DayOfMonthField
|
|
|
|
# Create your models here.
|
|
|
|
class Habit(models.Model):
|
|
name = models.CharField(max_length=64)
|
|
icon = models.CharField(default="fas fa-user-clock", max_length=64)
|
|
color = ColorField(default="#000000")
|
|
description = models.TextField(null=True, blank=True)
|
|
active = models.BooleanField(default=True)
|
|
|
|
class HabitSchedule(PolymorphicModel):
|
|
habit = models.ForeignKey(Habit, models.CASCADE)
|
|
|
|
@property
|
|
def next_scheduled(self, today=True, now=timezone.now()):
|
|
raise NotImplementedError("%s does not implement next_scheduled." % self.__class__)
|
|
|
|
class MonthlyHabitSchedule(HabitSchedule):
|
|
day = DayOfMonthField()
|
|
|
|
@property
|
|
def next_scheduled(self, today=True, now=timezone.now()):
|
|
if self.day < now.day:
|
|
date = now.replace(day=self.day) + relativedelta(months=1)
|
|
elif self.day == now.day:
|
|
date = now if today else now + relativedelta(months=1)
|
|
else:
|
|
date = now + relativedelta(day=self.day)
|
|
|
|
return date.date()
|
|
|
|
class WeeklyHabitSchedule(HabitSchedule):
|
|
weekdays = WeekdayField()
|
|
|
|
class DailyHabitSchedule(HabitSchedule):
|
|
pass
|