ASJServer/Controller/EquipmentManager.py

from django.views.decorators.csrf import csrf_exempt
import traceback
from Service.ModelService import ModelService
from Model.models import Device_User, Device_Info, Device_Meal, UID_Bucket
from Service.CommonService import CommonService
import simplejson as json
from Object.TokenObject import TokenObject
from Object.ResponseObject import ResponseObject
import re


def addNewUserEquipment(userID, deviceContent, response):
    userIDValid = Device_User.objects.filter(userID=userID)
    if userIDValid:
        try:
            try:
                deviceData = json.loads(deviceContent)
            except Exception as e:
                return response.json(803, repr(e))
            else:
                UID = deviceData.get('UID', None)
                if UID != None:
                    dValid = Device_Info.objects.filter(userID=userID, UID=UID)
                    if dValid:
                        return response.json(174)
                    else:
                        UID = deviceData.get('UID', '')
                        re_uid = re.compile(r'^[A-Za-z0-9]{20}$')
                        if re_uid.match(UID):
                            is_bind = Device_Info.objects.filter(UID=UID, isShare=False)
                            # is_bind = ''
                            # 判断是否有已绑定用户
                            if is_bind:
                                # return response.json(175)
                                # 判断用户密码是否正确
                                dev_user = deviceData.get('View_Account', '')
                                dev_pass = deviceData.get('View_Password', '')
                                if is_bind[0].View_Account == dev_user and is_bind[0].View_Password == dev_pass:
                                    deviceData['NickName'] = is_bind[0].NickName
                                    if 'isShare' in deviceData:
                                        deviceData['isShare'] = True
                                    userDevice = Device_Info(id=CommonService.getUserID(getUser=False),
                                                             primaryUserID=is_bind[0].userID_id, userID_id=userID,
                                                             **deviceData)
                                    userDevice.save()
                                else:
                                    return response.json(111)
                            else:
                                userDevice = Device_Info(id=CommonService.getUserID(getUser=False), userID_id=userID,
                                                         **deviceData)
                                userDevice.save()
                        else:
                            return response.json(180)
                else:
                    return response.json(806)

        except Exception as e:
            errorInfo = traceback.format_exc()
            print('添加设备错误: %s ' % errorInfo)
            return response.json(178, repr(e))
        else:
            sqlDict = CommonService.qs_to_dict([userDevice])
            return response.json(0, sqlDict)
    else:
        return response.json(104)


def delUserEquipment(userID, id, response):
    try:
        deviceValid = Device_Info.objects.filter(userID_id=userID, id=id)
    except Exception as e:
        errorInfo = traceback.format_exc()
        print('查询数据库错误: %s' % errorInfo)
        return response.json(500, repr(e))
    else:
        if deviceValid:
            try:
                Device_Info.objects.filter(userID_id=userID, id=id).delete()
            except Exception as e:
                errorInfo = traceback.format_exc()
                print('删除数据库记录错误: %s' % errorInfo)
                return response.json(176, repr(e))
            else:
                return response.json(0)
        else:
            UserValid = Device_User.objects.filter(userID=userID)
            if UserValid.exists():
                return response.json(172)
            else:
                return response.json(104)


def showAllUserEquipment(userID, response):
    try:
        userValid = Device_User.objects.filter(userID=userID).order_by('-data_joined')
    except Exception as e:
        errorInfo = traceback.format_exc()
        print('查询数据库错误: %s' % errorInfo)
        return response.json(500, repr(e))
    else:
        if not userValid.exists():
            return response.json(104)
        own_perm = ModelService.check_permission(userID=userID, permID=30)
        if own_perm is not True:
            return response.json(404)
        qs = Device_Info.objects.all()
        res = CommonService.qs_to_dict(qs)
        return response.json(0, res)


def showAllUserEquipmentPC(userID, fieldDict, response):
    try:
        user_valid = Device_User.objects.filter(userID=userID).order_by('-data_joined')
    except Exception as e:
        errorInfo = traceback.format_exc()
        print('查询数据库错误: %s' % errorInfo)
        return response.json(500, repr(e))
    else:
        if not user_valid:
            return response.json(104)
        own_permission = ModelService.check_permission(userID=userID, permID=30)
        if not own_permission:
            return response.json(404)
        page = int(fieldDict['page'])
        line = int(fieldDict['line'])
        device_info_query_set = Device_Info.objects.all()
        device_info_count = device_info_query_set.count()
        device_info_res = device_info_query_set[(page - 1) * line:page * line]
        sqlDict = CommonService.qs_to_dict(query_set=device_info_res)
        sqlDict['count'] = device_info_count
        return response.json(0, sqlDict)


def findEquipmentInfo(content, type, fieldDict):
    if type == 1:
        Device_Info_QuerySet = Device_Info.objects.all()
    if type == 2:
        searchCondition = content
        kwargs = CommonService.get_kwargs(data=searchCondition)
        Device_Info_QuerySet = Device_Info.objects.filter(**kwargs)
    page = int(fieldDict['page'])
    line = int(fieldDict['line'])
    device_info_count = Device_Info_QuerySet.count()
    res = Device_Info_QuerySet[(page - 1) * line:page * line]
    send_dict = CommonService.qs_to_dict(query_set=res)
    for k, v in enumerate(send_dict["datas"]):
        for val in res:
            if v['pk'] == val.id:
                username = ModelService.get_user_name(send_dict["datas"][k]['fields']['userID'])
                send_dict["datas"][k]['fields']['username'] = username
                primary = ModelService.get_user_name(send_dict["datas"][k]['fields']['primaryUserID'])
                send_dict["datas"][k]['fields']['primaryusername'] = primary
    send_dict['count'] = device_info_count
    return send_dict


#     查询用户设备
@csrf_exempt
def queryUserEquipmentInterface(request):
    response = ResponseObject()
    if request.method == 'POST':
        request.encoding = 'utf-8'
        request_dict = request.POST
    elif request.method == 'GET':
        request.encoding = 'utf-8'
        request_dict = request.GET
    else:
        return response.json(801)
    token = request_dict.get('token', None)
    tko = TokenObject(token)
    response.lang = tko.lang
    if tko.code != 0:
        return response.json(tko.code)
    userID = tko.userID
    if not userID:
        return response.json(104)
    dvqs = Device_Info.objects.filter(userID_id=userID)
    res = CommonService.qs_to_dict(dvqs)
    datas = res['datas']
    uid_list = []
    pm_master = []
    for q in datas:
        uid_list.append(q['fields']['UID'])
        if q['fields']['UID']:
            pm_master.append(q['fields']['primaryUserID'])
    ubqs = UID_Bucket.objects.filter(uid__in=uid_list). \
        values('bucket__content', 'status', 'channel', 'endTime', 'uid')
    ubql = CommonService.qs_to_list(ubqs)
    res = []
    pm_names = Device_User.objects.filter(userID__in=pm_master).values_list('userID', 'username').distinct()
    pm_dict = {}
    for pm in pm_names:
        pm_dict[pm[0]] = pm[1]
    print(pm_names)
    # 获取设备主用户名字
    for p in datas:
        p['fields']['vod'] = []
        for dm in ubql:
            if p['fields']['UID'] == dm['uid']:
                p['fields']['vod'].append(dm)
        if p['fields']['isShare']:
            p['fields']['primaryMaster'] = ''
        else:
            if p['fields']['primaryUserID'] in pm_dict.keys():
                p['fields']['primaryMaster'] = pm_dict[p['fields']['primaryUserID']]
            else:
                p['fields']['primaryMaster'] = ''
        res.append(p)
    return response.json(0, {'datas': res})


@csrf_exempt
def addNewUserEquipmentInterface(request, *callback_args,
                                 **callback_kwargs):
    response = ResponseObject()
    if request.method == 'POST':
        request.encoding = 'utf-8'
        request_dict = request.POST
    elif request.method == 'GET':
        request.encoding = 'utf-8'
        request_dict = request.GET
    else:
        return response.json(801)
    token = request_dict.get('token', None)
    deviceContent = request_dict.get('content', None)
    if token is not None and deviceContent is not None:
        tko = TokenObject(token)
        response.lang = tko.lang
        if tko.code == 0:
            userID = tko.userID
            if userID is not None:
                return addNewUserEquipment(userID, deviceContent, response)
            else:
                return response.json(309)
        else:
            return response.json(tko.code)
    else:
        return response.json(444, 'token,content')


@csrf_exempt
def delUserEquipmentInterface(request, *callback_args,
                              **callback_kwargs):
    '''
    删除用户设备
    :param request:
    :param callback_args: 表示任何多个无名参数，tuple类型
    :param callback_kwargs: 表示关键字参数，dict类型
    :return:
    '''
    response = ResponseObject()
    if request.method == 'POST':
        request.encoding = 'utf-8'
        request_dict = request.POST
    elif request.method == 'GET':
        request.encoding = 'gb2312'
        request_dict = request.GET
    else:
        return response.json(801)
    token = request_dict.get('token', None)
    id = request_dict.get('id', None)
    if token is not None and id is not None:
        tko = TokenObject(token)
        response.lang = tko.lang
        if tko.code == 0:
            userID = tko.userID
            if userID is not None:
                return delUserEquipment(userID, id, response)
            else:
                return response.json(309)
        else:
            return response.json(tko.code)
    else:
        return response.json(800)


@csrf_exempt
def modifyUserEquipmentInterface(request, *callback_args,
                                 **callback_kwargs):
    '''
    修改用户设备
    :param request:
    :param callback_args:
    :param callback_kwargs:
    :return:
    '''
    response = ResponseObject()
    if request.method == 'POST':
        request.encoding = 'utf-8'
        request_dict = request.POST
    elif request.method == 'GET':
        request.encoding = 'utf-8'
        request_dict = request.GET
    token = request_dict.get('token', None)
    deviceContent = request_dict.get('content', None)
    id = request_dict.get('id', None)
    if not deviceContent or not id:
        return response.json(444, 'content,id')
    tko = TokenObject(token)
    response.lang = tko.lang
    if tko.code != 0:
        return response.json(tko.code)
    userID = tko.userID
    if userID is None:
        return response.json(309)
    try:
        deviceValid = Device_Info.objects.filter(userID_id=userID, id=id)
    except Exception as e:
        return response.json(500, repr(e))
    else:
        if deviceValid.exists():
            deviceData = json.loads(deviceContent)
            try:
                Device_Info.objects.filter(userID_id=userID, id=id).update(**deviceData)
            except Exception as e:
                return response.json(177, repr(e))
            else:
                qs = Device_Info.objects.filter(userID_id=userID, id=id)
                res = CommonService.qs_to_dict(qs)
                return response.json(0, res)
        else:
            UserValid = Device_User.objects.filter(userID=userID)
            if UserValid.exists():
                return response.json(172)
            else:
                return response.json(104)


@csrf_exempt
def showAllUserEquipmentInterface(request, *callback_args, **callback_kwargs):
    request.encoding = 'utf-8'
    if request.method == 'POST':
        fieldDict = request.POST
    if request.method == 'GET':
        fieldDict = request.GET
    token = request.POST.get('token', None)
    type = request.POST.get('type', None)
    response = ResponseObject()
    tko = TokenObject(token)
    response.lang = tko.lang
    if tko.code != 0:
        return response.json(tko.code)
    userID = tko.userID
    if not userID:
        return response.json(104)
    if type == 'PC':
        return showAllUserEquipmentPC(userID, fieldDict, response)
    return showAllUserEquipment(userID, response)


@csrf_exempt
def findEquipmentInfoInterface(request, *callback_args, **callback_kwargs):
    if request.method == 'GET':
        request.encoding = 'gb2312'
        fieldDict = request.GET
    if request.method == 'POST':
        request.encoding = 'utf-8'
        fieldDict = request.POST
    deviceContent = fieldDict.get('content', None)
    token = fieldDict.get('token', None)
    response = ResponseObject()

    if token != None:
        tko = TokenObject(token)
        response.lang = tko.lang
        if tko.code == 0:
            if deviceContent:
                try:
                    deviceContent = json.loads(deviceContent)
                except Exception as e:
                    print(repr(e))
                    return response.json(10, repr(e))
                else:
                    resultDict = findEquipmentInfo(content=deviceContent, type=2, fieldDict=fieldDict)
            else:
                resultDict = findEquipmentInfo(content='', type=1, fieldDict=fieldDict)
            return response.json(0, resultDict)
        else:
            return response.json(tko.code)
    else:
        return response.json(444, 'token')