mirror of
https://github.com/himool/HimoolERP.git
synced 2025-02-25 14:44:02 +08:00
32 lines
1.1 KiB
Python
32 lines
1.1 KiB
Python
from django.contrib.auth.models import AbstractBaseUser, BaseUserManager
|
|
from django_mysql.models import JSONField
|
|
from django.db import models
|
|
|
|
|
|
class Teams(models.Model):
|
|
phone = models.CharField(max_length=12, unique=True)
|
|
company_name = models.CharField(max_length=48)
|
|
# end_datetime = models.DateTimeField()
|
|
|
|
|
|
class UserManager(BaseUserManager):
|
|
def create(self, username, password, **kwargs):
|
|
user = self.model(username=username, **kwargs)
|
|
user.set_password(password)
|
|
user.save(self._db)
|
|
return user
|
|
|
|
|
|
class User(AbstractBaseUser):
|
|
username = models.CharField(primary_key=True, max_length=24)
|
|
name = models.CharField(max_length=48)
|
|
phone = models.CharField(max_length=12)
|
|
teams = models.ForeignKey('user.Teams', models.CASCADE, related_name='users')
|
|
|
|
roles = models.ManyToManyField('account.Role', related_name='users')
|
|
create_date = models.DateTimeField(auto_now_add=True)
|
|
is_boss = models.BooleanField(default=True)
|
|
is_delete = models.BooleanField(default=False)
|
|
|
|
USERNAME_FIELD = 'username'
|
|
objects = UserManager()
|