from django.db import models
from django.contrib.auth import get_user_model

User = get_user_model()


class UserNotification(models.Model):
    user = models.ForeignKey(User, on_delete=models.CASCADE)
    certificate_number = models.SmallIntegerField(blank=True, null=True)
    certificate_last_email = models.DateTimeField(blank=True, null=True)
    camp_certificate_last_email = models.DateTimeField(blank=True, null=True)


class CourseNotification(models.Model):
    user = models.ForeignKey(User, on_delete=models.CASCADE)
    course = models.ForeignKey('course.Course', on_delete=models.CASCADE)
    access_expire_last_email = models.DateTimeField(blank=True, null=True)