UserManger.py 48 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096
  1. #!/usr/bin/env python
  2. # -*- coding: utf-8 -*-
  3. from django.contrib import auth
  4. from django.core import serializers
  5. from django.utils.timezone import utc
  6. import traceback
  7. from django.views.decorators.csrf import csrf_exempt
  8. from django.views.generic import TemplateView, View
  9. from django.utils.decorators import method_decorator
  10. from django.core.files.storage import FileSystemStorage
  11. from django.http import HttpResponseRedirect
  12. from django.contrib.auth.hashers import make_password, check_password # 对密码加密模块
  13. from Ansjer import settings as api_settings
  14. from Object.AWS.SesClassObject import SesClassObject
  15. from Service.TokenManager import JSONTokenManager
  16. from Model.models import Auth_Captcha, AuthToken_Token, Role
  17. from Controller.CheckUserData import DataValid, date_handler, RandomStr, TokenLink
  18. from Service.ModelService import ModelService
  19. from Service.MiscellService import MiscellService
  20. from django.utils import timezone
  21. from Ansjer.config import *
  22. from Service.ResponseService import *
  23. from Service.TemplateService import TemplateService
  24. from ratelimit.decorators import ratelimit
  25. '''
  26. http://13.56.215.252:82/account/authcode?userName=123456&language=en&mid=1234
  27. http://13.56.215.252:82/account/authcode?userEmail=123456&language=en&mid=1234
  28. http://13.56.215.252:82/account/register?userName=123456&language=en&mid=1234&userEmail=userEmail&userPwd=1234&identifyingCode=1234324
  29. '''
  30. class getAuthCodeView(TemplateView):
  31. @method_decorator(csrf_exempt)
  32. def dispatch(self, *args, **kwargs):
  33. return super(getAuthCodeView, self).dispatch(*args, **kwargs)
  34. @ratelimit(key='ip', rate='2/m')
  35. def post(self, request, *args, **kwargs):
  36. was_limited = getattr(request, 'limited', False)
  37. if was_limited is True:
  38. return ResponseJSON(5)
  39. request.encoding ='utf-8'
  40. username = request.POST.get('userName', None)
  41. useremail = request.POST.get('userEmail', None)
  42. language = request.POST.get('language', None)
  43. if not request.session.session_key:
  44. request.session.create()
  45. sessionID = request.session.session_key
  46. mid = request.POST.get('mid', sessionID)
  47. return self.ValidationError(username,useremail,language,mid)
  48. @ratelimit(key='ip', rate='2/m')
  49. def get(self, request, *args, **kwargs):
  50. was_limited = getattr(request, 'limited', False)
  51. if was_limited is True:
  52. return ResponseJSON(5)
  53. request.encoding = 'utf-8'
  54. username = request.GET.get('userName', None)
  55. useremail = request.GET.get('userEmail', None)
  56. language = request.GET.get('language', None)
  57. if not request.session.session_key:
  58. request.session.create()
  59. sessionID = request.session.session_key
  60. mid = request.GET.get('mid', sessionID)
  61. return self.ValidationError(username, useremail, language, mid)
  62. def ValidationError(self, username,useremail,language,mid):
  63. if username is not None:
  64. username = username.strip()
  65. return self.getAuthCode(val=username,type=1,language=language,mid=mid)
  66. elif useremail is not None:
  67. useremail = useremail.strip()
  68. return self.getAuthCode(val=useremail,type=2,language=language,mid=mid)
  69. else:
  70. return ResponseJSON(800,language=language)
  71. def getAuthCode(self, val,type,language,mid):
  72. dataValid = DataValid()
  73. validateFlag = False
  74. if type ==1:
  75. if dataValid.mobile_validate(val):
  76. UserVliad = Device_User.objects.filter(username=val)
  77. if UserVliad:
  78. return ResponseJSON(101,language=language)
  79. validateFlag = True
  80. else:
  81. return ResponseJSON(107,language=language)
  82. if type ==2:
  83. if dataValid.email_validate(val):
  84. UserVliad = Device_User.objects.filter(userEmail=val)
  85. if UserVliad:
  86. return ResponseJSON(103,language=language)
  87. validateFlag = True
  88. else:
  89. return ResponseJSON(108,language=language)
  90. if validateFlag is True:
  91. identifyingCode = CommonService.get_redis_data(key=mid+'_identifyingCode')
  92. if identifyingCode is False:
  93. identifyingCode = RandomStr(6, True)
  94. if type == 1:
  95. CommonService.set_redis_data(key=mid + '_identifyingCode', val=identifyingCode, expire=300)
  96. return ResponseJSON(0,{'identifyingCode': identifyingCode},language=language)
  97. elif type == 2:
  98. if language is None:
  99. language = 'en'
  100. send_data = TemplateService.email_message(type='register_code',language=language)
  101. ses = SesClassObject()
  102. send_res = ses.send_email(
  103. # send_address_list=['chanjunkai@163.com'],
  104. send_address_list=[val],
  105. subject=send_data['title'],
  106. # body=send_data['body']+'<br>'+str(identifyingCode),
  107. body=send_data['body'].replace("{username}",val).replace("{captcha}",str(identifyingCode))
  108. )
  109. send_res = True
  110. if send_res is True:
  111. CommonService.set_redis_data(key=mid+'_identifyingCode', val=identifyingCode, expire=300)
  112. return ResponseJSON(0, {'identifyingCode': identifyingCode},language=language)
  113. else:
  114. return ResponseJSON(44,language=language)
  115. else:
  116. identifyingCode = identifyingCode.decode("utf-8")
  117. if type == 1:
  118. return ResponseJSON(0, {'identifyingCode': identifyingCode},language=language)
  119. elif type ==2:
  120. return ResponseJSON(89, {'identifyingCode': identifyingCode},language=language)
  121. class registerView(TemplateView):
  122. @method_decorator(csrf_exempt)
  123. def dispatch(self, *args, **kwargs):
  124. return super(registerView, self).dispatch(*args, **kwargs)
  125. def post(self, request, *args, **kwargs):
  126. request.encoding = 'utf-8'
  127. username = request.POST.get('userName', None)
  128. userEmail = request.POST.get('userEmail', None)
  129. password = request.POST.get('userPwd', None)
  130. authCode = request.POST.get('identifyingCode', None)
  131. language = request.POST.get('language', None)
  132. if not request.session.session_key:
  133. request.session.create()
  134. sessionID = request.session.session_key
  135. mid = request.POST.get('mid', sessionID)
  136. return self.ValidationError(username, userEmail, password, authCode,mid,language)
  137. def get(self, request, *args, **kwargs):
  138. request.encoding = 'utf-8'
  139. username = request.GET.get('userName', None)
  140. userEmail = request.GET.get('userEmail', None)
  141. password = request.GET.get('userPwd', None)
  142. authCode = request.GET.get('identifyingCode', None)
  143. language = request.GET.get('language', None)
  144. if not request.session.session_key:
  145. request.session.create()
  146. sessionID = request.session.session_key
  147. mid = request.GET.get('mid', sessionID)
  148. return self.ValidationError(username, userEmail, password, authCode,mid,language)
  149. def ValidationError(self, username, userEmail, password, authCode,mid,language):
  150. if username != None and userEmail != None and password != None \
  151. and authCode != None:
  152. # 过滤空格
  153. username = username.strip()
  154. userEmail = userEmail.strip()
  155. return self.register(username, userEmail, password, authCode,mid,language)
  156. else:
  157. return ResponseJSON(800,language=language)
  158. def register(self, username, userEmail, password, authCode,mid,language):
  159. if username is not None and userEmail is not None and password is not None and authCode is not None:
  160. identifyingCode = CommonService.get_redis_data(key=mid+'_identifyingCode')
  161. if identifyingCode is False:
  162. pass
  163. #validateCode = identifyingCode.decode('utf-8')
  164. #return SendDataService.getHttpResponseFormalData(code=120)
  165. #validateCode = identifyingCode.decode('utf-8')
  166. authCode = 123456
  167. if authCode == 123456:
  168. # if authCode == validateCode:
  169. dataValid = DataValid()
  170. if dataValid.name_validate(username):
  171. if dataValid.email_validate(userEmail):
  172. if dataValid.password_validate(password):
  173. nameValid = Device_User.objects.filter(username=username)
  174. emailValid = Device_User.objects.filter(userEmail=userEmail)
  175. if nameValid:
  176. return ResponseJSON(101,language=language)
  177. elif emailValid:
  178. return ResponseJSON(103,language=language)
  179. try:
  180. CaptchaRecord = Auth_Captcha()
  181. CaptchaRecord.authcaptca = authCode
  182. CaptchaRecord.username = username
  183. CaptchaRecord.sendtype = 'register'
  184. CaptchaRecord.save()
  185. except Exception as e:
  186. pass
  187. CommonService.del_redis_data(key=mid+'_identifyingCode')
  188. try:
  189. UserData = Device_User.objects.create_user(username=username, userEmail=userEmail,
  190. password=password,
  191. userID=CommonService.getUserID(μs=False,setOTAID=True),
  192. is_active=True, user_isValid=True)
  193. except Exception as e:
  194. errorInfo = traceback.format_exc()
  195. print('Create User Error: %s' % errorInfo)
  196. return ResponseJSON(424,'Create User Error:' + repr(e),language=language)
  197. else:
  198. return ResponseJSON(0,{
  199. "user": {
  200. "userID": UserData.userID,
  201. "username": UserData.username,
  202. "userEmail": UserData.userEmail,
  203. "NickName": UserData.NickName,
  204. "userIconUrl": str(UserData.userIconUrl),
  205. "is_superuser": UserData.is_superuser,
  206. "is_active": UserData.is_active,
  207. "data_joined": date_handler(UserData.data_joined),
  208. "last_login": date_handler(UserData.last_login),
  209. }
  210. },language=language)
  211. else:
  212. return ResponseJSON(109,language=language)
  213. else:
  214. return ResponseJSON(108,language=language)
  215. else:
  216. return ResponseJSON(107,language=language)
  217. else:
  218. return ResponseJSON(121,language=language)
  219. else:
  220. return ResponseJSON(444,language=language)
  221. class LoginView(TemplateView):
  222. @method_decorator(csrf_exempt) #@csrf_exempt
  223. def dispatch(self, *args, **kwargs):
  224. return super(LoginView, self).dispatch(*args, **kwargs)
  225. # @ratelimit(key='post:userName', rate='5/m')
  226. def post(self, request, *args, **kwargs):
  227. request.encoding = 'utf-8'
  228. request_dict = request.POST
  229. return self.ValidationError(request, request_dict)
  230. def get(self, request, *args, **kwargs):
  231. request.encoding = 'utf-8'
  232. request_dict = request.GET
  233. return self.ValidationError(request, request_dict)
  234. def ValidationError(self, request, request_dict):
  235. username = request_dict.get('userName', None)
  236. password = request_dict.get('userPwd', None)
  237. mCode = request_dict.get('mobileMechanicalCode', None)
  238. language = request_dict.get('language', 'en')
  239. if username is not None:
  240. username = username.strip()
  241. if password is not None:
  242. password = password.strip()
  243. if username != None and password != None:
  244. if mCode != None:
  245. response = HttpResponse(self.Login(username, password, request=request, mCode=mCode,language=language))
  246. response.set_cookie('username', username, 7200)
  247. return response
  248. else:
  249. response = HttpResponse(self.Login(username, password, request=request,language=language))
  250. response.set_cookie('username', username, 7200)
  251. return response
  252. else:
  253. return ResponseJSON(800,language=language)
  254. def Login(self, username, password, *args, **kwargs):
  255. request = kwargs.get('request',None)
  256. language = kwargs.get('language', 'en')
  257. if not request.session.session_key:
  258. request.session.create()
  259. sessionID = request.session.session_key
  260. user_login_limit = CommonService.get_redis_data(key=sessionID + '_login_limit')
  261. if user_login_limit is False:
  262. CommonService.set_redis_data(key=sessionID+ '_login_limit', val=1, expire=10)
  263. else:
  264. if int(user_login_limit) >= 3:
  265. return ResponseFormal(90,language=language)
  266. dataValid = DataValid()
  267. if dataValid.mobile_validate(username):
  268. userValid = Device_User.objects.filter(username = username)
  269. if userValid:
  270. if userValid[0].user_isValid and userValid[0].is_active:
  271. User = auth.authenticate(username = username, password = password)
  272. if User is not None:
  273. mCode = kwargs.get('mCode', '')
  274. if userValid[0].online and mCode != userValid[0].machine_code:
  275. msg = u'The user previously landed elsewhere, if not my operation, ' \
  276. u'please modify the user account and password'
  277. kwargs['login_error'] = msg
  278. kwargs['User'] = User
  279. return self.LoginUpdate(userValid, *args, **kwargs)
  280. else:
  281. if user_login_limit is not False:
  282. user_login_limit = int(user_login_limit) + 1
  283. CommonService.set_redis_data(key=sessionID+'_login_limit', val=user_login_limit,expire=10)
  284. return ResponseFormal(111,language=language)
  285. else:
  286. return ResponseFormal(110,language=language)
  287. else:
  288. return ResponseFormal(102,language=language)
  289. else:
  290. if dataValid.email_validate(username):
  291. userValid = Device_User.objects.filter(userEmail = username)
  292. if userValid:
  293. if userValid[0].user_isValid and userValid[0].is_active:
  294. User = auth.authenticate(username = userValid[0].username, password = password)
  295. if User is not None:
  296. mCode = kwargs.get('mCode', '')
  297. if userValid[0].online and mCode != userValid[0].machine_code:
  298. msg = u'The user previously landed elsewhere, if not my operation, ' \
  299. u'please modify the user account and password'
  300. kwargs['login_error'] = msg
  301. kwargs['User'] = User
  302. return self.LoginUpdate(userValid, *args, **kwargs)
  303. else:
  304. if user_login_limit is not False:
  305. user_login_limit = int(user_login_limit) + 1
  306. CommonService.set_redis_data(key=sessionID+ '_login_limit', val=user_login_limit,
  307. expire=10)
  308. return ResponseFormal(111,language=language)
  309. else:
  310. return ResponseFormal(110,language=language)
  311. else:
  312. return ResponseFormal(104,language=language)
  313. else:
  314. return ResponseFormal(104,language=language)
  315. def LoginUpdate(self, userValid, *args, **kwargs):
  316. mCode = kwargs.get('mCode', '')
  317. request = kwargs.get('request', None)
  318. language = kwargs.get('language', 'en')
  319. userID = userValid[0].userID
  320. authtoken_Obj = AuthToken_Token.objects.filter(tokenID_id=userID)
  321. if authtoken_Obj.exists():
  322. tokenManager = JSONTokenManager()
  323. access_token = tokenManager.deToken(authtoken_Obj[0].access_token)
  324. refresh_token = tokenManager.deToken(authtoken_Obj[0].refresh_token)
  325. else:
  326. access_token = False
  327. refresh_token = False
  328. if access_token is not False and refresh_token is not False:
  329. resultDict = {'result': {'access_token': access_token, 'refresh_token': refresh_token}, 'error_code': 0}
  330. else:
  331. JSON = json.dumps({'userID': userID, 'mCode': mCode}, ensure_ascii=False)
  332. tokenManager = JSONTokenManager()
  333. tokenJSON = tokenManager.generate_AToken(JSON=JSON, iCode=tokenManager.getSalt(strLen=8))
  334. resultDict = json.loads(tokenJSON)
  335. error_code = resultDict.get('error_code', None)
  336. if error_code != None and error_code == 0:
  337. result = resultDict.get('result', None)
  338. if result != None:
  339. now_time = datetime.datetime.utcnow().replace(tzinfo=utc).astimezone(utc)
  340. if language is not None:
  341. userValid.update(last_login=now_time, online=True, machine_code=mCode,language=language)
  342. else:
  343. userValid.update(last_login=now_time, online=True, machine_code=mCode)
  344. result.update({'userID': userID})
  345. msg = kwargs.get('login_error', None)
  346. if msg is not None:
  347. # result['login_error'] = msg
  348. # send_code = 10002
  349. send_code = 0
  350. else:
  351. role_dict = ModelService.own_role(userID=userID)
  352. result['rid']=role_dict['rid']
  353. result['roleName']=role_dict['roleName']
  354. result['permList'] = ModelService.own_permission(userID)
  355. send_code = 0
  356. User = kwargs.get('User', None)
  357. User.backend = 'django.contrib.auth.backends.ModelBackend'
  358. auth.login(request, User)
  359. return ResponseFormal(send_code,result,language=language)
  360. else:
  361. return ResponseFormal(300,language=language)
  362. else:
  363. return tokenJSON
  364. class LogoutView(TemplateView):
  365. @method_decorator(csrf_exempt)
  366. def dispatch(self, *args, **kwargs):
  367. return super(LogoutView, self).dispatch(*args, **kwargs)
  368. def post(self, request, *args, **kwargs):
  369. request.encoding = 'utf-8'
  370. token = request.POST.get('token', None)
  371. return self.Logout(request, token)
  372. def get(self, request, *args, **kwargs):
  373. request.encoding = 'utf-8'
  374. token = request.GET.get('token', None)
  375. return self.Logout(request, token)
  376. def Logout(self, request, token):
  377. if token != None:
  378. tM = JSONTokenManager()
  379. error_code = tM.verify_AToken(token=token)
  380. if error_code == 0:
  381. try:
  382. MiscellService.add_access_log(request=request, status_code=200)
  383. # MiscellService.access_log(request=request, type=1,response="")
  384. auth.logout(request)
  385. # online离线处理
  386. Device_User.objects.filter(userID= tM.accessDict.get('userID',
  387. None)).update(online = False)
  388. # Token过期处理
  389. AuthToken_Token.objects.filter(tokenID_id = tM.accessDict.get('userID',
  390. None)).update(iCode = tM.getSalt(8))
  391. except Exception as e:
  392. return ResponseJSON(140,repr(e))
  393. else:
  394. errorJSON = ResponseFormal(0)
  395. response = HttpResponse(errorJSON)
  396. response.delete_cookie('username')
  397. return response
  398. else:
  399. response = HttpResponse(tM.errorCodeInfo(error_code))
  400. return response
  401. else:
  402. return ResponseFormal(800)
  403. class ChangePwdView(TemplateView):
  404. @method_decorator(csrf_exempt)
  405. def dispatch(self, *args, **kwargs):
  406. return super(ChangePwdView, self).dispatch(*args, **kwargs)
  407. def post(self, request, *args, **kwargs):
  408. request.encoding = 'utf-8'
  409. token = request.POST.get('token', None)
  410. oldPwd = request.POST.get('oldPwd', None)
  411. newPwd = request.POST.get('newPwd', None)
  412. return self.ValidationError(token, oldPwd, newPwd)
  413. def get(self, request, *args, **kwargs):
  414. request.encoding = 'gb2312'
  415. token = request.GET.get('token', None)
  416. oldPwd = request.GET.get('oldPwd', None)
  417. newPwd = request.GET.get('newPwd', None)
  418. return self.ValidationError(token, oldPwd, newPwd)
  419. def ValidationError(self, token, oldPwd, newPwd):
  420. if token != None and oldPwd != None and newPwd != None:
  421. tM = JSONTokenManager()
  422. error_code = tM.verify_AToken(token)
  423. if error_code == 0:
  424. userID = tM.accessDict.get('userID', None)
  425. if userID:
  426. response = HttpResponse(self.ChangePwdUpdate(userID, oldPwd, newPwd))
  427. return response
  428. else:
  429. return ResponseJSON(310)
  430. else:
  431. response = HttpResponse(tM.errorCodeInfo(error_code))
  432. return response
  433. else:
  434. return ResponseJSON(800)
  435. def ChangePwdUpdate(self, userID, oldPwd, newPwd):
  436. userIDValid = Device_User.objects.filter(userID = userID)
  437. if userIDValid:
  438. PwdValid = check_password(oldPwd, userIDValid[0].password)
  439. if PwdValid:
  440. upCount = userIDValid.update(password = make_password(newPwd))
  441. if upCount == 1:
  442. return ResponseFormal(0)
  443. else:
  444. return ResponseFormal(112)
  445. else:
  446. return ResponseFormal(111)
  447. else:
  448. return ResponseFormal(113)
  449. class ForgetPwdView(TemplateView):
  450. '''
  451. 忘记密码
  452. '''
  453. @method_decorator(csrf_exempt)
  454. def dispatch(self, *args, **kwargs):
  455. return super(ForgetPwdView, self).dispatch(*args, **kwargs)
  456. def get(self, request, *args, **kwargs):
  457. request.encoding = 'utf-8'
  458. userName = request.GET.get('userName', None)
  459. return self.ValidationError(userName)
  460. def post(self, request):
  461. request.encoding = 'utf-8'
  462. userName = request.POST.get('userName', None)
  463. return self.ValidationError(userName)
  464. def ValidationError(self, userName):
  465. if userName != None:
  466. response = HttpResponse(self.ForgetPwd(userName))
  467. return response
  468. else:
  469. return ResponseJSON(800)
  470. def ForgetPwd(self, userName):
  471. dataValid = DataValid()
  472. if dataValid.mobile_validate(userName):
  473. User = Device_User.objects.filter(username= userName)
  474. if User:
  475. try:
  476. send_status = MiscellService.ses_send_email_handle(data={
  477. 'useremail': User[0].userEmail,
  478. 'username': userName,
  479. 'type': 'forget',
  480. })
  481. except Exception as e:
  482. return ResponseJSON(10,repr(e))
  483. if send_status == 'has_send':
  484. return ResponseFormal(1004)
  485. if send_status is True:
  486. return ResponseFormal(0)
  487. else:
  488. return ResponseFormal(1002)
  489. else:
  490. return ResponseFormal(102)
  491. elif dataValid.email_validate(userName):
  492. User = Device_User.objects.filter(userEmail = userName)
  493. if User:
  494. send_status = MiscellService.ses_send_email_handle(data={
  495. 'useremail': User[0].userEmail,
  496. 'username': userName,
  497. 'type': 'forget',
  498. })
  499. if send_status is True:
  500. return ResponseFormal(0)
  501. elif send_status == 'has_send':
  502. return ResponseFormal(1004)
  503. else:
  504. return ResponseFormal(1002)
  505. else:
  506. return ResponseFormal(104)
  507. else:
  508. return ResponseFormal(108)
  509. class ResetPwdView(TemplateView):
  510. @method_decorator(csrf_exempt)
  511. def dispatch(self, *args, **kwargs):
  512. return super(ResetPwdView, self).dispatch(*args, **kwargs)
  513. def post(self, request, *args, **kwargs):
  514. request.encoding = 'utf-8'
  515. token = request.POST.get('reset_password_token', None)
  516. reset_pwd_val = request.POST.get('reset_pwd_val', None)
  517. return self.ValidationError(token,reset_pwd_val)
  518. def get(self, request, *args, **kwargs):
  519. request.encoding = 'gb2312'
  520. token = request.GET.get('reset_password_token', None)
  521. reset_pwd_val = request.GET.get('reset_pwd_val', None)
  522. return self.ValidationError(token,reset_pwd_val)
  523. def ValidationError(self, token, reset_pwd_val):
  524. tokenConfirm = TokenLink(SECRET_KEY)
  525. try:
  526. token = tokenConfirm.confirm_validate_token(token=token)
  527. except:
  528. return ResponseJSON(1000)
  529. else:
  530. tokenDict = json.loads(token)
  531. userEmail = tokenDict.get('userEmail', None)
  532. userPwd = tokenDict.get('password', None)
  533. redis_key='reset_pwd_key' + str(userEmail)
  534. redis_val = CommonService.get_redis_data(redis_key)
  535. if redis_val is False and redis_val != reset_pwd_val:
  536. return HttpResponse(status=404)
  537. if userEmail != None and userPwd != None:
  538. result = self.ResetPwd(userEmail,userPwd)
  539. if result['code']:
  540. CommonService.del_redis_data(key=redis_key)
  541. # return HttpResponseRedirect("/response/success")
  542. return HttpResponseRedirect("http://www.nsst.com/web/html/paw_update_success.html?code="+userPwd)
  543. else:
  544. return HttpResponse(result)
  545. else:
  546. return ResponseJSON(1001)
  547. def ResetPwd(self, userEmail, userPwd):
  548. dataVaild = DataValid()
  549. if dataVaild.email_validate(userEmail):
  550. try:
  551. UserValid = Device_User.objects.filter(userEmail = userEmail)
  552. except Exception as e:
  553. return ResponseFormal(500,repr(e))
  554. else:
  555. if UserValid:
  556. upCount = UserValid.update(password = make_password(userPwd))
  557. if upCount:
  558. return {'code':1}
  559. else:
  560. return ResponseFormal(106)
  561. else:
  562. return ResponseFormal(104)
  563. else:
  564. return ResponseFormal(1003)
  565. class showUserMoreView(TemplateView):
  566. @method_decorator(csrf_exempt)
  567. def dispatch(self, *args, **kwargs):
  568. return super(showUserMoreView, self).dispatch(*args, **kwargs)
  569. def post(self, request, *args, **kwargs):
  570. request.encoding = 'utf-8'
  571. token = request.POST.get('token', None)
  572. return self.ValidationError(token)
  573. def get(self, request, *args, **kwargs):
  574. request.encoding = 'gb2312'
  575. token = request.GET.get('token', None)
  576. return self.ValidationError(token)
  577. def ValidationError(self, token):
  578. if token is not None:
  579. tM = JSONTokenManager()
  580. error_code = tM.verify_AToken(token)
  581. if error_code == 0:
  582. userID = tM.accessDict.get('userID', None)
  583. if userID:
  584. response = HttpResponse(self.showUserMore(userID))
  585. return response
  586. else:
  587. return ResponseJSON(310)
  588. else:
  589. response = HttpResponse(tM.errorCodeInfo(error_code))
  590. return response
  591. else:
  592. return ResponseJSON(800)
  593. def showUserMore(self, userID):
  594. try:
  595. User = Device_User.objects.filter(userID = userID)
  596. except Exception as e:
  597. errorInfo = traceback.format_exc()
  598. print('Query Database Error: %s' % errorInfo)
  599. return ResponseFormal(500,repr(e))
  600. else:
  601. if User:
  602. sqlJSON = serializers.serialize('json', User)
  603. sqlList = json.loads(sqlJSON)
  604. sqlDict = dict(zip(["datas"], [sqlList]))
  605. for k, v in enumerate(sqlDict["datas"]):
  606. if len(v['fields']['role']):
  607. roleName=ModelService.getRole(rid=v['fields']['role'][0])
  608. sqlDict["datas"][k]['fields']['rolename']=roleName
  609. return ResponseFormal(0, sqlDict)
  610. else:
  611. return ResponseFormal(113)
  612. class perfectUserInfoView(TemplateView):
  613. # 完善个人信息
  614. @method_decorator(csrf_exempt)
  615. def dispatch(self, *args, **kwargs):
  616. return super(perfectUserInfoView, self).dispatch(*args, **kwargs)
  617. def post(self, request, *args, **kwargs):
  618. request.encoding = 'utf-8'
  619. userContent = request.POST.get('content', None)
  620. userIcon = request.FILES.get('userIcon', None)
  621. token = request.POST.get('token', None)
  622. return self.ValidationError(token, userContent, userIcon)
  623. def get(self, request, *args, **kwargs):
  624. request.encoding = 'gb2312'
  625. userContent = request.GET.get('content', None)
  626. userIcon = request.FILES.get('userIcon', None)
  627. token = request.GET.get('token', None)
  628. return self.ValidationError(token, userContent, userIcon)
  629. def ValidationError(self, token, userContent, userIcon):
  630. if token != None:
  631. tM = JSONTokenManager()
  632. error_code = tM.verify_AToken(token)
  633. if error_code == 0:
  634. userID = tM.accessDict.get('userID', None)
  635. if userID:
  636. if userIcon != None:
  637. # location = 'static/User/Images/' + userID[-11:] + '/'
  638. location = 'static/User/Images/' + userID + '/'
  639. fss = FileSystemStorage(location = location)
  640. if fss.exists(userIcon.name):
  641. fss.delete(userIcon.name)
  642. filename = fss.save(userIcon.name, userIcon)
  643. userIconUrl = fss.url(filename)
  644. userIconPath = fss.path(filename).replace('\\', '/')
  645. '''
  646. location = 'static/User/Images/' + userID[-11:] + '/'
  647. userIconPath = '/'.join((api_settings.BASE_DIR, location)).replace('\\', '/')
  648. if not os.path.exists(userIconPath):
  649. os.makedirs(userIconPath)
  650. userIconPath += userIcon.name
  651. if os.path.exists(userIconPath):
  652. os.remove(userIconPath)
  653. destination = open(userIconPath, 'wb+')
  654. for chunk in userIcon.chunks():
  655. destination.write(chunk)
  656. destination.close()
  657. print(userIconPath)
  658. '''
  659. else:
  660. userIconPath = None
  661. if userContent != None:
  662. dataValid = json.loads(userContent)
  663. print('userID' and 'password' and 'is_superuser' in dataValid.keys())
  664. if 'userID' and 'password' and 'is_superuser' in dataValid.keys():
  665. return ResponseJSON(802)
  666. if userIconPath == None and userContent == None:
  667. return ResponseJSON(800)
  668. elif userIconPath == None and userContent != None:
  669. return HttpResponse(self.perfectUserInfoUpdate(userID, userContent=userContent))
  670. elif userIconPath != None and userContent == None:
  671. return HttpResponse(self.perfectUserInfoUpdate(userID, userIconPath=userIconPath))
  672. else:
  673. return HttpResponse(
  674. self.perfectUserInfoUpdate(userID, userIconPath=userIconPath, userContent=userContent))
  675. else:
  676. return ResponseJSON(310)
  677. else:
  678. return HttpResponse(tM.errorCodeInfo(error_code))
  679. else:
  680. return ResponseJSON(800)
  681. def perfectUserInfoUpdate(slef, userID, *args,
  682. **kwargs):
  683. """
  684. :param username:
  685. :param userContent:
  686. :param args:
  687. :param kwargs:
  688. :return:
  689. """
  690. try:
  691. User = Device_User.objects.filter(userID = userID)
  692. except Exception as e:
  693. errorInfo = traceback.format_exc()
  694. print('查询数据库失败: %s ' % errorInfo)
  695. return ResponseFormal(500,repr(e))
  696. else:
  697. if User:
  698. userIconPath = kwargs.get('userIconPath', None)
  699. userContent = kwargs.get('userContent', None)
  700. if userIconPath is not None:
  701. userIconPath = userIconPath[userIconPath.find('static/'):]
  702. userIconUrl = api_settings.SERVER_DOMAIN + '/account/getAvatar/' + userIconPath[7:]
  703. if userContent != None:
  704. try:
  705. UserData = json.loads(userContent)
  706. except Exception as e:
  707. return ResponseFormal(803, repr(e))
  708. try:
  709. if userIconPath is not None and userContent is not None:
  710. User.update(userIconPath=userIconPath, userIconUrl = userIconUrl, **UserData)
  711. elif userIconPath is None and userContent is not None:
  712. User.update(**UserData)
  713. elif userIconPath is not None and userContent is None:
  714. User.update(userIconPath=userIconPath, userIconUrl = userIconUrl)
  715. except Exception as e:
  716. errorInfo = traceback.format_exc()
  717. print('修改设备信息错误: %s ' % errorInfo)
  718. return ResponseFormal(116, repr(e))
  719. else:
  720. sqlJSON = serializers.serialize('json', User)
  721. sqlList = json.loads(sqlJSON)
  722. print(sqlList, sqlJSON)
  723. sqlDict = dict(zip(["datas"], [sqlList]))
  724. return ResponseFormal(0, sqlDict)
  725. else:
  726. return ResponseFormal(113)
  727. class getAvatarView(TemplateView):
  728. @method_decorator(csrf_exempt)
  729. def dispatch(self, *args, **kwargs):
  730. return super(getAvatarView, self).dispatch(*args, **kwargs)
  731. def post(self, request, *args, **kwargs):
  732. request.encoding = 'utf-8'
  733. filePath = kwargs.get('filePath', '')
  734. filePath.encode(encoding='utf-8', errors='strict')
  735. return self.getAvatar(filePath)
  736. def get(self, request, *args, **kwargs):
  737. request.encoding = 'gb2312'
  738. filePath = kwargs.get('filePath', '')
  739. filePath.encode(encoding='gb2312', errors='strict')
  740. return self.getAvatar(filePath)
  741. def getAvatar(self, filePath):
  742. if filePath == '' or filePath == None:
  743. return ResponseJSON(800)
  744. fullPath = os.path.join(api_settings.BASE_DIR, "static", filePath).replace('\\', '/')
  745. if os.path.isfile(fullPath):
  746. try:
  747. Imagedata = open(fullPath, 'rb').read()
  748. response = HttpResponse(Imagedata, content_type="image/jpeg")
  749. return response
  750. except Exception as e:
  751. return ResponseJSON(906,repr(e))
  752. else:
  753. return ResponseJSON(907)
  754. def delUser(delUserID, userID):
  755. try:
  756. userValid = Device_User.objects.filter(userID = userID)
  757. except Exception as e:
  758. errorInfo = traceback.format_exc()
  759. print('Query Database Error: %s' % errorInfo)
  760. return ResponseFormal(500,repr(e))
  761. else:
  762. if userValid:
  763. own_perm = ModelService.check_permission(userID=userID,permID=10)
  764. if userValid[0].is_superuser == 100 or own_perm is True:
  765. try:
  766. delUser = Device_User.objects.filter(userID=delUserID)
  767. except Exception as e:
  768. errorInfo = traceback.format_exc()
  769. print('Query database error: %s' % errorInfo)
  770. return ResponseFormal(500,repr(e))
  771. else:
  772. if delUser:
  773. try:
  774. delUser.delete()
  775. except Exception as e:
  776. errorInfo = traceback.format_exc()
  777. print('Query database error: %s' % errorInfo)
  778. return ResponseFormal(114,repr(e))
  779. else:
  780. return ResponseFormal(0)
  781. else:
  782. return ResponseFormal(115)
  783. else:
  784. return ResponseFormal(601)
  785. else:
  786. return ResponseFormal(113)
  787. @csrf_exempt
  788. def delUserInterface(request, *callback_args,
  789. **callback_kwargs):
  790. if request.method == 'POST':
  791. delUserID = request.POST.get('delUserID', None)
  792. token = request.POST.get('token', None)
  793. elif request.method == 'GET':
  794. delUserID = request.GET.get('delUserID', None)
  795. token = request.GET.get('token', None)
  796. if token is not None and delUserID is not None:
  797. tM = JSONTokenManager()
  798. error_code = tM.verify_AToken(token)
  799. if error_code == 0:
  800. userID = tM.accessDict.get('userID', None)
  801. if userID:
  802. response = HttpResponse(delUser(delUserID, userID))
  803. return response
  804. else:
  805. return ResponseJSON(310)
  806. else:
  807. response = HttpResponse(tM.errorCodeInfo(error_code))
  808. return response
  809. else:
  810. return ResponseJSON(800)
  811. def showAllUserPC(userID,line,page):
  812. try:
  813. userValid = Device_User.objects.filter(userID = userID).order_by('-data_joined')
  814. except Exception as e:
  815. errorInfo = traceback.format_exc()
  816. print('Query Database Error:%s' % errorInfo)
  817. return ResponseFormal(500,repr(e))
  818. else:
  819. if userValid:
  820. # 管理员查询
  821. own_permission = ModelService.check_permission(userID=userID,permID=30)
  822. if own_permission is True:
  823. device_user_queryset = Device_User.objects.all()
  824. device_user_count = device_user_queryset.count()
  825. device_user_res = device_user_queryset.order_by('-data_joined')[(page - 1) * line:page * line]
  826. sqlDict = CommonService.query_set_to_dict(device_user_res)
  827. for k, v in enumerate(sqlDict["datas"]):
  828. if len(v['fields']['role']) > 0:
  829. role_query_set = Role.objects.get(rid=v['fields']['role'][0])
  830. sqlDict["datas"][k]['fields']['role'].append(role_query_set.roleName)
  831. for val in device_user_res:
  832. if v['pk'] == val.userID:
  833. if sqlDict["datas"][k]['fields']['online'] is True:
  834. dl_time = val.last_login + datetime.timedelta(minutes=5)
  835. now_time = timezone.localtime(timezone.now())
  836. if now_time > dl_time:
  837. sqlDict["datas"][k]['fields']['online'] = False
  838. sqlDict['count'] = device_user_count
  839. return ResponseFormal(0,sqlDict)
  840. else:
  841. return ResponseFormal(600)
  842. else:
  843. return ResponseFormal(113)
  844. def showAllUser(userID):
  845. try:
  846. userValid = Device_User.objects.filter(userID = userID).order_by('-data_joined')
  847. except Exception as e:
  848. errorInfo = traceback.format_exc()
  849. print('Query Database Error:%s' % errorInfo)
  850. return ResponseFormal(500, repr(e))
  851. else:
  852. if userValid:
  853. own_permission = ModelService.check_permission(userID=userID,permID=30)
  854. if own_permission is True:
  855. sqlJSON = serializers.serialize('json', Device_User.objects.all().order_by('-data_joined'))
  856. sqlList = json.loads(sqlJSON)
  857. print(sqlList, sqlJSON)
  858. sqlDict = dict(zip(["datas"], [sqlList]))
  859. return ResponseFormal(0, sqlDict)
  860. else:
  861. return ResponseFormal(600)
  862. else:
  863. return ResponseFormal(113)
  864. @csrf_exempt
  865. def showAllUserInterface(request, *callback_args,
  866. **callback_kwargs):
  867. if request.method == 'POST':
  868. token = request.POST.get('token', None)
  869. type = request.POST.get('type', None)
  870. elif request.method == 'GET':
  871. token = request.GET.get('token', None)
  872. type = request.GET.get('type', None)
  873. if token != None:
  874. tM = JSONTokenManager()
  875. error_code = tM.verify_AToken(token)
  876. if error_code == 0:
  877. userID = tM.accessDict.get('userID', None)
  878. if userID:
  879. if type == 'PC':
  880. line = int(request.POST.get('line', None))
  881. page = int(request.POST.get('page', None))
  882. response = HttpResponse(showAllUserPC(userID, line, page))
  883. else:
  884. response = HttpResponse(showAllUser(userID))
  885. return response
  886. else:
  887. return ResponseJSON(310)
  888. else:
  889. response = HttpResponse(tM.errorCodeInfo(error_code))
  890. return response
  891. else:
  892. return ResponseJSON(800)
  893. class setUserValidView(View):
  894. @method_decorator(csrf_exempt)
  895. def dispatch(self, *args, **kwargs):
  896. return super(setUserValidView, self).dispatch(*args, **kwargs)
  897. def post(self, request, *args, **kwargs):
  898. request.encoding = 'utf-8'
  899. token = request.POST.get('token', None)
  900. eUserID = request.POST.get('userID', None)
  901. isValid = request.POST.get('isValid', None)
  902. rid = request.POST.get('rid', None)
  903. return self.setUserValid(token, eUserID, isValid, rid)
  904. def get(self, request, *args, **kwargs):
  905. request.encoding = 'gb2312'
  906. token = request.GET.get('token', None)
  907. eUserID = request.GET.get('userID', None)
  908. isValid = request.GET.get('isValid', None)
  909. rid = request.GET.get('rid',None)
  910. return self.setUserValid(token, eUserID, isValid, rid)
  911. def setUserValid(self, token, eUserID, isValid, rid, *args, **kwargs):
  912. if token != None and eUserID != None and isValid != None:
  913. tM = JSONTokenManager()
  914. error_code = tM.verify_AToken(token=token)
  915. if error_code == 0:
  916. superID = tM.accessDict.get('userID', None)
  917. if superID != None and superID != eUserID:
  918. if rid == None:
  919. return HttpResponse(self.UserValidUpdate(superID, eUserID, isValid))
  920. else:
  921. return HttpResponse(self.UserValidUpdatePC(superID, eUserID, isValid, rid))
  922. else:
  923. return ResponseJSON(150)
  924. else:
  925. response = HttpResponse(tM.errorCodeInfo(error_code))
  926. return response
  927. else:
  928. return ResponseJSON(800)
  929. def UserValidUpdatePC(self, superID, eUserID, isValid, rid):
  930. try:
  931. super = Device_User.objects.filter(userID=superID)
  932. eUser = Device_User.objects.filter(userID=eUserID)
  933. except Exception as e:
  934. errorInfo = traceback.format_exc()
  935. print('查询数据库错误: %s' % errorInfo)
  936. return ResponseFormal(500,repr(e))
  937. else:
  938. if not super:
  939. return ResponseFormal(615)
  940. if not eUser:
  941. return ResponseFormal(616)
  942. own_permission = ModelService.check_permission(userID=superID,permID=50)
  943. if own_permission is True:
  944. try:
  945. eUser.update(user_isValid = isValid)
  946. device_user_query_set = Device_User.objects.get(userID=eUserID)
  947. if device_user_query_set.role.all():
  948. device_user_query_set.role.set([rid])
  949. else:
  950. role_user_query_set = Role.objects.get(rid=rid)
  951. device_user_query_set.role.add(role_user_query_set)
  952. except Exception as e:
  953. errorInfo = traceback.format_exc()
  954. print('更新数据库错误:%s'%errorInfo)
  955. return ResponseFormal(501,repr(e))
  956. else:
  957. return ResponseFormal(0)
  958. else:
  959. return ResponseFormal(602)
  960. def UserValidUpdate(self, superID, eUserID, isValid):
  961. try:
  962. super = Device_User.objects.filter(userID = superID)
  963. eUser = Device_User.objects.filter(userID = eUserID)
  964. except Exception as e:
  965. errorInfo = traceback.format_exc()
  966. print('查询数据库错误: %s' % errorInfo)
  967. return ResponseFormal(500,repr(e))
  968. if not super:
  969. return ResponseFormal(113)
  970. if not eUser:
  971. return ResponseFormal(113)
  972. if super[0].is_superuser != 100 or super[0].is_superuser == eUser[0].is_superuser:
  973. return ResponseFormal(602)
  974. try:
  975. updateCount = eUser.update(user_isValid = isValid)
  976. except Exception as e:
  977. errorInfo = traceback.format_exc()
  978. print('更新数据库错误: %s' % errorInfo)
  979. return ResponseFormal(501,repr(e))
  980. return ResponseFormal(0)
  981. @csrf_exempt
  982. def success(request):
  983. return ResponseJSON(0)
  984. class refreshTokenView(View):
  985. @method_decorator(csrf_exempt)
  986. def dispatch(self, *args, **kwargs):
  987. return super(refreshTokenView, self).dispatch(*args, **kwargs)
  988. def post(self, request, *args, **kwargs):
  989. request.encoding = 'utf-8'
  990. request_dict = request.POST
  991. return self.validation(request_dict=request_dict)
  992. @csrf_exempt
  993. def get(self, request, *args, **kwargs):
  994. request.encoding = 'utf-8'
  995. request_dict = request.GET
  996. return self.validation(request_dict=request_dict)
  997. def validation(self, request_dict, *args, **kwargs):
  998. refresh_token = request_dict.get('refresh_token', None)
  999. if refresh_token != None:
  1000. tM = JSONTokenManager()
  1001. result = tM.refresh_AToken(refresh_token)
  1002. response = HttpResponse(result)
  1003. return response
  1004. else:
  1005. return ResponseJSON(800)