You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
50 lines
1.6 KiB
50 lines
1.6 KiB
from pathlib import Path
|
|
import uuid
|
|
|
|
from django.db import models
|
|
from django.contrib.auth.models import BaseUserManager, AbstractBaseUser, PermissionsMixin
|
|
|
|
|
|
class UserManager(BaseUserManager):
|
|
|
|
def create_superuser(self, userid, username, password, **kwargs):
|
|
kwargs['is_superuser'] = True
|
|
user = self.model(userid=userid,
|
|
username=username,
|
|
**kwargs)
|
|
user.set_password(password)
|
|
user.save()
|
|
return user
|
|
|
|
|
|
class User(AbstractBaseUser, PermissionsMixin):
|
|
userid = models.CharField(verbose_name="账号", max_length=20, primary_key=True)
|
|
username = models.CharField(verbose_name="用户名", max_length=20, unique=True)
|
|
is_staff = models.BooleanField(default=False)
|
|
|
|
objects = UserManager()
|
|
|
|
USERNAME_FIELD = "userid"
|
|
|
|
REQUIRED_FIELDS = ["username", "is_staff"]
|
|
|
|
|
|
def user_avatar_path(instance, avatar_name):
|
|
ext = avatar_name.split('.')[-1]
|
|
avatar_name = '{}.{}'.format(uuid.uuid4().hex[:10], ext)
|
|
return Path("avatar", avatar_name)
|
|
|
|
|
|
class UserProfile(models.Model):
|
|
sex_choice = (
|
|
(0, "男"),
|
|
(1, "女")
|
|
)
|
|
user = models.OneToOneField(to=User, verbose_name="用户", on_delete=models.CASCADE, related_name="profile_user")
|
|
avatar = models.FileField(verbose_name="头像", upload_to=user_avatar_path)
|
|
sex = models.SmallIntegerField(verbose_name="性别", choices=sex_choice)
|
|
address = models.CharField(verbose_name="住址", max_length=20)
|
|
email = models.EmailField(verbose_name="邮箱", max_length=30)
|
|
telephone = models.CharField(verbose_name="手机号", max_length=11)
|
|
|