# coding=utf-8 from django.db import models from django.contrib.auth.models import AbstractBaseUser, PermissionsMixin, BaseUserManager, Group from django.utils import timezone from rest_framework.utils import model_meta from .consts import CONTENT_TYPE_SORTING, MENU_TO_MODEL from apps.system.models import SysLog from utils.exceptions import CustomError class Department(models.Model): name = models.CharField(max_length=100, verbose_name=u"名称") notes = models.CharField(max_length=500, verbose_name=u"备注",blank=True,null=True) parent_id = models.IntegerField(verbose_name=u"父部门",null=True,blank=True) lft = models.IntegerField(verbose_name=u"左值") rgt = models.IntegerField(verbose_name=u"右值") def __str__(self): return self.name def __unicode__(self): return self.name @staticmethod def getById(id): try: id = int(id) except: raise CustomError(u'无效的部门ID') instance = Department.objects.filter(pk=id).first() if not instance: raise CustomError(u'未找到相应的部门') return instance def getCompany(self): instance = Department.objects.filter(parent_id__isnull=True, lft__lte=self.lft, rgt__gte=self.rgt).first() if not instance: raise CustomError(u'未找到部门所属的公司') return instance @staticmethod def getLft(instance): if instance: return instance.lft return 1 class Meta: db_table = "staff_department" verbose_name = u"部门管理" ordering = ["parent_id", 'id'] default_permissions = () permissions = ( ("view_department", u"浏览"), ("add_department", u"添加"), ("edit_department", u"修改"), ("delete_department", u"删除"), ) class UserManager(BaseUserManager): def create_administrator(self, username, password=None, **extra_fields): return self.create_user(User.ADMINSTRATOR, username, password, **extra_fields) def create_staff(self, username, password=None, **extra_fields): return self.create_user(User.STUDENT, username, password, **extra_fields) def create_superuser(self, username, password, **extra_fields): u = self.create_administrator(username, password, **extra_fields) u.is_active = True u.is_superuser = True u.save(using=self._db) return u def create_user(self, type, username, password=None, **extra_fields): if not username: raise CustomError(u'请输入用户名！') count = User.objects.filter(username=username).count() if count > 0: raise CustomError(u'该用户名已存在！') user = self.model( type=type, username=username, is_superuser=False, last_login=timezone.now(), **extra_fields ) user.set_password(password) user.save(using=self._db) return user def sort_perms(self, perms): def get_index(app_label, model): try: return CONTENT_TYPE_SORTING.index('{}-{}'.format(app_label, model)) except: return 9999 perms = perms.order_by('content_type__model', 'id') perms = sorted(perms, key=lambda n: get_index(n.content_type.app_label, n.content_type.model)) return perms def get_menuname_of_contenttype(self, app_label, model): for menu in MENU_TO_MODEL: val = '{}-{}'.format(app_label, model) if val in menu[1]: return menu[0] return u'未分类' def save_group(self, id, name, permissions, user): name = name.strip(u' ') # old_permissions = None if id == None or id == '': is_exist = Group.objects.filter(name=name).first() if is_exist: raise CustomError(u'名称为[%s]的权限组已存在' % name) group = Group.objects.create(name=name) SysLog.objects.addnew(user, SysLog.INSERT, u"添加权限组[%s],id=%d" % (group.name, group.id)) else: is_exist = Group.objects.filter(name=name).exclude(pk=id).first() if is_exist: raise CustomError(u'名称为[%s]的权限组已存在' % name) group = Group.objects.filter(pk=id).first() if not group: raise CustomError(u'未找到相应的权限组') group.name = name group.save() # old_permissions = [p.id for p in group.permissions.all()] SysLog.objects.addnew(user, SysLog.UPDATE, u"修改权限组[%s],id=%d" % (group.name, group.id)) group.permissions = permissions class User(AbstractBaseUser, PermissionsMixin): ADMINSTRATOR = 1 STAFF = 2 type = models.PositiveSmallIntegerField(verbose_name=u"类型", editable=False) department = models.ForeignKey(Department, verbose_name=u"所属部门", null=True, blank=True, on_delete=models.PROTECT) username = models.CharField(verbose_name=u'帐号', max_length=30, unique=True, db_index=True) is_active = models.BooleanField(verbose_name=u'激活', default=True) date_joined = models.DateTimeField(verbose_name=u'注册时间', default=timezone.now, editable=False) objects = UserManager() USERNAME_FIELD = 'username' REQUIRED_FIELDS = [] class Meta: db_table = "staff_user" verbose_name = u"账户管理" unique_together = [ ('username') ] index_together = ( 'date_joined', ) ordering = ['-id'] default_permissions = () permissions = ( ("view_user", u"浏览"), ("add_user", u"添加"), ("edit_user", u"修改"), ("delete_user", u"删除"), ) def __unicode__(self): return self.username def add_administrator(self): self.type = self.type | self.ADMINSTRATOR def add_staff(self): self.type = self.type | self.STAFF def is_staff(self): if self.type & self.STAFF: return True return False def is_administrator(self): if self.type & self.ADMINSTRATOR: return True return False def change_password(self, new_password, confirm_password, old_password): if new_password != confirm_password: raise CustomError(u'两次输入的密码不一致， 请检查') if not self.check_password(old_password): raise CustomError(u'原密码输入错误， 请检查') self.set_password(new_password) def update_item(self, validated_data): def update(): info = model_meta.get_field_info(self) for attr, value in validated_data.items(): if attr in info.relations and info.relations[attr].to_many: field = getattr(self, attr) field.set(value) else: setattr(self, attr, value) if not 'username' in validated_data: raise CustomError(u'用户名不能为空！') count = User.objects.filter(username=validated_data['username']).exclude(id=self.id).count() if count > 0: raise CustomError(u'该用户名已存在！') if not 'password' in validated_data or not validated_data['password']: validated_data['password'] = self.password update() else: update() self.set_password(validated_data['password']) self.save() return self