ASJServer/Service/ModelService.py

from Model.models import *
from Service.CommonService import CommonService


# 针对模型封装的复用性代码
class ModelService:
    # 获取当前用户角色名
    @staticmethod
    def getRole(rid):
        return Role.objects.get(rid=rid).roleName

    # 获取用户所有权限
    @staticmethod
    def own_permission(userID):
        permission = Device_User.objects.get(userID=userID).role.values_list('permission',flat=True)
        if permission:
            return list(permission)
        return []

    # 获取用户角色相关
    @staticmethod
    def own_role(userID):
        try:
            role_qs = Device_User.objects.get(userID=userID).role.values('rid','roleName')
            if role_qs.exists():
                return {'rid': role_qs[0]['rid'], 'roleName': role_qs[0]['roleName']}
        except Exception as e:
            pass
        return {'rid': '', 'roleName': ''}

    # 检测权限有无
    @staticmethod
    def check_perm(userID, permID):
        try:
            perm_list = Device_User.objects.get(userID=userID).role.values_list('permission', flat=True)
            if perm_list:
                if permID in perm_list:
                    return True
        except Exception as e:
            return False
        return False

    # 根据设备主键ID判断是否拥有该设备
    @staticmethod
    def check_user_own_device(userID, deviceID):
        try:
            dvqs = Device_Info.objects.filter(userID_id=userID).values_list('id',flat=True)
            if dvqs:
                if deviceID in dvqs:
                    return True
        except Exception as e:
            return False
        return False

    # 根据设设备唯一名称UID判断是否拥有该设备
    @staticmethod
    def check_own_device(userID, UID):
        dvqs = Device_Info.objects.filter(userID_id=userID, UID=UID)
        if dvqs.exists():
            return True
        return False

    # 根据userID获取用户名
    @staticmethod
    def get_user_name(userID):
        try:
            if userID:
                device_user = Device_User.objects.get(userID=userID)
                return device_user.username
            else:
                return ''
        except Exception as e:
            return ''

    # 根据username获取userID
    @staticmethod
    def get_userID_byname(username):
        try:
            device_user = Device_User.objects.get(username=username)
        except Exception as e:
            return None
        else:
            return device_user.userID

    # 访问日志添加
    @staticmethod
    def addAccessLog(data):
        try:
            access_log = Access_Log.objects.create(**data)
        except Exception as e:
            return False
        else:
            if access_log.id:
                return True
            else:
                return False