diff --git a/users/models.py b/users/models.py index 2c75099..ac5c4ab 100644 --- a/users/models.py +++ b/users/models.py @@ -6,6 +6,8 @@ from common.utils.codeManger import set_expire from django.contrib.postgres.fields import ArrayField from django.contrib.auth.models import AbstractBaseUser, BaseUserManager, PermissionsMixin +from common.models.choiceModels import NotificationType + class UserManager(BaseUserManager): def create_user(self, email, password, **kwargs): user = self.model(email = email, **kwargs) @@ -47,6 +49,7 @@ class User(BaseModel, AbstractBaseUser, PermissionsMixin): external_links = ArrayField(models.TextField(), default=list, blank=True) short_bio = models.CharField(max_length=100, blank=True) profile_image = models.ImageField(upload_to='', blank=True) + banner_image = models.ImageField(upload_to='', blank=True) is_staff = models.BooleanField(default=False) is_active = models.BooleanField(default=True) @@ -58,4 +61,10 @@ class User(BaseModel, AbstractBaseUser, PermissionsMixin): def __str__(self): return self.nickname - \ No newline at end of file + + +class Notification(BaseModel): + user = models.ForeignKey(User, on_delete=models.CASCADE, related_name='notifications') + content = models.TextField() + is_read = models.BooleanField(default=False) + note_type = models.CharField(max_length=10, choices=NotificationType.choices)