UserController.py 168 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303230423052306230723082309231023112312231323142315231623172318231923202321232223232324232523262327232823292330233123322333233423352336233723382339234023412342234323442345234623472348234923502351235223532354235523562357235823592360236123622363236423652366236723682369237023712372237323742375237623772378237923802381238223832384238523862387238823892390239123922393239423952396239723982399240024012402240324042405240624072408240924102411241224132414241524162417241824192420242124222423242424252426242724282429243024312432243324342435243624372438243924402441244224432444244524462447244824492450245124522453245424552456245724582459246024612462246324642465246624672468246924702471247224732474247524762477247824792480248124822483248424852486248724882489249024912492249324942495249624972498249925002501250225032504250525062507250825092510251125122513251425152516251725182519252025212522252325242525252625272528252925302531253225332534253525362537253825392540254125422543254425452546254725482549255025512552255325542555255625572558255925602561256225632564256525662567256825692570257125722573257425752576257725782579258025812582258325842585258625872588258925902591259225932594259525962597259825992600260126022603260426052606260726082609261026112612261326142615261626172618261926202621262226232624262526262627262826292630263126322633263426352636263726382639264026412642264326442645264626472648264926502651265226532654265526562657265826592660266126622663266426652666266726682669267026712672267326742675267626772678267926802681268226832684268526862687268826892690269126922693269426952696269726982699270027012702270327042705270627072708270927102711271227132714271527162717271827192720272127222723272427252726272727282729273027312732273327342735273627372738273927402741274227432744274527462747274827492750275127522753275427552756275727582759276027612762276327642765276627672768276927702771277227732774277527762777277827792780278127822783278427852786278727882789279027912792279327942795279627972798279928002801280228032804280528062807280828092810281128122813281428152816281728182819282028212822282328242825282628272828282928302831283228332834283528362837283828392840284128422843284428452846284728482849285028512852285328542855285628572858285928602861286228632864286528662867286828692870287128722873287428752876287728782879288028812882288328842885288628872888288928902891289228932894289528962897289828992900290129022903290429052906290729082909291029112912291329142915291629172918291929202921292229232924292529262927292829292930293129322933293429352936293729382939294029412942294329442945294629472948294929502951295229532954295529562957295829592960296129622963296429652966296729682969297029712972297329742975297629772978297929802981298229832984298529862987298829892990299129922993299429952996299729982999300030013002300330043005300630073008300930103011301230133014301530163017301830193020302130223023302430253026302730283029303030313032303330343035303630373038303930403041304230433044304530463047304830493050305130523053305430553056305730583059306030613062306330643065306630673068306930703071307230733074307530763077307830793080308130823083308430853086308730883089309030913092309330943095309630973098309931003101310231033104310531063107310831093110311131123113311431153116311731183119312031213122312331243125312631273128312931303131313231333134313531363137313831393140314131423143314431453146314731483149315031513152315331543155315631573158315931603161316231633164316531663167316831693170317131723173317431753176317731783179318031813182318331843185318631873188318931903191319231933194319531963197319831993200320132023203320432053206320732083209321032113212321332143215321632173218321932203221322232233224322532263227322832293230323132323233323432353236323732383239324032413242324332443245324632473248324932503251325232533254325532563257325832593260326132623263326432653266326732683269327032713272327332743275327632773278327932803281328232833284328532863287328832893290329132923293329432953296329732983299330033013302330333043305330633073308330933103311331233133314331533163317331833193320332133223323332433253326332733283329333033313332333333343335333633373338333933403341334233433344334533463347334833493350335133523353335433553356335733583359336033613362336333643365336633673368336933703371337233733374337533763377337833793380338133823383338433853386338733883389339033913392339333943395339633973398339934003401340234033404340534063407340834093410341134123413341434153416341734183419342034213422342334243425342634273428342934303431343234333434343534363437343834393440344134423443344434453446344734483449345034513452345334543455345634573458345934603461346234633464346534663467346834693470347134723473347434753476347734783479348034813482348334843485348634873488348934903491349234933494349534963497349834993500350135023503350435053506350735083509351035113512351335143515351635173518351935203521352235233524352535263527352835293530353135323533353435353536353735383539354035413542354335443545354635473548354935503551355235533554355535563557355835593560356135623563356435653566356735683569357035713572357335743575357635773578357935803581358235833584358535863587358835893590359135923593359435953596359735983599360036013602360336043605360636073608360936103611361236133614361536163617361836193620362136223623362436253626362736283629363036313632363336343635363636373638363936403641364236433644364536463647364836493650365136523653365436553656365736583659366036613662366336643665366636673668366936703671367236733674367536763677367836793680368136823683368436853686368736883689369036913692369336943695369636973698369937003701370237033704370537063707370837093710371137123713371437153716371737183719372037213722372337243725372637273728372937303731373237333734373537363737373837393740374137423743374437453746374737483749375037513752375337543755375637573758375937603761376237633764376537663767376837693770377137723773377437753776377737783779378037813782378337843785378637873788378937903791379237933794379537963797379837993800380138023803380438053806380738083809381038113812381338143815381638173818381938203821382238233824382538263827382838293830383138323833383438353836383738383839384038413842384338443845384638473848384938503851385238533854385538563857385838593860386138623863386438653866386738683869387038713872387338743875387638773878387938803881388238833884388538863887388838893890389138923893389438953896389738983899390039013902390339043905390639073908390939103911391239133914391539163917391839193920392139223923392439253926392739283929393039313932393339343935393639373938393939403941394239433944394539463947394839493950395139523953395439553956395739583959396039613962396339643965396639673968396939703971397239733974397539763977397839793980398139823983398439853986398739883989399039913992399339943995399639973998399940004001400240034004400540064007400840094010401140124013401440154016401740184019402040214022402340244025402640274028402940304031403240334034403540364037403840394040404140424043404440454046
  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 logging
  18. import jwt
  19. import simplejson
  20. import simplejson as json
  21. import requests
  22. from django.contrib.auth.hashers import make_password, check_password # 对密码加密模块
  23. from django.db.models import Q
  24. from django.http import HttpResponseRedirect
  25. from django.utils.decorators import method_decorator
  26. from django.utils.timezone import utc
  27. from django.views.decorators.csrf import csrf_exempt
  28. from django.views.generic import TemplateView
  29. from jwt.algorithms import RSAAlgorithm
  30. from ratelimit.decorators import ratelimit
  31. from Ansjer.config import AuthCode_Expire, SERVER_DOMAIN, APNS_CONFIG, JPUSH_CONFIG, FCM_CONFIG, TUTK_PUSH_DOMAIN
  32. from Controller.CheckUserData import DataValid, date_handler, RandomStr
  33. from Model.models import Device_User, Role, UidPushModel, UserOauth2Model, UserExModel, Device_Info, UidSetModel, \
  34. UserAppFrequencyModel, CountryIPModel, CountryModel, UidChannelSetModel, Order_Model, UID_Bucket, Unused_Uid_Meal
  35. from Object.AWS.SesClassObject import SesClassObject
  36. from Object.AliSmsObject import AliSmsObject
  37. from Object.RedisObject import RedisObject
  38. from Object.ResponseObject import ResponseObject
  39. from Object.TokenObject import TokenObject
  40. from Service.CommonService import CommonService
  41. from Service.ModelService import ModelService
  42. from Service.TemplateService import TemplateService
  43. from django.views.generic import View
  44. import base64
  45. import random
  46. from io import BytesIO
  47. from PIL import Image, ImageDraw, ImageFont
  48. from django.shortcuts import HttpResponse
  49. from Ansjer.config import BASE_DIR
  50. # 获取验证码
  51. class authCodeView(TemplateView):
  52. @method_decorator(csrf_exempt)
  53. def dispatch(self, *args, **kwargs):
  54. # testtest11111111111111
  55. return super(authCodeView, self).dispatch(*args, **kwargs)
  56. @ratelimit(key='ip', rate='2/m')
  57. def post(self, request, *args, **kwargs):
  58. request.encoding = 'utf-8'
  59. lang = request.POST.get('language', None)
  60. response = ResponseObject(lang)
  61. was_limited = getattr(request, 'limited', False)
  62. if was_limited is True:
  63. return response.json(5)
  64. username = request.POST.get('userName', None)
  65. useremail = request.POST.get('userEmail', None)
  66. return self.ValidationError(username, useremail, response)
  67. # @ratelimit(key='ip', rate='2/m')
  68. def get(self, request, *args, **kwargs):
  69. # Device_User.objects.filter(userEmail='chanjunkai@163.com').delete()
  70. request.encoding = 'utf-8'
  71. lang = request.GET.get('language', None)
  72. response = ResponseObject(lang)
  73. was_limited = getattr(request, 'limited', False)
  74. if was_limited is True:
  75. return response.json(5)
  76. username = request.GET.get('userName', None)
  77. email = request.GET.get('userEmail', None)
  78. return self.ValidationError(username, email, response)
  79. def ValidationError(self, username, email, response):
  80. if username:
  81. username = username.strip()
  82. return self.phoneCode(username, response)
  83. elif email:
  84. email = email.strip()
  85. return self.emailCode(email, response)
  86. else:
  87. return response.json(800)
  88. def phoneCode(self, phone, response):
  89. dataValid = DataValid()
  90. if dataValid.mobile_validate(phone):
  91. reds = RedisObject()
  92. identifyingCode = reds.get_data(key=phone + '_identifyingCode')
  93. if identifyingCode is False:
  94. user_qs = Device_User.objects.filter(username=phone)
  95. if user_qs.exists():
  96. return response.json(101)
  97. else:
  98. identifyingCode = RandomStr(6, True)
  99. if reds.set_data(key=phone + '_identifyingCode', val=identifyingCode, expire=600):
  100. return response.json(0, {'identifyingCode': identifyingCode})
  101. else:
  102. return response.json(10, '生成缓存系统错误')
  103. else:
  104. return response.json(0, {'identifyingCode': identifyingCode})
  105. else:
  106. return response.json(107)
  107. def emailCode(self, email, response):
  108. dataValid = DataValid()
  109. if dataValid.email_validate(email):
  110. reds = RedisObject()
  111. identifyingCode = reds.get_data(key=email + '_identifyingCode')
  112. if identifyingCode is False:
  113. user_qs = Device_User.objects.filter(username=email)
  114. email_qs = Device_User.objects.filter(userEmail=email)
  115. if user_qs.exists():
  116. return response.json(103)
  117. elif email_qs.exists():
  118. return response.json(103)
  119. else:
  120. identifyingCode = RandomStr(6, True)
  121. if reds.set_data(key=email + '_identifyingCode', val=identifyingCode, expire=AuthCode_Expire):
  122. send_data = TemplateService.email_message(type='register_code', language=response.lang)
  123. ses = SesClassObject()
  124. send_res = ses.send_email(
  125. send_address_list=[email],
  126. subject=send_data['title'],
  127. body=send_data['body'].replace("{username}", email).replace("{captcha}",
  128. str(identifyingCode))
  129. )
  130. if send_res:
  131. return response.json(0, {'identifyingCode': identifyingCode})
  132. else:
  133. reds.del_data(key=email + '_identifyingCode')
  134. return response.json(44)
  135. else:
  136. return response.json(10, '生成缓存系统错误')
  137. else:
  138. return response.json(89, {'identifyingCode': identifyingCode})
  139. else:
  140. return response.json(107)
  141. # 验证码注册
  142. class registerView(TemplateView):
  143. @method_decorator(csrf_exempt)
  144. def dispatch(self, *args, **kwargs):
  145. return super(registerView, self).dispatch(*args, **kwargs)
  146. def post(self, request, *args, **kwargs):
  147. request.encoding = 'utf-8'
  148. request_dict = request.POST
  149. return self.validates(request_dict)
  150. def get(self, request, *args, **kwargs):
  151. request.encoding = 'utf-8'
  152. request_dict = request.GET
  153. return self.validates(request_dict)
  154. def validates(self, request_dict):
  155. username = request_dict.get('userName', None)
  156. userEmail = request_dict.get('userEmail', None)
  157. password = request_dict.get('userPwd', None)
  158. authCode = request_dict.get('identifyingCode', None)
  159. language = request_dict.get('language', None)
  160. unique = request_dict.get('unique', None)
  161. if unique:
  162. delete_local_account(unique)
  163. response = ResponseObject(language)
  164. if username and password and authCode:
  165. # 过滤空格
  166. username = username.strip()
  167. if userEmail:
  168. userEmail = userEmail.strip()
  169. return self.register(username, userEmail, password, authCode, response)
  170. else:
  171. return response.json(800)
  172. def register(self, username, userEmail, password, authCode, response):
  173. dataValid = DataValid()
  174. reds = RedisObject()
  175. identifyingCode = reds.get_data(key=username + '_identifyingCode')
  176. if identifyingCode is False:
  177. if userEmail:
  178. identifyingCode = reds.get_data(key=userEmail + '_identifyingCode')
  179. if identifyingCode is False:
  180. return response.json(120)
  181. else:
  182. username = userEmail
  183. else:
  184. return response.json(120)
  185. if authCode != identifyingCode:
  186. return response.json(121)
  187. if dataValid.password_validate(password):
  188. if dataValid.email_validate(username):
  189. if userEmail:
  190. print(userEmail)
  191. emailValid = Device_User.objects.filter(userEmail=userEmail)
  192. if emailValid.exists():
  193. return response.json(103)
  194. else:
  195. userEmail = username
  196. if username:
  197. nameValid = Device_User.objects.filter(username=username)
  198. if nameValid.exists():
  199. return response.json(101)
  200. try:
  201. create_data = {
  202. "username": username,
  203. "NickName": username,
  204. "userEmail": userEmail,
  205. "password": make_password(password),
  206. "userID": CommonService.getUserID(μs=False, setOTAID=True),
  207. "is_active": True,
  208. "user_isValid": True,
  209. }
  210. users = Device_User.objects.create(**create_data)
  211. except Exception as e:
  212. errorInfo = traceback.format_exc()
  213. print(errorInfo)
  214. return response.json(424, repr(e))
  215. else:
  216. if reds.del_data(key=username + '_identifyingCode'):
  217. return response.json(0, {
  218. "user": {
  219. "userID": users.userID,
  220. "username": users.username,
  221. "userEmail": users.userEmail,
  222. "NickName": users.NickName,
  223. "userIconUrl": str(users.userIconUrl),
  224. "is_superuser": users.is_superuser,
  225. "is_active": users.is_active,
  226. "data_joined": date_handler(users.data_joined),
  227. "last_login": date_handler(users.last_login),
  228. }
  229. })
  230. else:
  231. return response.json(10, '删除缓存验证码错误')
  232. elif dataValid.mobile_validate(username):
  233. nameValid = Device_User.objects.filter(username=username)
  234. if nameValid:
  235. return response.json(101)
  236. try:
  237. create_data = {
  238. "username": username,
  239. "NickName": username,
  240. "userEmail": userEmail,
  241. "password": make_password(password),
  242. "userID": CommonService.getUserID(μs=False, setOTAID=True),
  243. "is_active": True,
  244. "user_isValid": True,
  245. }
  246. users = Device_User.objects.create(**create_data)
  247. except Exception as e:
  248. errorInfo = traceback.format_exc()
  249. print(errorInfo)
  250. return response.json(424, repr(e))
  251. else:
  252. if reds.del_data(key=username + '_identifyingCode'):
  253. return response.json(0, {
  254. "user": {
  255. "userID": users.userID,
  256. "username": users.username,
  257. "userEmail": users.userEmail,
  258. "NickName": users.NickName,
  259. "userIconUrl": str(users.userIconUrl),
  260. "is_superuser": users.is_superuser,
  261. "is_active": users.is_active,
  262. "data_joined": date_handler(users.data_joined),
  263. "last_login": date_handler(users.last_login),
  264. }
  265. })
  266. else:
  267. return response.json(10, '删除缓存验证码错误')
  268. else:
  269. return response.json(107)
  270. else:
  271. return response.json(109)
  272. # 登出
  273. class LogoutView(TemplateView):
  274. @method_decorator(csrf_exempt)
  275. def dispatch(self, *args, **kwargs):
  276. return super(LogoutView, self).dispatch(*args, **kwargs)
  277. def post(self, request, *args, **kwargs):
  278. request.encoding = 'utf-8'
  279. request_dict = request.POST
  280. return self.Logout(request_dict)
  281. def get(self, request, *args, **kwargs):
  282. request.encoding = 'utf-8'
  283. request_dict = request.GET
  284. return self.Logout(request_dict)
  285. def Logout(self, request_dict):
  286. response = ResponseObject()
  287. token = request_dict.get('token')
  288. tko = TokenObject(token)
  289. if tko.code == 0:
  290. Device_User.objects.filter(userID=tko.userID).update(online=False)
  291. redisObj = RedisObject(db=3)
  292. redisObj.del_data(key=tko.userID)
  293. m_code = request_dict.get('m_code', None)
  294. if m_code:
  295. userID = tko.userID
  296. try:
  297. UidPushModel.objects.filter(userID_id=userID, m_code=m_code).delete()
  298. except Exception as e:
  299. pass
  300. else:
  301. pass
  302. return response.json(0)
  303. else:
  304. return response.json(tko.code)
  305. # 修改密码
  306. class ChangePwdView(TemplateView):
  307. @method_decorator(csrf_exempt)
  308. def dispatch(self, *args, **kwargs):
  309. return super(ChangePwdView, self).dispatch(*args, **kwargs)
  310. def post(self, request, *args, **kwargs):
  311. request.encoding = 'utf-8'
  312. request_dict = request.POST
  313. return self.validates(request_dict)
  314. def get(self, request, *args, **kwargs):
  315. request.encoding = 'utf-8'
  316. request_dict = request.GET
  317. return self.validates(request_dict)
  318. def validates(self, request_dict):
  319. token = request_dict.get('token', None)
  320. oldPwd = request_dict.get('oldPwd', None)
  321. newPwd = request_dict.get('newPwd', None)
  322. response = ResponseObject()
  323. if oldPwd is None and newPwd is None:
  324. return response.json(800)
  325. tko = TokenObject(token)
  326. response.lang = tko.lang
  327. if tko.code != 0:
  328. return response.json(tko.code)
  329. return self.updatePwd(tko.userID, oldPwd, newPwd, response)
  330. def updatePwd(self, userID, oldPwd, newPwd, response):
  331. user_qs = Device_User.objects.filter(userID=userID)
  332. if not user_qs.exists():
  333. return response.json(104)
  334. c_p = check_password(oldPwd, user_qs[0].password)
  335. # 密码是否正确
  336. if not c_p:
  337. return response.json(111)
  338. update = user_qs.update(password=make_password(newPwd))
  339. if update:
  340. return response.json(0)
  341. else:
  342. return response.json(177)
  343. # 修改密码v3
  344. class v3ChangePwdView(TemplateView):
  345. @method_decorator(csrf_exempt)
  346. def dispatch(self, *args, **kwargs):
  347. return super(v3ChangePwdView, self).dispatch(*args, **kwargs)
  348. def post(self, request, *args, **kwargs):
  349. request.encoding = 'utf-8'
  350. request_dict = request.POST
  351. return self.validates(request_dict)
  352. def get(self, request, *args, **kwargs):
  353. request.encoding = 'utf-8'
  354. request_dict = request.GET
  355. return self.validates(request_dict)
  356. def validates(self, request_dict):
  357. token = request_dict.get('token', None)
  358. oldPwd = request_dict.get('oldPwd', None)
  359. newPwd = request_dict.get('newPwd', None)
  360. response = ResponseObject()
  361. # 解密
  362. try:
  363. for i in range(1, 4):
  364. if i == 1:
  365. oldPwd = base64.b64decode(oldPwd)
  366. oldPwd = oldPwd.decode('utf-8')
  367. oldPwd = oldPwd[1:-1]
  368. if i == 2:
  369. oldPwd = base64.b64decode(oldPwd)
  370. oldPwd = oldPwd.decode('utf-8')
  371. oldPwd = oldPwd[2:-2]
  372. if i == 3:
  373. oldPwd = base64.b64decode(oldPwd)
  374. oldPwd = oldPwd.decode('utf-8')
  375. oldPwd = oldPwd[3:-3]
  376. for i in range(1, 4):
  377. if i == 1:
  378. newPwd = base64.b64decode(newPwd)
  379. newPwd = newPwd.decode('utf-8')
  380. newPwd = newPwd[1:-1]
  381. if i == 2:
  382. newPwd = base64.b64decode(newPwd)
  383. newPwd = newPwd.decode('utf-8')
  384. newPwd = newPwd[2:-2]
  385. if i == 3:
  386. newPwd = base64.b64decode(newPwd)
  387. newPwd = newPwd.decode('utf-8')
  388. newPwd = newPwd[3:-3]
  389. except Exception as e:
  390. return response.json(111)
  391. else:
  392. if oldPwd is None and newPwd is None:
  393. return response.json(800)
  394. tko = TokenObject(token)
  395. response.lang = tko.lang
  396. if tko.code != 0:
  397. return response.json(tko.code)
  398. return self.updatePwd(tko.userID, oldPwd, newPwd, response)
  399. def updatePwd(self, userID, oldPwd, newPwd, response):
  400. user_qs = Device_User.objects.filter(userID=userID)
  401. if not user_qs.exists():
  402. return response.json(104)
  403. c_p = check_password(oldPwd, user_qs[0].password)
  404. # 密码是否正确
  405. if not c_p:
  406. return response.json(111)
  407. update = user_qs.update(password=make_password(newPwd))
  408. if update:
  409. return response.json(0)
  410. else:
  411. return response.json(177)
  412. class ForgetPwdView(TemplateView):
  413. '''
  414. 忘记密码
  415. '''
  416. @method_decorator(csrf_exempt)
  417. def dispatch(self, *args, **kwargs):
  418. return super(ForgetPwdView, self).dispatch(*args, **kwargs)
  419. @ratelimit(key='ip', rate='1/m')
  420. def get(self, request, *args, **kwargs):
  421. request.encoding = 'utf-8'
  422. response = ResponseObject()
  423. was_limited = getattr(request, 'limited', False)
  424. if was_limited is True:
  425. return response.json(5)
  426. userName = request.GET.get('userName', None)
  427. return self.ValidationError(userName, response)
  428. @ratelimit(key='ip', rate='1/m')
  429. def post(self, request):
  430. request.encoding = 'utf-8'
  431. userName = request.POST.get('userName', None)
  432. response = ResponseObject()
  433. was_limited = getattr(request, 'limited', False)
  434. if was_limited is True:
  435. return response.json(5)
  436. return self.ValidationError(userName, response)
  437. def ValidationError(self, userName, response):
  438. # return response.json(475)
  439. if userName != None:
  440. userName = userName.strip()
  441. return self.ForgetPwd(userName, response)
  442. else:
  443. return response.json(800)
  444. def ForgetPwd(self, userName, response):
  445. dataValid = DataValid()
  446. if dataValid.mobile_validate(userName):
  447. User = Device_User.objects.filter(username=userName)
  448. elif dataValid.email_validate(userName):
  449. User = Device_User.objects.filter(username=userName)
  450. if not User.exists():
  451. User = Device_User.objects.filter(userEmail=userName)
  452. else:
  453. return response.json(104)
  454. if User:
  455. email = User[0].userEmail
  456. userID = User[0].userID
  457. if email:
  458. redisObj = RedisObject()
  459. reset_pwd = redisObj.get_data(key=userID + '_email_reset_pwd')
  460. if reset_pwd is False:
  461. tko = TokenObject()
  462. rest = tko.generate(data={'userID': userID, 'user': userName})
  463. token = rest['access_token']
  464. reset_pwd = CommonService.RandomStr(6)
  465. send_data = TemplateService.email_message(type='forget', language='en')
  466. reset_link = '{server_host}account/email-re-pwd?token={token}'.format(
  467. server_host=SERVER_DOMAIN, token=token)
  468. send_body = send_data['body'].format(username=email, reset_pwd=reset_pwd, reset_link=reset_link)
  469. ses = SesClassObject()
  470. send_res = ses.send_email(send_address_list=[email], subject=send_data['title'], body=send_body)
  471. if send_res is True:
  472. if redisObj.set_data(key=userID + '_email_reset_pwd', val=reset_pwd, expire=3600):
  473. return response.json(0)
  474. else:
  475. return response.json(10, '存储验证失败')
  476. else:
  477. return response.json(89)
  478. else:
  479. return response.json(103)
  480. else:
  481. return response.json(104)
  482. class EmailResetPwdView(TemplateView):
  483. @method_decorator(csrf_exempt)
  484. def dispatch(self, *args, **kwargs):
  485. return super(EmailResetPwdView, self).dispatch(*args, **kwargs)
  486. # 查询
  487. def get(self, request, *args, **kwargs):
  488. response = ResponseObject()
  489. request_dict = request.GET
  490. return self.validate(request_dict, response, *args, **kwargs)
  491. # 认证登录
  492. def post(self, request, *args, **kwargs):
  493. response = ResponseObject()
  494. try:
  495. print(request.body.decode("utf-8"))
  496. json_data = json.loads(request.body.decode("utf-8"))
  497. except Exception as e:
  498. return response.json(10, repr(e))
  499. else:
  500. request_dict = json_data
  501. return self.validate(request_dict, response, *args, **kwargs)
  502. def validate(self, request_dict, response, *args, **kwargs):
  503. token = request_dict.get('token', None)
  504. tko = TokenObject(token)
  505. if tko.code == 0:
  506. redisObj = RedisObject()
  507. userID = tko.userID
  508. reset_pwd = redisObj.get_data(key=userID + '_email_reset_pwd')
  509. if reset_pwd is not False:
  510. user_qs = Device_User.objects.filter(userID=userID)
  511. if user_qs.exists():
  512. redisObj.del_data(key=userID + '_email_reset_pwd')
  513. is_update = user_qs.update(password=make_password(reset_pwd))
  514. if is_update:
  515. return HttpResponseRedirect(
  516. "http://www.zositechc.cn/web/html/paw_update_success.html?code=" + reset_pwd)
  517. else:
  518. return response.json(10)
  519. else:
  520. return response.json(104)
  521. else:
  522. return HttpResponseRedirect('http://www.zositechc.cn/web/html/paw_update_unsuccessful.html?lang=en')
  523. else:
  524. return HttpResponseRedirect('http://www.zositechc.cn/web/html/paw_update_unsuccessful.html?lang=en')
  525. class refreshTokenView(TemplateView):
  526. @method_decorator(csrf_exempt)
  527. def dispatch(self, *args, **kwargs):
  528. return super(refreshTokenView, self).dispatch(*args, **kwargs)
  529. def post(self, request, *args, **kwargs):
  530. request.encoding = 'utf-8'
  531. content_type = request.META.get('CONTENT_TYPE', None)
  532. if content_type == 'application/json':
  533. request_dict = json.loads(request.body.decode('utf-8'))
  534. else:
  535. request_dict = request.POST
  536. return self.validation(request_dict)
  537. def get(self, request, *args, **kwargs):
  538. request.encoding = 'utf-8'
  539. request_dict = request.GET
  540. return self.validation(request_dict)
  541. def validation(self, request_dict):
  542. token = request_dict.get('token', None)
  543. lang = request_dict.get('lang', None)
  544. language = request_dict.get('language', None)
  545. if lang is None:
  546. lang = language
  547. response = ResponseObject(lang)
  548. if token is not None:
  549. tko = TokenObject(token)
  550. tko.lang = lang
  551. res = tko.refresh()
  552. userID = tko.userID
  553. if tko.code == 0:
  554. # 更新用户扩展信息语言
  555. try:
  556. user_ex_qs = UserExModel.objects.filter(userID_id=userID)
  557. if user_ex_qs.exists():
  558. nowTime = int(time.time())
  559. update_dict = {
  560. 'updTime': nowTime,
  561. 'region': lang
  562. }
  563. user_ex_qs.update(**update_dict)
  564. except Exception as e:
  565. pass
  566. # # #
  567. return response.json(0, res)
  568. else:
  569. return response.json(tko.code)
  570. else:
  571. return response.json(444, 'token')
  572. class refreshTokenViewV3(TemplateView):
  573. @method_decorator(csrf_exempt)
  574. def dispatch(self, *args, **kwargs):
  575. return super(refreshTokenViewV3, self).dispatch(*args, **kwargs)
  576. def post(self, request, *args, **kwargs):
  577. request.encoding = 'utf-8'
  578. content_type = request.META.get('CONTENT_TYPE', None)
  579. if content_type == 'application/json':
  580. request_dict = json.loads(request.body.decode('utf-8'))
  581. else:
  582. request_dict = request.POST
  583. return self.validation(request_dict)
  584. def get(self, request, *args, **kwargs):
  585. request.encoding = 'utf-8'
  586. request_dict = request.GET
  587. return self.validation(request_dict)
  588. def validation(self, request_dict):
  589. token = request_dict.get('token', None)
  590. lang = request_dict.get('lang', None)
  591. language = request_dict.get('language', None)
  592. password = request_dict.get('userPwd', None)
  593. response = ResponseObject(lang)
  594. if not token:
  595. return response.json(444, 'token')
  596. if lang is None:
  597. lang = language
  598. try:
  599. # token检验
  600. tko = TokenObject(token)
  601. if tko.code != 0:
  602. return response.json(tko.code)
  603. userID = tko.userID
  604. tko.lang = lang
  605. if password: # 检验密码
  606. password = password.strip()
  607. # 解密
  608. for i in range(1, 4):
  609. password = base64.b64decode(password)
  610. password = password.decode('utf-8')
  611. password = password[i:-i]
  612. user_qs = Device_User.objects.filter(userID=userID).values('password')
  613. if not user_qs.exists():
  614. return response.json(104)
  615. if not check_password(password, user_qs[0].password):
  616. return response.json(111)
  617. res = tko.refresh() # 更新refresh_token
  618. # 更新用户扩展信息语言
  619. user_ex_qs = UserExModel.objects.filter(userID_id=userID)
  620. if user_ex_qs.exists():
  621. nowTime = int(time.time())
  622. update_dict = {
  623. 'updTime': nowTime,
  624. 'region': lang
  625. }
  626. user_ex_qs.update(**update_dict)
  627. return response.json(0, res)
  628. except Exception as e:
  629. print(e)
  630. return response.json(500, repr(e))
  631. # 获取验证码
  632. class v2authCodeView(TemplateView):
  633. @method_decorator(csrf_exempt)
  634. def dispatch(self, *args, **kwargs):
  635. return super(v2authCodeView, self).dispatch(*args, **kwargs)
  636. @ratelimit(key='ip', rate='2/m')
  637. def post(self, request, *args, **kwargs):
  638. request.encoding = 'utf-8'
  639. lang = request.POST.get('lang', None)
  640. if not lang:
  641. lang = request.POST.get('language', None)
  642. response = ResponseObject(lang)
  643. request_dict = request.POST
  644. phone = request_dict.get('phone', None)
  645. if phone is not None:
  646. was_limited = getattr(request, 'limited', False)
  647. if was_limited is True:
  648. return response.json(5)
  649. return self.ValidationError(request_dict, response)
  650. @ratelimit(key='ip', rate='2/m')
  651. def get(self, request, *args, **kwargs):
  652. # Device_User.objects.filter(userEmail='chanjunkai@163.com').delete()
  653. request.encoding = 'utf-8'
  654. lang = request.GET.get('lang', None)
  655. if not lang:
  656. lang = request.GET.get('language', None)
  657. response = ResponseObject(lang)
  658. was_limited = getattr(request, 'limited', False)
  659. if was_limited is True:
  660. return response.json(5)
  661. request_dict = request.GET
  662. return self.ValidationError(request_dict, response)
  663. def ValidationError(self, request_dict, response):
  664. email = request_dict.get('email', None)
  665. phone = request_dict.get('phone', None)
  666. country_code = request_dict.get('country_code', None)
  667. sign_name = request_dict.get('sign_name', None)
  668. if email is not None:
  669. email = email.strip()
  670. # 阿里云的发送邮箱的调用方法
  671. return self.aliyun_emailCode(email, response)
  672. # return self.emailCode(email, response)
  673. elif phone is not None:
  674. phone = phone.strip()
  675. if country_code is None:
  676. return self.phoneCode(phone, response, sign_name)
  677. else:
  678. country_code = str(country_code.strip())
  679. return self.phoneCodeV2(country_code, phone, response, sign_name)
  680. else:
  681. return response.json(444)
  682. def emailCode(self, email, response):
  683. dataValid = DataValid()
  684. # 邮箱匹配
  685. if dataValid.email_validate(email) is False:
  686. return response.json(107)
  687. reds = RedisObject()
  688. identifyingCode = reds.get_data(key=email + '_identifyingCode')
  689. # 是否以获取邮箱验证码
  690. if identifyingCode:
  691. return response.json(89)
  692. user_qs = Device_User.objects.filter(username=email)
  693. email_qs = Device_User.objects.filter(userEmail=email)
  694. # 邮箱用户是否已存在
  695. if user_qs.exists():
  696. return response.json(103)
  697. elif email_qs.exists():
  698. return response.json(103)
  699. # 生成随机6位数
  700. identifyingCode = RandomStr(6, True)
  701. # 设置随机数缓存生命周期
  702. send_data = TemplateService.email_message(type='register_code', language=response.lang)
  703. ses = SesClassObject()
  704. # 发送邮件
  705. send_res = ses.send_email(
  706. send_address_list=[email],
  707. subject=send_data['title'],
  708. body=send_data['body'].replace("{username}", email).replace("{captcha}",
  709. str(identifyingCode))
  710. )
  711. if send_res is not True:
  712. return response.json(44)
  713. if reds.set_data(key=email + '_identifyingCode', val=identifyingCode, expire=600) is not True:
  714. return response.json(10, 'error')
  715. return response.json(0)
  716. # return response.json(0, {'identifyingCode': identifyingCode})
  717. # 阿里云获取邮箱验证码
  718. def aliyun_emailCode(self, email, response):
  719. print('阿里云开始')
  720. dataValid = DataValid()
  721. # 邮箱匹配
  722. if dataValid.email_validate(email) is False:
  723. return response.json(107)
  724. reds = RedisObject()
  725. identifyingCode = reds.get_data(key=email + '_identifyingCode')
  726. # 是否以获取邮箱验证码
  727. if identifyingCode:
  728. return response.json(89)
  729. user_qs = Device_User.objects.filter(username=email)
  730. email_qs = Device_User.objects.filter(userEmail=email)
  731. # 邮箱用户是否已存在
  732. if user_qs.exists():
  733. return response.json(103)
  734. elif email_qs.exists():
  735. return response.json(103)
  736. # 生成随机6位数
  737. identifyingCode = RandomStr(6, True)
  738. # 设置随机数缓存生命周期
  739. send_data = TemplateService.email_message(type='register_code', language=response.lang)
  740. ses = SesClassObject()
  741. # 发送邮件
  742. send_res = ses.alyEmailCode(
  743. send_address_list=[email],
  744. subject=send_data['title'],
  745. body=send_data['body'].replace("{username}", email).replace("{captcha}", str(identifyingCode))
  746. )
  747. if send_res is not True:
  748. return response.json(44)
  749. if reds.set_data(key=email + '_identifyingCode', val=identifyingCode, expire=600) is not True:
  750. return response.json(10, 'error')
  751. return response.json(0)
  752. def phoneCode(self, phone, response, sign_name):
  753. dataValid = DataValid()
  754. if dataValid.mobile_validate(phone) is not True:
  755. return response.json(107)
  756. reds = RedisObject()
  757. reds_key = str(phone) + '_identifyingCode'
  758. identifyingCode = reds.get_data(key=reds_key)
  759. reds_key_ttl = reds.get_ttl(key=reds_key)
  760. if reds_key_ttl > 240 and identifyingCode:
  761. # if identifyingCode :
  762. return response.json(90)
  763. user_qs = Device_User.objects.filter(username=phone)
  764. phone_qs = Device_User.objects.filter(phone=phone)
  765. if user_qs.exists() or phone_qs.exists():
  766. return response.json(101)
  767. identifyingCode = RandomStr(6, True)
  768. # 发送手机验证码
  769. aliSms = AliSmsObject()
  770. if sign_name == 'zosi':
  771. sign_ms = '周视'
  772. else:
  773. sign_ms = 'Ansjer'
  774. res = aliSms.send_code_sms(phone=phone, code=identifyingCode, sign_name=sign_ms,
  775. temp_msg='SMS_151600991')
  776. print(res)
  777. code = res["Code"]
  778. if code == "OK":
  779. if reds.set_data(key=reds_key, val=identifyingCode, expire=300) is not True:
  780. # if reds.set_data(key=phone + '_identifyingCode', val=identifyingCode, expire=60) is not True:
  781. return response.json(10, '生成缓存系统错误')
  782. return response.json(0)
  783. else:
  784. msg = res["Message"]
  785. if code == "isv.MOBILE_NUMBER_ILLEGAL":
  786. if response.lang == "cn":
  787. msg = phone+"非法手机"
  788. return response.json(10,msg)
  789. def phoneCodeV2(self, country_code, phone, response, sign_name):
  790. dataValid = DataValid()
  791. if dataValid.mobile_validate(phone) is not True:
  792. return response.json(107)
  793. reds = RedisObject()
  794. reds_key = str(phone) + '_identifyingCode'
  795. identifyingCode = reds.get_data(key=reds_key)
  796. reds_key_ttl = reds.get_ttl(key=reds_key)
  797. if reds_key_ttl > 240 and identifyingCode:
  798. # if identifyingCode :
  799. return response.json(90)
  800. user_qs = Device_User.objects.filter(username=phone)
  801. phone_qs = Device_User.objects.filter(phone=phone)
  802. if user_qs.exists() or phone_qs.exists():
  803. return response.json(101)
  804. identifyingCode = RandomStr(6, True)
  805. # 短信签名
  806. # sign_name_dict = {
  807. # 'ansjer':'Ansjer',
  808. # 'zosi':'周视'
  809. # }
  810. sign_ms = ''
  811. if country_code == '86':
  812. # 国内短信推送模板
  813. temp_msg = 'SMS_151600991'
  814. rec_phone = phone
  815. if sign_name == 'zosi':
  816. sign_ms = '周视'
  817. else:
  818. sign_ms = 'Ansjer'
  819. else:
  820. # 国际短信推送模板
  821. temp_msg = 'SMS_172165867'
  822. rec_phone = country_code + phone
  823. sign_ms = 'Ansjer'
  824. # 发送手机验证码
  825. aliSms = AliSmsObject()
  826. res = aliSms.send_code_sms(phone=rec_phone, code=identifyingCode, sign_name=sign_ms,
  827. temp_msg=temp_msg)
  828. print(res)
  829. if res["Code"] == "OK":
  830. # if reds.set_data(key=reds_key, val=identifyingCode, expire=60) is not True:
  831. if reds.set_data(key=reds_key, val=identifyingCode, expire=300) is not True:
  832. return response.json(10, '生成缓存系统错误')
  833. return response.json(0)
  834. else:
  835. return response.json(10, res["Message"])
  836. # 验证码注册
  837. class v2registerView(TemplateView):
  838. @method_decorator(csrf_exempt)
  839. def dispatch(self, *args, **kwargs):
  840. return super(v2registerView, self).dispatch(*args, **kwargs)
  841. def post(self, request, *args, **kwargs):
  842. request.encoding = 'utf-8'
  843. request_dict = request.POST
  844. return self.validates(request_dict)
  845. def get(self, request, *args, **kwargs):
  846. request.encoding = 'utf-8'
  847. request_dict = request.GET
  848. return self.validates(request_dict)
  849. def validates(self, request_dict):
  850. phone = request_dict.get('phone', None)
  851. email = request_dict.get('email', None)
  852. password = request_dict.get('password', None)
  853. authcode = request_dict.get('authcode', None)
  854. lang = request_dict.get('lang', None)
  855. unique = request_dict.get('unique', None)
  856. number = request_dict.get('number', None)
  857. if unique:
  858. delete_local_account(unique)
  859. response = ResponseObject(lang)
  860. if not lang:
  861. lang = request_dict.get('language', None)
  862. if password is None:
  863. return response.json(444, 'password')
  864. if authcode is None:
  865. return response.json(444, 'identifyingCode')
  866. if phone is not None:
  867. return self.do_phone_register(phone, password, authcode, number, response)
  868. elif email is not None:
  869. return self.do_email_register(email, password, authcode, number, response)
  870. else:
  871. return response.json(444, 'phone or email')
  872. def do_phone_register(self, phone, password, authcode, number, response):
  873. data_valid = DataValid()
  874. if data_valid.mobile_validate(phone) is not True:
  875. return response.json(100)
  876. if data_valid.password_validate(password) is not True:
  877. return response.json(109)
  878. reds = RedisObject()
  879. identifyingCode = reds.get_data(key=phone + '_identifyingCode')
  880. # 判断验证码是否过期
  881. if identifyingCode is False:
  882. return response.json(120)
  883. # 验证码是否正确
  884. if authcode != identifyingCode:
  885. return response.json(121)
  886. phone_qs = Device_User.objects.filter(Q(phone=phone) | Q(username=phone))
  887. # 是否已存在
  888. if phone_qs.exists():
  889. return response.json(101)
  890. try:
  891. create_data = {
  892. "username": phone,
  893. "NickName": phone,
  894. "phone": phone,
  895. "password": make_password(password),
  896. "userID": CommonService.getUserID(μs=False, setOTAID=True),
  897. "is_active": True,
  898. "user_isValid": True,
  899. }
  900. if number:
  901. create_data["region_country"] = number
  902. users = Device_User.objects.create(**create_data)
  903. except Exception as e:
  904. errorInfo = traceback.format_exc()
  905. print(errorInfo)
  906. return response.json(424, repr(e))
  907. else:
  908. if not reds.del_data(key=phone + '_identifyingCode'):
  909. return response.json(10, '删除缓存验证码错误')
  910. return self.do_login(phone_qs, response)
  911. def do_login(self, user_qs, response):
  912. now_time = datetime.datetime.utcnow().replace(tzinfo=utc).astimezone(utc)
  913. user_qs.update(last_login=now_time, online=True)
  914. userID = user_qs[0].userID
  915. print('userID' + userID)
  916. tko = TokenObject()
  917. user_list = user_qs.values("NickName", "userIconUrl", "userIconPath", "username", "userEmail", "phone")
  918. res = tko.generate(data={'userID': userID, 'lang': response.lang, 'user': user_list[0]["username"]})
  919. # 增加角色
  920. user_qs[0].role.add(Role.objects.get(rid=1))
  921. role_dict = ModelService.own_role(userID=userID)
  922. res['rid'] = role_dict['rid']
  923. res['roleName'] = role_dict['roleName']
  924. res['permList'] = ModelService.own_permission(userID)
  925. res['userID'] = userID
  926. # 昵称,邮箱,电话,刷新,头像
  927. userIconPath = str(user_list[0]["userIconPath"])
  928. if userIconPath and userIconPath.find('static/') != -1:
  929. userIconPath = userIconPath.replace('static/', '').replace('\\', '/')
  930. res['userIconUrl'] = SERVER_DOMAIN + 'account/getAvatar/' + userIconPath
  931. else:
  932. res['userIconUrl'] = ''
  933. res['NickName'] = user_list[0]["NickName"] if user_list[0]["NickName"] is not None else ''
  934. res['username'] = user_list[0]["username"] if user_list[0]["username"] is not None else ''
  935. res['userEmail'] = user_list[0]["userEmail"] if user_list[0]["userEmail"] is not None else ''
  936. res['phone'] = user_list[0]["phone"] if user_list[0]["phone"] is not None else ''
  937. print(res)
  938. return response.json(0, res)
  939. def do_email_register(self, email, password, authcode, number, response):
  940. data_valid = DataValid()
  941. if data_valid.email_validate(email) is not True:
  942. return response.json(105)
  943. if data_valid.password_validate(password) is not True:
  944. return response.json(109)
  945. reds = RedisObject()
  946. identifyingCode = reds.get_data(key=email + '_identifyingCode')
  947. # 判断验证码是否过期
  948. if identifyingCode is False:
  949. return response.json(120)
  950. # 验证码是否正确
  951. if authcode != identifyingCode:
  952. return response.json(121)
  953. email_qs = Device_User.objects.filter(Q(userEmail=email) | Q(username=email))
  954. # 是否已存在
  955. if email_qs.exists():
  956. return response.json(103)
  957. try:
  958. create_data = {
  959. "username": email,
  960. "NickName": email,
  961. "userEmail": email,
  962. "password": make_password(password),
  963. "userID": CommonService.getUserID(μs=False, setOTAID=True),
  964. "is_active": True,
  965. "user_isValid": True,
  966. }
  967. if number:
  968. create_data["region_country"] = number
  969. users = Device_User.objects.create(**create_data)
  970. except Exception as e:
  971. errorInfo = traceback.format_exc()
  972. print(errorInfo)
  973. return response.json(424, repr(e))
  974. else:
  975. if not reds.del_data(key=email + '_identifyingCode'):
  976. return response.json(10, '删除缓存验证码错误')
  977. return self.do_login(email_qs, response)
  978. # 验证码注册v3
  979. class v3registerView(TemplateView):
  980. @method_decorator(csrf_exempt)
  981. def dispatch(self, *args, **kwargs):
  982. return super(v3registerView, self).dispatch(*args, **kwargs)
  983. def post(self, request, *args, **kwargs):
  984. request.encoding = 'utf-8'
  985. request_dict = request.POST
  986. return self.validates(request_dict)
  987. def get(self, request, *args, **kwargs):
  988. request.encoding = 'utf-8'
  989. request_dict = request.GET
  990. return self.validates(request_dict)
  991. def validates(self, request_dict):
  992. phone = request_dict.get('phone', None)
  993. email = request_dict.get('email', None)
  994. password = request_dict.get('password', None)
  995. authcode = request_dict.get('authcode', None)
  996. lang = request_dict.get('lang', None)
  997. unique = request_dict.get('unique', None)
  998. number = request_dict.get('number', None)
  999. if unique:
  1000. delete_local_account(unique)
  1001. response = ResponseObject(lang)
  1002. # 解密
  1003. try:
  1004. for i in range(1, 4):
  1005. if i == 1:
  1006. password = base64.b64decode(password)
  1007. password = password.decode('utf-8')
  1008. password = password[1:-1]
  1009. if i == 2:
  1010. password = base64.b64decode(password)
  1011. password = password.decode('utf-8')
  1012. password = password[2:-2]
  1013. if i == 3:
  1014. password = base64.b64decode(password)
  1015. password = password.decode('utf-8')
  1016. password = password[3:-3]
  1017. print(password)
  1018. except Exception as e:
  1019. return response.json(111)
  1020. try:
  1021. for i in range(1, 4):
  1022. if i == 1:
  1023. authcode = base64.b64decode(authcode)
  1024. authcode = authcode.decode('utf-8')
  1025. authcode = authcode[1:-1]
  1026. if i == 2:
  1027. authcode = base64.b64decode(authcode)
  1028. authcode = authcode.decode('utf-8')
  1029. authcode = authcode[2:-2]
  1030. if i == 3:
  1031. authcode = base64.b64decode(authcode)
  1032. authcode = authcode.decode('utf-8')
  1033. authcode = authcode[3:-3]
  1034. print(authcode)
  1035. except Exception as e:
  1036. return response.json(121)
  1037. else:
  1038. if not lang:
  1039. lang = request_dict.get('language', None)
  1040. if password is None:
  1041. return response.json(444, 'password')
  1042. if authcode is None:
  1043. return response.json(444, 'identifyingCode')
  1044. if phone is not None:
  1045. return self.do_phone_register(phone, password, authcode, number, response)
  1046. elif email is not None:
  1047. return self.do_email_register(email, password, authcode, number, response)
  1048. else:
  1049. return response.json(444, 'phone or email')
  1050. def do_phone_register(self, phone, password, authcode, number, response):
  1051. data_valid = DataValid()
  1052. if data_valid.mobile_validate(phone) is not True:
  1053. return response.json(100)
  1054. if data_valid.password_validate(password) is not True:
  1055. return response.json(109)
  1056. reds = RedisObject()
  1057. identifyingCode = reds.get_data(key=phone + '_identifyingCode')
  1058. # 判断验证码是否过期
  1059. if identifyingCode is False:
  1060. return response.json(120)
  1061. # 验证码是否正确
  1062. if authcode != identifyingCode:
  1063. return response.json(121)
  1064. phone_qs = Device_User.objects.filter(Q(phone=phone) | Q(username=phone))
  1065. # 是否已存在
  1066. if phone_qs.exists():
  1067. return response.json(101)
  1068. try:
  1069. create_data = {
  1070. "username": phone,
  1071. "NickName": phone,
  1072. "phone": phone,
  1073. "password": make_password(password),
  1074. "userID": CommonService.getUserID(μs=False, setOTAID=True),
  1075. "is_active": True,
  1076. "user_isValid": True,
  1077. }
  1078. if number:
  1079. create_data["region_country"] = number
  1080. users = Device_User.objects.create(**create_data)
  1081. except Exception as e:
  1082. errorInfo = traceback.format_exc()
  1083. print(errorInfo)
  1084. return response.json(424, repr(e))
  1085. else:
  1086. if not reds.del_data(key=phone + '_identifyingCode'):
  1087. return response.json(10, '删除缓存验证码错误')
  1088. return self.do_login(phone_qs, response)
  1089. def do_login(self, user_qs, response):
  1090. now_time = datetime.datetime.utcnow().replace(tzinfo=utc).astimezone(utc)
  1091. user_qs.update(last_login=now_time, online=True)
  1092. userID = user_qs[0].userID
  1093. print('userID' + userID)
  1094. tko = TokenObject()
  1095. user_list = user_qs.values("NickName", "userIconUrl", "userIconPath", "username", "userEmail", "phone")
  1096. res = tko.generate(data={'userID': userID, 'lang': response.lang, 'user': user_list[0]["username"]})
  1097. # 增加角色
  1098. user_qs[0].role.add(Role.objects.get(rid=1))
  1099. role_dict = ModelService.own_role(userID=userID)
  1100. res['rid'] = role_dict['rid']
  1101. res['roleName'] = role_dict['roleName']
  1102. res['permList'] = ModelService.own_permission(userID)
  1103. res['userID'] = userID
  1104. # 昵称,邮箱,电话,刷新,头像
  1105. userIconPath = str(user_list[0]["userIconPath"])
  1106. if userIconPath and userIconPath.find('static/') != -1:
  1107. userIconPath = userIconPath.replace('static/', '').replace('\\', '/')
  1108. res['userIconUrl'] = SERVER_DOMAIN + 'account/getAvatar/' + userIconPath
  1109. else:
  1110. res['userIconUrl'] = ''
  1111. res['NickName'] = user_list[0]["NickName"] if user_list[0]["NickName"] is not None else ''
  1112. res['username'] = user_list[0]["username"] if user_list[0]["username"] is not None else ''
  1113. res['userEmail'] = user_list[0]["userEmail"] if user_list[0]["userEmail"] is not None else ''
  1114. res['phone'] = user_list[0]["phone"] if user_list[0]["phone"] is not None else ''
  1115. return response.json(0, res)
  1116. def do_email_register(self, email, password, authcode, number, response):
  1117. data_valid = DataValid()
  1118. if data_valid.email_validate(email) is not True:
  1119. return response.json(105)
  1120. if data_valid.password_validate(password) is not True:
  1121. return response.json(109)
  1122. reds = RedisObject()
  1123. identifyingCode = reds.get_data(key=email + '_identifyingCode')
  1124. # 判断验证码是否过期
  1125. if identifyingCode is False:
  1126. return response.json(120)
  1127. # 验证码是否正确
  1128. if authcode != identifyingCode:
  1129. return response.json(121)
  1130. email_qs = Device_User.objects.filter(Q(userEmail=email) | Q(username=email))
  1131. # 是否已存在
  1132. if email_qs.exists():
  1133. return response.json(103)
  1134. try:
  1135. create_data = {
  1136. "username": email,
  1137. "NickName": email,
  1138. "userEmail": email,
  1139. "password": make_password(password),
  1140. "userID": CommonService.getUserID(μs=False, setOTAID=True),
  1141. "is_active": True,
  1142. "user_isValid": True,
  1143. }
  1144. if number:
  1145. create_data["region_country"] = number
  1146. users = Device_User.objects.create(**create_data)
  1147. except Exception as e:
  1148. errorInfo = traceback.format_exc()
  1149. print(errorInfo)
  1150. return response.json(424, repr(e))
  1151. else:
  1152. if not reds.del_data(key=email + '_identifyingCode'):
  1153. return response.json(10, '删除缓存验证码错误')
  1154. return self.do_login(email_qs, response)
  1155. # 重置密码
  1156. # 忘记密码获取验证码v2
  1157. class v2forgetPwdCodeView(TemplateView):
  1158. @method_decorator(csrf_exempt)
  1159. def dispatch(self, *args, **kwargs):
  1160. return super(v2forgetPwdCodeView, self).dispatch(*args, **kwargs)
  1161. @ratelimit(key='ip', rate='1/m')
  1162. def get(self, request, *args, **kwargs):
  1163. request.encoding = 'utf-8'
  1164. request_dict = request.GET
  1165. lang = request_dict.get('lang')
  1166. response = ResponseObject(lang)
  1167. was_limited = getattr(request, 'limited', False)
  1168. if was_limited is True:
  1169. return response.json(5)
  1170. return self.ValidationError(request_dict, response)
  1171. @ratelimit(key='ip', rate='1/m')
  1172. def post(self, request):
  1173. request.encoding = 'utf-8'
  1174. request_dict = request.POST
  1175. lang = request_dict.get('lang')
  1176. response = ResponseObject(lang)
  1177. was_limited = getattr(request, 'limited', False)
  1178. if was_limited is True:
  1179. return response.json(5)
  1180. return self.ValidationError(request_dict, response)
  1181. def ValidationError(self, request_dict, response):
  1182. phone = request_dict.get('phone', None)
  1183. email = request_dict.get('email', None)
  1184. country_code = request_dict.get('country_code', None)
  1185. sign_name = request_dict.get('sign_name', None)
  1186. if phone is not None:
  1187. phone = phone.strip()
  1188. if country_code is None:
  1189. return self.do_send_phone_code(phone, response, sign_name)
  1190. else:
  1191. return self.do_v2_send_phone_code(country_code, phone, response, sign_name)
  1192. elif email is not None:
  1193. return self.do_send_email_code(email, response)
  1194. else:
  1195. return response.json(444, 'phone')
  1196. def do_send_phone_code(self, phone, response, sign_name):
  1197. data_valid = DataValid()
  1198. if data_valid.mobile_validate(phone) is not True:
  1199. return response.json(100)
  1200. user_qs = Device_User.objects.filter(Q(phone=phone) | Q(username=phone))
  1201. if not user_qs.exists():
  1202. return response.json(102)
  1203. reds = RedisObject()
  1204. reds_key = str(phone) + '_forgetPwdResetCode'
  1205. resetCode = reds.get_data(key=reds_key)
  1206. reds_key_ttl = reds.get_ttl(key=reds_key)
  1207. if reds_key_ttl > 240 and resetCode:
  1208. # if identifyingCode :
  1209. return response.json(90)
  1210. # if resetCode is True:
  1211. # return response.json(120)
  1212. resetCode = RandomStr(6, True)
  1213. if sign_name == 'zosi':
  1214. sign_ms = '周视'
  1215. else:
  1216. sign_ms = 'Ansjer'
  1217. aliSms = AliSmsObject()
  1218. res = aliSms.send_code_sms(phone=phone, code=resetCode, sign_name=sign_ms,
  1219. temp_msg='SMS_151675019')
  1220. if res["Code"] == "OK":
  1221. if not reds.set_data(key=reds_key, val=resetCode, expire=300):
  1222. return response.json(10, '生成缓存错误')
  1223. return response.json(0)
  1224. else:
  1225. return response.json(10, res["Message"])
  1226. def do_v2_send_phone_code(self, country_code, phone, response, sign_name):
  1227. data_valid = DataValid()
  1228. if data_valid.mobile_validate(phone) is not True:
  1229. return response.json(100)
  1230. user_qs = Device_User.objects.filter(Q(phone=phone) | Q(username=phone))
  1231. if not user_qs.exists():
  1232. return response.json(102)
  1233. reds = RedisObject()
  1234. reds_key = str(phone) + '_forgetPwdResetCode'
  1235. resetCode = reds.get_data(key=reds_key)
  1236. reds_key_ttl = reds.get_ttl(key=reds_key)
  1237. if reds_key_ttl > 240 and resetCode:
  1238. # if identifyingCode :
  1239. return response.json(90)
  1240. # if resetCode is True:
  1241. # return response.json(120)
  1242. resetCode = RandomStr(6, True)
  1243. aliSms = AliSmsObject()
  1244. if country_code == '86':
  1245. rec_phone = phone
  1246. temp_msg = 'SMS_151675019'
  1247. if sign_name == 'zosi':
  1248. sign_ms = '周视'
  1249. else:
  1250. sign_ms = 'Ansjer'
  1251. else:
  1252. temp_msg = 'SMS_172200051'
  1253. rec_phone = str(country_code) + str(phone)
  1254. sign_ms = 'Ansjer'
  1255. res = aliSms.send_code_sms(phone=rec_phone, code=resetCode, sign_name=sign_ms,
  1256. temp_msg=temp_msg)
  1257. if res["Code"] == "OK":
  1258. if not reds.set_data(key=reds_key, val=resetCode, expire=300):
  1259. return response.json(10, '生成缓存错误')
  1260. return response.json(0)
  1261. else:
  1262. return response.json(10, res["Message"])
  1263. def do_send_email_code(self, email, response):
  1264. data_valid = DataValid()
  1265. if data_valid.email_validate(email) is not True:
  1266. return response.json(105)
  1267. user_qs = Device_User.objects.filter(Q(userEmail=email) | Q(username=email))
  1268. if not user_qs.exists():
  1269. return response.json(104)
  1270. reds = RedisObject()
  1271. resetCode = reds.get_data(key=email + '_forgetPwdResetCode')
  1272. if resetCode is True:
  1273. return response.json(120)
  1274. resetCode = RandomStr(6, True)
  1275. if not reds.set_data(key=email + '_forgetPwdResetCode', val=resetCode, expire=600):
  1276. return response.json(10, '生成缓存错误')
  1277. send_data = TemplateService.email_message(type='forgetCode', language=response.lang)
  1278. send_body = send_data['body'].format(userEmail=email, email_valid_code=resetCode)
  1279. ses = SesClassObject()
  1280. # send_res = ses.send_email(send_address_list=[email], subject=send_data['title'], body=send_body)
  1281. # 阿里云发送邮箱
  1282. send_res = ses.alyEmailCode(send_address_list=[email], subject=send_data['title'], body=send_body)
  1283. if send_res is not True:
  1284. reds.del_data(key=email + '_forgetPwdResetCode')
  1285. return response.json(44)
  1286. return response.json(0)
  1287. # 忘记密码v2
  1288. class v2resetPwdByCodeView(TemplateView):
  1289. @method_decorator(csrf_exempt)
  1290. def dispatch(self, *args, **kwargs):
  1291. return super(v2resetPwdByCodeView, self).dispatch(*args, **kwargs)
  1292. def get(self, request, *args, **kwargs):
  1293. request.encoding = 'utf-8'
  1294. request_dict = request.GET
  1295. lang = request_dict.get('lang')
  1296. if not lang:
  1297. lang = request_dict.get('language', None)
  1298. response = ResponseObject(lang)
  1299. was_limited = getattr(request, 'limited', False)
  1300. if was_limited is True:
  1301. return response.json(5)
  1302. return self.ValidationError(request_dict, response)
  1303. def post(self, request):
  1304. request.encoding = 'utf-8'
  1305. request_dict = request.POST
  1306. lang = request_dict.get('lang')
  1307. if not lang:
  1308. lang = request_dict.get('language', None)
  1309. response = ResponseObject(lang)
  1310. was_limited = getattr(request, 'limited', False)
  1311. if was_limited is True:
  1312. return response.json(5)
  1313. return self.ValidationError(request_dict, response)
  1314. def ValidationError(self, request_dict, response):
  1315. phone = request_dict.get('phone', None)
  1316. email = request_dict.get('email', None)
  1317. password = request_dict.get('password', None)
  1318. authcode = request_dict.get('authcode', None)
  1319. print("1111111111111111111111")
  1320. if password is None or authcode is None:
  1321. return response.json(444, 'password,authcode')
  1322. authcode = authcode.strip()
  1323. password = password.strip()
  1324. if phone is not None:
  1325. phone = phone.strip()
  1326. return self.do_phone_pwd_reset(phone, authcode, password, response)
  1327. elif email is not None:
  1328. email = email.strip()
  1329. return self.do_email_pwd_reset(email, authcode, password, response)
  1330. else:
  1331. return response.json(444, 'phone')
  1332. def do_email_pwd_reset(self, email, authcode, password, response):
  1333. data_valid = DataValid()
  1334. if data_valid.email_validate(email) is not True:
  1335. return response.json(105)
  1336. if data_valid.password_validate(password) is not True:
  1337. return response.json(109)
  1338. user_qs = Device_User.objects.filter(Q(userEmail=email) | Q(username=email))
  1339. if not user_qs.exists():
  1340. return response.json(104)
  1341. reds = RedisObject()
  1342. resetCode = reds.get_data(key=email + '_forgetPwdResetCode')
  1343. if resetCode is False:
  1344. return response.json(90)
  1345. if authcode != resetCode:
  1346. return response.json(121)
  1347. # if not reds.set_data(key=email + '_forgetPwdResetCode', val=resetCode, expire=300):
  1348. # return response.json(10, '生成缓存错误')
  1349. user_qs.update(password=make_password(password))
  1350. if not reds.del_data(email + '_forgetPwdResetCode'):
  1351. return response.json(10, '删除缓存失败')
  1352. return response.json(0)
  1353. def do_phone_pwd_reset(self, phone, authcode, password, response):
  1354. data_valid = DataValid()
  1355. if data_valid.mobile_validate(phone) is not True:
  1356. return response.json(100)
  1357. if data_valid.password_validate(password) is not True:
  1358. return response.json(109)
  1359. user_qs = Device_User.objects.filter(Q(phone=phone) | Q(username=phone))
  1360. if not user_qs.exists():
  1361. return response.json(102)
  1362. reds = RedisObject()
  1363. resetCode = reds.get_data(key=phone + '_forgetPwdResetCode')
  1364. if resetCode is False:
  1365. return response.json(90)
  1366. if authcode != resetCode:
  1367. return response.json(121)
  1368. # if not reds.set_data(key=phone + '_forgetPwdResetCode', val=resetCode, expire=300):
  1369. # return response.json(10, '生成缓存错误')
  1370. user_qs.update(password=make_password(password))
  1371. if not reds.del_data(phone + '_forgetPwdResetCode'):
  1372. return response.json(10, '删除缓存失败')
  1373. return response.json(0)
  1374. def do_login(self, user_qs, response):
  1375. now_time = datetime.datetime.utcnow().replace(tzinfo=utc).astimezone(utc)
  1376. user_qs.update(last_login=now_time, online=True)
  1377. userID = user_qs[0].userID
  1378. print('userID' + userID)
  1379. tko = TokenObject()
  1380. user_list = user_qs.values("NickName", "userIconUrl", "userIconPath", "username", "userEmail", "phone")
  1381. res = tko.generate(data={'userID': userID, 'lang': response.lang, 'user': user_list[0]["username"]})
  1382. # 增加角色
  1383. user_qs[0].role.add(Role.objects.get(rid=1))
  1384. role_dict = ModelService.own_role(userID=userID)
  1385. res['rid'] = role_dict['rid']
  1386. res['roleName'] = role_dict['roleName']
  1387. res['permList'] = ModelService.own_permission(userID)
  1388. res['userID'] = userID
  1389. # 昵称,邮箱,电话,刷新,头像
  1390. userIconPath = str(user_list[0]["userIconPath"])
  1391. if userIconPath and userIconPath.find('static/') != -1:
  1392. userIconPath = userIconPath.replace('static/', '').replace('\\', '/')
  1393. res['userIconUrl'] = SERVER_DOMAIN + 'account/getAvatar/' + userIconPath
  1394. else:
  1395. res['userIconUrl'] = ''
  1396. res['NickName'] = user_list[0]["NickName"] if user_list[0]["NickName"] is not None else ''
  1397. res['username'] = user_list[0]["username"] if user_list[0]["username"] is not None else ''
  1398. res['userEmail'] = user_list[0]["userEmail"] if user_list[0]["userEmail"] is not None else ''
  1399. res['phone'] = user_list[0]["phone"] if user_list[0]["phone"] is not None else ''
  1400. print(res)
  1401. return response.json(0, res)
  1402. # 忘记密码v3
  1403. class v3resetPwdByCodeView(TemplateView):
  1404. @method_decorator(csrf_exempt)
  1405. def dispatch(self, *args, **kwargs):
  1406. return super(v3resetPwdByCodeView, self).dispatch(*args, **kwargs)
  1407. def get(self, request, *args, **kwargs):
  1408. request.encoding = 'utf-8'
  1409. request_dict = request.GET
  1410. lang = request_dict.get('lang')
  1411. if not lang:
  1412. lang = request_dict.get('language', None)
  1413. response = ResponseObject(lang)
  1414. was_limited = getattr(request, 'limited', False)
  1415. if was_limited is True:
  1416. return response.json(5)
  1417. return self.ValidationError(request_dict, response)
  1418. def post(self, request):
  1419. request.encoding = 'utf-8'
  1420. request_dict = request.POST
  1421. lang = request_dict.get('lang')
  1422. if not lang:
  1423. lang = request_dict.get('language', None)
  1424. response = ResponseObject(lang)
  1425. was_limited = getattr(request, 'limited', False)
  1426. if was_limited is True:
  1427. return response.json(5)
  1428. return self.ValidationError(request_dict, response)
  1429. def ValidationError(self, request_dict, response):
  1430. phone = request_dict.get('phone', None)
  1431. email = request_dict.get('email', None)
  1432. password = request_dict.get('password', None)
  1433. authcode = request_dict.get('authcode', None)
  1434. if password is None or authcode is None:
  1435. return response.json(444, 'password,authcode')
  1436. authcode = authcode.strip()
  1437. password = password.strip()
  1438. # 解密
  1439. try:
  1440. for i in range(1, 4):
  1441. if i == 1:
  1442. password = base64.b64decode(password)
  1443. password = password.decode('utf-8')
  1444. password = password[1:-1]
  1445. if i == 2:
  1446. password = base64.b64decode(password)
  1447. password = password.decode('utf-8')
  1448. password = password[2:-2]
  1449. if i == 3:
  1450. password = base64.b64decode(password)
  1451. password = password.decode('utf-8')
  1452. password = password[3:-3]
  1453. print(password)
  1454. except Exception as e:
  1455. return response.json(111)
  1456. try:
  1457. for i in range(1, 4):
  1458. if i == 1:
  1459. authcode = base64.b64decode(authcode)
  1460. authcode = authcode.decode('utf-8')
  1461. authcode = authcode[1:-1]
  1462. if i == 2:
  1463. authcode = base64.b64decode(authcode)
  1464. authcode = authcode.decode('utf-8')
  1465. authcode = authcode[2:-2]
  1466. if i == 3:
  1467. authcode = base64.b64decode(authcode)
  1468. authcode = authcode.decode('utf-8')
  1469. authcode = authcode[3:-3]
  1470. print(authcode)
  1471. except Exception as e:
  1472. return response.json(121)
  1473. if phone is not None:
  1474. phone = phone.strip()
  1475. return self.do_phone_pwd_reset(phone, authcode, password, response)
  1476. elif email is not None:
  1477. email = email.strip()
  1478. return self.do_email_pwd_reset(email, authcode, password, response)
  1479. else:
  1480. return response.json(444, 'phone')
  1481. def do_email_pwd_reset(self, email, authcode, password, response):
  1482. data_valid = DataValid()
  1483. if data_valid.email_validate(email) is not True:
  1484. return response.json(105)
  1485. if data_valid.password_validate(password) is not True:
  1486. return response.json(109)
  1487. user_qs = Device_User.objects.filter(Q(userEmail=email) | Q(username=email))
  1488. if not user_qs.exists():
  1489. return response.json(104)
  1490. reds = RedisObject()
  1491. resetCode = reds.get_data(key=email + '_forgetPwdResetCode')
  1492. if resetCode is False:
  1493. return response.json(90)
  1494. if authcode != resetCode:
  1495. return response.json(121)
  1496. # if not reds.set_data(key=email + '_forgetPwdResetCode', val=resetCode, expire=300):
  1497. # return response.json(10, '生成缓存错误')
  1498. user_qs.update(password=make_password(password))
  1499. if not reds.del_data(email + '_forgetPwdResetCode'):
  1500. return response.json(10, '删除缓存失败')
  1501. return response.json(0)
  1502. def do_phone_pwd_reset(self, phone, authcode, password, response):
  1503. data_valid = DataValid()
  1504. if data_valid.mobile_validate(phone) is not True:
  1505. return response.json(100)
  1506. if data_valid.password_validate(password) is not True:
  1507. return response.json(109)
  1508. user_qs = Device_User.objects.filter(Q(phone=phone) | Q(username=phone))
  1509. if not user_qs.exists():
  1510. return response.json(102)
  1511. reds = RedisObject()
  1512. resetCode = reds.get_data(key=phone + '_forgetPwdResetCode')
  1513. if resetCode is False:
  1514. return response.json(90)
  1515. if authcode != resetCode:
  1516. return response.json(121)
  1517. # if not reds.set_data(key=phone + '_forgetPwdResetCode', val=resetCode, expire=300):
  1518. # return response.json(10, '生成缓存错误')
  1519. user_qs.update(password=make_password(password))
  1520. if not reds.del_data(phone + '_forgetPwdResetCode'):
  1521. return response.json(10, '删除缓存失败')
  1522. return response.json(0)
  1523. def do_login(self, user_qs, response):
  1524. now_time = datetime.datetime.utcnow().replace(tzinfo=utc).astimezone(utc)
  1525. user_qs.update(last_login=now_time, online=True)
  1526. userID = user_qs[0].userID
  1527. print('userID' + userID)
  1528. tko = TokenObject()
  1529. user_list = user_qs.values("NickName", "userIconUrl", "userIconPath", "username", "userEmail", "phone")
  1530. res = tko.generate(data={'userID': userID, 'lang': response.lang, 'user': user_list[0]["username"]})
  1531. # 增加角色
  1532. user_qs[0].role.add(Role.objects.get(rid=1))
  1533. role_dict = ModelService.own_role(userID=userID)
  1534. res['rid'] = role_dict['rid']
  1535. res['roleName'] = role_dict['roleName']
  1536. res['permList'] = ModelService.own_permission(userID)
  1537. res['userID'] = userID
  1538. # 昵称,邮箱,电话,刷新,头像
  1539. userIconPath = str(user_list[0]["userIconPath"])
  1540. if userIconPath and userIconPath.find('static/') != -1:
  1541. userIconPath = userIconPath.replace('static/', '').replace('\\', '/')
  1542. res['userIconUrl'] = SERVER_DOMAIN + 'account/getAvatar/' + userIconPath
  1543. else:
  1544. res['userIconUrl'] = ''
  1545. res['NickName'] = user_list[0]["NickName"] if user_list[0]["NickName"] is not None else ''
  1546. res['username'] = user_list[0]["username"] if user_list[0]["username"] is not None else ''
  1547. res['userEmail'] = user_list[0]["userEmail"] if user_list[0]["userEmail"] is not None else ''
  1548. res['phone'] = user_list[0]["phone"] if user_list[0]["phone"] is not None else ''
  1549. print(res)
  1550. return response.json(0, res)
  1551. # 登录
  1552. class v2LoginView(TemplateView):
  1553. @method_decorator(csrf_exempt) # @csrf_exempt
  1554. def dispatch(self, *args, **kwargs):
  1555. return super(v2LoginView, self).dispatch(*args, **kwargs)
  1556. @ratelimit(key='ip', rate='5/m')
  1557. def post(self, request, *args, **kwargs):
  1558. request.encoding = 'utf-8'
  1559. request_dict = request.POST
  1560. language = request_dict.get('language', 'en')
  1561. response = ResponseObject(language)
  1562. was_limited = getattr(request, 'limited', False)
  1563. if was_limited is True:
  1564. return response.json(5)
  1565. return self.validates(request_dict, response)
  1566. # @ratelimit(key='ip', rate='5/m')
  1567. def get(self, request, *args, **kwargs):
  1568. print("进来了")
  1569. request.encoding = 'utf-8'
  1570. request_dict = request.GET
  1571. language = request_dict.get('language', 'en')
  1572. response = ResponseObject(language)
  1573. was_limited = getattr(request, 'limited', False)
  1574. if was_limited is True:
  1575. return response.json(5)
  1576. return self.validates(request_dict, response)
  1577. def validates(self, request_dict, response):
  1578. username = request_dict.get('userName', None)
  1579. password = request_dict.get('userPwd', None)
  1580. number = request_dict.get('number', None)
  1581. if not username or not password:
  1582. return response.json(111)
  1583. username = username.strip()
  1584. password = password.strip()
  1585. data_valid = DataValid()
  1586. if data_valid.email_validate(username):
  1587. return self.do_email_login(username, password, number, response)
  1588. elif data_valid.mobile_validate(username):
  1589. return self.do_phone_login(username, password, number, response)
  1590. elif data_valid.name_validate(username):
  1591. return self.do_name_login(username, password, number, response)
  1592. else:
  1593. return response.json(107)
  1594. def do_email_login(self, email, password, number, response):
  1595. user_qs = Device_User.objects.filter(Q(username=email) | Q(userEmail=email))
  1596. return self.valid_login(user_qs, password, number, response)
  1597. def do_phone_login(self, phone, password, number, response):
  1598. user_qs = Device_User.objects.filter(Q(phone=phone) | Q(username=phone), is_active=True, user_isValid=True)
  1599. return self.valid_login(user_qs, password, number, response)
  1600. def do_name_login(self, username, password, number, response):
  1601. user_qs = Device_User.objects.filter(Q(username=username) | Q(phone=username) | Q(userEmail=username),
  1602. is_active=True, user_isValid=True)
  1603. return self.valid_login(user_qs, password, number, response)
  1604. def valid_login(self, user_qs, password, number, response):
  1605. if not user_qs.exists():
  1606. return response.json(104)
  1607. # users = user_qs.values('role__rid', 'role__roleName', 'userID', 'role', 'NickName', 'username', 'userEmail',
  1608. # 'phone', 'password', 'userIconPath', 'user_isValid', 'is_active')[0]
  1609. users = user_qs.values('role__rid', 'role__roleName', 'userID', 'NickName', 'username', 'userEmail',
  1610. 'phone', 'password', 'userIconPath')[0]
  1611. if not check_password(password, users['password']):
  1612. return response.json(111)
  1613. userID = users['userID']
  1614. tko = TokenObject()
  1615. res = tko.generate(
  1616. data={'userID': userID, 'lang': response.lang, 'user': users['username'], 'm_code': '123413243214'})
  1617. if tko.code == 0:
  1618. now_time = datetime.datetime.utcnow().replace(tzinfo=utc).astimezone(utc)
  1619. if not number:
  1620. user_qs.update(last_login=now_time, language=response.lang)
  1621. else:
  1622. user_qs.update(last_login=now_time, language=response.lang, region_country=number)
  1623. user_qs.update(last_login=now_time, language=response.lang)
  1624. res['rid'] = users['role__rid']
  1625. res['roleName'] = users['role__roleName']
  1626. res['permList'] = ModelService.own_permission(userID)
  1627. res['userID'] = userID
  1628. # 昵称,邮箱,电话,刷新,头像
  1629. userIconPath = str(users['userIconPath'])
  1630. if userIconPath and userIconPath.find('static/') != -1:
  1631. userIconPath = userIconPath.replace('static/', '').replace('\\', '/')
  1632. res['userIconUrl'] = SERVER_DOMAIN + 'account/getAvatar/' + userIconPath
  1633. else:
  1634. res['userIconUrl'] = ''
  1635. res['NickName'] = users['NickName'] if users['NickName'] is not None else ''
  1636. res['username'] = users['username'] if users['username'] is not None else ''
  1637. res['userEmail'] = users['userEmail'] if users['userEmail'] is not None else ''
  1638. res['phone'] = users['phone'] if users['phone'] is not None else ''
  1639. return response.json(0, res)
  1640. else:
  1641. return response.json(tko.code)
  1642. # 登录
  1643. class noPasslogin(TemplateView):
  1644. @method_decorator(csrf_exempt) # @csrf_exempt
  1645. def dispatch(self, *args, **kwargs):
  1646. return super(noPasslogin, self).dispatch(*args, **kwargs)
  1647. @ratelimit(key='ip', rate='5/m')
  1648. def post(self, request, *args, **kwargs):
  1649. request.encoding = 'utf-8'
  1650. request_dict = request.POST
  1651. language = request_dict.get('language', 'en')
  1652. response = ResponseObject(language)
  1653. was_limited = getattr(request, 'limited', False)
  1654. if was_limited is True:
  1655. return response.json(5)
  1656. return self.validates(request_dict, response)
  1657. # @ratelimit(key='ip', rate='5/m')
  1658. def get(self, request, *args, **kwargs):
  1659. print("进来了")
  1660. request.encoding = 'utf-8'
  1661. request_dict = request.GET
  1662. language = request_dict.get('language', 'en')
  1663. response = ResponseObject(language)
  1664. was_limited = getattr(request, 'limited', False)
  1665. if was_limited is True:
  1666. return response.json(5)
  1667. return self.validates(request_dict, response)
  1668. def validates(self, request_dict, response):
  1669. username = request_dict.get('userName', None)
  1670. if not username:
  1671. return response.json(111)
  1672. username = username.strip()
  1673. data_valid = DataValid()
  1674. if data_valid.email_validate(username):
  1675. return self.do_email_login(username, response)
  1676. elif data_valid.mobile_validate(username):
  1677. return self.do_phone_login(username, response)
  1678. elif data_valid.name_validate(username):
  1679. return self.do_name_login(username, response)
  1680. else:
  1681. return response.json(107)
  1682. def do_email_login(self, email, response):
  1683. user_qs = Device_User.objects.filter(Q(username=email) | Q(userEmail=email))
  1684. return self.valid_login(user_qs, response)
  1685. def do_phone_login(self, phone, response):
  1686. user_qs = Device_User.objects.filter(Q(phone=phone) | Q(username=phone), is_active=True, user_isValid=True)
  1687. return self.valid_login(user_qs, response)
  1688. def do_name_login(self, username, response):
  1689. user_qs = Device_User.objects.filter(Q(username=username) | Q(phone=username) | Q(userEmail=username),
  1690. is_active=True, user_isValid=True)
  1691. return self.valid_login(user_qs, response)
  1692. def valid_login(self, user_qs, response):
  1693. if not user_qs.exists():
  1694. return response.json(104)
  1695. # users = user_qs.values('role__rid', 'role__roleName', 'userID', 'role', 'NickName', 'username', 'userEmail',
  1696. # 'phone', 'password', 'userIconPath', 'user_isValid', 'is_active')[0]
  1697. users = user_qs.values('role__rid', 'role__roleName', 'userID', 'NickName', 'username', 'userEmail',
  1698. 'phone', 'password', 'userIconPath')[0]
  1699. userID = users['userID']
  1700. tko = TokenObject()
  1701. res = tko.generate(
  1702. data={'userID': userID, 'lang': response.lang, 'user': users['username'], 'm_code': '123413243214'})
  1703. if tko.code == 0:
  1704. now_time = datetime.datetime.utcnow().replace(tzinfo=utc).astimezone(utc)
  1705. user_qs.update(last_login=now_time, language=response.lang)
  1706. res['rid'] = users['role__rid']
  1707. res['roleName'] = users['role__roleName']
  1708. res['permList'] = ModelService.own_permission(userID)
  1709. res['userID'] = userID
  1710. # 昵称,邮箱,电话,刷新,头像
  1711. userIconPath = str(users['userIconPath'])
  1712. if userIconPath and userIconPath.find('static/') != -1:
  1713. userIconPath = userIconPath.replace('static/', '').replace('\\', '/')
  1714. res['userIconUrl'] = SERVER_DOMAIN + 'account/getAvatar/' + userIconPath
  1715. else:
  1716. res['userIconUrl'] = ''
  1717. res['NickName'] = users['NickName'] if users['NickName'] is not None else ''
  1718. res['username'] = users['username'] if users['username'] is not None else ''
  1719. res['userEmail'] = users['userEmail'] if users['userEmail'] is not None else ''
  1720. res['phone'] = users['phone'] if users['phone'] is not None else ''
  1721. return response.json(0, res)
  1722. else:
  1723. return response.json(tko.code)
  1724. # 密码加密新登录
  1725. class v3LoginView(TemplateView):
  1726. @method_decorator(csrf_exempt) # @csrf_exempt
  1727. def dispatch(self, *args, **kwargs):
  1728. # chong
  1729. return super(v3LoginView, self).dispatch(*args, **kwargs)
  1730. @ratelimit(key='ip', rate='5/m')
  1731. def post(self, request, *args, **kwargs):
  1732. request.encoding = 'utf-8'
  1733. request_dict = request.POST
  1734. language = request_dict.get('language', 'en')
  1735. response = ResponseObject(language)
  1736. was_limited = getattr(request, 'limited', False)
  1737. if was_limited is True:
  1738. return response.json(5)
  1739. return self.validates(request_dict, response)
  1740. # @ratelimit(key='ip', rate='5/m')
  1741. def get(self, request, *args, **kwargs):
  1742. request.encoding = 'utf-8'
  1743. request_dict = request.GET
  1744. language = request_dict.get('language', 'en')
  1745. response = ResponseObject(language)
  1746. was_limited = getattr(request, 'limited', False)
  1747. if was_limited is True:
  1748. return response.json(5)
  1749. return self.validates(request_dict, response)
  1750. def validates(self, request_dict, response):
  1751. username = request_dict.get('userName', None)
  1752. password = request_dict.get('userPwd', None)
  1753. subscribe = request_dict.get('subscribe', None)
  1754. number = request_dict.get('number', None)
  1755. if not username or not password:
  1756. return response.json(111)
  1757. username = username.strip()
  1758. password = password.strip()
  1759. print("准备解密")
  1760. # 解密
  1761. try:
  1762. for i in range(1, 4):
  1763. if i == 1:
  1764. # 第一次先解密
  1765. password = base64.b64decode(password)
  1766. password = password.decode('utf-8')
  1767. # 截去第一位,最后一位
  1768. password = password[1:-1]
  1769. if i == 2:
  1770. # 第2次先解密
  1771. password = base64.b64decode(password)
  1772. password = password.decode('utf-8')
  1773. # 去前2位,后2位
  1774. password = password[2:-2]
  1775. if i == 3:
  1776. # 第3次先解密
  1777. password = base64.b64decode(password)
  1778. password = password.decode('utf-8')
  1779. # 去前3位,后3位
  1780. password = password[3:-3]
  1781. except Exception as e:
  1782. return response.json(111)
  1783. else:
  1784. data_valid = DataValid()
  1785. if data_valid.email_validate(username):
  1786. return self.do_email_login(username, password, response, subscribe, number)
  1787. elif data_valid.mobile_validate(username):
  1788. return self.do_phone_login(username, password, response, subscribe, number)
  1789. elif data_valid.name_validate(username):
  1790. return self.do_name_login(username, password, response, subscribe, number)
  1791. else:
  1792. return response.json(107)
  1793. def do_email_login(self, email, password, response, subscribe, number):
  1794. user_qs = Device_User.objects.filter(Q(username=email) | Q(userEmail=email))
  1795. return self.valid_login(user_qs, password, response, subscribe, number)
  1796. def do_phone_login(self, phone, password, response, subscribe, number):
  1797. user_qs = Device_User.objects.filter(Q(phone=phone) | Q(username=phone), is_active=True, user_isValid=True)
  1798. return self.valid_login(user_qs, password, response, subscribe, number)
  1799. def do_name_login(self, username, password, response, subscribe, number):
  1800. user_qs = Device_User.objects.filter(Q(username=username) | Q(phone=username) | Q(userEmail=username),
  1801. is_active=True, user_isValid=True)
  1802. return self.valid_login(user_qs, password, response, subscribe, number)
  1803. def valid_login(self, user_qs, password, response, subscribe, number):
  1804. if not user_qs.exists():
  1805. return response.json(104)
  1806. # users = user_qs.values('role__rid', 'role__roleName', 'userID', 'role', 'NickName', 'username', 'userEmail',
  1807. # 'phone', 'password', 'userIconPath', 'user_isValid', 'is_active')[0]
  1808. if subscribe:
  1809. user_qs.update(subscribe_email=subscribe)
  1810. users = user_qs.values('role__rid', 'role__roleName', 'userID', 'NickName', 'username', 'userEmail',
  1811. 'phone', 'password', 'userIconPath', 'fingerprint_enable', 'fingerprint_key', 'subscribe_email')[0]
  1812. if not check_password(password, users['password']):
  1813. return response.json(111)
  1814. userID = users['userID']
  1815. tko = TokenObject()
  1816. res = tko.generate(
  1817. data={'userID': userID, 'lang': response.lang, 'user': users['username'], 'm_code': '123413243214'})
  1818. oauth_qs = UserOauth2Model.objects.filter(userID__userID=userID)
  1819. auth_type = 0
  1820. if oauth_qs.exists():
  1821. auth_type = oauth_qs[0].authType
  1822. if tko.code == 0:
  1823. now_time = datetime.datetime.utcnow().replace(tzinfo=utc).astimezone(utc)
  1824. if not number:
  1825. user_qs.update(last_login=now_time, language=response.lang)
  1826. else:
  1827. user_qs.update(last_login=now_time, language=response.lang, region_country=number)
  1828. res['rid'] = users['role__rid']
  1829. res['roleName'] = users['role__roleName']
  1830. res['permList'] = ModelService.own_permission(userID)
  1831. res['userID'] = userID
  1832. # 昵称,邮箱,电话,刷新,头像
  1833. userIconPath = str(users['userIconPath'])
  1834. if userIconPath and userIconPath.find('static/') != -1:
  1835. userIconPath = userIconPath.replace('static/', '').replace('\\', '/')
  1836. res['userIconUrl'] = SERVER_DOMAIN + 'account/getAvatar/' + userIconPath
  1837. else:
  1838. res['userIconUrl'] = ''
  1839. res['NickName'] = users['NickName'] if users['NickName'] is not None else ''
  1840. res['username'] = users['username'] if users['username'] is not None else ''
  1841. res['userEmail'] = users['userEmail'] if users['userEmail'] is not None else ''
  1842. res['phone'] = users['phone'] if users['phone'] is not None else ''
  1843. # res['fingerprint_enable'] = users['fingerprint_enable']
  1844. # res['fingerprint_key'] = CommonService.encode_data(content=users['fingerprint_key'], start=2)
  1845. res['authType'] = auth_type
  1846. res['subscribe_email'] = users['subscribe_email'] if users['subscribe_email'] is not None else ''
  1847. return response.json(0, res)
  1848. else:
  1849. return response.json(tko.code)
  1850. # 用户登录后初始化接口
  1851. class InitInfoView(View):
  1852. def get(self, request, *args, **kwargs):
  1853. request.encoding = 'utf-8'
  1854. return self.validation(request.GET, request)
  1855. def post(self, request, *args, **kwargs):
  1856. request.encoding = 'utf-8'
  1857. return self.validation(request.POST, request)
  1858. def validation(self, request_dict, request):
  1859. response = ResponseObject()
  1860. token = request_dict.get('token', None)
  1861. unique = request_dict.get('unique', None)
  1862. tko = TokenObject(token)
  1863. if tko.code == 0:
  1864. userID = tko.userID
  1865. return self.init_info(request_dict, userID, response, request)
  1866. elif unique:
  1867. return self.update_country(request_dict, response, request)
  1868. else:
  1869. return response.json(tko.code)
  1870. # 初始化设备token
  1871. def init_info(self, request_dict, userID, response, request):
  1872. # 未读的系统消息
  1873. token_val = request_dict.get('token_val', None)
  1874. m_code = request_dict.get('m_code', None)
  1875. push_type = request_dict.get('push_type', None)
  1876. appBundleId = request_dict.get('appBundleId', None)
  1877. tz = request_dict.get('tz', '0')
  1878. lang = request_dict.get('lang', '') # 语言区域
  1879. now_time = int(time.time())
  1880. if all([token_val, push_type, appBundleId, userID]):
  1881. push_type = int(push_type)
  1882. if push_type == 0:
  1883. if appBundleId not in APNS_CONFIG.keys():
  1884. return response.json(904)
  1885. elif push_type == 1:
  1886. if appBundleId not in FCM_CONFIG.keys():
  1887. return response.json(904)
  1888. elif push_type == 2:
  1889. if appBundleId not in JPUSH_CONFIG.keys():
  1890. return response.json(904)
  1891. else:
  1892. return response.json(444, 'push_type')
  1893. if m_code:
  1894. # 获取设备推送状态
  1895. update_dict = {
  1896. 'token_val': token_val,
  1897. 'push_type': push_type,
  1898. 'tz': tz,
  1899. }
  1900. # 更新当前用户推送设备状态
  1901. UidPushModel.objects.filter(userID_id=userID, m_code=m_code). \
  1902. update(**update_dict)
  1903. if appBundleId:
  1904. user_ex_qs = UserExModel.objects.filter(userID_id=userID)
  1905. user_ex = None
  1906. if user_ex_qs.exists():
  1907. update_dict = {
  1908. 'updTime': now_time,
  1909. 'appBundleId': appBundleId,
  1910. 'region': lang
  1911. }
  1912. user_ex_qs.update(**update_dict)
  1913. user_ex = user_ex_qs[0]
  1914. else:
  1915. create_dict = {
  1916. 'addTime': now_time,
  1917. 'updTime': now_time,
  1918. 'appBundleId': appBundleId,
  1919. 'userID_id': userID,
  1920. 'region': lang
  1921. }
  1922. user_ex = UserExModel.objects.create(**create_dict)
  1923. country_ip_qs = CountryIPModel.objects.filter(user_ex_id=user_ex.id)
  1924. if not country_ip_qs.exists():
  1925. countryIp = CountryIPModel(
  1926. ip=CommonService.get_ip_address(request),
  1927. add_time=now_time,
  1928. user_ex_id=user_ex_qs[0].id
  1929. )
  1930. countryIp.save()
  1931. # 获取设备是否存在有已被删除
  1932. res = {'usmsg': 0} # 预留字段, 有版本app该字段去掉会报错
  1933. return response.json(0, res)
  1934. def update_country(self, request_dict, response, request):
  1935. username = request_dict.get('unique', None)
  1936. appBundleId = request_dict.get('appBundleId', None)
  1937. lang = request_dict.get('lang', '') # 语言区域
  1938. if username and appBundleId:
  1939. user_qs = Device_User.objects.filter(username=username)
  1940. if user_qs.exists():
  1941. user = user_qs[0]
  1942. user_ex_qs = UserExModel.objects.filter(userID_id=user.userID)
  1943. now_time = int(time.time())
  1944. if user_ex_qs.exists():
  1945. update_dict = {
  1946. 'updTime': now_time,
  1947. 'appBundleId': appBundleId,
  1948. 'region': lang
  1949. }
  1950. user_ex_qs.update(**update_dict)
  1951. user_ex = user_ex_qs[0]
  1952. else:
  1953. create_dict = {
  1954. 'addTime': now_time,
  1955. 'updTime': now_time,
  1956. 'appBundleId': appBundleId,
  1957. 'userID_id': user.userID,
  1958. 'region': lang
  1959. }
  1960. user_ex = UserExModel.objects.create(**create_dict)
  1961. country_ip_qs = CountryIPModel.objects.filter(user_ex_id=user_ex.id)
  1962. if not country_ip_qs.exists():
  1963. countryIp = CountryIPModel(
  1964. ip=CommonService.get_ip_address(request),
  1965. add_time=now_time,
  1966. user_ex_id=user_ex_qs[0].id
  1967. )
  1968. countryIp.save()
  1969. return response.json(0)
  1970. else:
  1971. return response.json(104)
  1972. else:
  1973. return response.json(444)
  1974. # 获取验证码
  1975. class verifyAuthcode(TemplateView):
  1976. def post(self, request, *args, **kwargs):
  1977. request.encoding = 'utf-8'
  1978. lang = request.POST.get('lang', None)
  1979. if not lang:
  1980. lang = request.POST.get('language', None)
  1981. response = ResponseObject(lang)
  1982. request_dict = request.POST
  1983. return self.ValidationError(request_dict, response)
  1984. def get(self, request, *args, **kwargs):
  1985. # Device_User.objects.filter(userEmail='chanjunkai@163.com').delete()
  1986. request.encoding = 'utf-8'
  1987. lang = request.GET.get('lang', None)
  1988. if not lang:
  1989. lang = request.GET.get('language', None)
  1990. response = ResponseObject(lang)
  1991. request_dict = request.GET
  1992. return self.ValidationError(request_dict, response)
  1993. def ValidationError(self, request_dict, response):
  1994. email = request_dict.get('email', None)
  1995. phone = request_dict.get('phone', None)
  1996. authcode = request_dict.get('authcode', None)
  1997. if email is not None:
  1998. email = email.strip()
  1999. return self.email_validate(email, authcode, response)
  2000. elif phone is not None:
  2001. phone = phone.strip()
  2002. return self.phone_validate(phone, authcode, response)
  2003. else:
  2004. return response.json(444)
  2005. def email_validate(self, email, authcode, response):
  2006. data_valid = DataValid()
  2007. if data_valid.email_validate(email) is not True:
  2008. return response.json(105)
  2009. reds = RedisObject()
  2010. resetCode = reds.get_data(key=email + '_forgetPwdResetCode')
  2011. if resetCode is False:
  2012. return response.json(120)
  2013. if authcode != resetCode:
  2014. return response.json(121)
  2015. return response.json(0)
  2016. def phone_validate(self, phone, authcode, response):
  2017. data_valid = DataValid()
  2018. if data_valid.mobile_validate(phone) is not True:
  2019. return response.json(100)
  2020. reds = RedisObject()
  2021. resetCode = reds.get_data(key=phone + '_forgetPwdResetCode')
  2022. if resetCode is False:
  2023. return response.json(120)
  2024. print(resetCode)
  2025. if authcode != resetCode:
  2026. return response.json(121)
  2027. return response.json(0)
  2028. # 微信登录
  2029. class wxAuthSignView(TemplateView):
  2030. def post(self, request, *args, **kwargs):
  2031. request.encoding = 'utf-8'
  2032. lang = request.POST.get('lang', None)
  2033. response = ResponseObject(lang)
  2034. request_dict = request.POST
  2035. return self.ValidationError(request_dict, response)
  2036. def get(self, request, *args, **kwargs):
  2037. # Device_User.objects.filter(userEmail='chanjunkai@163.com').delete()
  2038. request.encoding = 'utf-8'
  2039. lang = request.GET.get('lang', None)
  2040. response = ResponseObject(lang)
  2041. request_dict = request.GET
  2042. # return self.do_register('157113010663213800138000', '157113010663213800138000', response, 'xx')
  2043. return self.ValidationError(request_dict, response)
  2044. def ValidationError(self, request_dict, response):
  2045. grant_code = request_dict.get('grant_code', None) # 微信授权code
  2046. appBundleID = request_dict.get('appBundleID', None) # 包名
  2047. print('grant_code')
  2048. print(grant_code)
  2049. print('appBundleID')
  2050. print(appBundleID)
  2051. if all([grant_code, appBundleID]):
  2052. app_config = {
  2053. 'com.ansjer.loocamccloud': {'appid': 'wx9f6d6ce63f85b367',
  2054. 'secret': 'fe495884cd24637f1ae516c7f53d1b97', },
  2055. 'com.ansjer.zccloud': {'appid': 'wx2a9f5ef9baf2760f', 'secret': '5d38c7079676463149ffea593c58f2ed'},
  2056. 'com.ansjer.customizede': {'appid': 'wx2a9f5ef9baf2760f', 'secret': '5d38c7079676463149ffea593c58f2ed'},
  2057. # ios
  2058. 'com.ansjer.zccloud_ab': {'appid': 'wx2a9f5ef9baf2760f', 'secret': '5d38c7079676463149ffea593c58f2ed'},
  2059. # android
  2060. }
  2061. if appBundleID in app_config.keys():
  2062. appid = app_config[appBundleID]['appid']
  2063. secret = app_config[appBundleID]['secret']
  2064. # 获取access_token请求
  2065. at_url = 'https://api.weixin.qq.com/sns/oauth2/access_token?appid={appid}&secret={secret}&code={code}&grant_type=authorization_code'.format(
  2066. appid=appid, secret=secret, code=grant_code)
  2067. res_req = requests.get(url=at_url)
  2068. res_json = res_req.json()
  2069. print(res_json)
  2070. if 'access_token' not in res_json.keys():
  2071. # 授权过期
  2072. return response.json(717)
  2073. access_token = res_json['access_token']
  2074. openid = res_json['openid']
  2075. if access_token and openid:
  2076. info_url = 'https://api.weixin.qq.com/sns/userinfo?access_token={access_token}&openid={openid}'.format(
  2077. access_token=access_token, openid=openid)
  2078. res_req = requests.get(url=info_url)
  2079. res_req.encoding = res_req.apparent_encoding
  2080. res_json = res_req.json()
  2081. print(res_json)
  2082. unionID = res_json['unionid']
  2083. user_extend_qs = UserOauth2Model.objects.filter(unionID=unionID, authType=1)
  2084. if user_extend_qs.exists():
  2085. # 如果用户绑定过则直接登录
  2086. userID = user_extend_qs[0].userID_id
  2087. print(userID)
  2088. user_qs = Device_User.objects.filter(userID=userID)
  2089. return self.do_login(user_qs, response)
  2090. else:
  2091. # 如果用户为绑定过则创建用户并进行登录返回token
  2092. userID = CommonService.getUserID(getUser=False)
  2093. nickname = res_json['nickname'] + '_' + CommonService.RandomStr(4)
  2094. return self.do_register(userID, nickname, response, appBundleID, unionID)
  2095. else:
  2096. return response.json(414, 'access_token,openid')
  2097. else:
  2098. return response.json(414, 'appBundleID is wrong!')
  2099. else:
  2100. return response.json(414)
  2101. # if wxcode
  2102. # 登录
  2103. def do_login(self, user_qs, response):
  2104. now_time = datetime.datetime.utcnow().replace(tzinfo=utc).astimezone(utc)
  2105. userID = user_qs[0].userID
  2106. print('userID' + userID)
  2107. tko = TokenObject()
  2108. user_list = user_qs.values("NickName", "userIconUrl", "userIconPath", "username", "userEmail", "phone")
  2109. res = tko.generate(data={'userID': userID, 'lang': response.lang, 'user': user_list[0]["username"]})
  2110. # 增加角色
  2111. user_qs[0].role.add(Role.objects.get(rid=1))
  2112. role_dict = ModelService.own_role(userID=userID)
  2113. res['rid'] = role_dict['rid']
  2114. res['roleName'] = role_dict['roleName']
  2115. res['permList'] = ModelService.own_permission(userID)
  2116. res['userID'] = userID
  2117. # 昵称,邮箱,电话,刷新,头像
  2118. userIconPath = str(user_list[0]["userIconPath"])
  2119. if userIconPath and userIconPath.find('static/') != -1:
  2120. userIconPath = userIconPath.replace('static/', '').replace('\\', '/')
  2121. res['userIconUrl'] = SERVER_DOMAIN + 'account/getAvatar/' + userIconPath
  2122. else:
  2123. res['userIconUrl'] = ''
  2124. res['NickName'] = user_list[0]["NickName"] if user_list[0]["NickName"] is not None else ''
  2125. res['username'] = user_list[0]["username"] if user_list[0]["username"] is not None else ''
  2126. res['userEmail'] = user_list[0]["userEmail"] if user_list[0]["userEmail"] is not None else ''
  2127. res['phone'] = user_list[0]["phone"] if user_list[0]["phone"] is not None else ''
  2128. print(res)
  2129. # 添加用户登录类型
  2130. oauth_qs = UserOauth2Model.objects.filter(userID__userID=userID)
  2131. auth_type = 0
  2132. if oauth_qs.exists():
  2133. auth_type = oauth_qs[0].authType
  2134. res['authType'] = auth_type
  2135. user_qs.update(last_login=now_time, online=True)
  2136. return response.json(0, res)
  2137. def do_register(self, userID, nickname, response, appBundleId, unionID):
  2138. data_valid = DataValid()
  2139. if data_valid.name_validate(userID) is not True:
  2140. return response.json(105)
  2141. try:
  2142. users = Device_User.objects.create(
  2143. username=userID,
  2144. NickName=nickname,
  2145. password=make_password('123456'),
  2146. userID=userID,
  2147. is_active=True,
  2148. user_isValid=True,
  2149. )
  2150. nowTime = int(time.time())
  2151. UserOauth2Model.objects.create(
  2152. addTime=nowTime,
  2153. updTime=nowTime,
  2154. userID_id=users.userID,
  2155. authType=1,
  2156. unionID=unionID
  2157. )
  2158. except Exception as e:
  2159. errorInfo = traceback.format_exc()
  2160. print(errorInfo)
  2161. return response.json(424, repr(e))
  2162. else:
  2163. user_qs = Device_User.objects.filter(Q(userID=userID))
  2164. print('---')
  2165. print(user_qs)
  2166. return self.do_login(user_qs, response)
  2167. # 获取验证码
  2168. class wxPerfectView(TemplateView):
  2169. def post(self, request, *args, **kwargs):
  2170. request.encoding = 'utf-8'
  2171. lang = request.POST.get('lang', None)
  2172. response = ResponseObject(lang)
  2173. request_dict = request.POST
  2174. return self.ValidationError(request_dict, response)
  2175. def get(self, request, *args, **kwargs):
  2176. # Device_User.objects.filter(userEmail='chanjunkai@163.com').delete()
  2177. request.encoding = 'utf-8'
  2178. lang = request.GET.get('lang', None)
  2179. response = ResponseObject(lang)
  2180. request_dict = request.GET
  2181. # return self.do_register('157113010663213800138000', '157113010663213800138000', response, 'xx')
  2182. return self.ValidationError(request_dict, response)
  2183. def ValidationError(self, request_dict, response):
  2184. grant_code = request_dict.get('grant_code', None) # 微信授权code
  2185. appBundleID = request_dict.get('appBundleID', None) # 包名
  2186. token = request_dict.get('token', None) # 包名
  2187. if all([grant_code, appBundleID, token]):
  2188. tko = TokenObject(token)
  2189. if tko.code == 0:
  2190. userID = tko.userID
  2191. app_config = {
  2192. # ios
  2193. 'com.ansjer.loocamccloud': {'appid': 'wx9f6d6ce63f85b367',
  2194. 'secret': 'fe495884cd24637f1ae516c7f53d1b97', },
  2195. 'com.ansjer.zccloud': {'appid': 'wx2a9f5ef9baf2760f', 'secret': '5d38c7079676463149ffea593c58f2ed'},
  2196. 'com.ansjer.customizede': {'appid': 'wx2a9f5ef9baf2760f',
  2197. 'secret': '5d38c7079676463149ffea593c58f2ed'},
  2198. # android
  2199. 'com.ansjer.zccloud_ab': {'appid': 'wx2a9f5ef9baf2760f',
  2200. 'secret': '5d38c7079676463149ffea593c58f2ed'},
  2201. }
  2202. if appBundleID in app_config.keys():
  2203. appid = app_config[appBundleID]['appid']
  2204. secret = app_config[appBundleID]['secret']
  2205. # 获取access_token请求
  2206. at_url = 'https://api.weixin.qq.com/sns/oauth2/access_token?appid={appid}&secret={secret}&code={code}&grant_type=authorization_code'.format(
  2207. appid=appid, secret=secret, code=grant_code)
  2208. res_req = requests.get(url=at_url)
  2209. res_json = res_req.json()
  2210. print(res_json)
  2211. if 'access_token' not in res_json.keys():
  2212. # 授权过期
  2213. return response.json(717)
  2214. access_token = res_json['access_token']
  2215. openid = res_json['openid']
  2216. if access_token and openid:
  2217. info_url = 'https://api.weixin.qq.com/sns/userinfo?access_token={access_token}&openid={openid}'.format(
  2218. access_token=access_token, openid=openid)
  2219. res_req = requests.get(url=info_url)
  2220. res_req.encoding = res_req.apparent_encoding
  2221. res_json = res_req.json()
  2222. print(res_json)
  2223. unionID = res_json['unionid']
  2224. user_has_bind = UserOauth2Model.objects.filter(unionID=unionID)
  2225. if not user_has_bind.exists():
  2226. user_oauth2_qs = UserOauth2Model.objects. \
  2227. filter(userID_id=userID, authType=1)
  2228. if user_oauth2_qs.exists():
  2229. user_oauth2_qs.update(unionID=unionID)
  2230. return response.json(0)
  2231. else:
  2232. try:
  2233. nowTime = int(time.time())
  2234. UserOauth2Model.objects.create(
  2235. addTime=nowTime,
  2236. updTime=nowTime,
  2237. userID_id=userID,
  2238. authType=1,
  2239. unionID=unionID)
  2240. except Exception as e:
  2241. return response.json(424, repr(e))
  2242. else:
  2243. return response.json(0)
  2244. else:
  2245. return response.json(16)
  2246. else:
  2247. return response.json(414, 'access_token,openid')
  2248. else:
  2249. return response.json(414, 'appBundleID is wrong!')
  2250. else:
  2251. return response.json(tko.code)
  2252. else:
  2253. return response.json(414)
  2254. # 获取验证码
  2255. class OauthAuthCodeView(TemplateView):
  2256. @method_decorator(csrf_exempt)
  2257. def dispatch(self, *args, **kwargs):
  2258. return super(OauthAuthCodeView, self).dispatch(*args, **kwargs)
  2259. @ratelimit(key='ip', rate='2/m')
  2260. def post(self, request, *args, **kwargs):
  2261. request.encoding = 'utf-8'
  2262. lang = request.POST.get('lang', None)
  2263. if not lang:
  2264. lang = request.POST.get('language', None)
  2265. response = ResponseObject(lang)
  2266. request_dict = request.POST
  2267. phone = request_dict.get('phone', None)
  2268. if phone is not None:
  2269. was_limited = getattr(request, 'limited', False)
  2270. if was_limited is True:
  2271. return response.json(5)
  2272. return self.ValidationError(request_dict, response)
  2273. @ratelimit(key='ip', rate='2/m')
  2274. def get(self, request, *args, **kwargs):
  2275. # Device_User.objects.filter(userEmail='chanjunkai@163.com').delete()
  2276. request.encoding = 'utf-8'
  2277. lang = request.GET.get('lang', None)
  2278. if not lang:
  2279. lang = request.GET.get('language', None)
  2280. response = ResponseObject(lang)
  2281. was_limited = getattr(request, 'limited', False)
  2282. if was_limited is True:
  2283. return response.json(5)
  2284. request_dict = request.GET
  2285. return self.ValidationError(request_dict, response)
  2286. def ValidationError(self, request_dict, response):
  2287. email = request_dict.get('email', None)
  2288. phone = request_dict.get('phone', None)
  2289. country_code = request_dict.get('country_code', None)
  2290. sign_name = request_dict.get('sign_name', None)
  2291. token = request_dict.get('token', None)
  2292. print(token)
  2293. if email is not None:
  2294. email = email.strip()
  2295. # 阿里云的发送邮箱的调用方法
  2296. return self.aliyun_emailCode(email, response)
  2297. # return self.emailCode(email, response)
  2298. elif phone is not None:
  2299. phone = phone.strip()
  2300. if country_code is None:
  2301. return self.phoneCode(phone, response, sign_name)
  2302. else:
  2303. country_code = str(country_code.strip())
  2304. return self.phoneCodeV2(country_code, phone, response, sign_name)
  2305. else:
  2306. return response.json(444)
  2307. def emailCode(self, email, response):
  2308. dataValid = DataValid()
  2309. # 邮箱匹配
  2310. if dataValid.email_validate(email) is False:
  2311. return response.json(107)
  2312. reds = RedisObject()
  2313. identifyingCode = reds.get_data(key=email + '_OauthPerfect')
  2314. # 是否以获取邮箱验证码
  2315. if identifyingCode:
  2316. return response.json(89)
  2317. user_qs = Device_User.objects.filter(username=email)
  2318. email_qs = Device_User.objects.filter(userEmail=email)
  2319. # 邮箱用户是否已存在
  2320. if user_qs.exists():
  2321. return response.json(103)
  2322. elif email_qs.exists():
  2323. return response.json(103)
  2324. # 生成随机6位数
  2325. identifyingCode = RandomStr(6, True)
  2326. # 设置随机数缓存生命周期
  2327. send_data = TemplateService.email_message(type='register_code', language=response.lang)
  2328. ses = SesClassObject()
  2329. # 发送邮件
  2330. send_res = ses.send_email(
  2331. send_address_list=[email],
  2332. subject=send_data['title'],
  2333. body=send_data['body'].replace("{username}", email).replace("{captcha}",
  2334. str(identifyingCode))
  2335. )
  2336. if send_res is not True:
  2337. return response.json(44)
  2338. if reds.set_data(key=email + '_OauthPerfect', val=identifyingCode, expire=600) is not True:
  2339. return response.json(10, 'error')
  2340. return response.json(0)
  2341. # return response.json(0, {'identifyingCode': identifyingCode})
  2342. # 阿里云获取邮箱验证码
  2343. def aliyun_emailCode(self, email, response):
  2344. print('阿里云开始')
  2345. dataValid = DataValid()
  2346. # 邮箱匹配
  2347. if dataValid.email_validate(email) is False:
  2348. return response.json(107)
  2349. reds = RedisObject()
  2350. identifyingCode = reds.get_data(key=email + '_OauthPerfect')
  2351. # 是否以获取邮箱验证码
  2352. if identifyingCode:
  2353. return response.json(89)
  2354. user_qs = Device_User.objects.filter(username=email)
  2355. email_qs = Device_User.objects.filter(userEmail=email)
  2356. # 邮箱用户是否已存在
  2357. if user_qs.exists():
  2358. return response.json(103)
  2359. elif email_qs.exists():
  2360. return response.json(103)
  2361. # 生成随机6位数
  2362. identifyingCode = RandomStr(6, True)
  2363. # 设置随机数缓存生命周期
  2364. send_data = TemplateService.email_message(type='register_code', language=response.lang)
  2365. ses = SesClassObject()
  2366. # 发送邮件
  2367. send_res = ses.alyEmailCode(
  2368. send_address_list=[email],
  2369. subject=send_data['title'],
  2370. body=send_data['body'].replace("{username}", email).replace("{captcha}", str(identifyingCode))
  2371. )
  2372. if send_res is not True:
  2373. return response.json(44)
  2374. if reds.set_data(key=email + '_OauthPerfect', val=identifyingCode, expire=600) is not True:
  2375. return response.json(10, 'error')
  2376. return response.json(0)
  2377. def phoneCode(self, phone, response, sign_name):
  2378. dataValid = DataValid()
  2379. if dataValid.mobile_validate(phone) is not True:
  2380. return response.json(107)
  2381. reds = RedisObject()
  2382. reds_key = str(phone) + '_OauthPerfect'
  2383. identifyingCode = reds.get_data(key=reds_key)
  2384. reds_key_ttl = reds.get_ttl(key=reds_key)
  2385. if reds_key_ttl > 240 and identifyingCode:
  2386. # if identifyingCode :
  2387. return response.json(90)
  2388. user_qs = Device_User.objects.filter(username=phone)
  2389. phone_qs = Device_User.objects.filter(phone=phone)
  2390. if user_qs.exists() or phone_qs.exists():
  2391. return response.json(101)
  2392. identifyingCode = RandomStr(6, True)
  2393. # 发送手机验证码
  2394. aliSms = AliSmsObject()
  2395. if sign_name == 'zosi':
  2396. sign_ms = '周视'
  2397. else:
  2398. sign_ms = 'Ansjer'
  2399. res = aliSms.send_code_sms(phone=phone, code=identifyingCode, sign_name=sign_ms,
  2400. temp_msg='SMS_151600991')
  2401. print(res)
  2402. if res["Code"] == "OK":
  2403. if reds.set_data(key=reds_key, val=identifyingCode, expire=300) is not True:
  2404. # if reds.set_data(key=phone + '_identifyingCode', val=identifyingCode, expire=60) is not True:
  2405. return response.json(10, '生成缓存系统错误')
  2406. return response.json(0)
  2407. else:
  2408. return response.json(10, res["Message"])
  2409. def phoneCodeV2(self, country_code, phone, response, sign_name):
  2410. dataValid = DataValid()
  2411. if dataValid.mobile_validate(phone) is not True:
  2412. return response.json(107)
  2413. reds = RedisObject()
  2414. reds_key = str(phone) + '_OauthPerfect'
  2415. identifyingCode = reds.get_data(key=reds_key)
  2416. reds_key_ttl = reds.get_ttl(key=reds_key)
  2417. if reds_key_ttl > 240 and identifyingCode:
  2418. # if identifyingCode :
  2419. return response.json(90)
  2420. user_qs = Device_User.objects.filter(username=phone)
  2421. phone_qs = Device_User.objects.filter(phone=phone)
  2422. if user_qs.exists() or phone_qs.exists():
  2423. return response.json(101)
  2424. identifyingCode = RandomStr(6, True)
  2425. # 短信签名
  2426. # sign_name_dict = {
  2427. # 'ansjer':'Ansjer',
  2428. # 'zosi':'周视'
  2429. # }
  2430. sign_ms = ''
  2431. if country_code == '86':
  2432. # 国内短信推送模板
  2433. temp_msg = 'SMS_151600991'
  2434. rec_phone = phone
  2435. if sign_name == 'zosi':
  2436. sign_ms = '周视'
  2437. else:
  2438. sign_ms = 'Ansjer'
  2439. else:
  2440. # 国际短信推送模板
  2441. temp_msg = 'SMS_172165867'
  2442. rec_phone = country_code + phone
  2443. sign_ms = 'Ansjer'
  2444. # 发送手机验证码
  2445. aliSms = AliSmsObject()
  2446. res = aliSms.send_code_sms(phone=rec_phone, code=identifyingCode, sign_name=sign_ms,
  2447. temp_msg=temp_msg)
  2448. print(res)
  2449. if res["Code"] == "OK":
  2450. # if reds.set_data(key=reds_key, val=identifyingCode, expire=60) is not True:
  2451. if reds.set_data(key=reds_key, val=identifyingCode, expire=300) is not True:
  2452. return response.json(10, '生成缓存系统错误')
  2453. return response.json(0)
  2454. else:
  2455. return response.json(10, res["Message"])
  2456. class OauthPerfectView(TemplateView):
  2457. @method_decorator(csrf_exempt)
  2458. def dispatch(self, *args, **kwargs):
  2459. return super(OauthPerfectView, self).dispatch(*args, **kwargs)
  2460. def get(self, request, *args, **kwargs):
  2461. request.encoding = 'utf-8'
  2462. request_dict = request.GET
  2463. lang = request_dict.get('lang')
  2464. if not lang:
  2465. lang = request_dict.get('language', None)
  2466. response = ResponseObject(lang)
  2467. was_limited = getattr(request, 'limited', False)
  2468. if was_limited is True:
  2469. return response.json(5)
  2470. return self.ValidationError(request_dict, response)
  2471. def post(self, request):
  2472. request.encoding = 'utf-8'
  2473. request_dict = request.POST
  2474. lang = request_dict.get('lang')
  2475. if not lang:
  2476. lang = request_dict.get('language', None)
  2477. response = ResponseObject(lang)
  2478. was_limited = getattr(request, 'limited', False)
  2479. if was_limited is True:
  2480. return response.json(5)
  2481. return self.ValidationError(request_dict, response)
  2482. def ValidationError(self, request_dict, response):
  2483. phone = request_dict.get('phone', None)
  2484. email = request_dict.get('email', None)
  2485. password = request_dict.get('password', None)
  2486. authcode = request_dict.get('authcode', None)
  2487. token = request_dict.get('token', None)
  2488. token = request_dict.get('token')
  2489. tko = TokenObject(token)
  2490. if password is None or authcode is None:
  2491. return response.json(444, 'password,authcode')
  2492. authcode = CommonService.decode_data(authcode.strip())
  2493. if authcode is None:
  2494. return response.json(444, 'password,authcode')
  2495. password = CommonService.decode_data(password.strip())
  2496. if password is None:
  2497. return response.json(444, 'password,authcode')
  2498. if phone is not None:
  2499. phone = phone.strip()
  2500. return self.do_phone(tko, phone, authcode, password, response)
  2501. elif email is not None:
  2502. email = email.strip()
  2503. return self.do_email(tko, email, authcode, password, response)
  2504. else:
  2505. return response.json(444, 'phone')
  2506. def do_email(self, tko, email, authcode, password, response):
  2507. data_valid = DataValid()
  2508. if data_valid.email_validate(email) is not True:
  2509. return response.json(105)
  2510. if data_valid.password_validate(password) is not True:
  2511. return response.json(109)
  2512. if tko.code == 0:
  2513. user_qs = Device_User.objects.filter(userID=tko.userID)
  2514. else:
  2515. return response.json(tko.code)
  2516. if not user_qs.exists():
  2517. return response.json(104)
  2518. reds = RedisObject()
  2519. resetCode = reds.get_data(key=email + '_OauthPerfect')
  2520. if resetCode is False:
  2521. return response.json(90)
  2522. if authcode != resetCode:
  2523. return response.json(121)
  2524. # if not reds.set_data(key=email + '_forgetPwdResetCode', val=resetCode, expire=300):
  2525. # return response.json(10, '生成缓存错误')
  2526. user_qs.update(userEmail=email, password=make_password(password))
  2527. if not reds.del_data(email + '_OauthPerfect'):
  2528. return response.json(10, '删除缓存失败')
  2529. return response.json(0)
  2530. def do_phone(self, tko, phone, authcode, password, response):
  2531. data_valid = DataValid()
  2532. if data_valid.mobile_validate(phone) is not True:
  2533. return response.json(100)
  2534. if data_valid.password_validate(password) is not True:
  2535. return response.json(109)
  2536. if tko.code == 0:
  2537. user_qs = Device_User.objects.filter(userID=tko.userID)
  2538. else:
  2539. return response.json(tko.code)
  2540. if not user_qs.exists():
  2541. return response.json(102)
  2542. reds = RedisObject()
  2543. resetCode = reds.get_data(key=str(phone) + '_OauthPerfect')
  2544. print(resetCode)
  2545. if resetCode is False:
  2546. return response.json(90)
  2547. if authcode != resetCode:
  2548. return response.json(121)
  2549. # if not reds.set_data(key=phone + '_forgetPwdResetCode', val=resetCode, expire=300):
  2550. # return response.json(10, '生成缓存错误')
  2551. user_qs.update(phone=phone, password=make_password(password))
  2552. if not reds.del_data(str(phone) + '_OauthPerfect'):
  2553. return response.json(10, '删除缓存失败')
  2554. return response.json(0)
  2555. class UnbundingWXView(TemplateView):
  2556. def get(self, request, *args, **kwargs):
  2557. request.encoding = 'utf-8'
  2558. request_dict = request.GET
  2559. lang = request_dict.get('lang')
  2560. if not lang:
  2561. lang = request_dict.get('language', None)
  2562. response = ResponseObject(lang)
  2563. return self.ValidationError(request_dict, response)
  2564. def post(self, request, *args, **kwargs):
  2565. request.encoding = 'utf-8'
  2566. request_dict = request.POST
  2567. lang = request_dict.get('lang')
  2568. if not lang:
  2569. lang = request_dict.get('language', None)
  2570. response = ResponseObject(lang)
  2571. return self.ValidationError(request_dict, response)
  2572. def ValidationError(self, request_dict, response):
  2573. token = request_dict.get('token', None)
  2574. auth_type = request_dict.get('auth_type', None)
  2575. if not all([token, auth_type]):
  2576. return response.json(444, 'token,auth_type')
  2577. tko = TokenObject(token)
  2578. if tko.code == 0:
  2579. userID = tko.userID
  2580. user_oauth2_qs = UserOauth2Model.objects.filter(
  2581. ~Q(unionID='') & Q(userID_id=userID) & Q(authType=auth_type))
  2582. if user_oauth2_qs.exists():
  2583. user_qs = Device_User.objects.filter(phone='', userEmail='', userID=userID)
  2584. if user_qs.exists():
  2585. return response.json(17)
  2586. else:
  2587. user_oauth2_qs.delete()
  2588. return response.json(0)
  2589. else:
  2590. return response.json(173)
  2591. else:
  2592. return response.json(tko.code)
  2593. class alexaAuthView(TemplateView):
  2594. def post(self, request, *args, **kwargs):
  2595. request.encoding = 'utf-8'
  2596. # request_dict = json.loads(request.body.decode('utf-8'))
  2597. request_dict = request.POST
  2598. response = ResponseObject()
  2599. return self.validates(request_dict, response)
  2600. def get(self, request, *args, **kwargs):
  2601. request.encoding = 'utf-8'
  2602. request_dict = request.GET
  2603. response = ResponseObject()
  2604. return self.validates(request_dict, response)
  2605. def validates(self, request_dict, response):
  2606. username = request_dict.get('userName', None)
  2607. password = request_dict.get('userPwd', None)
  2608. if not username or not password:
  2609. return response.json(111)
  2610. username = username.strip()
  2611. password = password.strip()
  2612. data_valid = DataValid()
  2613. if data_valid.email_validate(username):
  2614. return self.do_email_login(username, password, response)
  2615. elif data_valid.mobile_validate(username):
  2616. return self.do_phone_login(username, password, response)
  2617. elif data_valid.name_validate(username):
  2618. return self.do_name_login(username, password, response)
  2619. else:
  2620. return response.json(107)
  2621. def do_email_login(self, email, password, response):
  2622. user_qs = Device_User.objects.filter(Q(username=email) | Q(userEmail=email))
  2623. return self.valid_login(user_qs, password, response)
  2624. def do_phone_login(self, phone, password, response):
  2625. user_qs = Device_User.objects.filter(Q(phone=phone) | Q(username=phone), is_active=True, user_isValid=True)
  2626. return self.valid_login(user_qs, password, response)
  2627. def do_name_login(self, username, password, response):
  2628. user_qs = Device_User.objects.filter(Q(username=username) | Q(phone=username) | Q(userEmail=username),
  2629. is_active=True, user_isValid=True)
  2630. return self.valid_login(user_qs, password, response)
  2631. def valid_login(self, user_qs, password, response):
  2632. if not user_qs.exists():
  2633. return response.json(104)
  2634. users = user_qs.values('userID', 'password')[0]
  2635. if not check_password(password, users['password']):
  2636. return response.json(111)
  2637. userID = users['userID']
  2638. # uid_qs = Device_Info.objects.filter(userID_id=userID, isExist=1).values('UID', 'NickName', 'View_Password')
  2639. # uid_arr = []
  2640. # for uid_q in uid_qs:
  2641. # uid_arr.append({'uid': uid_q['UID'], 'nick': uid_q['NickName'], 'password': uid_q['View_Password']})
  2642. res = {
  2643. 'userID': userID,
  2644. # 'uid_arr': uid_arr
  2645. }
  2646. return response.json(0, res)
  2647. # return response.json(0, res)
  2648. class alexaUidView(TemplateView):
  2649. def post(self, request, *args, **kwargs):
  2650. request.encoding = 'utf-8'
  2651. # request_dict = json.loads(request.body.decode('utf-8'))
  2652. request_dict = request.POST
  2653. response = ResponseObject()
  2654. return self.validates(request_dict, response)
  2655. def get(self, request, *args, **kwargs):
  2656. request.encoding = 'utf-8'
  2657. request_dict = request.GET
  2658. response = ResponseObject()
  2659. return self.validates(request_dict, response)
  2660. def validates1(self, request_dict, response):
  2661. userID = request_dict.get('alexa_user_id')
  2662. sid = request_dict.get('sid')
  2663. sst = request_dict.get('sst')
  2664. if sid == 'admin' and sst == 'admin':
  2665. uid_qs = Device_Info.objects.filter(userID_id=userID, isExist=1).values('UID', 'NickName', 'View_Password')
  2666. uid_arr = []
  2667. uid_list = []
  2668. for uid_q in uid_qs:
  2669. uid_list.append(uid_q['UID'])
  2670. uid_arr.append({'uid': uid_q['UID'], 'nick': uid_q['NickName'], 'password': uid_q['View_Password'], })
  2671. res = {
  2672. 'uid_arr': uid_arr
  2673. }
  2674. return response.json(0, res)
  2675. else:
  2676. return response.json(107)
  2677. def validates(self, request_dict, response):
  2678. sid = request_dict.get('sid')
  2679. sst = request_dict.get('sst')
  2680. userID = request_dict.get('alexa_user_id')
  2681. if sid != 'admin' or sst != 'admin':
  2682. return response.json(107)
  2683. uid_qs = Device_Info.objects.filter(userID_id=userID, isExist=1).values('UID', 'NickName', 'View_Password')
  2684. if not uid_qs.exists():
  2685. return response.json(107)
  2686. try:
  2687. uid_dict = {}
  2688. uid_list = []
  2689. for uid_q in uid_qs:
  2690. # uid转大写
  2691. UID = uid_q['UID'].upper()
  2692. uid_list.append(UID)
  2693. uid_dict[UID] = {'nick': uid_q['NickName'], 'password': uid_q['View_Password']}
  2694. us_qs = UidSetModel.objects.filter(uid__in=uid_list, is_alexa=1).values('id', 'uid', 'region_alexa', 'channel')
  2695. if not us_qs.exists():
  2696. return response.json(173)
  2697. # uid,password,region的列表
  2698. uid_arr = []
  2699. for us in us_qs:
  2700. uid = us['uid']
  2701. channel = us['channel']
  2702. # 设备alexa区域
  2703. region_alexa = us['region_alexa']
  2704. if region_alexa == '':
  2705. region_alexa = 'EN'
  2706. # 多通道设备获取通道名
  2707. if channel > 1:
  2708. uid_channel_set_qs = UidChannelSetModel.objects.filter(uid_id=us['id']).values('channel', 'channel_name')
  2709. if uid_channel_set_qs.exists():
  2710. # DVR设备名为 UidChannelSetModel 的 channel_name
  2711. for uid_channel_set in uid_channel_set_qs:
  2712. uid_arr.append({'uid': uid, 'nick': uid_channel_set['channel_name'], 'region': region_alexa,
  2713. 'password': uid_dict[uid]['password'], 'multi_channel': 1,
  2714. 'channel': uid_channel_set['channel']})
  2715. else:
  2716. uid_arr.append({'uid': uid, 'nick': uid_dict[uid]['nick'], 'region': region_alexa,
  2717. 'password': uid_dict[uid]['password'], 'multi_channel': 0})
  2718. res = {
  2719. 'uid_arr': uid_arr
  2720. }
  2721. return response.json(0, res)
  2722. except Exception as e:
  2723. return response.json(500, repr(e))
  2724. # 登出
  2725. class V2LogoutView(TemplateView):
  2726. @method_decorator(csrf_exempt)
  2727. def dispatch(self, *args, **kwargs):
  2728. return super(V2LogoutView, self).dispatch(*args, **kwargs)
  2729. def post(self, request, *args, **kwargs):
  2730. request.encoding = 'utf-8'
  2731. request_dict = request.POST
  2732. return self.Logout(request_dict)
  2733. def get(self, request, *args, **kwargs):
  2734. request.encoding = 'utf-8'
  2735. request_dict = request.GET
  2736. return self.Logout(request_dict)
  2737. def Logout(self, request_dict):
  2738. response = ResponseObject()
  2739. token = request_dict.get('token')
  2740. tko = TokenObject(token)
  2741. userID = tko.userID
  2742. if tko.code == 0:
  2743. Device_User.objects.filter(userID=tko.userID).update(online=False)
  2744. redisObj = RedisObject(db=3)
  2745. redisObj.del_data(key=tko.userID)
  2746. m_code = request_dict.get('m_code', None)
  2747. if m_code:
  2748. try:
  2749. UidPushModel.objects.filter(userID_id=userID, m_code=m_code).delete()
  2750. except Exception as e:
  2751. pass
  2752. else:
  2753. pass
  2754. os_type = request_dict.get('os_type', None)
  2755. appid = request_dict.get('appid', None)
  2756. udid = request_dict.get('udid', None)
  2757. if os_type and appid and udid:
  2758. tutk_push_url = "{tutk_push_domain}?cmd=mapsync&os={os_type}&appid={appid}&udid={udid}". \
  2759. format(os_type=os_type, appid=appid, udid=udid, tutk_push_domain=TUTK_PUSH_DOMAIN)
  2760. res = requests.get(url=tutk_push_url)
  2761. res_s = res.text
  2762. res_s = res_s.rstrip()
  2763. if res_s == '200 Success.':
  2764. return response.json(0)
  2765. else:
  2766. return response.json(10, 'not yes')
  2767. return response.json(0)
  2768. else:
  2769. return response.json(tko.code)
  2770. # 图片验证码生成
  2771. class generatePictureCodeView(TemplateView):
  2772. @method_decorator(csrf_exempt)
  2773. def dispatch(self, *args, **kwargs):
  2774. return super(generatePictureCodeView, self).dispatch(*args, **kwargs)
  2775. def post(self, request, *args, **kwargs):
  2776. request.encoding = 'utf-8'
  2777. request_dict = request.POST
  2778. return self.validates(request_dict)
  2779. def get(self, request, *args, **kwargs):
  2780. request.encoding = 'utf-8'
  2781. request_dict = request.GET
  2782. return self.validates(request_dict)
  2783. # 生成验证码和验证码图片
  2784. def get_valid_code_img(self, request_dict):
  2785. # ---------生成背景图片-----------
  2786. # img = Image.new('RGB', (260, 34), color=get_random_color())
  2787. img = Image.new('RGB', (260, 34),
  2788. color=(random.randint(0, 255), random.randint(0, 255), random.randint(0, 255)))
  2789. # -------/生成背景图片-----------
  2790. # --------生成随机验证码,设置字体格式,大小等属性------------
  2791. draw = ImageDraw.Draw(img)
  2792. path = BASE_DIR + '/Ansjer/file/font/simhei.ttf'
  2793. kumo_font = ImageFont.truetype(path, 40, encoding="unic") # 设置字体
  2794. print("字体可以")
  2795. valid_code_str = ''
  2796. for i in range(6):
  2797. random_num = str(random.randint(0, 9))
  2798. random_low_alpha = chr(random.randint(97, 122))
  2799. random_high_alpha = chr(random.randint(65, 90))
  2800. random_char = random.choice([random_num, random_low_alpha, random_high_alpha])
  2801. draw.text((i * 40 + 20, -3), random_char, 'white', kumo_font)
  2802. # 保存验证码字符串
  2803. valid_code_str += random_char
  2804. print(valid_code_str)
  2805. # --------/生成随机验证码,设置字体格式,大小等属性------------
  2806. # -------增加干扰线,点----------
  2807. # 噪点噪线
  2808. width = 260
  2809. height = 34
  2810. for i in range(10):
  2811. x1 = random.randint(0, width)
  2812. x2 = random.randint(0, width)
  2813. y1 = random.randint(0, height)
  2814. y2 = random.randint(0, height)
  2815. draw.line((x1, y1, x2, y2), fill=(random.randint(0, 255), random.randint(0, 255), random.randint(0, 255)))
  2816. for i in range(10):
  2817. draw.point([random.randint(0, width), random.randint(0, height)],
  2818. fill=(random.randint(0, 255), random.randint(0, 255), random.randint(0, 255)))
  2819. x = random.randint(0, width)
  2820. y = random.randint(0, height)
  2821. draw.arc((x, y, x + 4, y + 4), 0, 90,
  2822. fill=(random.randint(0, 255), random.randint(0, 255), random.randint(0, 255)))
  2823. # -------/增加干扰线,点-----------
  2824. # ------存入内存---------
  2825. f = BytesIO() # 用完之后,BytesIO会自动清掉
  2826. img.save(f, 'png')
  2827. data = f.getvalue()
  2828. return data, valid_code_str
  2829. # 生成验证码方法
  2830. def validates(self, request_dict):
  2831. # 页面传过来的uuid
  2832. imageCodeId = request_dict.get('imageCodeId', '')
  2833. response = ResponseObject()
  2834. if not imageCodeId:
  2835. return response.json(444)
  2836. # 存入redis的key
  2837. image_code_id = "image_code_%s" % imageCodeId
  2838. """
  2839. 基于PIL模块动态生成响应状态码图片
  2840. :param request:
  2841. :return:
  2842. """
  2843. # 图片data,验证码
  2844. data, valid_code_str = self.get_valid_code_img(request_dict)
  2845. if imageCodeId:
  2846. redisObj = RedisObject(db=6)
  2847. # 单条维护
  2848. redisObj.set_data(key=image_code_id, val=valid_code_str, expire=120)
  2849. image_code_val = redisObj.get_data(key=image_code_id)
  2850. print("________获取验证码的值" + image_code_val)
  2851. return HttpResponse(data)
  2852. # 图片验证码校验注册
  2853. class Image_Code_RegisterView(TemplateView):
  2854. @method_decorator(csrf_exempt)
  2855. def dispatch(self, *args, **kwargs):
  2856. return super(Image_Code_RegisterView, self).dispatch(*args, **kwargs)
  2857. def post(self, request, *args, **kwargs):
  2858. request.encoding = 'utf-8'
  2859. request_dict = request.POST
  2860. return self.validates(request_dict)
  2861. def get(self, request, *args, **kwargs):
  2862. request.encoding = 'utf-8'
  2863. request_dict = request.GET
  2864. return self.validates(request_dict)
  2865. # 检测验证码,并注册
  2866. def validates(self, request_dict):
  2867. print("__________request_dict:%s" % request_dict)
  2868. userEmail = request_dict.get('userEmail', None)
  2869. password = request_dict.get('userPwd', None)
  2870. lang = request_dict.get('lang', None)
  2871. # 前端传进来的uuid
  2872. imageCodeId = request_dict.get('imageCodeId', None)
  2873. # 页面输入的验证码
  2874. response = ResponseObject(lang)
  2875. valid_code = request_dict.get('id_v_code', None)
  2876. unique = request_dict.get('unique', None)
  2877. number = request_dict.get('number', None)
  2878. if unique:
  2879. delete_local_account(unique)
  2880. if not all([userEmail, password, lang, imageCodeId, valid_code]):
  2881. return response.json(444)
  2882. try:
  2883. for i in range(1, 4):
  2884. if i == 1:
  2885. password = base64.b64decode(password)
  2886. password = password.decode('utf-8')
  2887. password = password[1:-1]
  2888. if i == 2:
  2889. password = base64.b64decode(password)
  2890. password = password.decode('utf-8')
  2891. password = password[2:-2]
  2892. if i == 3:
  2893. password = base64.b64decode(password)
  2894. password = password.decode('utf-8')
  2895. password = password[3:-3]
  2896. print("password%s" % password)
  2897. except Exception as e:
  2898. return response.json(111)
  2899. try:
  2900. for i in range(1, 4):
  2901. if i == 1:
  2902. valid_code = base64.b64decode(valid_code)
  2903. valid_code = valid_code.decode('utf-8')
  2904. valid_code = valid_code[1:-1]
  2905. if i == 2:
  2906. valid_code = base64.b64decode(valid_code)
  2907. valid_code = valid_code.decode('utf-8')
  2908. valid_code = valid_code[2:-2]
  2909. if i == 3:
  2910. valid_code = base64.b64decode(valid_code)
  2911. valid_code = valid_code.decode('utf-8')
  2912. valid_code = valid_code[3:-3]
  2913. print("valid_code:%s" % valid_code)
  2914. except Exception as e:
  2915. return response.json(121)
  2916. if not userEmail:
  2917. return response.json(105)
  2918. if not password:
  2919. return response.json(109)
  2920. userEmail = userEmail.strip()
  2921. password = password.strip()
  2922. # 注释
  2923. if userEmail:
  2924. emailValid = Device_User.objects.filter(userEmail=userEmail)
  2925. if emailValid:
  2926. return response.json(103)
  2927. # 根据uuid拼接的key
  2928. image_code_key = "image_code_%s" % imageCodeId
  2929. # 判断验证码是否过期
  2930. if image_code_key is None:
  2931. return response.json(120)
  2932. redisObj = RedisObject(db=6)
  2933. # redis里面的验证码
  2934. redis_image_code = redisObj.get_data(key=image_code_key)
  2935. # 验证用户输入的验证码和redis中的验证码
  2936. if redis_image_code is False or valid_code.lower() != redis_image_code.lower():
  2937. return response.json(121)
  2938. # 删除redis中的图片验证码,防止用户使用同一个图片验证码验证多次
  2939. redisObj.del_data(key=image_code_key)
  2940. username = userEmail
  2941. email_qs = Device_User.objects.filter(Q(userEmail=userEmail) | Q(username=userEmail))
  2942. if email_qs:
  2943. return response.json(103)
  2944. # #存用户名和密码
  2945. create_data = {
  2946. "username": username,
  2947. "NickName": username,
  2948. "userEmail": userEmail,
  2949. "password": make_password(password),
  2950. "userID": CommonService.getUserID(μs=False, setOTAID=True),
  2951. "is_active": True,
  2952. "user_isValid": True,
  2953. }
  2954. if number:
  2955. create_data["region_country"] = number
  2956. users = Device_User.objects.create(**create_data)
  2957. return self.do_login(email_qs, response)
  2958. def do_login(self, user_qs, response):
  2959. now_time = datetime.datetime.utcnow().replace(tzinfo=utc).astimezone(utc)
  2960. user_qs.update(last_login=now_time, online=True)
  2961. userID = user_qs[0].userID
  2962. print('userID' + userID)
  2963. tko = TokenObject()
  2964. user_list = user_qs.values("NickName", "userIconUrl", "userIconPath", "username", "userEmail", "phone")
  2965. res = tko.generate(data={'userID': userID, 'lang': response.lang, 'user': user_list[0]["username"]})
  2966. # 增加角色
  2967. user_qs[0].role.add(Role.objects.get(rid=1))
  2968. role_dict = ModelService.own_role(userID=userID)
  2969. res['rid'] = role_dict['rid']
  2970. res['roleName'] = role_dict['roleName']
  2971. res['permList'] = ModelService.own_permission(userID)
  2972. res['userID'] = userID
  2973. # 昵称,邮箱,电话,刷新,头像
  2974. userIconPath = str(user_list[0]["userIconPath"])
  2975. if userIconPath and userIconPath.find('static/') != -1:
  2976. userIconPath = userIconPath.replace('static/', '').replace('\\', '/')
  2977. res['userIconUrl'] = SERVER_DOMAIN + 'account/getAvatar/' + userIconPath
  2978. else:
  2979. res['userIconUrl'] = ''
  2980. res['NickName'] = user_list[0]["NickName"] if user_list[0]["NickName"] is not None else ''
  2981. res['username'] = user_list[0]["username"] if user_list[0]["username"] is not None else ''
  2982. res['userEmail'] = user_list[0]["userEmail"] if user_list[0]["userEmail"] is not None else ''
  2983. res['phone'] = user_list[0]["phone"] if user_list[0]["phone"] is not None else ''
  2984. print(res)
  2985. return response.json(0, res)
  2986. class UserAppFrequencyView(TemplateView):
  2987. @method_decorator(csrf_exempt)
  2988. def dispatch(self, *args, **kwargs):
  2989. return super(UserAppFrequencyView, self).dispatch(*args, **kwargs)
  2990. def post(self, request, *args, **kwargs):
  2991. request.encoding = 'utf-8'
  2992. request_dict = request.POST
  2993. operation = kwargs.get('operation')
  2994. return self.validates(request_dict, operation)
  2995. def get(self, request, *args, **kwargs):
  2996. request.encoding = 'utf-8'
  2997. request_dict = request.GET
  2998. operation = kwargs.get('operation')
  2999. return self.validates(request_dict, operation)
  3000. def validates(self, request_dict, operation):
  3001. token = request_dict.get('token', None)
  3002. response = ResponseObject()
  3003. token = TokenObject(token)
  3004. if token.code != 0:
  3005. return response.json(token.code)
  3006. if operation == 'refresh':
  3007. return self.do_refresh(request_dict, token.userID, response)
  3008. else:
  3009. return response.json(404)
  3010. def do_refresh(self, request_dict, userID, response):
  3011. type = request_dict.get('type', None)
  3012. month = request_dict.get('month', None)
  3013. if not type or not month:
  3014. return response.json(444, 'type')
  3015. else:
  3016. type = int(type)
  3017. now_time = int(time.time())
  3018. month = int(month)
  3019. uaf_qs = UserAppFrequencyModel.objects.filter(user__userID=userID)
  3020. if not uaf_qs.exists():
  3021. user = Device_User.objects.filter(userID=userID)[0]
  3022. data = {
  3023. 'user': user,
  3024. 'type': type,
  3025. 'data_time': month,
  3026. 'add_time': now_time,
  3027. 'update_time': now_time,
  3028. }
  3029. UserAppFrequencyModel.objects.create(**data)
  3030. return response.json(0)
  3031. else:
  3032. updateMonth = time.strftime('%m', time.localtime(month))
  3033. uaf = uaf_qs.values('id', 'type', 'data_time')[0]
  3034. dbMonth = time.strftime('%m', time.localtime(int(uaf['data_time'])))
  3035. print('update month is ' + updateMonth)
  3036. print('db month is ' + dbMonth)
  3037. if updateMonth == dbMonth:
  3038. UserAppFrequencyModel.objects.filter(id=uaf['id']).update(type=type)
  3039. return response.json(0)
  3040. elif updateMonth > dbMonth:
  3041. user = Device_User.objects.filter(userID=userID)[0]
  3042. data = {
  3043. 'user': user,
  3044. 'type': type,
  3045. 'data_time': month,
  3046. 'add_time': now_time,
  3047. 'update_time': now_time,
  3048. }
  3049. UserAppFrequencyModel.objects.create(**data)
  3050. return response.json(0)
  3051. else:
  3052. return response.json(444, 'month')
  3053. class loginCodeView(View):
  3054. @method_decorator(csrf_exempt) # @csrf_exempt
  3055. def dispatch(self, *args, **kwargs):
  3056. return super(loginCodeView, self).dispatch(*args, **kwargs)
  3057. @ratelimit(key='ip', rate='2/m')
  3058. def post(self, request, *args, **kwargs):
  3059. request.encoding = 'utf-8'
  3060. lang = request.POST.get('lang', None)
  3061. if not lang:
  3062. lang = request.POST.get('language', None)
  3063. response = ResponseObject(lang)
  3064. request_dict = request.POST
  3065. phone = request_dict.get('phone', None)
  3066. if phone is not None:
  3067. was_limited = getattr(request, 'limited', False)
  3068. if was_limited is True:
  3069. return response.json(5)
  3070. return self.validate(request_dict, response)
  3071. @ratelimit(key='ip', rate='2/m')
  3072. def get(self, request, *args, **kwargs):
  3073. request.encoding = 'utf-8'
  3074. lang = request.GET.get('lang', None)
  3075. if not lang:
  3076. lang = request.GET.get('language', None)
  3077. response = ResponseObject(lang)
  3078. was_limited = getattr(request, 'limited', False)
  3079. if was_limited is True:
  3080. return response.json(5)
  3081. request_dict = request.GET
  3082. return self.validate(request_dict, response)
  3083. def validate(self, request_dict, response):
  3084. phone = request_dict.get('phone', None)
  3085. country_code = request_dict.get('country_code', None)
  3086. sign_name = request_dict.get('sign_name', None)
  3087. if phone and sign_name:
  3088. du_qs = Device_User.objects.filter(Q(phone=phone) | Q(username=phone))
  3089. if not du_qs.exists():
  3090. return response.json(104)
  3091. else:
  3092. redisObject = RedisObject()
  3093. login_code_key = '{phone}_login_code'.format(phone=phone)
  3094. login_code = redisObject.get_data(key=login_code_key)
  3095. login_code_ttl = redisObject.get_ttl(key=login_code_key)
  3096. if login_code_ttl > 240 and login_code:
  3097. return response.json(90)
  3098. login_code = RandomStr(6, True)
  3099. aliSms = AliSmsObject()
  3100. if sign_name == 'zosi':
  3101. sign_sms = '周视'
  3102. else:
  3103. sign_sms = 'Ansjer'
  3104. res = aliSms.send_code_sms(phone=phone, code=login_code, sign_name=sign_sms, temp_msg='SMS_151675022')
  3105. if res['Code'] == 'OK':
  3106. if redisObject.set_data(key=login_code_key, val=login_code, expire=300) is not True:
  3107. return response.json(48)
  3108. return response.json(0)
  3109. else:
  3110. return response.json(10, res['Message'])
  3111. else:
  3112. return response.json(444)
  3113. class v3LoginByCodeView(View):
  3114. @method_decorator(csrf_exempt) # @csrf_exempt
  3115. def dispatch(self, *args, **kwargs):
  3116. return super(v3LoginByCodeView, self).dispatch(*args, **kwargs)
  3117. def post(self, request, *args, **kwargs):
  3118. request.encoding = 'utf-8'
  3119. lang = request.POST.get('lang', None)
  3120. if not lang:
  3121. lang = request.POST.get('language', None)
  3122. response = ResponseObject(lang)
  3123. request_dict = request.POST
  3124. phone = request_dict.get('phone', None)
  3125. if phone is not None:
  3126. was_limited = getattr(request, 'limited', False)
  3127. if was_limited is True:
  3128. return response.json(5)
  3129. return self.validate(request_dict, response)
  3130. def get(self, request, *args, **kwargs):
  3131. request.encoding = 'utf-8'
  3132. lang = request.GET.get('lang', None)
  3133. if not lang:
  3134. lang = request.GET.get('language', None)
  3135. response = ResponseObject(lang)
  3136. was_limited = getattr(request, 'limited', False)
  3137. if was_limited is True:
  3138. return response.json(5)
  3139. request_dict = request.GET
  3140. return self.validate(request_dict, response)
  3141. def validate(self, request_dict, response):
  3142. phone = request_dict.get('phone', None)
  3143. code = request_dict.get('code', None)
  3144. if phone and code:
  3145. redisObject = RedisObject()
  3146. login_code_key = '{phone}_login_code'.format(phone=phone)
  3147. login_code = redisObject.get_data(key=login_code_key)
  3148. if login_code is not False:
  3149. print(code)
  3150. code = CommonService.decode_data(code)
  3151. print(code)
  3152. if login_code == code:
  3153. if response.lang is None:
  3154. response.lang = 'en'
  3155. return self.do_phone_login(phone, response)
  3156. else:
  3157. return response.json(121)
  3158. else:
  3159. return response.json(120)
  3160. def do_phone_login(self, phone, response):
  3161. user_qs = Device_User.objects.filter(Q(phone=phone) | Q(username=phone), is_active=True,
  3162. user_isValid=True)
  3163. return self.valid_login(user_qs, response)
  3164. def valid_login(self, user_qs, response):
  3165. if not user_qs.exists():
  3166. return response.json(104)
  3167. # users = user_qs.values('role__rid', 'role__roleName', 'userID', 'role', 'NickName', 'username', 'userEmail',
  3168. # 'phone', 'password', 'userIconPath', 'user_isValid', 'is_active')[0]
  3169. users = user_qs.values('role__rid', 'role__roleName', 'userID', 'NickName', 'username', 'userEmail',
  3170. 'phone', 'password', 'userIconPath')[0]
  3171. userID = users['userID']
  3172. tko = TokenObject()
  3173. res = tko.generate(
  3174. data={'userID': userID, 'lang': response.lang, 'user': users['username'],
  3175. 'm_code': '123413243214'})
  3176. # 添加用户登录类型
  3177. oauth_qs = UserOauth2Model.objects.filter(userID__userID=userID)
  3178. auth_type = 0
  3179. if oauth_qs.exists():
  3180. auth_type = oauth_qs[0].authType
  3181. if tko.code == 0:
  3182. now_time = datetime.datetime.utcnow().replace(tzinfo=utc).astimezone(utc)
  3183. user_qs.update(last_login=now_time, language=response.lang)
  3184. res['rid'] = users['role__rid']
  3185. res['roleName'] = users['role__roleName']
  3186. res['permList'] = ModelService.own_permission(userID)
  3187. res['userID'] = userID
  3188. # 昵称,邮箱,电话,刷新,头像
  3189. userIconPath = str(users['userIconPath'])
  3190. if userIconPath and userIconPath.find('static/') != -1:
  3191. userIconPath = userIconPath.replace('static/', '').replace('\\', '/')
  3192. res['userIconUrl'] = SERVER_DOMAIN + 'account/getAvatar/' + userIconPath
  3193. else:
  3194. res['userIconUrl'] = ''
  3195. res['NickName'] = users['NickName'] if users['NickName'] is not None else ''
  3196. res['username'] = users['username'] if users['username'] is not None else ''
  3197. res['userEmail'] = users['userEmail'] if users['userEmail'] is not None else ''
  3198. res['phone'] = users['phone'] if users['phone'] is not None else ''
  3199. res['authType'] = auth_type
  3200. return response.json(0, res)
  3201. else:
  3202. return response.json(tko.code)
  3203. class v3LoginByFingerprintView(View):
  3204. @method_decorator(csrf_exempt) # @csrf_exempt
  3205. def dispatch(self, *args, **kwargs):
  3206. return super(v3LoginByFingerprintView, self).dispatch(*args, **kwargs)
  3207. def post(self, request, *args, **kwargs):
  3208. request.encoding = 'utf-8'
  3209. lang = request.POST.get('lang', None)
  3210. if not lang:
  3211. lang = request.POST.get('language', None)
  3212. response = ResponseObject(lang)
  3213. request_dict = request.POST
  3214. was_limited = getattr(request, 'limited', False)
  3215. if was_limited is True:
  3216. return response.json(5)
  3217. return self.validate(request_dict, response)
  3218. def get(self, request, *args, **kwargs):
  3219. request.encoding = 'utf-8'
  3220. lang = request.GET.get('lang', None)
  3221. if not lang:
  3222. lang = request.GET.get('language', None)
  3223. response = ResponseObject(lang)
  3224. was_limited = getattr(request, 'limited', False)
  3225. if was_limited is True:
  3226. return response.json(5)
  3227. request_dict = request.GET
  3228. return self.validate(request_dict, response)
  3229. def validate(self, request_dict, response):
  3230. password = request_dict.get("password", None)
  3231. if password:
  3232. password = CommonService.decode_data(password)
  3233. if password is None:
  3234. return response.json(444)
  3235. else:
  3236. user_qs = Device_User.objects.filter(username=password, is_active=True, user_isValid=True)
  3237. if not user_qs.exists():
  3238. return response.json(104)
  3239. else:
  3240. users = user_qs.values('role__rid', 'role__roleName', 'userID', 'NickName', 'username', 'userEmail',
  3241. 'phone', 'password', 'userIconPath', 'fingerprint_enable',
  3242. 'fingerprint_key')[0]
  3243. if users['fingerprint_enable'] == 0:
  3244. return response.json(112)
  3245. else:
  3246. userID = users['userID']
  3247. tko = TokenObject()
  3248. res = tko.generate(
  3249. data={'userID': userID, 'lang': response.lang, 'user': users['username'],
  3250. 'm_code': '123413243214'})
  3251. if tko.code == 0:
  3252. now_time = datetime.datetime.utcnow().replace(tzinfo=utc).astimezone(utc)
  3253. user_qs.update(last_login=now_time, language=response.lang)
  3254. res['rid'] = users['role__rid']
  3255. res['roleName'] = users['role__roleName']
  3256. res['permList'] = ModelService.own_permission(userID)
  3257. res['userID'] = userID
  3258. # 昵称,邮箱,电话,刷新,头像
  3259. userIconPath = str(users['userIconPath'])
  3260. if userIconPath and userIconPath.find('static/') != -1:
  3261. userIconPath = userIconPath.replace('static/', '').replace('\\', '/')
  3262. res['userIconUrl'] = SERVER_DOMAIN + 'account/getAvatar/' + userIconPath
  3263. else:
  3264. res['userIconUrl'] = ''
  3265. res['NickName'] = users['NickName'] if users['NickName'] is not None else ''
  3266. res['username'] = users['username'] if users['username'] is not None else ''
  3267. res['userEmail'] = users['userEmail'] if users['userEmail'] is not None else ''
  3268. res['phone'] = users['phone'] if users['phone'] is not None else ''
  3269. res['fingerprint_enable'] = users['fingerprint_enable']
  3270. res['fingerprint_key'] = CommonService.encode_data(content=users['fingerprint_key'],
  3271. start=2)
  3272. return response.json(0, res)
  3273. else:
  3274. return response.json(tko.code)
  3275. else:
  3276. return response.json(444)
  3277. class v3SetFingerprintView(View):
  3278. @method_decorator(csrf_exempt)
  3279. def dispatch(self, *args, **kwargs):
  3280. return super(v3SetFingerprintView, self).dispatch(*args, **kwargs)
  3281. def post(self, request, *args, **kwargs):
  3282. request.encoding = 'utf-8'
  3283. request_dict = request.POST
  3284. return self.validate(request_dict)
  3285. def get(self, request, *args, **kwargs):
  3286. request.encoding = 'utf-8'
  3287. request_dict = request.GET
  3288. return self.validate(request_dict)
  3289. def validate(self, request_dict):
  3290. lang = request_dict.get('lang', None)
  3291. token = request_dict.get('token', None)
  3292. fingerprint_enable = request_dict.get('fingerprint_enable', None)
  3293. fingerprint_key = request_dict.get('fingerprint_key', None)
  3294. response = ResponseObject()
  3295. token = TokenObject(token)
  3296. if token.code != 0:
  3297. return response.json(token.code)
  3298. if not lang:
  3299. return response.json(444, 'lang')
  3300. response.lang = lang
  3301. data = {}
  3302. if fingerprint_enable:
  3303. data['fingerprint_enable'] = int(fingerprint_enable)
  3304. if fingerprint_key:
  3305. data['fingerprint_key'] = CommonService.decode_data(fingerprint_key, end=3)
  3306. if len(data) > 0:
  3307. Device_User.objects.filter(userID=token.userID).update(**data)
  3308. return response.json(0)
  3309. class AppleAuthLogin(View):
  3310. def post(self, request, *args, **kwargs):
  3311. request.encoding = 'utf-8'
  3312. request_dict = request.POST
  3313. return self.validate(request_dict)
  3314. def get(self, request, *args, **kwargs):
  3315. request.encoding = 'utf-8'
  3316. request_dict = request.GET
  3317. return self.validate(request_dict)
  3318. def validate(self, request_dict):
  3319. lang = request_dict.get('lang', None)
  3320. identity_token = request_dict.get('identity_token', None)
  3321. app_bundle_id = request_dict.get('app_bundle_id', None) # 包名
  3322. response = ResponseObject(lang)
  3323. identity_token = CommonService.decode_data(identity_token)
  3324. # print(identity_token)
  3325. if identity_token:
  3326. key_url = 'https://appleid.apple.com/auth/keys'
  3327. key_response = requests.get(key_url).json()
  3328. # print(key_response)
  3329. head = jwt.get_unverified_header(identity_token)
  3330. # print(head)
  3331. token_key = head['kid']
  3332. key_object = None
  3333. alg = None
  3334. for pub_key in key_response['keys']:
  3335. if pub_key['kid'] == token_key:
  3336. key_object = simplejson.dumps(pub_key)
  3337. key_object = RSAAlgorithm.from_jwk(key_object)
  3338. alg = pub_key['alg']
  3339. break
  3340. if key_object:
  3341. try:
  3342. claims = jwt.decode(identity_token, key=key_object, verify=True, algorithms=[alg],
  3343. audience=app_bundle_id)
  3344. unionID = claims['sub']
  3345. print(claims)
  3346. user_extend_qs = UserOauth2Model.objects.filter(unionID=unionID, authType=2)
  3347. if user_extend_qs.exists():
  3348. # 如果用户绑定过则直接登录
  3349. userID = user_extend_qs[0].userID_id
  3350. print(userID)
  3351. user_qs = Device_User.objects.filter(userID=userID)
  3352. return self.do_login(user_qs, response)
  3353. else:
  3354. # 如果用户为绑定过则创建用户并进行登录返回token
  3355. userID = CommonService.getUserID(getUser=False)
  3356. if claims.__contains__('email'):
  3357. nickname = claims['email']
  3358. else:
  3359. nickname = 'apple_{num}'.format(num=CommonService.RandomStr(6, False))
  3360. return self.do_register(userID, nickname, response, app_bundle_id, unionID)
  3361. except Exception as e:
  3362. print(e)
  3363. return response.json(717)
  3364. else:
  3365. return response.json(444)
  3366. else:
  3367. return response.json(444)
  3368. # 登录
  3369. def do_login(self, user_qs, response):
  3370. now_time = datetime.datetime.utcnow().replace(tzinfo=utc).astimezone(utc)
  3371. userID = user_qs[0].userID
  3372. print('userID' + userID)
  3373. tko = TokenObject()
  3374. user_list = user_qs.values("NickName", "userIconUrl", "userIconPath", "username", "userEmail", "phone")
  3375. res = tko.generate(data={'userID': userID, 'lang': response.lang, 'user': user_list[0]["username"]})
  3376. # 增加角色
  3377. user_qs[0].role.add(Role.objects.get(rid=1))
  3378. role_dict = ModelService.own_role(userID=userID)
  3379. res['rid'] = role_dict['rid']
  3380. res['roleName'] = role_dict['roleName']
  3381. res['permList'] = ModelService.own_permission(userID)
  3382. res['userID'] = userID
  3383. # 昵称,邮箱,电话,刷新,头像
  3384. userIconPath = str(user_list[0]["userIconPath"])
  3385. if userIconPath and userIconPath.find('static/') != -1:
  3386. userIconPath = userIconPath.replace('static/', '').replace('\\', '/')
  3387. res['userIconUrl'] = SERVER_DOMAIN + 'account/getAvatar/' + userIconPath
  3388. else:
  3389. res['userIconUrl'] = ''
  3390. res['NickName'] = user_list[0]["NickName"] if user_list[0]["NickName"] is not None else ''
  3391. res['username'] = user_list[0]["username"] if user_list[0]["username"] is not None else ''
  3392. res['userEmail'] = user_list[0]["userEmail"] if user_list[0]["userEmail"] is not None else ''
  3393. res['phone'] = user_list[0]["phone"] if user_list[0]["phone"] is not None else ''
  3394. print(res)
  3395. # 添加用户登录类型
  3396. oauth_qs = UserOauth2Model.objects.filter(userID__userID=userID)
  3397. auth_type = 0
  3398. if oauth_qs.exists():
  3399. auth_type = oauth_qs[0].authType
  3400. res['authType'] = auth_type
  3401. user_qs.update(last_login=now_time, online=True)
  3402. return response.json(0, res)
  3403. def do_register(self, userID, nickname, response, appBundleId, unionID):
  3404. data_valid = DataValid()
  3405. if data_valid.name_validate(userID) is not True:
  3406. return response.json(105)
  3407. try:
  3408. users = Device_User.objects.create(
  3409. username=userID,
  3410. NickName=nickname,
  3411. password=make_password('123456'),
  3412. userID=userID,
  3413. is_active=True,
  3414. user_isValid=True,
  3415. )
  3416. nowTime = int(time.time())
  3417. UserOauth2Model.objects.create(
  3418. addTime=nowTime,
  3419. updTime=nowTime,
  3420. userID_id=users.userID,
  3421. authType=2,
  3422. unionID=unionID
  3423. )
  3424. except Exception as e:
  3425. errorInfo = traceback.format_exc()
  3426. print(errorInfo)
  3427. return response.json(424, repr(e))
  3428. else:
  3429. user_qs = Device_User.objects.filter(Q(userID=userID))
  3430. print('---')
  3431. print(user_qs)
  3432. return self.do_login(user_qs, response)
  3433. class LocalUserView(View):
  3434. def get(self, request, *args, **kwargs):
  3435. request.encoding = 'utf-8'
  3436. request_dict = request.GET
  3437. operation = kwargs.get('operation', None)
  3438. return self.validate(request_dict, operation)
  3439. def post(self, request, *args, **kwargs):
  3440. request.encoding = 'utf-8'
  3441. request_dict = request.POST
  3442. operation = kwargs.get('operation', None)
  3443. print('start_time=' + str((time.time())))
  3444. ip = CommonService.get_ip_address(request)
  3445. print('end_time=' + str((time.time())))
  3446. return self.validate(request_dict, operation)
  3447. def validate(self, request_dict, operation):
  3448. language = request_dict.get('language', None)
  3449. response = ResponseObject(lang=language)
  3450. if operation == 'login':
  3451. return self.do_local_login(request_dict, response)
  3452. else:
  3453. return response.json(404)
  3454. def do_local_login(self, request_dict, response):
  3455. username = request_dict.get('username', None)
  3456. app_bundle_id = request_dict.get('app_bundle_id', None)
  3457. if username is None:
  3458. return response.json(444)
  3459. user_qs = Device_User.objects.filter(username=username)
  3460. if user_qs.exists():
  3461. return self.do_login(user_qs, response)
  3462. else:
  3463. # 如果用户为绑定过则创建用户并进行登录返回token
  3464. userID = CommonService.getUserID(μs=False, setOTAID=True)
  3465. nickname = 'local_{num}'.format(num=CommonService.RandomStr(6, False))
  3466. return self.do_register(userID, username, nickname, response, app_bundle_id)
  3467. # 登录
  3468. def do_login(self, user_qs, response):
  3469. now_time = datetime.datetime.utcnow().replace(tzinfo=utc).astimezone(utc)
  3470. userID = user_qs[0].userID
  3471. print('userID' + userID)
  3472. tko = TokenObject()
  3473. user_list = user_qs.values("NickName", "userIconUrl", "userIconPath", "username", "userEmail", "phone")
  3474. res = tko.generate(data={'userID': userID, 'lang': response.lang, 'user': user_list[0]["username"]})
  3475. # 增加角色
  3476. user_qs[0].role.add(Role.objects.get(rid=1))
  3477. role_dict = ModelService.own_role(userID=userID)
  3478. res['rid'] = role_dict['rid']
  3479. res['roleName'] = role_dict['roleName']
  3480. res['permList'] = ModelService.own_permission(userID)
  3481. res['userID'] = userID
  3482. # 昵称,邮箱,电话,刷新,头像
  3483. userIconPath = str(user_list[0]["userIconPath"])
  3484. if userIconPath and userIconPath.find('static/') != -1:
  3485. userIconPath = userIconPath.replace('static/', '').replace('\\', '/')
  3486. res['userIconUrl'] = SERVER_DOMAIN + 'account/getAvatar/' + userIconPath
  3487. else:
  3488. res['userIconUrl'] = ''
  3489. res['NickName'] = user_list[0]["NickName"] if user_list[0]["NickName"] is not None else ''
  3490. res['username'] = user_list[0]["username"] if user_list[0]["username"] is not None else ''
  3491. res['userEmail'] = user_list[0]["userEmail"] if user_list[0]["userEmail"] is not None else ''
  3492. res['phone'] = user_list[0]["phone"] if user_list[0]["phone"] is not None else ''
  3493. print(res)
  3494. # 添加用户登录类型
  3495. oauth_qs = UserOauth2Model.objects.filter(userID__userID=userID)
  3496. auth_type = 0
  3497. if oauth_qs.exists():
  3498. auth_type = oauth_qs[0].authType
  3499. res['authType'] = auth_type
  3500. user_qs.update(last_login=now_time, online=True)
  3501. return response.json(0, res)
  3502. def do_register(self, userID, username, nickname, response, appBundleId):
  3503. data_valid = DataValid()
  3504. if data_valid.name_validate(userID) is not True:
  3505. return response.json(105)
  3506. try:
  3507. Device_User.objects.create(
  3508. username=username,
  3509. NickName=nickname,
  3510. password=make_password('123456'),
  3511. userID=userID,
  3512. is_active=True,
  3513. user_isValid=True,
  3514. is_local=True
  3515. )
  3516. except Exception as e:
  3517. errorInfo = traceback.format_exc()
  3518. print(errorInfo)
  3519. return response.json(424, repr(e))
  3520. else:
  3521. user_qs = Device_User.objects.filter(Q(userID=userID))
  3522. print('---')
  3523. print(user_qs)
  3524. return self.do_login(user_qs, response)
  3525. class SubscribeEmailView(View):
  3526. def get(self, request, *args, **kwargs):
  3527. request.encoding = 'utf-8'
  3528. request_dict = request.GET
  3529. return self.validate(request_dict)
  3530. def post(self, request, *args, **kwargs):
  3531. request.encoding = 'utf-8'
  3532. request_dict = request.POST
  3533. return self.validate(request_dict)
  3534. def validate(self, request_dict):
  3535. token = request_dict.get('token', None)
  3536. lang = request_dict.get('lang', None)
  3537. token = TokenObject(token)
  3538. response = ResponseObject(lang=lang)
  3539. if token.code != 0:
  3540. return response.json(token.code)
  3541. status = request_dict.get('subscribe', None)
  3542. if status is None:
  3543. return response.json(444)
  3544. status = int(status)
  3545. user_qs = Device_User.objects.filter(userID=token.userID)
  3546. if user_qs.exists():
  3547. user_qs.update(subscribe_email=status)
  3548. return response.json(0)
  3549. else:
  3550. return response.json(104)
  3551. def delete_local_account(username):
  3552. user_qs = Device_User.objects.filter(username=username)
  3553. print(user_qs)
  3554. if user_qs.exists():
  3555. user = user_qs[0]
  3556. if user.is_local:
  3557. user.delete()
  3558. Device_Info.objects.filter(userID__userID=user.userID).delete()
  3559. def updateUserCountry(request):
  3560. country_ip_qs = CountryIPModel.objects.filter(status=0)[0: 100]
  3561. if country_ip_qs.exists():
  3562. country_ip_qs = country_ip_qs.values()
  3563. redisObject = RedisObject(db=6)
  3564. for country_ip in country_ip_qs:
  3565. key = 'ip_country_{ip}'.format(ip=country_ip['ip'])
  3566. data = redisObject.get_data(key=key)
  3567. if data:
  3568. country = data
  3569. else:
  3570. country = CommonService.getIpIpInfo(country_ip['ip'], lang='CN')
  3571. country = country['country_name']
  3572. redisObject.set_data(key=key, val=country, expire=3600)
  3573. country_ip['country'] = country
  3574. CountryIPModel.objects.filter(id=country_ip['id']).update(country=country, status=1)
  3575. # ids.append(country_ip['id'])
  3576. # CountryIPModel.objects.filter(id__in=tuple(ids)).update(status=1)
  3577. response = ResponseObject()
  3578. return response.json(0)
  3579. def confirmRegion(request):
  3580. response = ResponseObject()
  3581. request.encoding = 'utf-8'
  3582. # test
  3583. number = request.POST.get('number', None)
  3584. if number is None:
  3585. return response.json(309)
  3586. number = request.POST.get('number', None)
  3587. selectRegion = CountryModel.objects.filter(number=number).values('region__api')
  3588. if not selectRegion.exists():
  3589. return response.json(0,{"request_url":"https://www.dvema.com"})
  3590. else:
  3591. return response.json(0,{"request_url":selectRegion[0]['region__api']})
  3592. def deleteAccount(request):
  3593. response = ResponseObject()
  3594. request.encoding = 'utf-8'
  3595. #test
  3596. token = request.POST.get('token',None)
  3597. password = request.POST.get('userPwd',None)
  3598. try:
  3599. for i in range(1, 4):
  3600. if i == 1:
  3601. # 第一次先解密
  3602. password = base64.b64decode(password)
  3603. password = password.decode('utf-8')
  3604. # 截去第一位,最后一位
  3605. password = password[1:-1]
  3606. if i == 2:
  3607. # 第2次先解密
  3608. password = base64.b64decode(password)
  3609. password = password.decode('utf-8')
  3610. # 去前2位,后2位
  3611. password = password[2:-2]
  3612. if i == 3:
  3613. # 第3次先解密
  3614. password = base64.b64decode(password)
  3615. password = password.decode('utf-8')
  3616. # 去前3位,后3位
  3617. password = password[3:-3]
  3618. except Exception as e:
  3619. return response.json(111)
  3620. else:
  3621. if token is None:
  3622. return response.json(309)
  3623. tko = TokenObject(token)
  3624. response.lang = tko.lang
  3625. if tko.code != 0:
  3626. return response.json(tko.code)
  3627. userID = tko.userID
  3628. if not userID:
  3629. return response.json(309)
  3630. delUser = Device_User.objects.filter(userID=userID)
  3631. if not delUser.exists():
  3632. return response.json(104)
  3633. userPWD = delUser.values('password')[0]
  3634. if not check_password(password, userPWD['password']):
  3635. return response.json(111)
  3636. hasDevices = Device_Info.objects.filter(userID=userID)
  3637. if hasDevices.exists():
  3638. return response.json(10047)
  3639. orderUserIds = Order_Model.objects.filter(userID=userID,status=1).values_list('UID').distinct().order_by("UID")
  3640. hadUseSevice = UID_Bucket.objects.filter(uid__in=orderUserIds,use_status=1)
  3641. if hadUseSevice.exists():
  3642. return response.json(10046)
  3643. hadUnUseSevice = Unused_Uid_Meal.objects.filter(uid__in=orderUserIds)
  3644. if hadUnUseSevice.exists():
  3645. return response.json(10046)
  3646. # return HttpResponse(list(hadUseSevice.values()))
  3647. delUser.delete()
  3648. return response.json(0)
  3649. class InitUserInformationView(View):
  3650. def get(self, request, *args, **kwargs):
  3651. request.encoding = 'utf-8'
  3652. operation = kwargs.get('operation', None)
  3653. request_dict = request.GET
  3654. return self.validate(request_dict, operation)
  3655. def post(self, request, *args, **kwargs):
  3656. request.encoding = 'utf-8'
  3657. operation = kwargs.get('operation', None)
  3658. request_dict = request.POST
  3659. return self.validate(request_dict, operation)
  3660. def validate(self, request_dict, operation):
  3661. token = TokenObject(request_dict.get('token', None))
  3662. response = ResponseObject()
  3663. if token.code != 0:
  3664. return response.json(token.code)
  3665. if operation == 'init':
  3666. return self.do_init(token.userID, request_dict, response)
  3667. else:
  3668. return response.json(444)
  3669. def do_init(self, userID, request_dict, response):
  3670. appBundleId = request_dict.get('appBundleId', None)
  3671. if appBundleId:
  3672. user_ex_qs = UserExModel.objects.filter(userID_id=userID)
  3673. now_time = int(time.time())
  3674. if user_ex_qs.exists():
  3675. update = {
  3676. 'appBundleId':appBundleId,
  3677. 'updTime': now_time
  3678. }
  3679. user_ex_qs.update(**update)
  3680. else:
  3681. user_ex = UserExModel(userID_id=userID, appBundleId=appBundleId, addTime=now_time, updTime=now_time)
  3682. user_ex.save()
  3683. return response.json(0)
  3684. else:
  3685. return response.json(444)