ASJServer/Controller/UserManger.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
import datetime
import os
import traceback

import simplejson as json
from django.core import serializers
from django.core.files.storage import FileSystemStorage
from django.http import HttpResponse
from django.utils import timezone
from django.utils.decorators import method_decorator
from django.views.decorators.csrf import csrf_exempt
from django.views.generic import TemplateView, View

from Ansjer.config import SERVER_DOMAIN
from Ansjer.settings import BASE_DIR
from Model.models import Role, Device_User
from Object.ResponseObject import ResponseObject
from Object.TokenObject import TokenObject
from Service.CommonService import CommonService
from Service.ModelService import ModelService


# http://192.168.136.40:8077/account/showUserMore?token=test
class showUserMoreView(TemplateView):
    @method_decorator(csrf_exempt)
    def dispatch(self, *args, **kwargs):
        return super(showUserMoreView, self).dispatch(*args, **kwargs)

    def post(self, request, *args, **kwargs):
        request.encoding = 'utf-8'
        token = request.POST.get('token', None)

        return self.ValidationError(token)

    def get(self, request, *args, **kwargs):
        request.encoding = 'gb2312'
        token = request.GET.get('token', None)

        return self.ValidationError(token)

    def ValidationError(self, token):
        response = ResponseObject()
        if token is not None:
            tko = TokenObject(token)
            tko.valid()
            response.lang = tko.lang
            if tko.code == 0:
                userID = tko.userID()
                if userID:
                    return self.showUserMore(userID, response)
                else:
                    return response.json(310)
            else:
                return response.json(tko.code)
        else:
            return response.json(800)

    def showUserMore(self, userID, response):
        try:
            User = Device_User.objects.filter(userID=userID)
        except Exception as e:
            errorInfo = traceback.format_exc()
            print('Query Database Error: %s' % errorInfo)
            return response.json(500, repr(e))
        else:
            if User:
                sqlJSON = serializers.serialize('json', User)
                sqlList = json.loads(sqlJSON)
                sqlDict = dict(zip(["datas"], [sqlList]))
                for k, v in enumerate(sqlDict["datas"]):
                    sqlDict['datas'][k]['fields'].pop('password')
                    if len(v['fields']['role']):
                        roleName = ModelService.getRole(rid=v['fields']['role'][0])
                        sqlDict["datas"][k]['fields']['rolename'] = roleName
                return response.json(0, sqlDict)
            else:
                return response.json(113)


# http://192.168.136.40:8077/account/perfectUserInfo?token=test&content={"NickName":"333"}
class perfectUserInfoView(TemplateView):
    # 完善个人信息
    @method_decorator(csrf_exempt)
    def dispatch(self, *args, **kwargs):
        return super(perfectUserInfoView, self).dispatch(*args, **kwargs)

    def post(self, request, *args, **kwargs):
        request.encoding = 'utf-8'
        userContent = request.POST.get('content', None)
        userIcon = request.FILES.get('userIcon', None)
        token = request.POST.get('token', None)
        return self.ValidationError(token, userContent, userIcon)

    def get(self, request, *args, **kwargs):
        request.encoding = 'gb2312'
        userContent = request.GET.get('content', None)
        userIcon = request.FILES.get('userIcon', None)
        token = request.GET.get('token', None)
        return self.ValidationError(token, userContent, userIcon)

    def ValidationError(self, token, userContent, userIcon):
        response = ResponseObject()
        if token is not None:
            tko = TokenObject(token)
            tko.valid()
            response.lang = tko.lang
            if tko.code == 0:
                userID = tko.userID()
                if userID:
                    if userIcon != None:
                        # location = 'static/User/Images/' + userID[-11:] + '/'
                        location = 'static/User/Images/' + userID + '/'
                        fss = FileSystemStorage(location=location)
                        if fss.exists(userIcon.name):
                            fss.delete(userIcon.name)
                        filename = fss.save(userIcon.name, userIcon)
                        userIconUrl = fss.url(filename)
                        userIconPath = fss.path(filename).replace('\\', '/')
                    else:
                        userIconPath = None

                    if userContent != None:
                        dataValid = json.loads(userContent)
                        print('userID' and 'password' and 'is_superuser' in dataValid.keys())
                        if 'userID' and 'password' and 'is_superuser' in dataValid.keys():
                            return response.json(802)
                    if userIconPath == None and userContent == None:
                        return response.json(800)
                    elif userIconPath == None and userContent != None:
                        return self.perfectUserInfoUpdate(userID, response, userContent=userContent)
                    elif userIconPath != None and userContent == None:
                        return self.perfectUserInfoUpdate(userID, response, userIconPath=userIconPath)
                    else:
                        return self.perfectUserInfoUpdate(userID, response, userIconPath=userIconPath,
                                                          userContent=userContent)
                else:
                    return response.json(310)
            else:
                return response.json(tko.code)
        else:
            return response.json(800)

    def perfectUserInfoUpdate(slef, userID, response, *args,
                              **kwargs):
        """
        :param username:
        :param userContent:
        :param args:
        :param kwargs:
        :return:
        """
        try:
            User = Device_User.objects.filter(userID=userID)
        except Exception as e:
            errorInfo = traceback.format_exc()
            print('查询数据库失败: %s ' % errorInfo)
            return response.json(500, repr(e))
        else:
            if User:
                userIconPath = kwargs.get('userIconPath', None)
                userContent = kwargs.get('userContent', None)
                if userIconPath is not None:
                    userIconPath = userIconPath[userIconPath.find('static/'):]
                    userIconUrl = SERVER_DOMAIN + '/account/getAvatar/' + userIconPath[7:]
                if userContent != None:
                    try:
                        UserData = json.loads(userContent)
                    except Exception as e:
                        return response.json(803, repr(e))
                try:
                    if userIconPath is not None and userContent is not None:
                        User.update(userIconPath=userIconPath, userIconUrl=userIconUrl, **UserData)
                    elif userIconPath is None and userContent is not None:
                        User.update(**UserData)
                    elif userIconPath is not None and userContent is None:
                        User.update(userIconPath=userIconPath, userIconUrl=userIconUrl)
                except Exception as e:
                    errorInfo = traceback.format_exc()
                    print('修改设备信息错误: %s ' % errorInfo)
                    return response.json(116, repr(e))
                else:
                    sqlJSON = serializers.serialize('json', User)
                    sqlList = json.loads(sqlJSON)
                    print(sqlList, sqlJSON)
                    sqlDict = dict(zip(["datas"], [sqlList]))
                    return response.json(0, sqlDict)
            else:
                return response.json(113)


class getAvatarView(TemplateView):
    @method_decorator(csrf_exempt)
    def dispatch(self, *args, **kwargs):
        return super(getAvatarView, self).dispatch(*args, **kwargs)

    def post(self, request, *args, **kwargs):
        request.encoding = 'utf-8'
        filePath = kwargs.get('filePath', '')
        filePath.encode(encoding='utf-8', errors='strict')
        return self.getAvatar(filePath)

    def get(self, request, *args, **kwargs):
        request.encoding = 'gb2312'
        filePath = kwargs.get('filePath', '')
        filePath.encode(encoding='gb2312', errors='strict')
        return self.getAvatar(filePath)

    def getAvatar(self, filePath):
        response = ResponseObject()
        if filePath == '' or filePath == None:
            return response.json(800)
        fullPath = os.path.join(BASE_DIR, "static", filePath).replace('\\', '/')
        if os.path.isfile(fullPath):
            try:
                Imagedata = open(fullPath, 'rb').read()
                return HttpResponse(Imagedata, content_type="image/jpeg")
            except Exception as e:
                return response.json(906, repr(e))
        else:
            return response.json(907)


@csrf_exempt
def delUserInterface(request):
    if request.method == 'POST':
        delUserID = request.POST.get('delUserID', None)
        token = request.POST.get('token', None)
    elif request.method == 'GET':
        delUserID = request.GET.get('delUserID', None)
        token = request.GET.get('token', None)
    response = ResponseObject()
    if token is not None and delUserID is not None:
        tko = TokenObject(token)
        tko.valid()
        response.lang = tko.lang
        if tko.code == 0:
            userID = tko.userID()
            if userID:
                try:
                    userValid = Device_User.objects.filter(userID=userID)
                except Exception as e:
                    errorInfo = traceback.format_exc()
                    print('Query Database Error: %s' % errorInfo)
                    return response.json(500, repr(e))
                else:
                    if userValid:
                        own_perm = ModelService.check_permission(userID=userID, permID=10)
                        if userValid[0].is_superuser == 100 or own_perm is True:
                            try:
                                delUser = Device_User.objects.filter(userID=delUserID)
                            except Exception as e:
                                errorInfo = traceback.format_exc()
                                print('Query database error: %s' % errorInfo)
                                return response.json(500, repr(e))
                            else:
                                if delUser:
                                    try:
                                        delUser.delete()
                                    except Exception as e:
                                        errorInfo = traceback.format_exc()
                                        print('Query database error: %s' % errorInfo)
                                        return response.json(114, repr(e))
                                    else:
                                        return response.json(0)
                                else:
                                    return response.json(115)
                        else:
                            return response.json(601)
                    else:
                        return response.json(113)
            else:
                return response.json(310)
        else:
            return response.json(tko.code)
    else:
        return response.json(800)


def showAllUserPC(userID, line, page, response):
    try:
        userValid = Device_User.objects.filter(userID=userID).order_by('-data_joined')
    except Exception as e:
        errorInfo = traceback.format_exc()
        print('Query Database Error:%s' % errorInfo)
        return response.json(500, repr(e))
    else:
        if userValid:
            # 管理员查询
            own_permission = ModelService.check_permission(userID=userID, permID=30)
            if own_permission is True:
                device_user_queryset = Device_User.objects.all()
                device_user_count = device_user_queryset.count()
                device_user_res = device_user_queryset.order_by('-data_joined')[(page - 1) * line:page * line]
                sqlDict = CommonService.qs_to_dict(device_user_res)
                for k, v in enumerate(sqlDict["datas"]):
                    if len(v['fields']['role']) > 0:
                        role_query_set = Role.objects.get(rid=v['fields']['role'][0])
                        sqlDict["datas"][k]['fields']['role'].append(role_query_set.roleName)
                    for val in device_user_res:
                        if v['pk'] == val.userID:
                            if sqlDict["datas"][k]['fields']['online'] is True:
                                dl_time = val.last_login + datetime.timedelta(minutes=5)
                                now_time = timezone.localtime(timezone.now())
                                if now_time > dl_time:
                                    sqlDict["datas"][k]['fields']['online'] = False
                sqlDict['count'] = device_user_count
                return response.json(0, sqlDict)
            else:
                return response.json(600)
        else:
            return response.json(113)


def showAllUser(userID, response):
    try:
        userValid = Device_User.objects.filter(userID=userID).order_by('-data_joined')
    except Exception as e:
        errorInfo = traceback.format_exc()
        print('Query Database Error:%s' % errorInfo)
        return response.json(500, repr(e))
    else:
        if userValid:
            own_permission = ModelService.check_permission(userID=userID, permID=30)
            if own_permission is True:
                sqlJSON = serializers.serialize('json', Device_User.objects.all().order_by('-data_joined'))
                sqlList = json.loads(sqlJSON)
                print(sqlList, sqlJSON)
                sqlDict = dict(zip(["datas"], [sqlList]))
                return response.json(0, sqlDict)
            else:
                return response.json(600)
        else:
            return response.json(113)


@csrf_exempt
def showAllUserInterface(request):
    if request.method == 'POST':
        token = request.POST.get('token', None)
        type = request.POST.get('type', None)
    elif request.method == 'GET':
        token = request.GET.get('token', None)
        type = request.GET.get('type', None)
    response = ResponseObject()
    if token is not None:
        tko = TokenObject(token)
        tko.valid()
        response.lang = tko.lang
        if tko.code == 0:
            userID = tko.userID()
            if userID:
                if type == 'PC':
                    line = int(request.POST.get('line', None))
                    page = int(request.POST.get('page', None))
                    return showAllUserPC(userID, line, page, response)
                else:
                    return showAllUser(userID, response)
            else:
                return response.json(310)
        else:
            return response.json(tko.code)
    else:
        return response.json(800)


class setUserValidView(View):
    @method_decorator(csrf_exempt)
    def dispatch(self, *args, **kwargs):
        return super(setUserValidView, self).dispatch(*args, **kwargs)

    def post(self, request, *args, **kwargs):
        request.encoding = 'utf-8'

        token = request.POST.get('token', None)
        eUserID = request.POST.get('userID', None)
        isValid = request.POST.get('isValid', None)
        rid = request.POST.get('rid', None)
        return self.setUserValid(token, eUserID, isValid, rid)

    def get(self, request, *args, **kwargs):
        request.encoding = 'gb2312'

        token = request.GET.get('token', None)
        eUserID = request.GET.get('userID', None)
        isValid = request.GET.get('isValid', None)
        rid = request.GET.get('rid', None)

        return self.setUserValid(token, eUserID, isValid, rid)

    def setUserValid(self, token, eUserID, isValid, rid, *args, **kwargs):
        response = ResponseObject()
        if token is not None:
            tko = TokenObject(token)
            tko.valid()
            response.lang = tko.lang
            if tko.code == 0:
                superID = tko.userID()
                if superID != None and superID != eUserID:
                    if rid == None:
                        return self.UserValidUpdate(superID, eUserID, isValid, response)
                    else:
                        return self.UserValidUpdatePC(superID, eUserID, isValid, rid, response)
                else:
                    return response.json(150)
            else:
                return response.json(tko.code)
        else:
            return response.json(800)

    def UserValidUpdatePC(self, superID, eUserID, isValid, rid, response):
        try:
            super = Device_User.objects.filter(userID=superID)
            eUser = Device_User.objects.filter(userID=eUserID)
        except Exception as e:
            errorInfo = traceback.format_exc()
            print('查询数据库错误: %s' % errorInfo)
            return response.json(500, repr(e))
        else:
            if not super:
                return response.json(615)
            if not eUser:
                return response.json(616)
            own_permission = ModelService.check_permission(userID=superID, permID=50)
            if own_permission is True:
                try:
                    eUser.update(user_isValid=isValid)
                    device_user_query_set = Device_User.objects.get(userID=eUserID)
                    if device_user_query_set.role.all():
                        device_user_query_set.role.set([rid])
                    else:
                        role_user_query_set = Role.objects.get(rid=rid)
                        device_user_query_set.role.add(role_user_query_set)
                except Exception as e:
                    errorInfo = traceback.format_exc()
                    print('更新数据库错误:%s' % errorInfo)
                    return response.json(501, repr(e))
                else:
                    return response.json(0)
            else:
                return response.json(602)

    def UserValidUpdate(self, superID, eUserID, isValid, response):
        try:
            super = Device_User.objects.filter(userID=superID)
            eUser = Device_User.objects.filter(userID=eUserID)
        except Exception as e:
            errorInfo = traceback.format_exc()
            print('查询数据库错误: %s' % errorInfo)
            return response.json(500, repr(e))
        if not super:
            return response.json(113)
        if not eUser:
            return response.json(113)
        if super[0].is_superuser != 100 or super[0].is_superuser == eUser[0].is_superuser:
            return response.json(602)
        try:
            updateCount = eUser.update(user_isValid=isValid)
        except Exception as e:
            errorInfo = traceback.format_exc()
            print('更新数据库错误: %s' % errorInfo)
            return response.json(501, repr(e))
        return response.json(0)


@csrf_exempt
def success(request):
    return HttpResponse(status=200)