Add save() method
This commit is contained in:
parent
8b4ca48944
commit
2629d3ade7
|
@ -4,7 +4,10 @@ from django.db import models
|
|||
from django.utils import timezone
|
||||
|
||||
from datetime import date
|
||||
from .date_week_transition import get_number_of_weeks_between
|
||||
from .date_week_transition import (
|
||||
get_number_of_weeks_between,
|
||||
from_date_to_week_number,
|
||||
)
|
||||
import markdown
|
||||
import pendulum
|
||||
|
||||
|
@ -17,6 +20,14 @@ class Seasonisable(models.Model):
|
|||
season = models.CharField(max_length=9)
|
||||
week_number = models.PositiveSmallIntegerField()
|
||||
|
||||
def save(self, *args, **kwargs):
|
||||
""" Calcule les valeurs `season` et `week_number` sur base d'une date lors de l'
|
||||
enregistrement d'un object enfant.
|
||||
"""
|
||||
if self.date is not None:
|
||||
self.season, self.week_number = from_date_to_week_number(self.date)
|
||||
super().save(*args, **kwargs)
|
||||
|
||||
|
||||
class TemporizableQuerySet(models.QuerySet):
|
||||
"""
|
||||
|
|
Loading…
Reference in New Issue