1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084 |
- #!/usr/bin/env python
- # -*- coding: utf-8 -*-
- from django.contrib import auth
- from django.core import serializers
- from django.utils.timezone import utc
- import traceback
- from django.views.decorators.csrf import csrf_exempt
- from django.views.generic import TemplateView, View
- from django.utils.decorators import method_decorator
- from django.core.files.storage import FileSystemStorage
- from django.http import HttpResponseRedirect
- from django.contrib.auth.hashers import make_password, check_password # 对密码加密模块
- from Ansjer import settings as api_settings
- from Object.AWS.SesClassObject import SesClassObject
- from Service.TokenManager import JSONTokenManager
- from Model.models import Auth_Captcha, AuthToken_Token, Role
- from Controller.CheckUserData import DataValid, date_handler, RandomStr, TokenLink
- from Service.ModelService import ModelService
- from Service.MiscellService import MiscellService
- from django.utils import timezone
- from Ansjer.config import *
- from Service.ResponseService import *
- from Service.TemplateService import TemplateService
- '''
- http://13.56.215.252:82/account/authcode?userName=123456&language=en&mid=1234
- http://13.56.215.252:82/account/authcode?userEmail=123456&language=en&mid=1234
- http://13.56.215.252:82/account/register?userName=123456&language=en&mid=1234&userEmail=userEmail&userPwd=1234&identifyingCode=1234324
- '''
- class getAuthCodeView(TemplateView):
- @method_decorator(csrf_exempt)
- def dispatch(self, *args, **kwargs):
- return super(getAuthCodeView, self).dispatch(*args, **kwargs)
- def post(self, request, *args, **kwargs):
- request.encoding ='utf-8'
- username = request.POST.get('userName', None)
- useremail = request.POST.get('userEmail', None)
- language = request.POST.get('language', None)
- if not request.session.session_key:
- request.session.create()
- sessionID = request.session.session_key
- mid = request.POST.get('mid', sessionID)
- return self.ValidationError(username,useremail,language,mid)
- def get(self, request, *args, **kwargs):
- request.encoding = 'utf-8'
- username = request.GET.get('userName', None)
- useremail = request.GET.get('userEmail', None)
- language = request.GET.get('language', None)
- if not request.session.session_key:
- request.session.create()
- sessionID = request.session.session_key
- mid = request.GET.get('mid', sessionID)
- return self.ValidationError(username, useremail, language, mid)
- def ValidationError(self, username,useremail,language,mid):
- if username is not None:
- username = username.strip()
- return self.getAuthCode(val=username,type=1,language=language,mid=mid)
- elif useremail is not None:
- useremail = useremail.strip()
- return self.getAuthCode(val=useremail,type=2,language=language,mid=mid)
- else:
- return ResponseJSON(800,language=language)
- def getAuthCode(self, val,type,language,mid):
- dataValid = DataValid()
- validateFlag = False
- if type ==1:
- if dataValid.mobile_validate(val):
- UserVliad = Device_User.objects.filter(username=val)
- if UserVliad:
- return ResponseJSON(101,language=language)
- validateFlag = True
- else:
- return ResponseJSON(107,language=language)
- if type ==2:
- if dataValid.email_validate(val):
- UserVliad = Device_User.objects.filter(userEmail=val)
- if UserVliad:
- return ResponseJSON(103,language=language)
- validateFlag = True
- else:
- return ResponseJSON(108,language=language)
- if validateFlag is True:
- identifyingCode = CommonService.get_redis_data(key=mid+'_identifyingCode')
- getcodename = CommonService.get_redis_data(key=mid + '_getcodename')
- if getcodename is not False:
- getcodename = getcodename.decode("utf-8")
- if identifyingCode is False or getcodename != val:
- identifyingCode = RandomStr(6, True)
- if type == 1:
- CommonService.set_redis_data(key=mid + '_identifyingCode', val=identifyingCode, expire=300)
- return ResponseJSON(0,{'identifyingCode': identifyingCode},language=language)
- elif type == 2:
- if language is None:
- language = 'en'
- send_data = TemplateService.email_message(type='register_code',language=language)
- ses = SesClassObject()
- send_res = ses.send_email(
- # send_address_list=['chanjunkai@163.com'],
- send_address_list=[val],
- subject=send_data['title'],
- # body=send_data['body']+'<br>'+str(identifyingCode),
- body=send_data['body'].replace("{username}",val).replace("{captcha}",str(identifyingCode))
- )
- send_res = True
- if send_res is True:
- CommonService.set_redis_data(key=mid+'_identifyingCode', val=identifyingCode, expire=300)
- CommonService.set_redis_data(key=mid+'_getcodename', val=val, expire=300)
- return ResponseJSON(0, {'identifyingCode': identifyingCode},language=language)
- else:
- return ResponseJSON(44,language=language)
- else:
- identifyingCode = identifyingCode.decode("utf-8")
- if type == 1:
- return ResponseJSON(0, {'identifyingCode': identifyingCode},language=language)
- elif type ==2:
- return ResponseJSON(89, {'identifyingCode': identifyingCode},language=language)
- class registerView(TemplateView):
- @method_decorator(csrf_exempt)
- def dispatch(self, *args, **kwargs):
- return super(registerView, self).dispatch(*args, **kwargs)
- def post(self, request, *args, **kwargs):
- request.encoding = 'utf-8'
- username = request.POST.get('userName', None)
- userEmail = request.POST.get('userEmail', None)
- password = request.POST.get('userPwd', None)
- authCode = request.POST.get('identifyingCode', None)
- language = request.POST.get('language', None)
- if not request.session.session_key:
- request.session.create()
- sessionID = request.session.session_key
- mid = request.POST.get('mid', sessionID)
- return self.ValidationError(username, userEmail, password, authCode,mid,language)
- def get(self, request, *args, **kwargs):
- request.encoding = 'utf-8'
- username = request.GET.get('userName', None)
- userEmail = request.GET.get('userEmail', None)
- password = request.GET.get('userPwd', None)
- authCode = request.GET.get('identifyingCode', None)
- language = request.GET.get('language', None)
- if not request.session.session_key:
- request.session.create()
- sessionID = request.session.session_key
- mid = request.GET.get('mid', sessionID)
- return self.ValidationError(username, userEmail, password, authCode,mid,language)
- def ValidationError(self, username, userEmail, password, authCode,mid,language):
- if username != None and userEmail != None and password != None \
- and authCode != None:
- # 过滤空格
- username = username.strip()
- userEmail = userEmail.strip()
- return self.register(username, userEmail, password, authCode,mid,language)
- else:
- return ResponseJSON(800,language=language)
- def register(self, username, userEmail, password, authCode,mid,language):
- if username is not None and userEmail is not None and password is not None and authCode is not None:
- identifyingCode = CommonService.get_redis_data(key=mid+'_identifyingCode')
- if identifyingCode is False:
- pass
- #validateCode = identifyingCode.decode('utf-8')
- #return SendDataService.getHttpResponseFormalData(code=120)
- #validateCode = identifyingCode.decode('utf-8')
- authCode = 123456
- if authCode == 123456:
- # if authCode == validateCode:
- dataValid = DataValid()
- if dataValid.name_validate(username):
- if dataValid.email_validate(userEmail):
- if dataValid.password_validate(password):
- nameValid = Device_User.objects.filter(username=username)
- emailValid = Device_User.objects.filter(userEmail=userEmail)
- if nameValid:
- return ResponseJSON(101,language=language)
- elif emailValid:
- return ResponseJSON(103,language=language)
- try:
- CaptchaRecord = Auth_Captcha()
- CaptchaRecord.authcaptca = authCode
- CaptchaRecord.username = username
- CaptchaRecord.sendtype = 'register'
- CaptchaRecord.save()
- except Exception as e:
- pass
- CommonService.del_redis_data(key=mid+'_identifyingCode')
- try:
- UserData = Device_User.objects.create_user(username=username, userEmail=userEmail,
- password=password,
- userID=CommonService.getUserID(μs=False,setOTAID=True),
- is_active=True, user_isValid=True)
- except Exception as e:
- errorInfo = traceback.format_exc()
- print('Create User Error: %s' % errorInfo)
- return ResponseJSON(424,'Create User Error:' + repr(e),language=language)
- else:
- return ResponseJSON(0,{
- "user": {
- "userID": UserData.userID,
- "username": UserData.username,
- "userEmail": UserData.userEmail,
- "NickName": UserData.NickName,
- "userIconUrl": str(UserData.userIconUrl),
- "is_superuser": UserData.is_superuser,
- "is_active": UserData.is_active,
- "data_joined": date_handler(UserData.data_joined),
- "last_login": date_handler(UserData.last_login),
- }
- },language=language)
- else:
- return ResponseJSON(109,language=language)
- else:
- return ResponseJSON(108,language=language)
- else:
- return ResponseJSON(107,language=language)
- else:
- return ResponseJSON(121,language=language)
- else:
- return ResponseJSON(444,language=language)
- class LoginView(TemplateView):
- @method_decorator(csrf_exempt) #@csrf_exempt
- def dispatch(self, *args, **kwargs):
- return super(LoginView, self).dispatch(*args, **kwargs)
- # @ratelimit(key='post:userName', rate='5/m')
- def post(self, request, *args, **kwargs):
- request.encoding = 'utf-8'
- request_dict = request.POST
- return self.ValidationError(request, request_dict)
- def get(self, request, *args, **kwargs):
- request.encoding = 'utf-8'
- request_dict = request.GET
- return self.ValidationError(request, request_dict)
- def ValidationError(self, request, request_dict):
- username = request_dict.get('userName', None)
- password = request_dict.get('userPwd', None)
- mCode = request_dict.get('mobileMechanicalCode', None)
- language = request_dict.get('language', 'en')
- if username != None and password != None:
- if mCode != None:
- response = HttpResponse(self.Login(username, password, request=request, mCode=mCode,language=language))
- response.set_cookie('username', username, 7200)
- return response
- else:
- response = HttpResponse(self.Login(username, password, request=request,language=language))
- response.set_cookie('username', username, 7200)
- return response
- else:
- return ResponseJSON(800,language=language)
- def Login(self, username, password, *args, **kwargs):
- request = kwargs.get('request',None)
- language = kwargs.get('language', 'en')
- if not request.session.session_key:
- request.session.create()
- sessionID = request.session.session_key
- user_login_limit = CommonService.get_redis_data(key=sessionID + '_login_limit')
- if user_login_limit is False:
- CommonService.set_redis_data(key=sessionID+ '_login_limit', val=1, expire=10)
- else:
- if int(user_login_limit) >= 3:
- return ResponseFormal(90,language=language)
- dataValid = DataValid()
- if dataValid.mobile_validate(username):
- userValid = Device_User.objects.filter(username = username)
- if userValid:
- if userValid[0].user_isValid and userValid[0].is_active:
- User = auth.authenticate(username = username, password = password)
- if User is not None:
- mCode = kwargs.get('mCode', '')
- if userValid[0].online and mCode != userValid[0].machine_code:
- msg = u'The user previously landed elsewhere, if not my operation, ' \
- u'please modify the user account and password'
- kwargs['login_error'] = msg
- kwargs['User'] = User
- return self.LoginUpdate(userValid, *args, **kwargs)
- else:
- if user_login_limit is not False:
- user_login_limit = int(user_login_limit) + 1
- CommonService.set_redis_data(key=sessionID+'_login_limit', val=user_login_limit,expire=10)
- return ResponseFormal(111,language=language)
- else:
- return ResponseFormal(110,language=language)
- else:
- return ResponseFormal(102,language=language)
- else:
- if dataValid.email_validate(username):
- userValid = Device_User.objects.filter(userEmail = username)
- if userValid:
- if userValid[0].user_isValid and userValid[0].is_active:
- User = auth.authenticate(username = userValid[0].username, password = password)
- if User is not None:
- mCode = kwargs.get('mCode', '')
- if userValid[0].online and mCode != userValid[0].machine_code:
- msg = u'The user previously landed elsewhere, if not my operation, ' \
- u'please modify the user account and password'
- kwargs['login_error'] = msg
- kwargs['User'] = User
- return self.LoginUpdate(userValid, *args, **kwargs)
- else:
- if user_login_limit is not False:
- user_login_limit = int(user_login_limit) + 1
- CommonService.set_redis_data(key=sessionID+ '_login_limit', val=user_login_limit,
- expire=10)
- return ResponseFormal(111,language=language)
- else:
- return ResponseFormal(110,language=language)
- else:
- return ResponseFormal(104,language=language)
- else:
- return ResponseFormal(104,language=language)
- def LoginUpdate(self, userValid, *args, **kwargs):
- mCode = kwargs.get('mCode', '')
- request = kwargs.get('request', None)
- language = kwargs.get('language', 'en')
- userID = userValid[0].userID
- authtoken_Obj = AuthToken_Token.objects.filter(tokenID_id=userID)
- if authtoken_Obj.exists():
- tokenManager = JSONTokenManager()
- access_token = tokenManager.deToken(authtoken_Obj[0].access_token)
- refresh_token = tokenManager.deToken(authtoken_Obj[0].refresh_token)
- else:
- access_token = False
- refresh_token = False
- if access_token is not False and refresh_token is not False:
- resultDict = {'result': {'access_token': access_token, 'refresh_token': refresh_token}, 'error_code': 0}
- else:
- JSON = json.dumps({'userID': userID, 'mCode': mCode}, ensure_ascii=False)
- tokenManager = JSONTokenManager()
- tokenJSON = tokenManager.generate_AToken(JSON=JSON, iCode=tokenManager.getSalt(strLen=8))
- resultDict = json.loads(tokenJSON)
- error_code = resultDict.get('error_code', None)
- if error_code != None and error_code == 0:
- result = resultDict.get('result', None)
- if result != None:
- now_time = datetime.datetime.utcnow().replace(tzinfo=utc).astimezone(utc)
- if language is not None:
- userValid.update(last_login=now_time, online=True, machine_code=mCode,language=language)
- else:
- userValid.update(last_login=now_time, online=True, machine_code=mCode)
- result.update({'userID': userID})
- msg = kwargs.get('login_error', None)
- if msg is not None:
- # result['login_error'] = msg
- # send_code = 10002
- send_code = 0
- else:
- role_dict = ModelService.own_role(userID=userID)
- result['rid']=role_dict['rid']
- result['roleName']=role_dict['roleName']
- result['permList'] = ModelService.own_permission(userID)
- send_code = 0
- User = kwargs.get('User', None)
- User.backend = 'django.contrib.auth.backends.ModelBackend'
- auth.login(request, User)
- return ResponseFormal(send_code,result,language=language)
- else:
- return ResponseFormal(300,language=language)
- else:
- return tokenJSON
- class LogoutView(TemplateView):
- @method_decorator(csrf_exempt)
- def dispatch(self, *args, **kwargs):
- return super(LogoutView, self).dispatch(*args, **kwargs)
- def post(self, request, *args, **kwargs):
- request.encoding = 'utf-8'
- token = request.POST.get('token', None)
- return self.Logout(request, token)
- def get(self, request, *args, **kwargs):
- request.encoding = 'utf-8'
- token = request.GET.get('token', None)
- return self.Logout(request, token)
- def Logout(self, request, token):
- if token != None:
- tM = JSONTokenManager()
- error_code = tM.verify_AToken(token=token)
- if error_code == 0:
- try:
- MiscellService.add_access_log(request=request, status_code=200)
- # MiscellService.access_log(request=request, type=1,response="")
- auth.logout(request)
- # online离线处理
- Device_User.objects.filter(userID= tM.accessDict.get('userID',
- None)).update(online = False)
- # Token过期处理
- AuthToken_Token.objects.filter(tokenID_id = tM.accessDict.get('userID',
- None)).update(iCode = tM.getSalt(8))
- except Exception as e:
- return ResponseJSON(140,repr(e))
- else:
- errorJSON = ResponseFormal(0)
- response = HttpResponse(errorJSON)
- response.delete_cookie('username')
- return response
- else:
- response = HttpResponse(tM.errorCodeInfo(error_code))
- return response
- else:
- return ResponseFormal(800)
- class ChangePwdView(TemplateView):
- @method_decorator(csrf_exempt)
- def dispatch(self, *args, **kwargs):
- return super(ChangePwdView, self).dispatch(*args, **kwargs)
- def post(self, request, *args, **kwargs):
- request.encoding = 'utf-8'
- token = request.POST.get('token', None)
- oldPwd = request.POST.get('oldPwd', None)
- newPwd = request.POST.get('newPwd', None)
- return self.ValidationError(token, oldPwd, newPwd)
- def get(self, request, *args, **kwargs):
- request.encoding = 'gb2312'
- token = request.GET.get('token', None)
- oldPwd = request.GET.get('oldPwd', None)
- newPwd = request.GET.get('newPwd', None)
- return self.ValidationError(token, oldPwd, newPwd)
- def ValidationError(self, token, oldPwd, newPwd):
- if token != None and oldPwd != None and newPwd != None:
- tM = JSONTokenManager()
- error_code = tM.verify_AToken(token)
- if error_code == 0:
- userID = tM.accessDict.get('userID', None)
- if userID:
- response = HttpResponse(self.ChangePwdUpdate(userID, oldPwd, newPwd))
- return response
- else:
- return ResponseJSON(310)
- else:
- response = HttpResponse(tM.errorCodeInfo(error_code))
- return response
- else:
- return ResponseJSON(800)
- def ChangePwdUpdate(self, userID, oldPwd, newPwd):
- userIDValid = Device_User.objects.filter(userID = userID)
- if userIDValid:
- PwdValid = check_password(oldPwd, userIDValid[0].password)
- if PwdValid:
- upCount = userIDValid.update(password = make_password(newPwd))
- if upCount == 1:
- return ResponseFormal(0)
- else:
- return ResponseFormal(112)
- else:
- return ResponseFormal(111)
- else:
- return ResponseFormal(113)
- class ForgetPwdView(TemplateView):
- '''
- 忘记密码
- '''
- @method_decorator(csrf_exempt)
- def dispatch(self, *args, **kwargs):
- return super(ForgetPwdView, self).dispatch(*args, **kwargs)
- def get(self, request, *args, **kwargs):
- request.encoding = 'utf-8'
- userName = request.GET.get('userName', None)
- return self.ValidationError(userName)
- def post(self, request):
- request.encoding = 'utf-8'
- userName = request.POST.get('userName', None)
- return self.ValidationError(userName)
- def ValidationError(self, userName):
- if userName != None:
- response = HttpResponse(self.ForgetPwd(userName))
- return response
- else:
- return ResponseJSON(800)
- def ForgetPwd(self, userName):
- dataValid = DataValid()
- if dataValid.mobile_validate(userName):
- User = Device_User.objects.filter(username = userName)
- if User:
- send_status = MiscellService.ses_send_email_handle(data={
- 'useremail':User[0].userEmail,
- 'username':userName,
- 'type':'forget',
- })
- if send_status == 'has_send':
- return ResponseFormal(1004)
- if send_status is True:
- return ResponseFormal(0)
- else:
- return ResponseFormal(1002)
- else:
- return ResponseFormal(102)
- elif dataValid.email_validate(userName):
- User = Device_User.objects.filter(userEmail = userName)
- if User:
- send_status = MiscellService.ses_send_email_handle(data={
- 'useremail': User[0].userEmail,
- 'username': userName,
- 'type': 'forget',
- })
- if send_status is True:
- return ResponseFormal(0)
- elif send_status == 'has_send':
- return ResponseFormal(1004)
- else:
- return ResponseFormal(1002)
- else:
- return ResponseFormal(104)
- else:
- return ResponseFormal(108)
- class ResetPwdView(TemplateView):
- @method_decorator(csrf_exempt)
- def dispatch(self, *args, **kwargs):
- return super(ResetPwdView, self).dispatch(*args, **kwargs)
- def post(self, request, *args, **kwargs):
- request.encoding = 'utf-8'
- token = request.POST.get('reset_password_token', None)
- reset_pwd_val = request.POST.get('reset_pwd_val', None)
- return self.ValidationError(token,reset_pwd_val)
- def get(self, request, *args, **kwargs):
- request.encoding = 'gb2312'
- token = request.GET.get('reset_password_token', None)
- reset_pwd_val = request.GET.get('reset_pwd_val', None)
- return self.ValidationError(token,reset_pwd_val)
- def ValidationError(self, token, reset_pwd_val):
- tokenConfirm = TokenLink(SECRET_KEY)
- try:
- token = tokenConfirm.confirm_validate_token(token=token)
- except:
- return ResponseJSON(1000)
- else:
- tokenDict = json.loads(token)
- userEmail = tokenDict.get('userEmail', None)
- userPwd = tokenDict.get('password', None)
- redis_key='reset_pwd_key' + str(userEmail)
- redis_val = CommonService.get_redis_data(redis_key)
- if redis_val is False and redis_val != reset_pwd_val:
- return HttpResponse(status=404)
- if userEmail != None and userPwd != None:
- result = self.ResetPwd(userEmail,userPwd)
- if result['code']:
- CommonService.del_redis_data(key=redis_key)
- # return HttpResponseRedirect("/response/success")
- return HttpResponseRedirect("http://www.nsst.com/web/html/paw_update_success.html?code="+userPwd)
- else:
- return HttpResponse(result)
- else:
- return ResponseJSON(1001)
- def ResetPwd(self, userEmail, userPwd):
- dataVaild = DataValid()
- if dataVaild.email_validate(userEmail):
- try:
- UserValid = Device_User.objects.filter(userEmail = userEmail)
- except Exception as e:
- return ResponseFormal(500,repr(e))
- else:
- if UserValid:
- upCount = UserValid.update(password = make_password(userPwd))
- if upCount:
- return {'code':1}
- else:
- return ResponseFormal(106)
- else:
- return ResponseFormal(104)
- else:
- return ResponseFormal(1003)
- 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):
- if token is not None:
- tM = JSONTokenManager()
- error_code = tM.verify_AToken(token)
- if error_code == 0:
- userID = tM.accessDict.get('userID', None)
- if userID:
- response = HttpResponse(self.showUserMore(userID))
- return response
- else:
- return ResponseJSON(310)
- else:
- response = HttpResponse(tM.errorCodeInfo(error_code))
- return response
- else:
- return ResponseJSON(800)
- def showUserMore(self, userID):
- try:
- User = Device_User.objects.filter(userID = userID)
- except Exception as e:
- errorInfo = traceback.format_exc()
- print('Query Database Error: %s' % errorInfo)
- return ResponseFormal(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"]):
- if len(v['fields']['role']):
- roleName=ModelService.getRole(rid=v['fields']['role'][0])
- sqlDict["datas"][k]['fields']['rolename']=roleName
- return ResponseFormal(0, sqlDict)
- else:
- return ResponseFormal(113)
- 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):
- if token != None:
- tM = JSONTokenManager()
- error_code = tM.verify_AToken(token)
- if error_code == 0:
- userID = tM.accessDict.get('userID', None)
- 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('\\', '/')
- '''
- location = 'static/User/Images/' + userID[-11:] + '/'
- userIconPath = '/'.join((api_settings.BASE_DIR, location)).replace('\\', '/')
- if not os.path.exists(userIconPath):
- os.makedirs(userIconPath)
- userIconPath += userIcon.name
- if os.path.exists(userIconPath):
- os.remove(userIconPath)
- destination = open(userIconPath, 'wb+')
- for chunk in userIcon.chunks():
- destination.write(chunk)
- destination.close()
- print(userIconPath)
- '''
- 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 ResponseJSON(802)
- if userIconPath == None and userContent == None:
- return ResponseJSON(800)
- elif userIconPath == None and userContent != None:
- return HttpResponse(self.perfectUserInfoUpdate(userID, userContent=userContent))
- elif userIconPath != None and userContent == None:
- return HttpResponse(self.perfectUserInfoUpdate(userID, userIconPath=userIconPath))
- else:
- return HttpResponse(
- self.perfectUserInfoUpdate(userID, userIconPath=userIconPath, userContent=userContent))
- else:
- return ResponseJSON(310)
- else:
- return HttpResponse(tM.errorCodeInfo(error_code))
- else:
- return ResponseJSON(800)
- def perfectUserInfoUpdate(slef, userID, *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 ResponseFormal(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 = api_settings.SERVER_DOMAIN + '/account/getAvatar/' + userIconPath[7:]
- if userContent != None:
- try:
- UserData = json.loads(userContent)
- except Exception as e:
- return ResponseFormal(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 ResponseFormal(116, repr(e))
- else:
- sqlJSON = serializers.serialize('json', User)
- sqlList = json.loads(sqlJSON)
- print(sqlList, sqlJSON)
- sqlDict = dict(zip(["datas"], [sqlList]))
- return ResponseFormal(0, sqlDict)
- else:
- return ResponseFormal(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):
- if filePath == '' or filePath == None:
- return ResponseJSON(800)
- fullPath = os.path.join(api_settings.BASE_DIR, "static", filePath).replace('\\', '/')
- if os.path.isfile(fullPath):
- try:
- Imagedata = open(fullPath, 'rb').read()
- response = HttpResponse(Imagedata, content_type="image/jpeg")
- return response
- except Exception as e:
- return ResponseJSON(906,repr(e))
- else:
- return ResponseJSON(907)
- def delUser(delUserID, userID):
- try:
- userValid = Device_User.objects.filter(userID = userID)
- except Exception as e:
- errorInfo = traceback.format_exc()
- print('Query Database Error: %s' % errorInfo)
- return ResponseFormal(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 ResponseFormal(500,repr(e))
- else:
- if delUser:
- try:
- delUser.delete()
- except Exception as e:
- errorInfo = traceback.format_exc()
- print('Query database error: %s' % errorInfo)
- return ResponseFormal(114,repr(e))
- else:
- return ResponseFormal(0)
- else:
- return ResponseFormal(115)
- else:
- return ResponseFormal(601)
- else:
- return ResponseFormal(113)
- @csrf_exempt
- def delUserInterface(request, *callback_args,
- **callback_kwargs):
- 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)
- if token is not None and delUserID is not None:
- tM = JSONTokenManager()
- error_code = tM.verify_AToken(token)
- if error_code == 0:
- userID = tM.accessDict.get('userID', None)
- if userID:
- response = HttpResponse(delUser(delUserID, userID))
- return response
- else:
- return ResponseJSON(310)
- else:
- response = HttpResponse(tM.errorCodeInfo(error_code))
- return response
- else:
- return ResponseJSON(800)
- def showAllUserPC(userID,line,page):
- 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 ResponseFormal(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.query_set_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=30)
- now_time = timezone.localtime(timezone.now())
- if now_time > dl_time:
- sqlDict["datas"][k]['fields']['online'] = False
- sqlDict['count'] = device_user_count
- return ResponseFormal(0,sqlDict)
- else:
- return ResponseFormal(600)
- else:
- return ResponseFormal(113)
- def showAllUser(userID):
- 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 ResponseFormal(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 ResponseFormal(0, sqlDict)
- else:
- return ResponseFormal(600)
- else:
- return ResponseFormal(113)
- @csrf_exempt
- def showAllUserInterface(request, *callback_args,
- **callback_kwargs):
- 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)
- if token != None:
- tM = JSONTokenManager()
- error_code = tM.verify_AToken(token)
- if error_code == 0:
- userID = tM.accessDict.get('userID', None)
- if userID:
- if type == 'PC':
- line = int(request.POST.get('line', None))
- page = int(request.POST.get('page', None))
- response = HttpResponse(showAllUserPC(userID, line, page))
- else:
- response = HttpResponse(showAllUser(userID))
- return response
- else:
- return ResponseJSON(310)
- else:
- response = HttpResponse(tM.errorCodeInfo(error_code))
- return response
- else:
- return ResponseJSON(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):
- if token != None and eUserID != None and isValid != None:
- tM = JSONTokenManager()
- error_code = tM.verify_AToken(token=token)
- if error_code == 0:
- superID = tM.accessDict.get('userID', None)
- if superID != None and superID != eUserID:
- if rid == None:
- return HttpResponse(self.UserValidUpdate(superID, eUserID, isValid))
- else:
- return HttpResponse(self.UserValidUpdatePC(superID, eUserID, isValid, rid))
- else:
- return ResponseJSON(150)
- else:
- response = HttpResponse(tM.errorCodeInfo(error_code))
- return response
- else:
- return ResponseJSON(800)
- def UserValidUpdatePC(self, superID, eUserID, isValid, rid):
- 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 ResponseFormal(500,repr(e))
- else:
- if not super:
- return ResponseFormal(615)
- if not eUser:
- return ResponseFormal(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 ResponseFormal(501,repr(e))
- else:
- return ResponseFormal(0)
- else:
- return ResponseFormal(602)
- def UserValidUpdate(self, superID, eUserID, isValid):
- 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 ResponseFormal(500,repr(e))
- if not super:
- return ResponseFormal(113)
- if not eUser:
- return ResponseFormal(113)
- if super[0].is_superuser != 100 or super[0].is_superuser == eUser[0].is_superuser:
- return ResponseFormal(602)
- try:
- updateCount = eUser.update(user_isValid = isValid)
- except Exception as e:
- errorInfo = traceback.format_exc()
- print('更新数据库错误: %s' % errorInfo)
- return ResponseFormal(501,repr(e))
- return ResponseFormal(0)
- @csrf_exempt
- def success(request):
- return ResponseJSON(0)
- class refreshTokenView(View):
- @method_decorator(csrf_exempt)
- def dispatch(self, *args, **kwargs):
- return super(refreshTokenView, self).dispatch(*args, **kwargs)
- def post(self, request, *args, **kwargs):
- request.encoding = 'utf-8'
- request_dict = request.POST
- return self.validation(request_dict=request_dict)
- @csrf_exempt
- def get(self, request, *args, **kwargs):
- request.encoding = 'utf-8'
- request_dict = request.GET
- return self.validation(request_dict=request_dict)
- def validation(self, request_dict, *args, **kwargs):
- refresh_token = request_dict.get('refresh_token', None)
- if refresh_token != None:
- tM = JSONTokenManager()
- result = tM.refresh_AToken(refresh_token)
- response = HttpResponse(result)
- return response
- else:
- return ResponseJSON(800)
|