-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathmodels.py
38 lines (31 loc) · 1.34 KB
/
models.py
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
from django.db import models
from django.contrib.auth.models import AbstractBaseUser, BaseUserManager, PermissionsMixin
from django.utils import timezone
class MyUserManager(BaseUserManager):
def create_user(self, email, nickname, password=None):
if not email:
raise ValueError('이메일 주소는 필수입니다.')
if not nickname:
raise ValueError('닉네임은 필수입니다.')
email = self.normalize_email(email)
user = self.model(email=email, nickname=nickname)
user.set_password(password)
user.save(using=self._db)
return user
def create_superuser(self, email, nickname, password=None):
user = self.create_user(email=email, nickname=nickname, password=password)
user.is_staff = True
user.is_superuser = True
user.save(using=self._db)
return user
class MyUser(AbstractBaseUser, PermissionsMixin):
email = models.EmailField(max_length=255, unique=True)
nickname = models.CharField(max_length=50, unique=True)
is_active = models.BooleanField(default=True)
is_staff = models.BooleanField(default=False)
date_joined = models.DateTimeField(default=timezone.now)
objects = MyUserManager()
USERNAME_FIELD = 'email'
REQUIRED_FIELDS = ['nickname']
def __str__(self):
return self.email