ASJServer/Controller/AdminManage.py

# -*- coding: utf-8 -*-
from django.views.decorators.csrf import csrf_exempt
from django.views.generic import TemplateView
from django.utils.decorators import method_decorator
from django.contrib.auth.hashers import make_password  # 对密码加密模块
from Service.TokenManager import JSONTokenManager
from Model.models import Device_Info,Role
from Service.ModelService import ModelService
from django.utils import timezone
import datetime
from Model.models import Access_Log
from Service.ResponseService import *
from django.views.decorators.http import require_http_methods
from Ansjer.config import *
'''
http://13.56.215.252:8222/adminManage/manage?operation=getAllDeviceArea&token=test
http://13.56.215.252:8222/adminManage/manage?operation=getAllUserName&token=test
http://13.56.215.252:8222/adminManage/manage?operation=getAllUID&token=test
http://127.0.0.1:8000/adminManage/manage?operation=getAllOnLine&token=stest
http://127.0.0.1:8000/adminManage/manage?operation=getOnLine&token=stest
'''


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

    def get(self, request, *args, **kwargs):
        request.encoding = 'utf-8'
        return self.validation(request_dict=request.GET)

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

    def validation(self, request_dict, *args, **kwargs):

        token = request_dict.get('token', None)
        if token is not None:
            tokenManager = JSONTokenManager()
            error_code = tokenManager.verify_AToken(token)
            if error_code == 0:
                userID = tokenManager.accessDict.get('userID', None)
                operation = request_dict.get('operation', None)
                param_flag = CommonService.get_param_flag(data=[userID, operation])
                if param_flag is True:
                    if operation == 'resetUserPwd':
                        return self.resetUserPwd(request_dict=request_dict, userID=userID)
                    if operation == 'getAllOnLine':
                        return self.getAllOnLine(userID=userID)
                    if operation == 'getOnLine':
                        return self.getOnLine(userID=userID)
                    if operation == 'getAllUserName':
                        return self.getAllUserName(userID=userID)
                    if operation == 'getStatisAccess':
                        return self.getStatisAccess(userID=userID,request_dict=request_dict)
                    if operation == 'getAllUID':
                        return self.getAllUID(userID=userID)
                    if operation == 'getAllDeviceArea':
                        return self.getAllDeviceArea(userID=userID)
                else:
                    return ResponseJSON(444)
            else:
                return HttpResponse(tokenManager.errorCodeInfo(error_code))
        else:
            return ResponseJSON(311)

    def resetUserPwd(self, request_dict, userID):
        own_permission = ModelService.check_permission(userID=userID, permID=50)
        if own_permission is True:
            duserID = request_dict.get('duserID', None)
            userPwd = request_dict.get('userPwd', None)
            param_flag = CommonService.get_param_flag(data=[duserID])
            if param_flag is True:
                UserValid = Device_User.objects.filter(userID=duserID)
                if UserValid:
                    if userPwd is None:
                        userPwd = '123456'
                    is_update = UserValid.update(password=make_password(userPwd))
                    if is_update:
                        return ResponseJSON(0)
                    else:
                        return ResponseJSON(106)
            else:
                return ResponseJSON(444)
        else:
            return ResponseJSON(404)

    def getAllUserName(self, userID):
        # 权限固定为30
        own_permission = ModelService.check_permission(userID=userID, permID=30)
        if own_permission is True:
            username_list = Device_User.objects.all().values_list('username', flat=True)
            if username_list:
                return ResponseJSON(0,{'username_list': list(username_list)})
            else:
                return ResponseJSON(0)
        else:
            return ResponseJSON(404)
    #    获取全部用户的在线个数
    def getAllOnLine(self, userID):
        # 权限固定为30
        own_permission = ModelService.check_permission(userID=userID, permID=30)
        if own_permission is True:
            allonline = Device_User.objects.all().values('online')
            # 两个变量，分别是在线，离线
            onlinenum = 0
            noonlinenum=0
            for q in allonline:
                if q['online'] ==True:
                    onlinenum+=1
                else:
                    noonlinenum+=1
            print("在线人数：")
            print(onlinenum)
            return ResponseJSON(0, {"onlinenum": onlinenum, "no_onlinenum": noonlinenum})
        else:
            return ResponseJSON(404)
    #    获取全部用户的在线人数
    def getOnLine(self, userID):
        # 权限固定为30
        own_permission = ModelService.check_permission(userID=userID, permID=30)
        if own_permission is True:
            online_list = Device_User.objects.all().values('online', 'last_login')
            # 两个变量，分别是在线，离线
            onlinenum=0
            noonlinenum=0
            for q in online_list:
                    # print(q['last_login'] )
                    # 最后访问时间加5分种
                    dl_time = q['last_login'] + datetime.timedelta(minutes=OFF_LINE_TIME_DELTA)
                    # print(dl_time)
                    # 当前时间
                    now_time = timezone.localtime(timezone.now())
                    # print(now_time)
                    # 如果当前时间大于最后访问的时间
                    if now_time < dl_time:
                        onlinenum += 1
                    else:
                        noonlinenum += 1
            print("在线人")
            print(onlinenum)
            return ResponseJSON(0, {"onlinenum": onlinenum,"no_onlinenum": noonlinenum})
        else:
            return ResponseJSON(404)

    # 获取所有设备地区
    def getAllDeviceArea(self, userID):
        own_permission = ModelService.check_permission(userID=userID, permID=30)
        if own_permission is True:
            qs = Device_Info.objects.all().values('area','UID')
            uid_area_dict = {}
            for q in qs:
                if q['UID'] and q['area']:
                    uid_area_dict[q['UID']]=q['area']
            if len(uid_area_dict):
                area_dict = {}

                for k,v in uid_area_dict.items():
                    if v in area_dict:
                        area_dict[v] += 1
                    else:
                        area_dict[v] = 1
                return ResponseJSON(0,{'area':area_dict})
            else:
                return ResponseJSON(0)
        else:
            return ResponseJSON(404)

    '''
    统计一天访问量
    http://192.168.136.45:8077/adminManage/manage?token=test&operation=getStatisAccess&timestamp=1528773308
    '''
    def getStatisAccess(self,userID,request_dict):
        own_permission = ModelService.check_permission(userID=userID, permID=30)
        if own_permission is True:
            time_stamp = int(request_dict.get('timestamp', None))
            times = datetime.datetime.fromtimestamp(time_stamp)
            time_dict = CommonService.getTimeDict(times)
            res = {}
            for k, v in time_dict.items():
                start_date = time_dict[k]
                end_date = time_dict[k] + datetime.timedelta(hours=1)
                count = Access_Log.objects.filter(time__range=(start_date, end_date)).count()
                if count:
                    res[k] = count
                else:
                    res[k] = 0
            return ResponseJSON(0, {'count': res})
        else:
            return ResponseJSON(404)

    def getAllUID(self,userID):
        own_permission = ModelService.check_permission(userID=userID, permID=30)
        if own_permission is True:
            uid_list = Device_Info.objects.all().values_list('UID', flat=True)
            if uid_list:
                return ResponseJSON(0, {'count': len(uid_list),'uid_list':list(uid_list)})
        else:
            return ResponseJSON(404)

@csrf_exempt
@require_http_methods(["GET"])
def getUserIds(request):
    token = request.GET.get('token', None)
    if token is not None:
        tokenManager = JSONTokenManager()
        error_code = tokenManager.verify_AToken(token)
        if error_code == 0:
            userID = tokenManager.accessDict.get('userID', None)
            own_perm = ModelService.check_permission(userID,30)
            if own_perm is True:
                # userID_list = Device_User.objects.all().values_list('userID', flat=True)
                dn = Device_User.objects.all().values('userID', 'username')
                return ResponseJSON(0,{"datas":list(dn)})
            else:
                return ResponseJSON(404)
        else:
            return HttpResponse(tokenManager.errorCodeInfo(error_code))
    else:
        return ResponseJSON(311)

@csrf_exempt
# @require_http_methods(['GET'])
def search_user_by_content(request):
    if request.method == 'GET':
        request_dict = request.GET
    if request.method == 'POST':
        request_dict = request.POST
    token = request_dict.get('token', None)
    page = request_dict.get('page', None)
    line = request_dict.get('line', None)
    content = request_dict.get('content', None)
    rstime = request_dict.get('rstime', None)
    retime = request_dict.get('retime', None)
    if page is not None and line is not None:
        page = int(page)
        line = int(line)
    else:
        return ResponseJSON(10,'page,line is none')
    if token is not None:
        tokenManager = JSONTokenManager()
        error_code = tokenManager.verify_AToken(token)
        if error_code == 0:
            userID = tokenManager.accessDict.get('userID', None)
            own_perm = ModelService.check_permission(userID,30)
            if own_perm is True:
                check_perm = ModelService.check_permission(userID=userID, permID=20)
                if check_perm is True:
                    try:
                        content = json.loads(content)
                        search_kwargs = CommonService.get_kwargs(data=content)
                        queryset = Device_User.objects.filter(**search_kwargs)
                    except Exception as e:
                        return ResponseJSON(444,repr(e))
                    if rstime is not None and rstime != '' and retime is not None and retime != '':
                        startt = datetime.datetime.fromtimestamp(int(rstime))
                        rstime = startt.strftime("%Y-%m-%d %H:%M:%S.%f")
                        endt = datetime.datetime.fromtimestamp(int(retime))
                        retime = endt.strftime("%Y-%m-%d %H:%M:%S.%f")
                        queryset = queryset.filter(data_joined__range=(rstime, retime))
                    elif rstime is not None and rstime != '':
                        startt = datetime.datetime.fromtimestamp(int(rstime))
                        rstime = startt.strftime("%Y-%m-%d %H:%M:%S.%f")
                        queryset = queryset.filter(data_joined__gte=rstime)
                    elif retime is not None and retime != '':
                        endt = datetime.datetime.fromtimestamp(int(retime))
                        retime = endt.strftime("%Y-%m-%d %H:%M:%S.%f")
                        queryset = queryset.filter(data_joined__lte=retime)
                    if queryset.exists():
                        count = queryset.count()
                        res = queryset[(page - 1) * line:page * line]
                        sqlDict = CommonService.query_set_to_dict(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 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'] = count
                        return ResponseJSON(0, sqlDict)
                    return ResponseJSON(0, {'datas': [], 'count': 0})
                else:
                    return ResponseJSON(404)
            else:
                return ResponseJSON(404)
        else:
            return HttpResponse(tokenManager.errorCodeInfo(error_code))
    else:
        return ResponseJSON(311)