ASJServer/Service/ModelService.py

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


# 针对模型封装的复用性代码
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:
            return True

    # 访问日志批量添加
    @staticmethod
    def add_batch_log(data_list):
        try:
            if data_list:
                querysetlist = []
                for i in data_list:
                    data = json.loads(i.decode('utf-8'))
                    querysetlist.append(Access_Log(**data))
                Access_Log.objects.bulk_create(querysetlist)
            else:
                return
        except Exception as e:
            print('ggga')
            print(repr(e))
            return False
        else:
            return True