UserManger.py 19 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446
  1. import os
  2. import traceback
  3. import boto3
  4. import botocore
  5. from botocore import client
  6. import simplejson as json
  7. from django.http import HttpResponse
  8. from django.utils.decorators import method_decorator
  9. from django.views.decorators.csrf import csrf_exempt
  10. from django.views.generic import TemplateView, View
  11. from Ansjer.config import BASE_DIR, ACCESS_KEY_ID, SECRET_ACCESS_KEY, REGION_NAME, AVATAR_BUCKET
  12. from Ansjer.config import SERVER_DOMAIN
  13. from Model.models import Role, Device_User, UserOauth2Model, UserExModel, CountryLanguageModel, LanguageModel, App_Info
  14. from Object.RedisObject import RedisObject
  15. from Object.ResponseObject import ResponseObject
  16. from Object.TokenObject import TokenObject
  17. from Service.CommonService import CommonService
  18. from Service.ModelService import ModelService
  19. class showUserMoreView(TemplateView):
  20. @method_decorator(csrf_exempt)
  21. def dispatch(self, *args, **kwargs):
  22. return super(showUserMoreView, self).dispatch(*args, **kwargs)
  23. def post(self, request, *args, **kwargs):
  24. request.encoding = 'utf-8'
  25. return self.validation(request.POST)
  26. def get(self, request, *args, **kwargs):
  27. request.encoding = 'gb2312'
  28. return self.validation(request.GET)
  29. def validation(self, request_dict):
  30. response = ResponseObject()
  31. token = request_dict.get('token', None)
  32. lang = request_dict.get('lang', 'en')
  33. app_bundle_id = request_dict.get('app_bundle_id', None)
  34. tko = TokenObject(token)
  35. response.lang = tko.lang
  36. if tko.code != 0:
  37. return response.json(tko.code)
  38. user_id = tko.userID
  39. if not user_id:
  40. return response.json(104)
  41. return self.show_user_more(user_id, lang, app_bundle_id, response)
  42. @staticmethod
  43. def show_user_more(user_id, lang, app_bundle_id, response):
  44. """
  45. 获取用户完整信息
  46. @param user_id: 用户id
  47. @param lang: 语言
  48. @param app_bundle_id: app包id
  49. @param response: 响应
  50. @return: response
  51. """
  52. device_user_qs = Device_User.objects.filter(userID=user_id)
  53. if not device_user_qs.exists():
  54. return response.json(104)
  55. try:
  56. sqlDict = CommonService.qs_to_dict(device_user_qs)
  57. for k, v in enumerate(sqlDict["datas"]):
  58. sqlDict['datas'][k]['fields'].pop('password')
  59. userIconPath = sqlDict['datas'][k]['fields']['userIconPath']
  60. region_status = sqlDict['datas'][k]['fields']['region_status']
  61. if userIconPath:
  62. if userIconPath.find('static/') != -1:
  63. userIconPath = userIconPath.replace('static/', '').replace('\\', '/')
  64. userIconUrl = SERVER_DOMAIN + 'account/getAvatar/' + userIconPath
  65. sqlDict['datas'][k]['fields']['userIconUrl'] = userIconUrl
  66. # 确认地区字段
  67. sqlDict['datas'][k]['fields']['region_status'] = 1 if region_status else 0
  68. if len(v['fields']['role']):
  69. roleName = ModelService.getRole(rid=v['fields']['role'][0])
  70. sqlDict["datas"][k]['fields']['rolename'] = roleName
  71. # 根据region_country的值返回api和region数据
  72. sqlDict["datas"][k]['fields']['api'] = ''
  73. sqlDict['datas'][k]['fields']['region'] = ''
  74. region_country = sqlDict["datas"][k]['fields']['region_country']
  75. if region_country != 0:
  76. # api数据
  77. country_qs = CountryLanguageModel.objects.filter(country_id=region_country).values(
  78. 'country__region__api', 'country__region__zosi_api', 'country__region__loocam_api')
  79. sqlDict["datas"][k]['fields']['api'] = country_qs[0]['country__region__api']
  80. if region_country == 1: # 中国返回美洲域名
  81. sqlDict["datas"][k]['fields']['api'] = 'https://www.dvema.com/'
  82. # 根据app_bundle_id返回对应域名
  83. if app_bundle_id:
  84. # 查询app名
  85. app_inf_qs = App_Info.objects.filter(appBundleId=app_bundle_id).values('appName')
  86. if app_inf_qs.exists():
  87. app_name = app_inf_qs[0]['appName']
  88. if 'Zosi' in app_name:
  89. sqlDict['datas'][k]['fields']['api'] = country_qs[0]['country__region__zosi_api']
  90. elif 'Loocam' in app_name:
  91. sqlDict['datas'][k]['fields']['api'] = country_qs[0]['country__region__loocam_api']
  92. # region数据
  93. region_country = sqlDict['datas'][k]['fields']['region_country']
  94. language_qs = LanguageModel.objects.filter(lang=lang).values('id')
  95. region_qs = CountryLanguageModel.objects.filter(country_id=region_country,
  96. language_id=language_qs[0]['id']).\
  97. values('country_name')
  98. sqlDict['datas'][k]['fields']['region'] = region_qs[0]['country_name'] if region_qs.exists() else ''
  99. # 增加oauth2关联数据
  100. ua_qs = UserOauth2Model.objects.filter(userID_id=user_id).values_list('authType', flat=True)
  101. sqlDict['oauth2'] = list(ua_qs)
  102. return response.json(0, sqlDict)
  103. except Exception as e:
  104. return response.json(500, repr(e))
  105. class perfectUserInfoView(TemplateView):
  106. # 完善个人信息
  107. @method_decorator(csrf_exempt)
  108. def dispatch(self, *args, **kwargs):
  109. return super(perfectUserInfoView, self).dispatch(*args, **kwargs)
  110. def post(self, request, *args, **kwargs):
  111. request.encoding = 'utf-8'
  112. userContent = request.POST.get('content', None)
  113. userIcon = request.FILES.get('userIcon', None)
  114. token = request.POST.get('token', None)
  115. return self.ValidationError(token, userContent, userIcon)
  116. def get(self, request, *args, **kwargs):
  117. request.encoding = 'gb2312'
  118. userContent = request.GET.get('content', None)
  119. userIcon = request.FILES.get('userIcon', None)
  120. token = request.GET.get('token', None)
  121. return self.ValidationError(token, userContent, userIcon)
  122. def ValidationError(self, token, userContent, userIcon):
  123. response = ResponseObject()
  124. tko = TokenObject(token)
  125. response.lang = tko.lang
  126. if tko.code != 0:
  127. return response.json(tko.code)
  128. userID = tko.userID
  129. if not userID:
  130. return response.json(309)
  131. userIconPath = ''
  132. if userIcon:
  133. # 上传头像到aws s3
  134. aws_s3_client = boto3.client(
  135. 's3',
  136. region_name=REGION_NAME,
  137. aws_access_key_id=ACCESS_KEY_ID,
  138. aws_secret_access_key=SECRET_ACCESS_KEY,
  139. config=botocore.client.Config(signature_version='s3v4'),
  140. )
  141. Key = userID + '/' + userIcon.name
  142. aws_s3_client.put_object(Bucket=AVATAR_BUCKET, Key=Key, Body=userIcon)
  143. userIconPath = userID + '/' + userIcon.name
  144. if userContent:
  145. dataValid = json.loads(userContent)
  146. if 'userID' and 'password' and 'is_superuser' in dataValid.keys():
  147. return response.json(444)
  148. if not userIconPath and not userContent:
  149. return response.json(444)
  150. elif not userIconPath and userContent:
  151. return self.perfectUserInfoUpdate(userID, response, userContent=userContent)
  152. elif userIconPath and not userContent:
  153. return self.perfectUserInfoUpdate(userID, response, userIconPath=userIconPath)
  154. else:
  155. return self.perfectUserInfoUpdate(userID, response, userIconPath=userIconPath, userContent=userContent)
  156. def perfectUserInfoUpdate(slef, userID, response, **kwargs):
  157. """
  158. :param username:
  159. :param userContent:
  160. :param args:
  161. :param kwargs:
  162. :return:
  163. """
  164. User = Device_User.objects.filter(userID=userID)
  165. if not User.exists():
  166. return response.json(104)
  167. userIconPath = kwargs.get('userIconPath', None)
  168. userContent = kwargs.get('userContent', None)
  169. userIconUrl = ""
  170. if userIconPath:
  171. userIconUrl = SERVER_DOMAIN + 'account/getAvatar/' + userIconPath
  172. if userContent:
  173. try:
  174. UserData = json.loads(userContent)
  175. except Exception as e:
  176. return response.json(444, repr(e))
  177. try:
  178. if userIconPath and userContent:
  179. User.update(userIconPath=userIconPath, userIconUrl=userIconUrl, **UserData)
  180. elif not userIconPath and userContent:
  181. User.update(**UserData)
  182. elif userIconPath and not userContent:
  183. User.update(userIconPath=userIconPath, userIconUrl=userIconUrl)
  184. except Exception as e:
  185. return response.json(117, repr(e))
  186. else:
  187. res = CommonService.qs_to_dict(User)
  188. for k, v in enumerate(res["datas"]):
  189. res['datas'][k]['fields'].pop('password')
  190. userIconPath = res['datas'][k]['fields']['userIconPath']
  191. if userIconPath and userIconUrl != '':
  192. res['datas'][k]['fields']['userIconUrl'] = userIconUrl
  193. return response.json(0, res)
  194. class getAvatarView(TemplateView):
  195. @method_decorator(csrf_exempt)
  196. def dispatch(self, *args, **kwargs):
  197. return super(getAvatarView, self).dispatch(*args, **kwargs)
  198. def post(self, request, *args, **kwargs):
  199. request.encoding = 'utf-8'
  200. filePath = kwargs.get('filePath', '')
  201. filePath.encode(encoding='utf-8', errors='strict')
  202. return self.getAvatar(filePath)
  203. def get(self, request, *args, **kwargs):
  204. request.encoding = 'utf-8'
  205. filePath = kwargs.get('filePath', '')
  206. filePath.encode(encoding='utf-8', errors='strict')
  207. return self.getAvatar(filePath)
  208. def getAvatar(self, filePath):
  209. response = ResponseObject()
  210. if not filePath:
  211. return response.json(800)
  212. if filePath == 'User/default.png' or filePath == 'User/defaultUser.png':
  213. # 使用默认头像
  214. try:
  215. aws_s3_client = boto3.client(
  216. 's3',
  217. region_name=REGION_NAME,
  218. aws_access_key_id=ACCESS_KEY_ID,
  219. aws_secret_access_key=SECRET_ACCESS_KEY,
  220. config=botocore.client.Config(signature_version='s3v4'),
  221. )
  222. get_object_response = aws_s3_client.get_object(Bucket=AVATAR_BUCKET, Key='default/default.png')
  223. return HttpResponse(get_object_response['Body'], content_type="image/jpeg")
  224. except Exception as e:
  225. print(e)
  226. return response.json(500, repr(e))
  227. fullPath = os.path.join(BASE_DIR, "static", filePath).replace('\\', '/')
  228. if os.path.isfile(fullPath):
  229. try:
  230. imageData = open(fullPath, 'rb').read()
  231. return HttpResponse(imageData, content_type="image/jpeg")
  232. except Exception as e:
  233. return response.json(906, repr(e))
  234. else:
  235. try:
  236. aws_s3_client = boto3.client(
  237. 's3',
  238. region_name=REGION_NAME,
  239. aws_access_key_id=ACCESS_KEY_ID,
  240. aws_secret_access_key=SECRET_ACCESS_KEY,
  241. config=botocore.client.Config(signature_version='s3v4'),
  242. )
  243. get_object_response = aws_s3_client.get_object(Bucket=AVATAR_BUCKET, Key=filePath)
  244. return HttpResponse(get_object_response['Body'], content_type="image/jpeg")
  245. except Exception as e:
  246. return response.json(906, repr(e))
  247. @csrf_exempt
  248. def delUserInterface(request):
  249. response = ResponseObject()
  250. if request.method == 'POST':
  251. request_dict = request.POST
  252. elif request.method == 'GET':
  253. request_dict = request.GET
  254. else:
  255. return response.json(404)
  256. token = request_dict.get('token', None)
  257. delUserID = request_dict.get('delUserID', None)
  258. if not token or not delUserID:
  259. return response.json(444, 'token,delUserID')
  260. tko = TokenObject(token)
  261. response.lang = tko.lang
  262. if tko.code != 0:
  263. return response.json(tko.code)
  264. userID = tko.userID
  265. if not userID:
  266. return response.json(309)
  267. userValid = Device_User.objects.filter(userID=userID)
  268. if not userValid.exists():
  269. return response.json(104)
  270. own_perm = ModelService.check_perm(userID=userID, permID=10)
  271. if own_perm is not True:
  272. return response.json(404)
  273. delUser = Device_User.objects.filter(userID=delUserID)
  274. if not delUser.exists():
  275. return response.json(104)
  276. delUser.delete()
  277. return response.json(0)
  278. @csrf_exempt
  279. def showAllUserInterface(request):
  280. response = ResponseObject()
  281. if request.method == 'POST':
  282. request_dict = request.POST
  283. elif request.method == 'GET':
  284. request_dict = request.GET
  285. else:
  286. return response.json(404)
  287. token = request_dict.get('token', None)
  288. type = request_dict.get('type', None)
  289. tko = TokenObject(token)
  290. response.lang = tko.lang
  291. if tko.code != 0:
  292. return response.json(tko.code)
  293. userID = tko.userID
  294. if not userID:
  295. return response.json(104)
  296. if type == 'PC':
  297. line = int(request.POST.get('line', None))
  298. page = int(request.POST.get('page', None))
  299. userValid = Device_User.objects.filter(userID=userID).order_by('-data_joined')
  300. if not userValid:
  301. return response.json(104)
  302. # 管理员查询
  303. own_permission = ModelService.check_perm(userID=userID, permID=30)
  304. if own_permission is not True:
  305. return response.json(404)
  306. device_user_queryset = Device_User.objects.all()
  307. device_user_count = device_user_queryset.count()
  308. device_user_res = device_user_queryset.order_by('-data_joined')[(page - 1) * line:page * line]
  309. sqlDict = CommonService.qs_to_dict(device_user_res)
  310. redisObj = RedisObject(db=3)
  311. for k, v in enumerate(sqlDict["datas"]):
  312. if len(v['fields']['role']) > 0:
  313. role_query_set = Role.objects.get(rid=v['fields']['role'][0])
  314. sqlDict["datas"][k]['fields']['role'].append(role_query_set.roleName)
  315. if redisObj.get_data(key=v['pk']):
  316. sqlDict["datas"][k]['fields']['online'] = True
  317. else:
  318. sqlDict["datas"][k]['fields']['online'] = False
  319. ue = UserExModel.objects.filter(userID=sqlDict["datas"][k]['pk'])
  320. if ue.exists():
  321. sqlDict["datas"][k]['fields']['appBundleId'] = ue[0].appBundleId
  322. else:
  323. sqlDict["datas"][k]['fields']['appBundleId'] = ''
  324. sqlDict['count'] = device_user_count
  325. return response.json(0, sqlDict)
  326. else:
  327. own_permission = ModelService.check_perm(userID=userID, permID=30)
  328. if own_permission is not True:
  329. return response.json(404)
  330. qs = Device_User.objects.all().order_by('-data_joined')
  331. res = CommonService.qs_to_dict(qs)
  332. return response.json(0, res)
  333. class setUserValidView(View):
  334. @method_decorator(csrf_exempt)
  335. def dispatch(self, *args, **kwargs):
  336. return super(setUserValidView, self).dispatch(*args, **kwargs)
  337. def post(self, request, *args, **kwargs):
  338. request.encoding = 'utf-8'
  339. request_dict = request.POST
  340. return self.setUserValid(request_dict)
  341. def get(self, request, *args, **kwargs):
  342. request.encoding = 'utf-8'
  343. request_dict = request.GET
  344. return self.setUserValid(request_dict)
  345. def setUserValid(self, request_dict):
  346. token = request_dict.get('token', None)
  347. eUserID = request_dict.get('userID', None)
  348. isValid = request_dict.get('isValid', None)
  349. rid = request_dict.get('rid', None)
  350. response = ResponseObject()
  351. tko = TokenObject(token)
  352. response.lang = tko.lang
  353. if tko.code != 0:
  354. return response.json(tko.code)
  355. superID = tko.userID
  356. own_perm = ModelService.check_perm(userID=superID, permID=50)
  357. if own_perm is True or superID != None and superID != eUserID:
  358. if rid == None:
  359. return self.UserValidUpdate(superID, eUserID, isValid, response)
  360. else:
  361. return self.UserValidUpdatePC(superID, eUserID, isValid, rid, response)
  362. else:
  363. return response.json(444, 'superID or userID')
  364. def UserValidUpdatePC(self, superID, eUserID, isValid, rid, response):
  365. super = Device_User.objects.filter(userID=superID)
  366. eUser = Device_User.objects.filter(userID=eUserID)
  367. if not super.exists() or not eUser.exists():
  368. return response.json(104)
  369. own_permission = ModelService.check_perm(userID=superID, permID=50)
  370. if own_permission is True:
  371. try:
  372. eUser.update(user_isValid=isValid)
  373. device_user_query_set = Device_User.objects.get(userID=eUserID)
  374. if device_user_query_set.role.all():
  375. device_user_query_set.role.set([rid])
  376. else:
  377. role_user_query_set = Role.objects.get(rid=rid)
  378. device_user_query_set.role.add(role_user_query_set)
  379. except Exception as e:
  380. errorInfo = traceback.format_exc()
  381. print('更新数据库错误:%s' % errorInfo)
  382. return response.json(177, repr(e))
  383. else:
  384. return response.json(0)
  385. else:
  386. return response.json(404)
  387. def UserValidUpdate(self, superID, eUserID, isValid, response):
  388. super = Device_User.objects.filter(userID=superID)
  389. eUser = Device_User.objects.filter(userID=eUserID)
  390. if not super.exists():
  391. return response.json(104)
  392. if not eUser.exists():
  393. return response.json(104)
  394. if super[0].is_superuser != 100 or super[0].is_superuser == eUser[0].is_superuser:
  395. return response.json(404)
  396. try:
  397. eUser.update(user_isValid=isValid)
  398. except Exception as e:
  399. errorInfo = traceback.format_exc()
  400. print('更新数据库错误: %s' % errorInfo)
  401. return response.json(177, repr(e))
  402. return response.json(0)
  403. @csrf_exempt
  404. def success(request):
  405. return HttpResponse(status=200)