mirror of
https://github.com/kingmo888/rustdesk-api-server.git
synced 2026-02-21 18:37:23 +08:00
90 lines
2.8 KiB
Python
90 lines
2.8 KiB
Python
# cython:language_level=3
|
|
from django.db import models
|
|
from django.contrib.auth.models import (
|
|
BaseUserManager,AbstractBaseUser,PermissionsMixin
|
|
)
|
|
from .models_work import *
|
|
from django.utils.translation import gettext as _
|
|
|
|
class MyUserManager(BaseUserManager):
|
|
def create_user(self, username, password=None):
|
|
if not username:
|
|
raise ValueError('Users must have an username')
|
|
|
|
user = self.model(username=username,
|
|
)
|
|
|
|
user.set_password(password)
|
|
user.save(using=self._db)
|
|
return user
|
|
|
|
def create_superuser(self, username, password):
|
|
user = self.create_user(username,
|
|
password=password,
|
|
|
|
)
|
|
user.is_admin = True
|
|
user.save(using=self._db)
|
|
return user
|
|
|
|
|
|
class UserProfile(AbstractBaseUser, PermissionsMixin):
|
|
username = models.CharField(_('用户名'),
|
|
unique=True,
|
|
max_length=50)
|
|
|
|
rid = models.CharField(verbose_name='RustDesk ID', max_length=16)
|
|
uuid = models.CharField(verbose_name='uuid', max_length=60)
|
|
autoLogin = models.BooleanField(verbose_name='autoLogin', default=True)
|
|
rtype = models.CharField(verbose_name='rtype', max_length=20)
|
|
deviceInfo = models.TextField(verbose_name=_('登录信息:'), blank=True)
|
|
|
|
is_active = models.BooleanField(verbose_name=_('是否激活'), default=True)
|
|
is_admin = models.BooleanField(verbose_name=_('是否管理员'), default=False)
|
|
|
|
objects = MyUserManager()
|
|
|
|
USERNAME_FIELD = 'username' # 用作用户名的字段
|
|
REQUIRED_FIELDS = ['password'] #必须填写的字段
|
|
|
|
|
|
def get_full_name(self):
|
|
# The user is identified by their email address
|
|
return self.username
|
|
|
|
def get_short_name(self):
|
|
# The user is identified by their email address
|
|
return self.username
|
|
|
|
def __str__(self): # __unicode__ on Python 2
|
|
return self.username
|
|
|
|
def has_perm(self, perm, obj=None): #有没有指定的权限
|
|
"Does the user have a specific permission?"
|
|
# Simplest possible answer: Yes, always
|
|
return True
|
|
|
|
def has_module_perms(self, app_label):
|
|
"Does the user have permissions to view the app `app_label`?"
|
|
# Simplest possible answer: Yes, always
|
|
return True
|
|
|
|
|
|
|
|
@property
|
|
def is_staff(self):
|
|
"Is the user a member of staff?"
|
|
# Simplest possible answer: All admins are staff
|
|
return self.is_admin
|
|
|
|
class Meta:
|
|
|
|
verbose_name = _("用户")
|
|
verbose_name_plural = _("用户列表")
|
|
permissions = (
|
|
("view_task", "Can see available tasks"),
|
|
("change_task_status", "Can change the status of tasks"),
|
|
("close_task", "Can remove a task by setting its status as closed"),
|
|
)
|
|
|