UserManger.py 48 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093
  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. send_status = MiscellService.ses_send_email_handle(data={
  476. 'useremail':User[0].userEmail,
  477. 'username':userName,
  478. 'type':'forget',
  479. })
  480. if send_status == 'has_send':
  481. return ResponseFormal(1004)
  482. if send_status is True:
  483. return ResponseFormal(0)
  484. else:
  485. return ResponseFormal(1002)
  486. else:
  487. return ResponseFormal(102)
  488. elif dataValid.email_validate(userName):
  489. User = Device_User.objects.filter(userEmail = userName)
  490. if User:
  491. send_status = MiscellService.ses_send_email_handle(data={
  492. 'useremail': User[0].userEmail,
  493. 'username': userName,
  494. 'type': 'forget',
  495. })
  496. if send_status is True:
  497. return ResponseFormal(0)
  498. elif send_status == 'has_send':
  499. return ResponseFormal(1004)
  500. else:
  501. return ResponseFormal(1002)
  502. else:
  503. return ResponseFormal(104)
  504. else:
  505. return ResponseFormal(108)
  506. class ResetPwdView(TemplateView):
  507. @method_decorator(csrf_exempt)
  508. def dispatch(self, *args, **kwargs):
  509. return super(ResetPwdView, self).dispatch(*args, **kwargs)
  510. def post(self, request, *args, **kwargs):
  511. request.encoding = 'utf-8'
  512. token = request.POST.get('reset_password_token', None)
  513. reset_pwd_val = request.POST.get('reset_pwd_val', None)
  514. return self.ValidationError(token,reset_pwd_val)
  515. def get(self, request, *args, **kwargs):
  516. request.encoding = 'gb2312'
  517. token = request.GET.get('reset_password_token', None)
  518. reset_pwd_val = request.GET.get('reset_pwd_val', None)
  519. return self.ValidationError(token,reset_pwd_val)
  520. def ValidationError(self, token, reset_pwd_val):
  521. tokenConfirm = TokenLink(SECRET_KEY)
  522. try:
  523. token = tokenConfirm.confirm_validate_token(token=token)
  524. except:
  525. return ResponseJSON(1000)
  526. else:
  527. tokenDict = json.loads(token)
  528. userEmail = tokenDict.get('userEmail', None)
  529. userPwd = tokenDict.get('password', None)
  530. redis_key='reset_pwd_key' + str(userEmail)
  531. redis_val = CommonService.get_redis_data(redis_key)
  532. if redis_val is False and redis_val != reset_pwd_val:
  533. return HttpResponse(status=404)
  534. if userEmail != None and userPwd != None:
  535. result = self.ResetPwd(userEmail,userPwd)
  536. if result['code']:
  537. CommonService.del_redis_data(key=redis_key)
  538. # return HttpResponseRedirect("/response/success")
  539. return HttpResponseRedirect("http://www.nsst.com/web/html/paw_update_success.html?code="+userPwd)
  540. else:
  541. return HttpResponse(result)
  542. else:
  543. return ResponseJSON(1001)
  544. def ResetPwd(self, userEmail, userPwd):
  545. dataVaild = DataValid()
  546. if dataVaild.email_validate(userEmail):
  547. try:
  548. UserValid = Device_User.objects.filter(userEmail = userEmail)
  549. except Exception as e:
  550. return ResponseFormal(500,repr(e))
  551. else:
  552. if UserValid:
  553. upCount = UserValid.update(password = make_password(userPwd))
  554. if upCount:
  555. return {'code':1}
  556. else:
  557. return ResponseFormal(106)
  558. else:
  559. return ResponseFormal(104)
  560. else:
  561. return ResponseFormal(1003)
  562. class showUserMoreView(TemplateView):
  563. @method_decorator(csrf_exempt)
  564. def dispatch(self, *args, **kwargs):
  565. return super(showUserMoreView, self).dispatch(*args, **kwargs)
  566. def post(self, request, *args, **kwargs):
  567. request.encoding = 'utf-8'
  568. token = request.POST.get('token', None)
  569. return self.ValidationError(token)
  570. def get(self, request, *args, **kwargs):
  571. request.encoding = 'gb2312'
  572. token = request.GET.get('token', None)
  573. return self.ValidationError(token)
  574. def ValidationError(self, token):
  575. if token is not None:
  576. tM = JSONTokenManager()
  577. error_code = tM.verify_AToken(token)
  578. if error_code == 0:
  579. userID = tM.accessDict.get('userID', None)
  580. if userID:
  581. response = HttpResponse(self.showUserMore(userID))
  582. return response
  583. else:
  584. return ResponseJSON(310)
  585. else:
  586. response = HttpResponse(tM.errorCodeInfo(error_code))
  587. return response
  588. else:
  589. return ResponseJSON(800)
  590. def showUserMore(self, userID):
  591. try:
  592. User = Device_User.objects.filter(userID = userID)
  593. except Exception as e:
  594. errorInfo = traceback.format_exc()
  595. print('Query Database Error: %s' % errorInfo)
  596. return ResponseFormal(500,repr(e))
  597. else:
  598. if User:
  599. sqlJSON = serializers.serialize('json', User)
  600. sqlList = json.loads(sqlJSON)
  601. sqlDict = dict(zip(["datas"], [sqlList]))
  602. for k, v in enumerate(sqlDict["datas"]):
  603. if len(v['fields']['role']):
  604. roleName=ModelService.getRole(rid=v['fields']['role'][0])
  605. sqlDict["datas"][k]['fields']['rolename']=roleName
  606. return ResponseFormal(0, sqlDict)
  607. else:
  608. return ResponseFormal(113)
  609. class perfectUserInfoView(TemplateView):
  610. # 完善个人信息
  611. @method_decorator(csrf_exempt)
  612. def dispatch(self, *args, **kwargs):
  613. return super(perfectUserInfoView, self).dispatch(*args, **kwargs)
  614. def post(self, request, *args, **kwargs):
  615. request.encoding = 'utf-8'
  616. userContent = request.POST.get('content', None)
  617. userIcon = request.FILES.get('userIcon', None)
  618. token = request.POST.get('token', None)
  619. return self.ValidationError(token, userContent, userIcon)
  620. def get(self, request, *args, **kwargs):
  621. request.encoding = 'gb2312'
  622. userContent = request.GET.get('content', None)
  623. userIcon = request.FILES.get('userIcon', None)
  624. token = request.GET.get('token', None)
  625. return self.ValidationError(token, userContent, userIcon)
  626. def ValidationError(self, token, userContent, userIcon):
  627. if token != None:
  628. tM = JSONTokenManager()
  629. error_code = tM.verify_AToken(token)
  630. if error_code == 0:
  631. userID = tM.accessDict.get('userID', None)
  632. if userID:
  633. if userIcon != None:
  634. # location = 'static/User/Images/' + userID[-11:] + '/'
  635. location = 'static/User/Images/' + userID + '/'
  636. fss = FileSystemStorage(location = location)
  637. if fss.exists(userIcon.name):
  638. fss.delete(userIcon.name)
  639. filename = fss.save(userIcon.name, userIcon)
  640. userIconUrl = fss.url(filename)
  641. userIconPath = fss.path(filename).replace('\\', '/')
  642. '''
  643. location = 'static/User/Images/' + userID[-11:] + '/'
  644. userIconPath = '/'.join((api_settings.BASE_DIR, location)).replace('\\', '/')
  645. if not os.path.exists(userIconPath):
  646. os.makedirs(userIconPath)
  647. userIconPath += userIcon.name
  648. if os.path.exists(userIconPath):
  649. os.remove(userIconPath)
  650. destination = open(userIconPath, 'wb+')
  651. for chunk in userIcon.chunks():
  652. destination.write(chunk)
  653. destination.close()
  654. print(userIconPath)
  655. '''
  656. else:
  657. userIconPath = None
  658. if userContent != None:
  659. dataValid = json.loads(userContent)
  660. print('userID' and 'password' and 'is_superuser' in dataValid.keys())
  661. if 'userID' and 'password' and 'is_superuser' in dataValid.keys():
  662. return ResponseJSON(802)
  663. if userIconPath == None and userContent == None:
  664. return ResponseJSON(800)
  665. elif userIconPath == None and userContent != None:
  666. return HttpResponse(self.perfectUserInfoUpdate(userID, userContent=userContent))
  667. elif userIconPath != None and userContent == None:
  668. return HttpResponse(self.perfectUserInfoUpdate(userID, userIconPath=userIconPath))
  669. else:
  670. return HttpResponse(
  671. self.perfectUserInfoUpdate(userID, userIconPath=userIconPath, userContent=userContent))
  672. else:
  673. return ResponseJSON(310)
  674. else:
  675. return HttpResponse(tM.errorCodeInfo(error_code))
  676. else:
  677. return ResponseJSON(800)
  678. def perfectUserInfoUpdate(slef, userID, *args,
  679. **kwargs):
  680. """
  681. :param username:
  682. :param userContent:
  683. :param args:
  684. :param kwargs:
  685. :return:
  686. """
  687. try:
  688. User = Device_User.objects.filter(userID = userID)
  689. except Exception as e:
  690. errorInfo = traceback.format_exc()
  691. print('查询数据库失败: %s ' % errorInfo)
  692. return ResponseFormal(500,repr(e))
  693. else:
  694. if User:
  695. userIconPath = kwargs.get('userIconPath', None)
  696. userContent = kwargs.get('userContent', None)
  697. if userIconPath is not None:
  698. userIconPath = userIconPath[userIconPath.find('static/'):]
  699. userIconUrl = api_settings.SERVER_DOMAIN + '/account/getAvatar/' + userIconPath[7:]
  700. if userContent != None:
  701. try:
  702. UserData = json.loads(userContent)
  703. except Exception as e:
  704. return ResponseFormal(803, repr(e))
  705. try:
  706. if userIconPath is not None and userContent is not None:
  707. User.update(userIconPath=userIconPath, userIconUrl = userIconUrl, **UserData)
  708. elif userIconPath is None and userContent is not None:
  709. User.update(**UserData)
  710. elif userIconPath is not None and userContent is None:
  711. User.update(userIconPath=userIconPath, userIconUrl = userIconUrl)
  712. except Exception as e:
  713. errorInfo = traceback.format_exc()
  714. print('修改设备信息错误: %s ' % errorInfo)
  715. return ResponseFormal(116, repr(e))
  716. else:
  717. sqlJSON = serializers.serialize('json', User)
  718. sqlList = json.loads(sqlJSON)
  719. print(sqlList, sqlJSON)
  720. sqlDict = dict(zip(["datas"], [sqlList]))
  721. return ResponseFormal(0, sqlDict)
  722. else:
  723. return ResponseFormal(113)
  724. class getAvatarView(TemplateView):
  725. @method_decorator(csrf_exempt)
  726. def dispatch(self, *args, **kwargs):
  727. return super(getAvatarView, self).dispatch(*args, **kwargs)
  728. def post(self, request, *args, **kwargs):
  729. request.encoding = 'utf-8'
  730. filePath = kwargs.get('filePath', '')
  731. filePath.encode(encoding='utf-8', errors='strict')
  732. return self.getAvatar(filePath)
  733. def get(self, request, *args, **kwargs):
  734. request.encoding = 'gb2312'
  735. filePath = kwargs.get('filePath', '')
  736. filePath.encode(encoding='gb2312', errors='strict')
  737. return self.getAvatar(filePath)
  738. def getAvatar(self, filePath):
  739. if filePath == '' or filePath == None:
  740. return ResponseJSON(800)
  741. fullPath = os.path.join(api_settings.BASE_DIR, "static", filePath).replace('\\', '/')
  742. if os.path.isfile(fullPath):
  743. try:
  744. Imagedata = open(fullPath, 'rb').read()
  745. response = HttpResponse(Imagedata, content_type="image/jpeg")
  746. return response
  747. except Exception as e:
  748. return ResponseJSON(906,repr(e))
  749. else:
  750. return ResponseJSON(907)
  751. def delUser(delUserID, userID):
  752. try:
  753. userValid = Device_User.objects.filter(userID = userID)
  754. except Exception as e:
  755. errorInfo = traceback.format_exc()
  756. print('Query Database Error: %s' % errorInfo)
  757. return ResponseFormal(500,repr(e))
  758. else:
  759. if userValid:
  760. own_perm = ModelService.check_permission(userID=userID,permID=10)
  761. if userValid[0].is_superuser == 100 or own_perm is True:
  762. try:
  763. delUser = Device_User.objects.filter(userID=delUserID)
  764. except Exception as e:
  765. errorInfo = traceback.format_exc()
  766. print('Query database error: %s' % errorInfo)
  767. return ResponseFormal(500,repr(e))
  768. else:
  769. if delUser:
  770. try:
  771. delUser.delete()
  772. except Exception as e:
  773. errorInfo = traceback.format_exc()
  774. print('Query database error: %s' % errorInfo)
  775. return ResponseFormal(114,repr(e))
  776. else:
  777. return ResponseFormal(0)
  778. else:
  779. return ResponseFormal(115)
  780. else:
  781. return ResponseFormal(601)
  782. else:
  783. return ResponseFormal(113)
  784. @csrf_exempt
  785. def delUserInterface(request, *callback_args,
  786. **callback_kwargs):
  787. if request.method == 'POST':
  788. delUserID = request.POST.get('delUserID', None)
  789. token = request.POST.get('token', None)
  790. elif request.method == 'GET':
  791. delUserID = request.GET.get('delUserID', None)
  792. token = request.GET.get('token', None)
  793. if token is not None and delUserID is not None:
  794. tM = JSONTokenManager()
  795. error_code = tM.verify_AToken(token)
  796. if error_code == 0:
  797. userID = tM.accessDict.get('userID', None)
  798. if userID:
  799. response = HttpResponse(delUser(delUserID, userID))
  800. return response
  801. else:
  802. return ResponseJSON(310)
  803. else:
  804. response = HttpResponse(tM.errorCodeInfo(error_code))
  805. return response
  806. else:
  807. return ResponseJSON(800)
  808. def showAllUserPC(userID,line,page):
  809. try:
  810. userValid = Device_User.objects.filter(userID = userID).order_by('-data_joined')
  811. except Exception as e:
  812. errorInfo = traceback.format_exc()
  813. print('Query Database Error:%s' % errorInfo)
  814. return ResponseFormal(500,repr(e))
  815. else:
  816. if userValid:
  817. # 管理员查询
  818. own_permission = ModelService.check_permission(userID=userID,permID=30)
  819. if own_permission is True:
  820. device_user_queryset = Device_User.objects.all()
  821. device_user_count = device_user_queryset.count()
  822. device_user_res = device_user_queryset.order_by('-data_joined')[(page - 1) * line:page * line]
  823. sqlDict = CommonService.query_set_to_dict(device_user_res)
  824. for k, v in enumerate(sqlDict["datas"]):
  825. if len(v['fields']['role']) > 0:
  826. role_query_set = Role.objects.get(rid=v['fields']['role'][0])
  827. sqlDict["datas"][k]['fields']['role'].append(role_query_set.roleName)
  828. for val in device_user_res:
  829. if v['pk'] == val.userID:
  830. if sqlDict["datas"][k]['fields']['online'] is True:
  831. dl_time = val.last_login + datetime.timedelta(minutes=30)
  832. now_time = timezone.localtime(timezone.now())
  833. if now_time > dl_time:
  834. sqlDict["datas"][k]['fields']['online'] = False
  835. sqlDict['count'] = device_user_count
  836. return ResponseFormal(0,sqlDict)
  837. else:
  838. return ResponseFormal(600)
  839. else:
  840. return ResponseFormal(113)
  841. def showAllUser(userID):
  842. try:
  843. userValid = Device_User.objects.filter(userID = userID).order_by('-data_joined')
  844. except Exception as e:
  845. errorInfo = traceback.format_exc()
  846. print('Query Database Error:%s' % errorInfo)
  847. return ResponseFormal(500, repr(e))
  848. else:
  849. if userValid:
  850. own_permission = ModelService.check_permission(userID=userID,permID=30)
  851. if own_permission is True:
  852. sqlJSON = serializers.serialize('json', Device_User.objects.all().order_by('-data_joined'))
  853. sqlList = json.loads(sqlJSON)
  854. print(sqlList, sqlJSON)
  855. sqlDict = dict(zip(["datas"], [sqlList]))
  856. return ResponseFormal(0, sqlDict)
  857. else:
  858. return ResponseFormal(600)
  859. else:
  860. return ResponseFormal(113)
  861. @csrf_exempt
  862. def showAllUserInterface(request, *callback_args,
  863. **callback_kwargs):
  864. if request.method == 'POST':
  865. token = request.POST.get('token', None)
  866. type = request.POST.get('type', None)
  867. elif request.method == 'GET':
  868. token = request.GET.get('token', None)
  869. type = request.GET.get('type', None)
  870. if token != None:
  871. tM = JSONTokenManager()
  872. error_code = tM.verify_AToken(token)
  873. if error_code == 0:
  874. userID = tM.accessDict.get('userID', None)
  875. if userID:
  876. if type == 'PC':
  877. line = int(request.POST.get('line', None))
  878. page = int(request.POST.get('page', None))
  879. response = HttpResponse(showAllUserPC(userID, line, page))
  880. else:
  881. response = HttpResponse(showAllUser(userID))
  882. return response
  883. else:
  884. return ResponseJSON(310)
  885. else:
  886. response = HttpResponse(tM.errorCodeInfo(error_code))
  887. return response
  888. else:
  889. return ResponseJSON(800)
  890. class setUserValidView(View):
  891. @method_decorator(csrf_exempt)
  892. def dispatch(self, *args, **kwargs):
  893. return super(setUserValidView, self).dispatch(*args, **kwargs)
  894. def post(self, request, *args, **kwargs):
  895. request.encoding = 'utf-8'
  896. token = request.POST.get('token', None)
  897. eUserID = request.POST.get('userID', None)
  898. isValid = request.POST.get('isValid', None)
  899. rid = request.POST.get('rid', None)
  900. return self.setUserValid(token, eUserID, isValid, rid)
  901. def get(self, request, *args, **kwargs):
  902. request.encoding = 'gb2312'
  903. token = request.GET.get('token', None)
  904. eUserID = request.GET.get('userID', None)
  905. isValid = request.GET.get('isValid', None)
  906. rid = request.GET.get('rid',None)
  907. return self.setUserValid(token, eUserID, isValid, rid)
  908. def setUserValid(self, token, eUserID, isValid, rid, *args, **kwargs):
  909. if token != None and eUserID != None and isValid != None:
  910. tM = JSONTokenManager()
  911. error_code = tM.verify_AToken(token=token)
  912. if error_code == 0:
  913. superID = tM.accessDict.get('userID', None)
  914. if superID != None and superID != eUserID:
  915. if rid == None:
  916. return HttpResponse(self.UserValidUpdate(superID, eUserID, isValid))
  917. else:
  918. return HttpResponse(self.UserValidUpdatePC(superID, eUserID, isValid, rid))
  919. else:
  920. return ResponseJSON(150)
  921. else:
  922. response = HttpResponse(tM.errorCodeInfo(error_code))
  923. return response
  924. else:
  925. return ResponseJSON(800)
  926. def UserValidUpdatePC(self, superID, eUserID, isValid, rid):
  927. try:
  928. super = Device_User.objects.filter(userID=superID)
  929. eUser = Device_User.objects.filter(userID=eUserID)
  930. except Exception as e:
  931. errorInfo = traceback.format_exc()
  932. print('查询数据库错误: %s' % errorInfo)
  933. return ResponseFormal(500,repr(e))
  934. else:
  935. if not super:
  936. return ResponseFormal(615)
  937. if not eUser:
  938. return ResponseFormal(616)
  939. own_permission = ModelService.check_permission(userID=superID,permID=50)
  940. if own_permission is True:
  941. try:
  942. eUser.update(user_isValid = isValid)
  943. device_user_query_set = Device_User.objects.get(userID=eUserID)
  944. if device_user_query_set.role.all():
  945. device_user_query_set.role.set([rid])
  946. else:
  947. role_user_query_set = Role.objects.get(rid=rid)
  948. device_user_query_set.role.add(role_user_query_set)
  949. except Exception as e:
  950. errorInfo = traceback.format_exc()
  951. print('更新数据库错误:%s'%errorInfo)
  952. return ResponseFormal(501,repr(e))
  953. else:
  954. return ResponseFormal(0)
  955. else:
  956. return ResponseFormal(602)
  957. def UserValidUpdate(self, superID, eUserID, isValid):
  958. try:
  959. super = Device_User.objects.filter(userID = superID)
  960. eUser = Device_User.objects.filter(userID = eUserID)
  961. except Exception as e:
  962. errorInfo = traceback.format_exc()
  963. print('查询数据库错误: %s' % errorInfo)
  964. return ResponseFormal(500,repr(e))
  965. if not super:
  966. return ResponseFormal(113)
  967. if not eUser:
  968. return ResponseFormal(113)
  969. if super[0].is_superuser != 100 or super[0].is_superuser == eUser[0].is_superuser:
  970. return ResponseFormal(602)
  971. try:
  972. updateCount = eUser.update(user_isValid = isValid)
  973. except Exception as e:
  974. errorInfo = traceback.format_exc()
  975. print('更新数据库错误: %s' % errorInfo)
  976. return ResponseFormal(501,repr(e))
  977. return ResponseFormal(0)
  978. @csrf_exempt
  979. def success(request):
  980. return ResponseJSON(0)
  981. class refreshTokenView(View):
  982. @method_decorator(csrf_exempt)
  983. def dispatch(self, *args, **kwargs):
  984. return super(refreshTokenView, self).dispatch(*args, **kwargs)
  985. def post(self, request, *args, **kwargs):
  986. request.encoding = 'utf-8'
  987. request_dict = request.POST
  988. return self.validation(request_dict=request_dict)
  989. @csrf_exempt
  990. def get(self, request, *args, **kwargs):
  991. request.encoding = 'utf-8'
  992. request_dict = request.GET
  993. return self.validation(request_dict=request_dict)
  994. def validation(self, request_dict, *args, **kwargs):
  995. refresh_token = request_dict.get('refresh_token', None)
  996. if refresh_token != None:
  997. tM = JSONTokenManager()
  998. result = tM.refresh_AToken(refresh_token)
  999. response = HttpResponse(result)
  1000. return response
  1001. else:
  1002. return ResponseJSON(800)