from datetime import timezone
def today():
return datetime.utcnow() + timezone.timedelta(days=1)
# Enrollment model as a lookup table with additional enrollment info
class Enrollment(models.Model):
AUDIT = 'audit'
HONOR = 'honor'
COURSE_MODES = [
(AUDIT, 'Audit'),
(HONOR, 'Honor'),
]
# Add a learner foreign key
learner = models.ForeignKey(Learner, on_delete=models.CASCADE)
# Add a course foreign key
course = models.ForeignKey(Course, on_delete=models.CASCADE)
# Enrollment date
date_enrolled = models.DateField(default=today)
# Enrollment mode
mode = models.CharField(max_length=5, choices=COURSE_MODES, default=AUDIT)