ASJServer/Service/ModelService.py

from Model.models import *
from Service.CommonService import CommonService
import json
from django.db.models import Q


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

    @staticmethod
    def get_user_mark(userID):
        if userID:
            qs = Device_User.objects.filter(userID=userID).values('username', 'userEmail', 'phone')
            if qs[0]['username']:
                return qs[0]['username']
            elif qs[0]['userEmail']:
                return qs[0]['userEmail']
            elif qs[0]['phone']:
                return qs[0]['phone']
            else:
                return ''
        else:
            return ''

    # 根据username获取userID
    @staticmethod
    def get_userID_byname(username):
        try:
            device_user = Device_User.objects.get(Q(username=username) | Q(userEmail=username) | Q(phone=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


    # 通过用户名获取userIDLIST
    @staticmethod
    def get_user_list_by_username(username):
        userID_list = Device_User.objects.filter(Q(username=username) | Q(userEmail=username) | Q(phone=username)). \
            values_list('userID', flat=True)
        return userID_list

    @staticmethod
    def del_eq_info(userID, uid):
        ei_qs = Equipment_Info.objects.filter(userID_id=userID, devUid=uid)
        ei_qs.delete()
        # ei_count = ei_qs.count()
        # while (ei_count > 1000):
        #     ei_qs[0:1000].delete()

        # 根据设备主键ID判断是否拥有该设备

    # 获取绑定用户设备列表
    @staticmethod
    def get_uid_list(userID):
        uid_list = Device_Info.objects.filter(userID_id=userID).values_list('UID', flat=True)
        return list(uid_list)