UserController.py 57 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375
  1. #!/usr/bin/env python3
  2. # -*- coding: utf-8 -*-
  3. """
  4. @Copyright (C) ansjer cop Video Technology Co.,Ltd.All rights reserved.
  5. @AUTHOR: ASJRD018
  6. @NAME: AnsjerFormal
  7. @software: PyCharm
  8. @DATE: 2018/9/11 15:08
  9. @Version: python3.6
  10. @MODIFY DECORD:ansjer dev
  11. @file: UserController.py
  12. @Contact: chanjunkai@163.com
  13. """
  14. import datetime
  15. import traceback
  16. import time
  17. import simplejson as json
  18. from django.contrib.auth.hashers import make_password, check_password # 对密码加密模块
  19. from django.db.models import Q
  20. from django.http import HttpResponseRedirect
  21. from django.utils.decorators import method_decorator
  22. from django.utils.timezone import utc
  23. from django.views.decorators.csrf import csrf_exempt
  24. from django.views.generic import TemplateView
  25. from ratelimit.decorators import ratelimit
  26. from Ansjer.config import AuthCode_Expire, SERVER_DOMAIN, APNS_CONFIG, JPUSH_CONFIG, FCM_CONFIG
  27. from Controller.CheckUserData import DataValid, date_handler, RandomStr
  28. from Model.models import Device_User, Role, SysMsgModel, UidPushModel, UidSetModel, UserExtendModel
  29. from Object.AWS.SesClassObject import SesClassObject
  30. from Object.AliSmsObject import AliSmsObject
  31. from Object.RedisObject import RedisObject
  32. from Object.ResponseObject import ResponseObject
  33. from Object.TokenObject import TokenObject
  34. from Service.CommonService import CommonService
  35. from Service.ModelService import ModelService
  36. from Service.TemplateService import TemplateService
  37. from django.views.generic import View
  38. # 获取验证码
  39. class authCodeView(TemplateView):
  40. @method_decorator(csrf_exempt)
  41. def dispatch(self, *args, **kwargs):
  42. return super(authCodeView, self).dispatch(*args, **kwargs)
  43. @ratelimit(key='ip', rate='2/m')
  44. def post(self, request, *args, **kwargs):
  45. request.encoding = 'utf-8'
  46. lang = request.POST.get('language', None)
  47. response = ResponseObject(lang)
  48. was_limited = getattr(request, 'limited', False)
  49. if was_limited is True:
  50. return response.json(5)
  51. username = request.POST.get('userName', None)
  52. useremail = request.POST.get('userEmail', None)
  53. return self.ValidationError(username, useremail, response)
  54. # @ratelimit(key='ip', rate='2/m')
  55. def get(self, request, *args, **kwargs):
  56. # Device_User.objects.filter(userEmail='chanjunkai@163.com').delete()
  57. request.encoding = 'utf-8'
  58. lang = request.GET.get('language', None)
  59. response = ResponseObject(lang)
  60. was_limited = getattr(request, 'limited', False)
  61. if was_limited is True:
  62. return response.json(5)
  63. username = request.GET.get('userName', None)
  64. email = request.GET.get('userEmail', None)
  65. return self.ValidationError(username, email, response)
  66. def ValidationError(self, username, email, response):
  67. if username:
  68. username = username.strip()
  69. return self.phoneCode(username, response)
  70. elif email:
  71. email = email.strip()
  72. return self.emailCode(email, response)
  73. else:
  74. return response.json(800)
  75. def phoneCode(self, phone, response):
  76. dataValid = DataValid()
  77. if dataValid.mobile_validate(phone):
  78. reds = RedisObject()
  79. identifyingCode = reds.get_data(key=phone + '_identifyingCode')
  80. if identifyingCode is False:
  81. user_qs = Device_User.objects.filter(username=phone)
  82. if user_qs.exists():
  83. return response.json(101)
  84. else:
  85. identifyingCode = RandomStr(6, True)
  86. if reds.set_data(key=phone + '_identifyingCode', val=identifyingCode, expire=600):
  87. return response.json(0, {'identifyingCode': identifyingCode})
  88. else:
  89. return response.json(10, '生成缓存系统错误')
  90. else:
  91. return response.json(0, {'identifyingCode': identifyingCode})
  92. else:
  93. return response.json(107)
  94. def emailCode(self, email, response):
  95. dataValid = DataValid()
  96. if dataValid.email_validate(email):
  97. reds = RedisObject()
  98. identifyingCode = reds.get_data(key=email + '_identifyingCode')
  99. if identifyingCode is False:
  100. user_qs = Device_User.objects.filter(username=email)
  101. email_qs = Device_User.objects.filter(userEmail=email)
  102. if user_qs.exists():
  103. return response.json(103)
  104. elif email_qs.exists():
  105. return response.json(103)
  106. else:
  107. identifyingCode = RandomStr(6, True)
  108. if reds.set_data(key=email + '_identifyingCode', val=identifyingCode, expire=AuthCode_Expire):
  109. send_data = TemplateService.email_message(type='register_code', language=response.lang)
  110. ses = SesClassObject()
  111. send_res = ses.send_email(
  112. send_address_list=[email],
  113. subject=send_data['title'],
  114. body=send_data['body'].replace("{username}", email).replace("{captcha}",
  115. str(identifyingCode))
  116. )
  117. if send_res:
  118. return response.json(0, {'identifyingCode': identifyingCode})
  119. else:
  120. reds.del_data(key=email + '_identifyingCode')
  121. return response.json(44)
  122. else:
  123. return response.json(10, '生成缓存系统错误')
  124. else:
  125. return response.json(89, {'identifyingCode': identifyingCode})
  126. else:
  127. return response.json(107)
  128. # 验证码注册
  129. class registerView(TemplateView):
  130. @method_decorator(csrf_exempt)
  131. def dispatch(self, *args, **kwargs):
  132. return super(registerView, self).dispatch(*args, **kwargs)
  133. def post(self, request, *args, **kwargs):
  134. request.encoding = 'utf-8'
  135. request_dict = request.POST
  136. return self.validates(request_dict)
  137. def get(self, request, *args, **kwargs):
  138. request.encoding = 'utf-8'
  139. request_dict = request.GET
  140. return self.validates(request_dict)
  141. def validates(self, request_dict):
  142. username = request_dict.get('userName', None)
  143. userEmail = request_dict.get('userEmail', None)
  144. password = request_dict.get('userPwd', None)
  145. authCode = request_dict.get('identifyingCode', None)
  146. language = request_dict.get('language', None)
  147. response = ResponseObject(language)
  148. if username and password and authCode:
  149. # 过滤空格
  150. username = username.strip()
  151. if userEmail:
  152. userEmail = userEmail.strip()
  153. return self.register(username, userEmail, password, authCode, response)
  154. else:
  155. return response.json(800)
  156. def register(self, username, userEmail, password, authCode, response):
  157. dataValid = DataValid()
  158. reds = RedisObject()
  159. identifyingCode = reds.get_data(key=username + '_identifyingCode')
  160. if identifyingCode is False:
  161. if userEmail:
  162. identifyingCode = reds.get_data(key=userEmail + '_identifyingCode')
  163. if identifyingCode is False:
  164. return response.json(120)
  165. else:
  166. username = userEmail
  167. else:
  168. return response.json(120)
  169. if authCode != identifyingCode:
  170. return response.json(121)
  171. if dataValid.password_validate(password):
  172. if dataValid.email_validate(username):
  173. if userEmail:
  174. print(userEmail)
  175. emailValid = Device_User.objects.filter(userEmail=userEmail)
  176. if emailValid.exists():
  177. return response.json(103)
  178. else:
  179. userEmail = username
  180. if username:
  181. nameValid = Device_User.objects.filter(username=username)
  182. if nameValid.exists():
  183. return response.json(101)
  184. try:
  185. create_data = {
  186. "username": username,
  187. "NickName": username,
  188. "userEmail": userEmail,
  189. "password": make_password(password),
  190. "userID": CommonService.getUserID(μs=False, setOTAID=True),
  191. "is_active": True,
  192. "user_isValid": True,
  193. }
  194. users = Device_User.objects.create(**create_data)
  195. except Exception as e:
  196. errorInfo = traceback.format_exc()
  197. print(errorInfo)
  198. return response.json(424, repr(e))
  199. else:
  200. if reds.del_data(key=username + '_identifyingCode'):
  201. return response.json(0, {
  202. "user": {
  203. "userID": users.userID,
  204. "username": users.username,
  205. "userEmail": users.userEmail,
  206. "NickName": users.NickName,
  207. "userIconUrl": str(users.userIconUrl),
  208. "is_superuser": users.is_superuser,
  209. "is_active": users.is_active,
  210. "data_joined": date_handler(users.data_joined),
  211. "last_login": date_handler(users.last_login),
  212. }
  213. })
  214. else:
  215. return response.json(10, '删除缓存验证码错误')
  216. elif dataValid.mobile_validate(username):
  217. nameValid = Device_User.objects.filter(username=username)
  218. if nameValid:
  219. return response.json(101)
  220. try:
  221. create_data = {
  222. "username": username,
  223. "NickName": username,
  224. "userEmail": userEmail,
  225. "password": make_password(password),
  226. "userID": CommonService.getUserID(μs=False, setOTAID=True),
  227. "is_active": True,
  228. "user_isValid": True,
  229. }
  230. users = Device_User.objects.create(**create_data)
  231. except Exception as e:
  232. errorInfo = traceback.format_exc()
  233. print(errorInfo)
  234. return response.json(424, repr(e))
  235. else:
  236. if reds.del_data(key=username + '_identifyingCode'):
  237. return response.json(0, {
  238. "user": {
  239. "userID": users.userID,
  240. "username": users.username,
  241. "userEmail": users.userEmail,
  242. "NickName": users.NickName,
  243. "userIconUrl": str(users.userIconUrl),
  244. "is_superuser": users.is_superuser,
  245. "is_active": users.is_active,
  246. "data_joined": date_handler(users.data_joined),
  247. "last_login": date_handler(users.last_login),
  248. }
  249. })
  250. else:
  251. return response.json(10, '删除缓存验证码错误')
  252. else:
  253. return response.json(107)
  254. else:
  255. return response.json(109)
  256. # 登出
  257. class LogoutView(TemplateView):
  258. @method_decorator(csrf_exempt)
  259. def dispatch(self, *args, **kwargs):
  260. return super(LogoutView, self).dispatch(*args, **kwargs)
  261. def post(self, request, *args, **kwargs):
  262. request.encoding = 'utf-8'
  263. request_dict = request.POST
  264. return self.Logout(request_dict)
  265. def get(self, request, *args, **kwargs):
  266. request.encoding = 'utf-8'
  267. request_dict = request.GET
  268. return self.Logout(request_dict)
  269. def Logout(self, request_dict):
  270. response = ResponseObject()
  271. token = request_dict.get('token')
  272. tko = TokenObject(token)
  273. if tko.code == 0:
  274. Device_User.objects.filter(userID=tko.userID).update(online=False)
  275. redisObj = RedisObject(db=3)
  276. redisObj.del_data(key=tko.userID)
  277. m_code = request_dict.get('m_code', None)
  278. if m_code:
  279. userID = tko.userID
  280. try:
  281. UidPushModel.objects.filter(userID_id=userID, m_code=m_code).delete()
  282. except Exception as e:
  283. pass
  284. else:
  285. pass
  286. return response.json(0)
  287. else:
  288. return response.json(tko.code)
  289. # 修改密码
  290. class ChangePwdView(TemplateView):
  291. @method_decorator(csrf_exempt)
  292. def dispatch(self, *args, **kwargs):
  293. return super(ChangePwdView, self).dispatch(*args, **kwargs)
  294. def post(self, request, *args, **kwargs):
  295. request.encoding = 'utf-8'
  296. request_dict = request.POST
  297. return self.validates(request_dict)
  298. def get(self, request, *args, **kwargs):
  299. request.encoding = 'utf-8'
  300. request_dict = request.GET
  301. return self.validates(request_dict)
  302. def validates(self, request_dict):
  303. token = request_dict.get('token', None)
  304. oldPwd = request_dict.get('oldPwd', None)
  305. newPwd = request_dict.get('newPwd', None)
  306. response = ResponseObject()
  307. if oldPwd is None and newPwd is None:
  308. return response.json(800)
  309. tko = TokenObject(token)
  310. response.lang = tko.lang
  311. if tko.code != 0:
  312. return response.json(tko.code)
  313. return self.updatePwd(tko.userID, oldPwd, newPwd, response)
  314. def updatePwd(self, userID, oldPwd, newPwd, response):
  315. user_qs = Device_User.objects.filter(userID=userID)
  316. if not user_qs.exists():
  317. return response.json(104)
  318. c_p = check_password(oldPwd, user_qs[0].password)
  319. # 密码是否正确
  320. if not c_p:
  321. return response.json(111)
  322. update = user_qs.update(password=make_password(newPwd))
  323. if update:
  324. return response.json(0)
  325. else:
  326. return response.json(177)
  327. class ForgetPwdView(TemplateView):
  328. '''
  329. 忘记密码
  330. '''
  331. @method_decorator(csrf_exempt)
  332. def dispatch(self, *args, **kwargs):
  333. return super(ForgetPwdView, self).dispatch(*args, **kwargs)
  334. @ratelimit(key='ip', rate='1/m')
  335. def get(self, request, *args, **kwargs):
  336. request.encoding = 'utf-8'
  337. response = ResponseObject()
  338. was_limited = getattr(request, 'limited', False)
  339. if was_limited is True:
  340. return response.json(5)
  341. userName = request.GET.get('userName', None)
  342. return self.ValidationError(userName, response)
  343. @ratelimit(key='ip', rate='1/m')
  344. def post(self, request):
  345. request.encoding = 'utf-8'
  346. userName = request.POST.get('userName', None)
  347. response = ResponseObject()
  348. was_limited = getattr(request, 'limited', False)
  349. if was_limited is True:
  350. return response.json(5)
  351. return self.ValidationError(userName, response)
  352. def ValidationError(self, userName, response):
  353. # return response.json(475)
  354. if userName != None:
  355. userName = userName.strip()
  356. return self.ForgetPwd(userName, response)
  357. else:
  358. return response.json(800)
  359. def ForgetPwd(self, userName, response):
  360. dataValid = DataValid()
  361. if dataValid.mobile_validate(userName):
  362. User = Device_User.objects.filter(username=userName)
  363. elif dataValid.email_validate(userName):
  364. User = Device_User.objects.filter(username=userName)
  365. if not User.exists():
  366. User = Device_User.objects.filter(userEmail=userName)
  367. else:
  368. return response.json(104)
  369. if User:
  370. email = User[0].userEmail
  371. userID = User[0].userID
  372. if email:
  373. redisObj = RedisObject()
  374. reset_pwd = redisObj.get_data(key=userID + '_email_reset_pwd')
  375. if reset_pwd is False:
  376. tko = TokenObject()
  377. rest = tko.generate(data={'userID': userID, 'user': userName})
  378. token = rest['access_token']
  379. reset_pwd = CommonService.RandomStr(6)
  380. send_data = TemplateService.email_message(type='forget', language='en')
  381. reset_link = '{server_host}account/email-re-pwd?token={token}'.format(
  382. server_host=SERVER_DOMAIN, token=token)
  383. send_body = send_data['body'].format(username=email, reset_pwd=reset_pwd, reset_link=reset_link)
  384. ses = SesClassObject()
  385. send_res = ses.send_email(send_address_list=[email], subject=send_data['title'], body=send_body)
  386. if send_res is True:
  387. if redisObj.set_data(key=userID + '_email_reset_pwd', val=reset_pwd, expire=3600):
  388. return response.json(0)
  389. else:
  390. return response.json(10, '存储验证失败')
  391. else:
  392. return response.json(89)
  393. else:
  394. return response.json(103)
  395. else:
  396. return response.json(104)
  397. class EmailResetPwdView(TemplateView):
  398. @method_decorator(csrf_exempt)
  399. def dispatch(self, *args, **kwargs):
  400. return super(EmailResetPwdView, self).dispatch(*args, **kwargs)
  401. # 查询
  402. def get(self, request, *args, **kwargs):
  403. response = ResponseObject()
  404. request_dict = request.GET
  405. return self.validate(request_dict, response, *args, **kwargs)
  406. # 认证登录
  407. def post(self, request, *args, **kwargs):
  408. response = ResponseObject()
  409. try:
  410. print(request.body.decode("utf-8"))
  411. json_data = json.loads(request.body.decode("utf-8"))
  412. except Exception as e:
  413. return response.json(10, repr(e))
  414. else:
  415. request_dict = json_data
  416. return self.validate(request_dict, response, *args, **kwargs)
  417. def validate(self, request_dict, response, *args, **kwargs):
  418. token = request_dict.get('token', None)
  419. tko = TokenObject(token)
  420. if tko.code == 0:
  421. redisObj = RedisObject()
  422. userID = tko.userID
  423. reset_pwd = redisObj.get_data(key=userID + '_email_reset_pwd')
  424. if reset_pwd is not False:
  425. user_qs = Device_User.objects.filter(userID=userID)
  426. if user_qs.exists():
  427. redisObj.del_data(key=userID + '_email_reset_pwd')
  428. is_update = user_qs.update(password=make_password(reset_pwd))
  429. if is_update:
  430. return HttpResponseRedirect(
  431. "http://www.dvema.com/web/html/paw_update_success.html?code=" + reset_pwd)
  432. else:
  433. return response.json(10)
  434. else:
  435. return response.json(104)
  436. else:
  437. return HttpResponseRedirect('http://www.dvema.com/web/html/paw_update_unsuccessful.html?lang=en')
  438. else:
  439. return HttpResponseRedirect('http://www.dvema.com/web/html/paw_update_unsuccessful.html?lang=en')
  440. class refreshTokenView(TemplateView):
  441. @method_decorator(csrf_exempt)
  442. def dispatch(self, *args, **kwargs):
  443. return super(refreshTokenView, self).dispatch(*args, **kwargs)
  444. def post(self, request, *args, **kwargs):
  445. request.encoding = 'utf-8'
  446. content_type = request.META.get('CONTENT_TYPE', None)
  447. if content_type == 'application/json':
  448. request_dict = json.loads(request.body.decode('utf-8'))
  449. else:
  450. request_dict = request.POST
  451. return self.validation(request_dict)
  452. def get(self, request, *args, **kwargs):
  453. request.encoding = 'utf-8'
  454. request_dict = request.GET
  455. return self.validation(request_dict)
  456. def validation(self, request_dict):
  457. token = request_dict.get('token', None)
  458. lang = request_dict.get('lang', None)
  459. language = request_dict.get('language', None)
  460. if lang is None:
  461. lang = language
  462. response = ResponseObject(lang)
  463. if token is not None:
  464. tko = TokenObject(token)
  465. tko.lang = lang
  466. res = tko.refresh()
  467. code = tko.code
  468. if code == 0:
  469. return response.json(0, res)
  470. else:
  471. return response.json(code)
  472. else:
  473. return response.json(444, 'token')
  474. # 获取验证码
  475. class v2authCodeView(TemplateView):
  476. @method_decorator(csrf_exempt)
  477. def dispatch(self, *args, **kwargs):
  478. return super(v2authCodeView, self).dispatch(*args, **kwargs)
  479. @ratelimit(key='ip', rate='2/m')
  480. def post(self, request, *args, **kwargs):
  481. request.encoding = 'utf-8'
  482. lang = request.POST.get('lang', None)
  483. if not lang:
  484. lang = request.POST.get('language', None)
  485. response = ResponseObject(lang)
  486. request_dict = request.POST
  487. phone = request_dict.get('phone', None)
  488. if phone is not None:
  489. was_limited = getattr(request, 'limited', False)
  490. if was_limited is True:
  491. return response.json(5)
  492. return self.ValidationError(request_dict, response)
  493. @ratelimit(key='ip', rate='2/m')
  494. def get(self, request, *args, **kwargs):
  495. # Device_User.objects.filter(userEmail='chanjunkai@163.com').delete()
  496. request.encoding = 'utf-8'
  497. lang = request.GET.get('lang', None)
  498. if not lang:
  499. lang = request.GET.get('language', None)
  500. response = ResponseObject(lang)
  501. was_limited = getattr(request, 'limited', False)
  502. if was_limited is True:
  503. return response.json(5)
  504. request_dict = request.GET
  505. return self.ValidationError(request_dict, response)
  506. def ValidationError(self, request_dict, response):
  507. email = request_dict.get('email', None)
  508. phone = request_dict.get('phone', None)
  509. country_code = request_dict.get('country_code', None)
  510. sign_name = request_dict.get('sign_name', None)
  511. if email is not None:
  512. email = email.strip()
  513. # 阿里云的发送邮箱的调用方法
  514. return self.aliyun_emailCode(email, response)
  515. # return self.emailCode(email, response)
  516. elif phone is not None:
  517. phone = phone.strip()
  518. if country_code is None:
  519. return self.phoneCode(phone, response, sign_name)
  520. else:
  521. country_code = str(country_code.strip())
  522. return self.phoneCodeV2(country_code, phone, response, sign_name)
  523. else:
  524. return response.json(444)
  525. def emailCode(self, email, response):
  526. dataValid = DataValid()
  527. # 邮箱匹配
  528. if dataValid.email_validate(email) is False:
  529. return response.json(107)
  530. reds = RedisObject()
  531. identifyingCode = reds.get_data(key=email + '_identifyingCode')
  532. # 是否以获取邮箱验证码
  533. if identifyingCode:
  534. return response.json(89)
  535. user_qs = Device_User.objects.filter(username=email)
  536. email_qs = Device_User.objects.filter(userEmail=email)
  537. # 邮箱用户是否已存在
  538. if user_qs.exists():
  539. return response.json(103)
  540. elif email_qs.exists():
  541. return response.json(103)
  542. # 生成随机6位数
  543. identifyingCode = RandomStr(6, True)
  544. # 设置随机数缓存生命周期
  545. send_data = TemplateService.email_message(type='register_code', language=response.lang)
  546. ses = SesClassObject()
  547. # 发送邮件
  548. send_res = ses.send_email(
  549. send_address_list=[email],
  550. subject=send_data['title'],
  551. body=send_data['body'].replace("{username}", email).replace("{captcha}",
  552. str(identifyingCode))
  553. )
  554. if send_res is not True:
  555. return response.json(44)
  556. if reds.set_data(key=email + '_identifyingCode', val=identifyingCode, expire=600) is not True:
  557. return response.json(10, 'error')
  558. return response.json(0)
  559. # return response.json(0, {'identifyingCode': identifyingCode})
  560. # 阿里云获取邮箱验证码
  561. def aliyun_emailCode(self, email, response):
  562. print('阿里云开始')
  563. dataValid = DataValid()
  564. # 邮箱匹配
  565. if dataValid.email_validate(email) is False:
  566. return response.json(107)
  567. reds = RedisObject()
  568. identifyingCode = reds.get_data(key=email + '_identifyingCode')
  569. # 是否以获取邮箱验证码
  570. if identifyingCode:
  571. return response.json(89)
  572. user_qs = Device_User.objects.filter(username=email)
  573. email_qs = Device_User.objects.filter(userEmail=email)
  574. # 邮箱用户是否已存在
  575. if user_qs.exists():
  576. return response.json(103)
  577. elif email_qs.exists():
  578. return response.json(103)
  579. # 生成随机6位数
  580. identifyingCode = RandomStr(6, True)
  581. # 设置随机数缓存生命周期
  582. send_data = TemplateService.email_message(type='register_code', language=response.lang)
  583. ses = SesClassObject()
  584. # 发送邮件
  585. send_res = ses.alyEmailCode(
  586. send_address_list=[email],
  587. subject=send_data['title'],
  588. body=send_data['body'].replace("{username}", email).replace("{captcha}", str(identifyingCode))
  589. )
  590. if send_res is not True:
  591. return response.json(44)
  592. if reds.set_data(key=email + '_identifyingCode', val=identifyingCode, expire=600) is not True:
  593. return response.json(10, 'error')
  594. return response.json(0)
  595. def phoneCode(self, phone, response, sign_name):
  596. dataValid = DataValid()
  597. if dataValid.mobile_validate(phone) is not True:
  598. return response.json(107)
  599. reds = RedisObject()
  600. reds_key = str(phone) + '_identifyingCode'
  601. identifyingCode = reds.get_data(key=reds_key)
  602. reds_key_ttl = reds.get_ttl(key=reds_key)
  603. if reds_key_ttl > 240 and identifyingCode:
  604. # if identifyingCode :
  605. return response.json(90)
  606. user_qs = Device_User.objects.filter(username=phone)
  607. phone_qs = Device_User.objects.filter(phone=phone)
  608. if user_qs.exists() or phone_qs.exists():
  609. return response.json(101)
  610. identifyingCode = RandomStr(6, True)
  611. # 发送手机验证码
  612. aliSms = AliSmsObject()
  613. if sign_name == 'zosi':
  614. sign_ms = '周视'
  615. else:
  616. sign_ms = 'Ansjer'
  617. res = aliSms.send_code_sms(phone=phone, code=identifyingCode, sign_name=sign_ms,
  618. temp_msg='SMS_151600991')
  619. print(res)
  620. if res["Code"] == "OK":
  621. if reds.set_data(key=reds_key, val=identifyingCode, expire=300) is not True:
  622. # if reds.set_data(key=phone + '_identifyingCode', val=identifyingCode, expire=60) is not True:
  623. return response.json(10, '生成缓存系统错误')
  624. return response.json(0)
  625. else:
  626. return response.json(10, res["Message"])
  627. def phoneCodeV2(self, country_code, phone, response, sign_name):
  628. dataValid = DataValid()
  629. if dataValid.mobile_validate(phone) is not True:
  630. return response.json(107)
  631. reds = RedisObject()
  632. reds_key = str(phone) + '_identifyingCode'
  633. identifyingCode = reds.get_data(key=reds_key)
  634. reds_key_ttl = reds.get_ttl(key=reds_key)
  635. if reds_key_ttl > 240 and identifyingCode:
  636. # if identifyingCode :
  637. return response.json(90)
  638. user_qs = Device_User.objects.filter(username=phone)
  639. phone_qs = Device_User.objects.filter(phone=phone)
  640. if user_qs.exists() or phone_qs.exists():
  641. return response.json(101)
  642. identifyingCode = RandomStr(6, True)
  643. # 短信签名
  644. # sign_name_dict = {
  645. # 'ansjer':'Ansjer',
  646. # 'zosi':'周视'
  647. # }
  648. sign_ms = ''
  649. if country_code == '86':
  650. # 国内短信推送模板
  651. temp_msg = 'SMS_151600991'
  652. rec_phone = phone
  653. if sign_name == 'zosi':
  654. sign_ms = '周视'
  655. else:
  656. sign_ms = 'Ansjer'
  657. else:
  658. # 国际短信推送模板
  659. temp_msg = 'SMS_172165867'
  660. rec_phone = country_code + phone
  661. sign_ms = 'Ansjer'
  662. # 发送手机验证码
  663. aliSms = AliSmsObject()
  664. res = aliSms.send_code_sms(phone=rec_phone, code=identifyingCode, sign_name=sign_ms,
  665. temp_msg=temp_msg)
  666. print(res)
  667. if res["Code"] == "OK":
  668. # if reds.set_data(key=reds_key, val=identifyingCode, expire=60) is not True:
  669. if reds.set_data(key=reds_key, val=identifyingCode, expire=300) is not True:
  670. return response.json(10, '生成缓存系统错误')
  671. return response.json(0)
  672. else:
  673. return response.json(10, res["Message"])
  674. # 验证码注册
  675. class v2registerView(TemplateView):
  676. @method_decorator(csrf_exempt)
  677. def dispatch(self, *args, **kwargs):
  678. return super(v2registerView, self).dispatch(*args, **kwargs)
  679. def post(self, request, *args, **kwargs):
  680. request.encoding = 'utf-8'
  681. request_dict = request.POST
  682. return self.validates(request_dict)
  683. def get(self, request, *args, **kwargs):
  684. request.encoding = 'utf-8'
  685. request_dict = request.GET
  686. return self.validates(request_dict)
  687. def validates(self, request_dict):
  688. phone = request_dict.get('phone', None)
  689. email = request_dict.get('email', None)
  690. password = request_dict.get('password', None)
  691. authcode = request_dict.get('authcode', None)
  692. lang = request_dict.get('lang', None)
  693. if not lang:
  694. lang = request_dict.get('language', None)
  695. response = ResponseObject(lang)
  696. if password is None:
  697. return response.json(444, 'password')
  698. if authcode is None:
  699. return response.json(444, 'identifyingCode')
  700. if phone is not None:
  701. return self.do_phone_register(phone, password, authcode, response)
  702. elif email is not None:
  703. return self.do_email_register(email, password, authcode, response)
  704. else:
  705. return response.json(444, 'phone or email')
  706. def do_phone_register(self, phone, password, authcode, response):
  707. data_valid = DataValid()
  708. if data_valid.mobile_validate(phone) is not True:
  709. return response.json(100)
  710. if data_valid.password_validate(password) is not True:
  711. return response.json(109)
  712. reds = RedisObject()
  713. identifyingCode = reds.get_data(key=phone + '_identifyingCode')
  714. # 判断验证码是否过期
  715. if identifyingCode is False:
  716. return response.json(120)
  717. # 验证码是否正确
  718. if authcode != identifyingCode:
  719. return response.json(121)
  720. phone_qs = Device_User.objects.filter(Q(phone=phone) | Q(username=phone))
  721. # 是否已存在
  722. if phone_qs.exists():
  723. return response.json(101)
  724. try:
  725. users = Device_User.objects.create(
  726. username=phone,
  727. NickName=phone,
  728. phone=phone,
  729. password=make_password(password),
  730. userID=CommonService.getUserID(μs=False, setOTAID=True),
  731. is_active=True,
  732. user_isValid=True,
  733. )
  734. except Exception as e:
  735. errorInfo = traceback.format_exc()
  736. print(errorInfo)
  737. return response.json(424, repr(e))
  738. else:
  739. if not reds.del_data(key=phone + '_identifyingCode'):
  740. return response.json(10, '删除缓存验证码错误')
  741. return self.do_login(phone_qs, response)
  742. def do_login(self, user_qs, response):
  743. now_time = datetime.datetime.utcnow().replace(tzinfo=utc).astimezone(utc)
  744. user_qs.update(last_login=now_time, online=True)
  745. userID = user_qs[0].userID
  746. print('userID' + userID)
  747. tko = TokenObject()
  748. user_list = user_qs.values("NickName", "userIconUrl", "userIconPath", "username", "userEmail", "phone")
  749. res = tko.generate(data={'userID': userID, 'lang': response.lang, 'user': user_list[0]["username"]})
  750. # 增加角色
  751. user_qs[0].role.add(Role.objects.get(rid=1))
  752. role_dict = ModelService.own_role(userID=userID)
  753. res['rid'] = role_dict['rid']
  754. res['roleName'] = role_dict['roleName']
  755. res['permList'] = ModelService.own_permission(userID)
  756. res['userID'] = userID
  757. # 昵称,邮箱,电话,刷新,头像
  758. userIconPath = str(user_list[0]["userIconPath"])
  759. if userIconPath and userIconPath.find('static/') != -1:
  760. userIconPath = userIconPath.replace('static/', '').replace('\\', '/')
  761. res['userIconUrl'] = SERVER_DOMAIN + 'account/getAvatar/' + userIconPath
  762. else:
  763. res['userIconUrl'] = ''
  764. res['NickName'] = user_list[0]["NickName"] if user_list[0]["NickName"] is not None else ''
  765. res['username'] = user_list[0]["username"] if user_list[0]["username"] is not None else ''
  766. res['userEmail'] = user_list[0]["userEmail"] if user_list[0]["userEmail"] is not None else ''
  767. res['phone'] = user_list[0]["phone"] if user_list[0]["phone"] is not None else ''
  768. print(res)
  769. return response.json(0, res)
  770. def do_email_register(self, email, password, authcode, response):
  771. data_valid = DataValid()
  772. if data_valid.email_validate(email) is not True:
  773. return response.json(105)
  774. if data_valid.password_validate(password) is not True:
  775. return response.json(109)
  776. reds = RedisObject()
  777. identifyingCode = reds.get_data(key=email + '_identifyingCode')
  778. # 判断验证码是否过期
  779. if identifyingCode is False:
  780. return response.json(120)
  781. # 验证码是否正确
  782. if authcode != identifyingCode:
  783. return response.json(121)
  784. email_qs = Device_User.objects.filter(Q(userEmail=email) | Q(username=email))
  785. # 是否已存在
  786. if email_qs.exists():
  787. return response.json(103)
  788. try:
  789. users = Device_User.objects.create(
  790. username=email,
  791. NickName=email,
  792. userEmail=email,
  793. password=make_password(password),
  794. userID=CommonService.getUserID(μs=False, setOTAID=True),
  795. is_active=True,
  796. user_isValid=True,
  797. )
  798. except Exception as e:
  799. errorInfo = traceback.format_exc()
  800. print(errorInfo)
  801. return response.json(424, repr(e))
  802. else:
  803. if not reds.del_data(key=email + '_identifyingCode'):
  804. return response.json(10, '删除缓存验证码错误')
  805. return self.do_login(email_qs, response)
  806. # 重置密码
  807. # 忘记密码获取验证码v2
  808. class v2forgetPwdCodeView(TemplateView):
  809. @method_decorator(csrf_exempt)
  810. def dispatch(self, *args, **kwargs):
  811. return super(v2forgetPwdCodeView, self).dispatch(*args, **kwargs)
  812. @ratelimit(key='ip', rate='1/m')
  813. def get(self, request, *args, **kwargs):
  814. request.encoding = 'utf-8'
  815. request_dict = request.GET
  816. lang = request_dict.get('lang')
  817. response = ResponseObject(lang)
  818. was_limited = getattr(request, 'limited', False)
  819. if was_limited is True:
  820. return response.json(5)
  821. return self.ValidationError(request_dict, response)
  822. @ratelimit(key='ip', rate='1/m')
  823. def post(self, request):
  824. request.encoding = 'utf-8'
  825. request_dict = request.POST
  826. lang = request_dict.get('lang')
  827. response = ResponseObject(lang)
  828. was_limited = getattr(request, 'limited', False)
  829. if was_limited is True:
  830. return response.json(5)
  831. return self.ValidationError(request_dict, response)
  832. def ValidationError(self, request_dict, response):
  833. phone = request_dict.get('phone', None)
  834. email = request_dict.get('email', None)
  835. country_code = request_dict.get('country_code', None)
  836. sign_name = request_dict.get('sign_name', None)
  837. if phone is not None:
  838. phone = phone.strip()
  839. if country_code is None:
  840. return self.do_send_phone_code(phone, response, sign_name)
  841. else:
  842. return self.do_v2_send_phone_code(country_code, phone, response, sign_name)
  843. elif email is not None:
  844. return self.do_send_email_code(email, response)
  845. else:
  846. return response.json(444, 'phone')
  847. def do_send_phone_code(self, phone, response, sign_name):
  848. data_valid = DataValid()
  849. if data_valid.mobile_validate(phone) is not True:
  850. return response.json(100)
  851. user_qs = Device_User.objects.filter(Q(phone=phone) | Q(username=phone))
  852. if not user_qs.exists():
  853. return response.json(102)
  854. reds = RedisObject()
  855. reds_key = str(phone) + '_forgetPwdResetCode'
  856. resetCode = reds.get_data(key=reds_key)
  857. reds_key_ttl = reds.get_ttl(key=reds_key)
  858. if reds_key_ttl > 240 and resetCode:
  859. # if identifyingCode :
  860. return response.json(90)
  861. # if resetCode is True:
  862. # return response.json(120)
  863. resetCode = RandomStr(6, True)
  864. if sign_name == 'zosi':
  865. sign_ms = '周视'
  866. else:
  867. sign_ms = 'Ansjer'
  868. aliSms = AliSmsObject()
  869. res = aliSms.send_code_sms(phone=phone, code=resetCode, sign_name=sign_ms,
  870. temp_msg='SMS_151675019')
  871. if res["Code"] == "OK":
  872. if not reds.set_data(key=reds_key, val=resetCode, expire=300):
  873. return response.json(10, '生成缓存错误')
  874. return response.json(0)
  875. else:
  876. return response.json(10, res["Message"])
  877. def do_v2_send_phone_code(self, country_code, phone, response, sign_name):
  878. data_valid = DataValid()
  879. if data_valid.mobile_validate(phone) is not True:
  880. return response.json(100)
  881. user_qs = Device_User.objects.filter(Q(phone=phone) | Q(username=phone))
  882. if not user_qs.exists():
  883. return response.json(102)
  884. reds = RedisObject()
  885. reds_key = str(phone) + '_forgetPwdResetCode'
  886. resetCode = reds.get_data(key=reds_key)
  887. reds_key_ttl = reds.get_ttl(key=reds_key)
  888. if reds_key_ttl > 240 and resetCode:
  889. # if identifyingCode :
  890. return response.json(90)
  891. # if resetCode is True:
  892. # return response.json(120)
  893. resetCode = RandomStr(6, True)
  894. aliSms = AliSmsObject()
  895. if country_code == '86':
  896. rec_phone = phone
  897. temp_msg = 'SMS_151675019'
  898. if sign_name == 'zosi':
  899. sign_ms = '周视'
  900. else:
  901. sign_ms = 'Ansjer'
  902. else:
  903. temp_msg = 'SMS_172200051'
  904. rec_phone = str(country_code) + str(phone)
  905. sign_ms = 'Ansjer'
  906. res = aliSms.send_code_sms(phone=rec_phone, code=resetCode, sign_name=sign_ms,
  907. temp_msg=temp_msg)
  908. if res["Code"] == "OK":
  909. if not reds.set_data(key=reds_key, val=resetCode, expire=300):
  910. return response.json(10, '生成缓存错误')
  911. return response.json(0)
  912. else:
  913. return response.json(10, res["Message"])
  914. def do_send_email_code(self, email, response):
  915. data_valid = DataValid()
  916. if data_valid.email_validate(email) is not True:
  917. return response.json(105)
  918. user_qs = Device_User.objects.filter(Q(userEmail=email) | Q(username=email))
  919. if not user_qs.exists():
  920. return response.json(104)
  921. reds = RedisObject()
  922. resetCode = reds.get_data(key=email + '_forgetPwdResetCode')
  923. if resetCode is True:
  924. return response.json(120)
  925. resetCode = RandomStr(6, True)
  926. if not reds.set_data(key=email + '_forgetPwdResetCode', val=resetCode, expire=600):
  927. return response.json(10, '生成缓存错误')
  928. send_data = TemplateService.email_message(type='forgetCode', language=response.lang)
  929. send_body = send_data['body'].format(userEmail=email, email_valid_code=resetCode)
  930. ses = SesClassObject()
  931. # send_res = ses.send_email(send_address_list=[email], subject=send_data['title'], body=send_body)
  932. # 阿里云发送邮箱
  933. send_res = ses.alyEmailCode(send_address_list=[email], subject=send_data['title'], body=send_body)
  934. if send_res is not True:
  935. reds.del_data(key=email + '_forgetPwdResetCode')
  936. return response.json(44)
  937. return response.json(0)
  938. # 忘记密码v2
  939. class v2resetPwdByCodeView(TemplateView):
  940. @method_decorator(csrf_exempt)
  941. def dispatch(self, *args, **kwargs):
  942. return super(v2resetPwdByCodeView, self).dispatch(*args, **kwargs)
  943. def get(self, request, *args, **kwargs):
  944. request.encoding = 'utf-8'
  945. request_dict = request.GET
  946. lang = request_dict.get('lang')
  947. if not lang:
  948. lang = request_dict.get('language', None)
  949. response = ResponseObject(lang)
  950. was_limited = getattr(request, 'limited', False)
  951. if was_limited is True:
  952. return response.json(5)
  953. return self.ValidationError(request_dict, response)
  954. def post(self, request):
  955. request.encoding = 'utf-8'
  956. request_dict = request.POST
  957. lang = request_dict.get('lang')
  958. if not lang:
  959. lang = request_dict.get('language', None)
  960. response = ResponseObject(lang)
  961. was_limited = getattr(request, 'limited', False)
  962. if was_limited is True:
  963. return response.json(5)
  964. return self.ValidationError(request_dict, response)
  965. def ValidationError(self, request_dict, response):
  966. phone = request_dict.get('phone', None)
  967. email = request_dict.get('email', None)
  968. password = request_dict.get('password', None)
  969. authcode = request_dict.get('authcode', None)
  970. if password is None or authcode is None:
  971. return response.json(444, 'password,authcode')
  972. authcode = authcode.strip()
  973. password = password.strip()
  974. if phone is not None:
  975. phone = phone.strip()
  976. return self.do_phone_pwd_reset(phone, authcode, password, response)
  977. elif email is not None:
  978. email = email.strip()
  979. return self.do_email_pwd_reset(email, authcode, password, response)
  980. else:
  981. return response.json(444, 'phone')
  982. def do_email_pwd_reset(self, email, authcode, password, response):
  983. data_valid = DataValid()
  984. if data_valid.email_validate(email) is not True:
  985. return response.json(105)
  986. if data_valid.password_validate(password) is not True:
  987. return response.json(109)
  988. user_qs = Device_User.objects.filter(Q(userEmail=email) | Q(username=email))
  989. if not user_qs.exists():
  990. return response.json(104)
  991. reds = RedisObject()
  992. resetCode = reds.get_data(key=email + '_forgetPwdResetCode')
  993. if resetCode is False:
  994. return response.json(90)
  995. if authcode != resetCode:
  996. return response.json(121)
  997. # if not reds.set_data(key=email + '_forgetPwdResetCode', val=resetCode, expire=300):
  998. # return response.json(10, '生成缓存错误')
  999. user_qs.update(password=make_password(password))
  1000. if not reds.del_data(email + '_forgetPwdResetCode'):
  1001. return response.json(10, '删除缓存失败')
  1002. return response.json(0)
  1003. def do_phone_pwd_reset(self, phone, authcode, password, response):
  1004. data_valid = DataValid()
  1005. if data_valid.mobile_validate(phone) is not True:
  1006. return response.json(100)
  1007. if data_valid.password_validate(password) is not True:
  1008. return response.json(109)
  1009. user_qs = Device_User.objects.filter(Q(phone=phone) | Q(username=phone))
  1010. if not user_qs.exists():
  1011. return response.json(102)
  1012. reds = RedisObject()
  1013. resetCode = reds.get_data(key=phone + '_forgetPwdResetCode')
  1014. if resetCode is False:
  1015. return response.json(90)
  1016. if authcode != resetCode:
  1017. return response.json(121)
  1018. # if not reds.set_data(key=phone + '_forgetPwdResetCode', val=resetCode, expire=300):
  1019. # return response.json(10, '生成缓存错误')
  1020. user_qs.update(password=make_password(password))
  1021. if not reds.del_data(phone + '_forgetPwdResetCode'):
  1022. return response.json(10, '删除缓存失败')
  1023. return response.json(0)
  1024. def do_login(self, user_qs, response):
  1025. now_time = datetime.datetime.utcnow().replace(tzinfo=utc).astimezone(utc)
  1026. user_qs.update(last_login=now_time, online=True)
  1027. userID = user_qs[0].userID
  1028. print('userID' + userID)
  1029. tko = TokenObject()
  1030. user_list = user_qs.values("NickName", "userIconUrl", "userIconPath", "username", "userEmail", "phone")
  1031. res = tko.generate(data={'userID': userID, 'lang': response.lang, 'user': user_list[0]["username"]})
  1032. # 增加角色
  1033. user_qs[0].role.add(Role.objects.get(rid=1))
  1034. role_dict = ModelService.own_role(userID=userID)
  1035. res['rid'] = role_dict['rid']
  1036. res['roleName'] = role_dict['roleName']
  1037. res['permList'] = ModelService.own_permission(userID)
  1038. res['userID'] = userID
  1039. # 昵称,邮箱,电话,刷新,头像
  1040. userIconPath = str(user_list[0]["userIconPath"])
  1041. if userIconPath and userIconPath.find('static/') != -1:
  1042. userIconPath = userIconPath.replace('static/', '').replace('\\', '/')
  1043. res['userIconUrl'] = SERVER_DOMAIN + 'account/getAvatar/' + userIconPath
  1044. else:
  1045. res['userIconUrl'] = ''
  1046. res['NickName'] = user_list[0]["NickName"] if user_list[0]["NickName"] is not None else ''
  1047. res['username'] = user_list[0]["username"] if user_list[0]["username"] is not None else ''
  1048. res['userEmail'] = user_list[0]["userEmail"] if user_list[0]["userEmail"] is not None else ''
  1049. res['phone'] = user_list[0]["phone"] if user_list[0]["phone"] is not None else ''
  1050. print(res)
  1051. return response.json(0, res)
  1052. # 登录
  1053. class v2LoginView(TemplateView):
  1054. @method_decorator(csrf_exempt) # @csrf_exempt
  1055. def dispatch(self, *args, **kwargs):
  1056. return super(v2LoginView, self).dispatch(*args, **kwargs)
  1057. @ratelimit(key='ip', rate='5/m')
  1058. def post(self, request, *args, **kwargs):
  1059. request.encoding = 'utf-8'
  1060. request_dict = request.POST
  1061. language = request_dict.get('language', 'en')
  1062. response = ResponseObject(language)
  1063. was_limited = getattr(request, 'limited', False)
  1064. if was_limited is True:
  1065. return response.json(5)
  1066. return self.validates(request_dict, response)
  1067. # @ratelimit(key='ip', rate='5/m')
  1068. def get(self, request, *args, **kwargs):
  1069. request.encoding = 'utf-8'
  1070. request_dict = request.GET
  1071. language = request_dict.get('language', 'en')
  1072. response = ResponseObject(language)
  1073. was_limited = getattr(request, 'limited', False)
  1074. if was_limited is True:
  1075. return response.json(5)
  1076. return self.validates(request_dict, response)
  1077. def validates(self, request_dict, response):
  1078. username = request_dict.get('userName', None)
  1079. password = request_dict.get('userPwd', None)
  1080. if not username or not password:
  1081. return response.json(111)
  1082. username = username.strip()
  1083. password = password.strip()
  1084. data_valid = DataValid()
  1085. if data_valid.email_validate(username):
  1086. return self.do_email_login(username, password, response)
  1087. elif data_valid.mobile_validate(username):
  1088. return self.do_phone_login(username, password, response)
  1089. elif data_valid.name_validate(username):
  1090. return self.do_name_login(username, password, response)
  1091. else:
  1092. return response.json(107)
  1093. def do_email_login(self, email, password, response):
  1094. user_qs = Device_User.objects.filter(Q(username=email) | Q(userEmail=email))
  1095. return self.valid_login(user_qs, password, response)
  1096. def do_phone_login(self, phone, password, response):
  1097. user_qs = Device_User.objects.filter(Q(phone=phone) | Q(username=phone), is_active=True, user_isValid=True)
  1098. return self.valid_login(user_qs, password, response)
  1099. def do_name_login(self, username, password, response):
  1100. user_qs = Device_User.objects.filter(Q(username=username) | Q(phone=username) | Q(userEmail=username),
  1101. is_active=True, user_isValid=True)
  1102. return self.valid_login(user_qs, password, response)
  1103. def valid_login(self, user_qs, password, response):
  1104. if not user_qs.exists():
  1105. return response.json(104)
  1106. # users = user_qs.values('role__rid', 'role__roleName', 'userID', 'role', 'NickName', 'username', 'userEmail',
  1107. # 'phone', 'password', 'userIconPath', 'user_isValid', 'is_active')[0]
  1108. users = user_qs.values('role__rid', 'role__roleName', 'userID', 'NickName', 'username', 'userEmail',
  1109. 'phone', 'password', 'userIconPath')[0]
  1110. if not check_password(password, users['password']):
  1111. return response.json(111)
  1112. userID = users['userID']
  1113. tko = TokenObject()
  1114. res = tko.generate(
  1115. data={'userID': userID, 'lang': response.lang, 'user': users['username'], 'm_code': '123413243214'})
  1116. if tko.code == 0:
  1117. now_time = datetime.datetime.utcnow().replace(tzinfo=utc).astimezone(utc)
  1118. user_qs.update(last_login=now_time, language=response.lang)
  1119. res['rid'] = users['role__rid']
  1120. res['roleName'] = users['role__roleName']
  1121. res['permList'] = ModelService.own_permission(userID)
  1122. res['userID'] = userID
  1123. # 昵称,邮箱,电话,刷新,头像
  1124. userIconPath = str(users['userIconPath'])
  1125. if userIconPath and userIconPath.find('static/') != -1:
  1126. userIconPath = userIconPath.replace('static/', '').replace('\\', '/')
  1127. res['userIconUrl'] = SERVER_DOMAIN + 'account/getAvatar/' + userIconPath
  1128. else:
  1129. res['userIconUrl'] = ''
  1130. res['NickName'] = users['NickName'] if users['NickName'] is not None else ''
  1131. res['username'] = users['username'] if users['username'] is not None else ''
  1132. res['userEmail'] = users['userEmail'] if users['userEmail'] is not None else ''
  1133. res['phone'] = users['phone'] if users['phone'] is not None else ''
  1134. return response.json(0, res)
  1135. else:
  1136. return response.json(tko.code)
  1137. # 用户登录后初始化接口
  1138. class InitInfoView(View):
  1139. def get(self, request, *args, **kwargs):
  1140. request.encoding = 'utf-8'
  1141. return self.validation(request.GET)
  1142. def post(self, request, *args, **kwargs):
  1143. request.encoding = 'utf-8'
  1144. return self.validation(request.POST)
  1145. def validation(self, request_dict):
  1146. response = ResponseObject()
  1147. token = request_dict.get('token', None)
  1148. tko = TokenObject(token)
  1149. if tko.code == 0:
  1150. userID = tko.userID
  1151. return self.init_uid(request_dict, userID, response)
  1152. else:
  1153. return response.json(tko.code)
  1154. # 初始化设备token
  1155. def init_uid(self, request_dict, userID, response):
  1156. # 未读的系统消息
  1157. sm_count = SysMsgModel.objects.filter(userID_id=userID, status=0).count()
  1158. token_val = request_dict.get('token_val', None)
  1159. m_code = request_dict.get('m_code', None)
  1160. push_type = request_dict.get('push_type', None)
  1161. appBundleId = request_dict.get('appBundleId', None)
  1162. tz = request_dict.get('tz', None)
  1163. is_cn = request_dict.get('is_cn', None)
  1164. nowTime = int(time.time())
  1165. if all([token_val, push_type, appBundleId]):
  1166. if push_type == 0:
  1167. if appBundleId not in APNS_CONFIG.keys():
  1168. return response.json(904)
  1169. elif push_type == 1:
  1170. if appBundleId not in FCM_CONFIG.keys():
  1171. return response.json(904)
  1172. elif push_type == 2:
  1173. if appBundleId not in JPUSH_CONFIG.keys():
  1174. return response.json(904)
  1175. else:
  1176. return response.json(173)
  1177. uid_list = ModelService.get_uid_list(userID=userID)
  1178. if uid_list:
  1179. uid_set_id_list = UidSetModel.objects.filter(uid__in=uid_list, detect_status=1).values_list('id',
  1180. flat=True)
  1181. if uid_set_id_list:
  1182. update_dict = {
  1183. 'token_val': token_val,
  1184. 'push_type': push_type,
  1185. 'tz': tz
  1186. }
  1187. UidPushModel.objects.filter(userID_id=userID, m_code=m_code, uid_set_id__in=uid_set_id_list). \
  1188. update(**update_dict)
  1189. if appBundleId:
  1190. user_extend_qs = UserExtendModel.objects.filter(userID_id=userID)
  1191. if user_extend_qs.exists():
  1192. update_dict = {
  1193. 'updTime': nowTime,
  1194. 'appBundleId': appBundleId,
  1195. # 'is_cn':
  1196. }
  1197. user_extend_qs.update(**update_dict)
  1198. else:
  1199. create_dict = {
  1200. 'addTime': nowTime,
  1201. 'updTime': nowTime,
  1202. 'appBundleId': appBundleId,
  1203. 'userID_id': userID
  1204. }
  1205. UserExtendModel.objects.create(**create_dict)
  1206. res = {
  1207. 'usmsg': sm_count
  1208. }
  1209. return response.json(0, res)
  1210. # 获取验证码
  1211. class verifyAuthcode(TemplateView):
  1212. def post(self, request, *args, **kwargs):
  1213. request.encoding = 'utf-8'
  1214. lang = request.POST.get('lang', None)
  1215. if not lang:
  1216. lang = request.POST.get('language', None)
  1217. response = ResponseObject(lang)
  1218. request_dict = request.POST
  1219. return self.ValidationError(request_dict, response)
  1220. def get(self, request, *args, **kwargs):
  1221. # Device_User.objects.filter(userEmail='chanjunkai@163.com').delete()
  1222. request.encoding = 'utf-8'
  1223. lang = request.GET.get('lang', None)
  1224. if not lang:
  1225. lang = request.GET.get('language', None)
  1226. response = ResponseObject(lang)
  1227. request_dict = request.GET
  1228. return self.ValidationError(request_dict, response)
  1229. def ValidationError(self, request_dict, response):
  1230. email = request_dict.get('email', None)
  1231. phone = request_dict.get('phone', None)
  1232. authcode = request_dict.get('authcode', None)
  1233. if email is not None:
  1234. email = email.strip()
  1235. return self.email_validate(email, authcode, response)
  1236. elif phone is not None:
  1237. phone = phone.strip()
  1238. return self.phone_validate(phone, authcode, response)
  1239. else:
  1240. return response.json(444)
  1241. def email_validate(self, email, authcode, response):
  1242. data_valid = DataValid()
  1243. if data_valid.email_validate(email) is not True:
  1244. return response.json(105)
  1245. reds = RedisObject()
  1246. resetCode = reds.get_data(key=email + '_forgetPwdResetCode')
  1247. if resetCode is False:
  1248. return response.json(120)
  1249. if authcode != resetCode:
  1250. return response.json(121)
  1251. return response.json(0)
  1252. def phone_validate(self, phone, authcode, response):
  1253. data_valid = DataValid()
  1254. if data_valid.mobile_validate(phone) is not True:
  1255. return response.json(100)
  1256. reds = RedisObject()
  1257. resetCode = reds.get_data(key=phone + '_forgetPwdResetCode')
  1258. if resetCode is False:
  1259. return response.json(120)
  1260. print(resetCode)
  1261. if authcode != resetCode:
  1262. return response.json(121)
  1263. return response.json(0)