-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathmodels.py
More file actions
72 lines (53 loc) · 2.12 KB
/
models.py
File metadata and controls
72 lines (53 loc) · 2.12 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
from django.db import models
from django.contrib.auth.models import PermissionsMixin
from django.contrib.auth.models import AbstractBaseUser
from django.contrib.auth.models import BaseUserManager
from django.conf import settings
#classes in python should have two spaces between them
class UserProfileManager(BaseUserManager):
"""Manager for user profile"""
def create_user(self, email, name, password=None):
"""create a new user profile"""
if not email: #checking for error
raise ValueError('user must have an email address')
email = self.normalize_email(email) # making email lower case
user = self.model(email=email, name=name)
user.set_password(password) #hashing the password
user.save(using=self._db) #saving to db, supports for multiple db
return user
def create_superuser(self, email, name, password):
"""create super user with given details"""
user = self.create_user(email, name, password)
user.is_superuser = True
user.is_staff = True
user.save(using=self._db)
return user
class UserProfile(AbstractBaseUser, PermissionsMixin):
"""Database model for users in the system"""
email = models.EmailField(max_length=255, unique=True)
name = models.CharField(max_length=255)
is_active = models.BooleanField(default=True)
is_staff = models.BooleanField(default=False)
objects = UserProfileManager()
USERNAME_FIELD = 'email' #replacing the default django user name field with the email
REQUIRED_FIELDS = ['name']
def get_full_name(self):
"""Retrieve full name user"""
return self.name
def get_short_name(self):
"""Retrieve short name of user"""
return self.name
def __str__(self):
"""return string representation of the user"""
return self.email
class ProfileFeedItem(models.Model):
"""Profile status update"""
user_profile = models.ForeignKey(
settings.AUTH_USER_MODEL,
on_delete=models.CASCADE
)
status_text = models.CharField(max_length=255)
created_on = models.DateTimeField(auto_now_add=True)
def __str__(self):
"""return the model as a string"""
return self.status_text