UserManageController.py 29 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694
  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 oss2
  20. import simplejson
  21. import simplejson as json
  22. import requests
  23. import xlwt
  24. from django.contrib.auth.hashers import make_password, check_password # 对密码加密模块
  25. from django.db import transaction
  26. from django.db.models import Q
  27. from django.http import HttpResponseRedirect
  28. from django.utils.decorators import method_decorator
  29. from django.utils.timezone import utc
  30. from django.views.decorators.csrf import csrf_exempt
  31. from django.views.generic import TemplateView
  32. from jwt.algorithms import RSAAlgorithm
  33. from ratelimit.decorators import ratelimit
  34. from Ansjer.config import AuthCode_Expire, SERVER_DOMAIN, APNS_CONFIG, JPUSH_CONFIG, FCM_CONFIG, TUTK_PUSH_DOMAIN, \
  35. OSS_STS_ACCESS_KEY, OSS_STS_ACCESS_SECRET
  36. from Controller.CheckUserData import DataValid, date_handler, RandomStr
  37. from Model.models import Device_User, Role, UidPushModel, UserOauth2Model, UserExModel, Device_Info, UidSetModel, \
  38. UserAppFrequencyModel, CountryIPModel, CountryModel, UidChannelSetModel, MenuModel, FeedBackModel, StatResModel, \
  39. SysMassModel, App_Info, SysMsgModel
  40. from Object.AWS.SesClassObject import SesClassObject
  41. from Object.AliSmsObject import AliSmsObject
  42. from Object.RedisObject import RedisObject
  43. from Object.ResponseObject import ResponseObject
  44. from Object.TokenObject import TokenObject
  45. from Service.CommonService import CommonService
  46. from Service.ModelService import ModelService
  47. from Service.TemplateService import TemplateService
  48. from django.views.generic import View
  49. import base64
  50. import random
  51. from io import BytesIO
  52. from PIL import Image, ImageDraw, ImageFont
  53. from django.shortcuts import HttpResponse
  54. from Ansjer.config import BASE_DIR
  55. # 登录
  56. class LoginView(TemplateView):
  57. @method_decorator(csrf_exempt) # @csrf_exempt
  58. def dispatch(self, *args, **kwargs):
  59. return super(LoginView, self).dispatch(*args, **kwargs)
  60. def post(self, request, *args, **kwargs):
  61. request.encoding = 'utf-8'
  62. request_dict = request.POST
  63. language = request_dict.get('language', 'en')
  64. response = ResponseObject(language, 'pc')
  65. return self.validates(request_dict, response)
  66. def validates(self, request_dict, response):
  67. username = request_dict.get('username', None)
  68. password = request_dict.get('password', None)
  69. if not username or not password:
  70. return response.json(111)
  71. username = username.strip()
  72. password = password.strip()
  73. data_valid = DataValid()
  74. if data_valid.email_validate(username):
  75. return self.do_email_login(username, password, response)
  76. elif data_valid.mobile_validate(username):
  77. return self.do_phone_login(username, password, response)
  78. elif data_valid.name_validate(username):
  79. return self.do_name_login(username, password, response)
  80. else:
  81. return response.json(107)
  82. def do_email_login(self, email, password, response):
  83. user_qs = Device_User.objects.filter(Q(username=email) | Q(userEmail=email))
  84. return self.valid_login(user_qs, password, response)
  85. def do_phone_login(self, phone, password, response):
  86. user_qs = Device_User.objects.filter(Q(phone=phone) | Q(username=phone), is_active=True, user_isValid=True)
  87. return self.valid_login(user_qs, password, response)
  88. def do_name_login(self, username, password, response):
  89. user_qs = Device_User.objects.filter(Q(username=username) | Q(phone=username) | Q(userEmail=username),
  90. is_active=True, user_isValid=True)
  91. return self.valid_login(user_qs, password, response)
  92. def valid_login(self, user_qs, password, response):
  93. if not user_qs.exists():
  94. return response.json(104)
  95. # users = user_qs.values('role__rid', 'role__roleName', 'userID', 'role', 'NickName', 'username', 'userEmail',
  96. # 'phone', 'password', 'userIconPath', 'user_isValid', 'is_active')[0]
  97. users = user_qs.values('role__rid', 'role__roleName', 'userID', 'NickName', 'username', 'userEmail',
  98. 'phone', 'password', 'userIconPath')[0]
  99. if not check_password(password, users['password']):
  100. return response.json(111)
  101. userID = users['userID']
  102. tko = TokenObject(returntpye='pc')
  103. res = tko.generate(
  104. data={'userID': userID, 'lang': response.lang, 'user': users['username'], 'm_code': '123413243214'})
  105. if tko.code == 0:
  106. now_time = datetime.datetime.utcnow().replace(tzinfo=utc).astimezone(utc)
  107. user_qs.update(last_login=now_time, language=response.lang)
  108. res['rid'] = users['role__rid']
  109. res['roleName'] = users['role__roleName']
  110. res['permList'] = ModelService.own_permission(userID)
  111. res['userID'] = userID
  112. # 昵称,邮箱,电话,刷新,头像
  113. userIconPath = str(users['userIconPath'])
  114. if userIconPath and userIconPath.find('static/') != -1:
  115. userIconPath = userIconPath.replace('static/', '').replace('\\', '/')
  116. res['userIconUrl'] = SERVER_DOMAIN + 'account/getAvatar/' + userIconPath
  117. else:
  118. res['userIconUrl'] = ''
  119. res['NickName'] = users['NickName'] if users['NickName'] is not None else ''
  120. res['username'] = users['username'] if users['username'] is not None else ''
  121. res['userEmail'] = users['userEmail'] if users['userEmail'] is not None else ''
  122. res['phone'] = users['phone'] if users['phone'] is not None else ''
  123. return response.json(0, res)
  124. else:
  125. return response.json(tko.code)
  126. # 获取登录权限
  127. class GetPermissions(TemplateView):
  128. @method_decorator(csrf_exempt) # @csrf_exempt
  129. def dispatch(self, *args, **kwargs):
  130. return super(GetPermissions, self).dispatch(*args, **kwargs)
  131. def get(self, request, *args, **kwargs):
  132. token = request.META.get('HTTP_AUTHORIZATION')
  133. request.encoding = 'utf-8'
  134. request_dict = request.GET
  135. language = request_dict.get('language', 'en')
  136. response = ResponseObject(language, 'pc')
  137. return self.validates(request_dict, token, response)
  138. def validates(self, request_dict, token, response):
  139. tko = TokenObject(token, returntpye='pc')
  140. response.lang = tko.lang
  141. if tko.code != 0:
  142. return response.json(tko.code)
  143. userID = tko.userID
  144. user_qs = Device_User.objects.filter(userID=userID)
  145. if not user_qs.exists():
  146. return response.json(104)
  147. # 待补充逻辑
  148. username = user_qs[0].username
  149. userIconPath = user_qs[0].userIconPath.url
  150. if userIconPath:
  151. if userIconPath.find('static/') != -1:
  152. userIconPath = userIconPath.replace('static/', '').replace('\\', '/')
  153. userIconUrl = SERVER_DOMAIN + 'account/getAvatar/' + userIconPath
  154. role_qs = Role.objects.filter(device_user=userID)
  155. menu_qs = MenuModel.objects.filter(role__in=role_qs, menutype=2);
  156. perms = []
  157. for menu in menu_qs:
  158. perms.append(menu.menu_code)
  159. res = {
  160. "code": 200,
  161. "msg": "success",
  162. "data": {
  163. "roles": ["admin"], # 一个用户可包含多个角色如["admin","editor","XXXX"],必须返回,如小项目用不到角色权限请返回 ["admin"]
  164. "ability": ["READ", "WRITE", "DELETE"], # 如果用不到rabc精细化权限可以不返回,建议返回
  165. "username": username, # 用户名,必须返回
  166. "avatar": userIconUrl, # 头像,必须返回
  167. "perms": perms
  168. }
  169. }
  170. return response.json(0, res)
  171. # 获取菜单
  172. class GetList(TemplateView):
  173. @method_decorator(csrf_exempt) # @csrf_exempt
  174. def dispatch(self, *args, **kwargs):
  175. return super(GetList, self).dispatch(*args, **kwargs)
  176. def get(self, request, *args, **kwargs):
  177. token = request.META.get('HTTP_AUTHORIZATION')
  178. request.encoding = 'utf-8'
  179. request_dict = request.GET
  180. language = request_dict.get('language', 'en')
  181. response = ResponseObject(language, 'pc')
  182. return self.validates(request_dict, token, response)
  183. def validates(self, request_dict, token, response):
  184. tko = TokenObject(token, returntpye='pc')
  185. response.lang = tko.lang
  186. if tko.code != 0:
  187. return response.json(tko.code)
  188. userID = tko.userID
  189. role_qs = Role.objects.filter(device_user=userID)
  190. menu_qs = MenuModel.objects.filter(parentId=0, role__in=role_qs, menutype=1);
  191. list = []
  192. for menu in menu_qs:
  193. list.append(
  194. {
  195. 'id': menu.id,
  196. 'parentId': menu.parentId,
  197. 'path': menu.path,
  198. 'name': menu.name,
  199. 'component': menu.component,
  200. 'meta': {
  201. 'hidden': menu.hidden,
  202. 'levelHidden': menu.levelHidden,
  203. 'title': menu.title,
  204. 'icon': menu.icon,
  205. 'isCustomSvg': menu.isCustomSvg,
  206. 'noKeepAlive': menu.noKeepAlive,
  207. 'noClosable': menu.noClosable,
  208. 'badge': menu.badge,
  209. 'tabHidden': menu.tabHidden,
  210. 'activeMenu': menu.activeMenu,
  211. 'dot': menu.dot,
  212. 'dynamicNewTab': menu.dynamicNewTab,
  213. 'sort': menu.sort
  214. }
  215. }
  216. )
  217. menu_qs = MenuModel.objects.filter(role__in=role_qs, menutype=1)
  218. menulist = []
  219. for objlist in list:
  220. menulist.append(self.menulist(menu_qs, objlist))
  221. return response.json(0, {'list': menulist})
  222. def menulist(self, menu_qs, objlist):
  223. if objlist is None:
  224. return
  225. for menu in menu_qs:
  226. if objlist['id'] == menu.parentId:
  227. if 'children' not in objlist:
  228. objlist['children'] = []
  229. obj = {
  230. 'id': menu.id,
  231. 'parentId': menu.parentId,
  232. 'path': menu.path,
  233. 'name': menu.name,
  234. 'component': menu.component,
  235. 'menutype': menu.menutype,
  236. 'menu_code': menu.menu_code,
  237. 'meta': {
  238. 'hidden': menu.hidden,
  239. 'levelHidden': menu.levelHidden,
  240. 'title': menu.title,
  241. 'icon': menu.icon,
  242. 'isCustomSvg': menu.isCustomSvg,
  243. 'noKeepAlive': menu.noKeepAlive,
  244. 'noClosable': menu.noClosable,
  245. 'badge': menu.badge,
  246. 'tabHidden': menu.tabHidden,
  247. 'activeMenu': menu.activeMenu,
  248. 'dot': menu.dot,
  249. 'dynamicNewTab': menu.dynamicNewTab,
  250. 'sort': menu.sort
  251. }
  252. }
  253. objlist['children'].append(
  254. obj
  255. )
  256. self.menulist(menu_qs, obj)
  257. return objlist
  258. class UserManagement(View):
  259. def get(self, request, *args, **kwargs):
  260. request.encoding = 'utf-8'
  261. operation = kwargs.get('operation')
  262. return self.validation(request.GET, request, operation)
  263. def post(self, request, *args, **kwargs):
  264. request.encoding = 'utf-8'
  265. operation = kwargs.get('operation')
  266. return self.validation(request.POST, request, operation)
  267. def validation(self, request_dict, request, operation):
  268. language = request_dict.get('language', 'en')
  269. response = ResponseObject(language, 'pc')
  270. if operation == 'sendSysMessage':
  271. return self.sendSysMessage(request_dict, response)
  272. elif operation == 'getAppDataList':
  273. return self.getAppDataList(request_dict, response)
  274. elif operation == 'replyFeedBack':
  275. return self.replyFeedBack(request_dict, response)
  276. elif operation == 'sendSysMsgToUser':
  277. return self.sendSysMsgToUser(request_dict, response)
  278. elif operation == 'collectFeedBack':
  279. return self.collectFeedBack(request_dict, response)
  280. else:
  281. tko = TokenObject(request.META.get('HTTP_AUTHORIZATION'), returntpye='pc')
  282. if tko.code != 0:
  283. return response.json(tko.code)
  284. response.lang = tko.lang
  285. userID = tko.userID
  286. if operation == 'getUserInfo':
  287. return self.getUserInfo(userID, request_dict, response)
  288. elif operation == 'AddOrEditAccount':
  289. return self.AddOrEditAccount(userID, request_dict, response)
  290. elif operation == 'doDelete':
  291. return self.doDelete(request_dict, response)
  292. elif operation == 'resetPassword':
  293. return self.resetPassword(request_dict, response)
  294. elif operation == 'getFeedbackList':
  295. return self.getFeedbackList(request_dict, response)
  296. elif operation == 'deleteFeedBack':
  297. return self.deleteFeedBack(request_dict, response)
  298. elif operation == 'getSysMessageList':
  299. return self.getSysMessageList(request_dict, response)
  300. else:
  301. return response.json(404)
  302. def getUserInfo(self, userID, request_dict, response):
  303. print('request_dict: ', request_dict)
  304. username = request_dict.get('username', '').strip() # 移除字符串头尾的空格
  305. NickName = request_dict.get('NickName', '').strip()
  306. phone = request_dict.get('phone', '').strip()
  307. userEmail = request_dict.get('userEmail', '').strip()
  308. pageNo = request_dict.get('pageNo', None)
  309. pageSize = request_dict.get('pageSize', None)
  310. if not all([pageNo, pageSize]):
  311. return response.json(444)
  312. page = int(pageNo)
  313. line = int(pageSize)
  314. try:
  315. if username or NickName or phone or userEmail:
  316. # 条件查询
  317. if username:
  318. device_user_qs = Device_User.objects.filter(username__icontains=username)
  319. if NickName:
  320. device_user_qs = Device_User.objects.filter(NickName__icontains=NickName)
  321. if phone:
  322. device_user_qs = Device_User.objects.filter(phone__icontains=phone)
  323. if userEmail:
  324. device_user_qs = Device_User.objects.filter(userEmail__icontains=userEmail)
  325. if not device_user_qs.exists():
  326. return response.json(0)
  327. total = len(device_user_qs)
  328. device_users = device_user_qs[(page - 1) * line:page * line]
  329. else:
  330. total = Device_User.objects.filter().count()
  331. device_users = Device_User.objects.filter()[(page - 1) * line:page * line]
  332. user_list = []
  333. for device_user in device_users:
  334. role = device_user.role.first()
  335. rid = role.rid if role else 1 # 不存在角色默认分配为'Users'
  336. user_ex_qs = UserExModel.objects.filter(userID_id=device_user.userID).values('appBundleId')
  337. app_bundle_id = user_ex_qs[0]['appBundleId'] if user_ex_qs.exists() else ''
  338. user_list.append({
  339. 'userID': device_user.userID,
  340. 'username': device_user.username,
  341. 'NickName': device_user.NickName,
  342. 'role': Role.objects.get(rid=rid).roleName,
  343. 'phone': device_user.phone,
  344. 'userEmail': device_user.userEmail,
  345. 'data_joined': device_user.data_joined.strftime("%Y-%m-%d %H:%M:%S"),
  346. 'last_login': device_user.last_login.strftime("%Y-%m-%d %H:%M:%S"),
  347. 'appBundleId': app_bundle_id,
  348. })
  349. print('user_list: ', user_list)
  350. return response.json(0, {'list': user_list, 'total': total})
  351. except Exception as e:
  352. print(e)
  353. return response.json(500, repr(e))
  354. def AddOrEditAccount(self, userID, request_dict, response):
  355. # 添加/编辑用户
  356. print('request_dict: ', request_dict)
  357. username = request_dict.get('username', '').strip() # 移除字符串头尾的空格
  358. userEmail = request_dict.get('userEmail', '').strip()
  359. roleName = request_dict.get('role', None)
  360. password = request_dict.get('password', None)
  361. isEdit = request_dict.get('isEdit', None)
  362. # 校验用户名,邮箱,密码是否符合规则
  363. dataValid = DataValid()
  364. if not username or not dataValid.name_validate(username):
  365. return response.json(444, {'Parameter error': 'username'})
  366. if userEmail and not dataValid.email_validate(userEmail):
  367. return response.json(444, {'Parameter error': 'userEmail'})
  368. if not isEdit: # 添加用户需要输入密码
  369. if not password or not dataValid.password_validate(password):
  370. return response.json(444, {'Parameter error': 'password'})
  371. try:
  372. if isEdit: # 编辑用户信息
  373. userID = request_dict.get('userID')
  374. user_data = {
  375. "username": username,
  376. "NickName": username,
  377. "userEmail": userEmail,
  378. "password": make_password(password),
  379. }
  380. device_user_qs = Device_User.objects.filter(userID=userID)
  381. device_user_qs.update(**user_data)
  382. # 如果角色改变,修改用户角色
  383. device_user_role = device_user_qs[0].role
  384. user_role = device_user_role.first()
  385. if not user_role or roleName != user_role.roleName:
  386. device_user_role.clear()
  387. role_qs = Role.objects.filter(roleName=roleName) # 账号角色
  388. device_user_qs[0].role.set(role_qs)
  389. else: # 添加用户
  390. # 查询邮箱是否已注册
  391. if Device_User.objects.filter(userEmail=userEmail).exists():
  392. return response.json(103)
  393. role_qs = Role.objects.filter(roleName=roleName) # 账号角色
  394. # 创建用户
  395. user_data = {
  396. "username": username,
  397. "NickName": username,
  398. "userEmail": userEmail,
  399. "password": make_password(password),
  400. "userID": CommonService.getUserID(μs=False, setOTAID=True),
  401. "is_active": True,
  402. "user_isValid": True,
  403. }
  404. Device_User.objects.create(**user_data).role.set(role_qs)
  405. return response.json(0)
  406. except Exception as e:
  407. print(e)
  408. return response.json(500, repr(e))
  409. def doDelete(self, request_dict, response):
  410. userID = request_dict.get('userID', '')
  411. if not userID:
  412. return response.json(444)
  413. try:
  414. Device_User.objects.filter(userID=userID).delete()
  415. return response.json(0)
  416. except Exception as e:
  417. print(e)
  418. return response.json(500, repr(e))
  419. def resetPassword(self, request_dict, response):
  420. userID = request_dict.get('userID', None)
  421. if not userID:
  422. return response.json(444)
  423. try:
  424. password = '123456'
  425. is_update = Device_User.objects.filter(userID=userID).update(password=make_password(password))
  426. if is_update:
  427. return response.json(0)
  428. else:
  429. return response.json(177)
  430. except Exception as e:
  431. print(e)
  432. return response.json(500, repr(e))
  433. def getFeedbackList(self, request_dict, response):
  434. status = request_dict.get('status', 0)
  435. username = request_dict.get('username', None)
  436. collect_status = request_dict.get('collectStatus', 0)
  437. fileType = request_dict.get('fileType', None)
  438. pageNo = request_dict.get('pageNo', None)
  439. pageSize = request_dict.get('pageSize', None)
  440. if not all([pageNo, pageSize]):
  441. return response.json(444)
  442. page = int(pageNo)
  443. line = int(pageSize)
  444. try:
  445. filter_data = {}
  446. if status or username or collect_status:
  447. if status:
  448. filter_data['status'] = int(status)
  449. if username:
  450. filter_data['userID_id'] = ModelService.get_userID_byname(username)
  451. if collect_status:
  452. filter_data['collect_status'] = int(collect_status)
  453. feed_back_qs = FeedBackModel.objects.filter()
  454. if filter_data:
  455. feed_back_qs = feed_back_qs.filter(**filter_data).values()
  456. total = feed_back_qs.count()
  457. feed_backs = feed_back_qs.values(
  458. 'id',
  459. 'userID__username',
  460. 'userID__phone',
  461. 'userID__userEmail',
  462. 'content',
  463. 'addTime',
  464. 'status',
  465. 'type',
  466. 'app',
  467. 'app_version',
  468. 'phone_model',
  469. 'os_version',
  470. 'uid',
  471. 'appBundleId',
  472. 'score',
  473. 'collect_status'
  474. )
  475. if fileType:
  476. feed_backs = feed_backs
  477. else:
  478. feed_backs = feed_backs[(page - 1) * line: page * line]
  479. feed_back_id_list = [feed_back['id'] for feed_back in feed_backs]
  480. stat_res_qs = StatResModel.objects.filter(
  481. feedbackmodel__id__in=feed_back_id_list).values(
  482. 'name', 'feedbackmodel__id')
  483. auth = oss2.Auth(OSS_STS_ACCESS_KEY, OSS_STS_ACCESS_SECRET)
  484. bucket = oss2.Bucket(
  485. auth, 'oss-cn-hongkong.aliyuncs.com', 'statres')
  486. feed_back_list = []
  487. for feed_back in feed_backs:
  488. feed_back['img_url'] = []
  489. for stat_res in stat_res_qs:
  490. if stat_res['feedbackmodel__id'] == feed_back['id']:
  491. obj = 'feedback/' + stat_res['name']
  492. img_url = bucket.sign_url('GET', obj, 3600)
  493. feed_back['img_url'].append(img_url)
  494. if not feed_back['userID__username']:
  495. feed_back['userID__username'] = feed_back['userID__userEmail'] if feed_back[
  496. 'userID__userEmail'] else feed_back['userID__phone']
  497. feed_back_list.append(feed_back)
  498. return response.json(0, {'list': feed_back_list, 'total': total})
  499. except Exception as e:
  500. print(e)
  501. return response.json(500, repr(e))
  502. def replyFeedBack(self, request_dict, response):
  503. feedBackID = request_dict.get('feedBackID', None)
  504. msg = request_dict.get('msg', None)
  505. uid = request_dict.get('uid', None)
  506. if not all([feedBackID, msg]):
  507. return response.json(444)
  508. try:
  509. nowTime = int(time.time())
  510. feed_back_qs = FeedBackModel.objects.filter(id=feedBackID)
  511. if not feed_back_qs.exists():
  512. return response.json(173)
  513. with transaction.atomic():
  514. SysMsgModel.objects.create(userID_id=feed_back_qs.first().userID_id, msg=msg, uid=uid,
  515. addTime=nowTime, updTime=nowTime) # 创建系统消息
  516. feed_back_qs.update(status=1) # 更新状态为已回复
  517. return response.json(0)
  518. except Exception as e:
  519. print(e)
  520. return response.json(500, repr(e))
  521. def sendSysMsgToUser(self, request_dict, response):
  522. userID = request_dict.get('userID', None)
  523. msg = request_dict.get('msg', None)
  524. if not all([userID, msg]):
  525. return response.json(444)
  526. try:
  527. nowTime = int(time.time())
  528. SysMsgModel.objects.create(userID_id=userID, msg=msg, addTime=nowTime, updTime=nowTime)
  529. return response.json(0)
  530. except Exception as e:
  531. print(e)
  532. return response.json(500, repr(e))
  533. def deleteFeedBack(self, request_dict, response):
  534. feedBackID = request_dict.get('feedBackID', None)
  535. if not feedBackID:
  536. return response.json(444)
  537. try:
  538. feed_back = FeedBackModel.objects.get(id=feedBackID)
  539. sr_id_list = feed_back.FS.values_list('id', flat=True)
  540. StatResModel.objects.filter(id__in=sr_id_list).delete()
  541. feed_back.delete()
  542. return response.json(0)
  543. except Exception as e:
  544. print(e)
  545. return response.json(500, repr(e))
  546. def getSysMessageList(self, request_dict, response):
  547. print('request_dict: ', request_dict)
  548. pageNo = request_dict.get('pageNo', None)
  549. pageSize = request_dict.get('pageSize', None)
  550. if not all([pageNo, pageSize]):
  551. return response.json(444)
  552. page = int(pageNo)
  553. line = int(pageSize)
  554. try:
  555. sys_mass_qs = SysMassModel.objects.filter()
  556. total = sys_mass_qs.count()
  557. sys_mass_qs = sys_mass_qs.values('id', 'sender__username', 'addTime', 'updTime', 'lang', 'recever', 'msg',
  558. 'platform')
  559. sys_mass_qs = sys_mass_qs[(page - 1) * line:page * line]
  560. return response.json(0, {'list': list(sys_mass_qs), 'total': total})
  561. except Exception as e:
  562. print(e)
  563. return response.json(500, repr(e))
  564. def getAppDataList(self, request_dict, response):
  565. print('request_dict: ', request_dict)
  566. platform = request_dict.get('platform', None)
  567. if platform == 'ios':
  568. app_type = [1]
  569. elif platform == 'android':
  570. app_type = [2]
  571. else:
  572. app_type = [1, 2]
  573. try:
  574. app_info_qs = App_Info.objects.filter(app_type__in=app_type).values('appBundleId', 'appName').distinct()
  575. appBundleId_list = []
  576. for app_info in app_info_qs:
  577. appBundleId_list.append(app_info['appBundleId'])
  578. return response.json(0, {'appBundleId_list': appBundleId_list})
  579. except Exception as e:
  580. print(e)
  581. return response.json(500, repr(e))
  582. def sendSysMessage(self, request_dict, response):
  583. print('request_dict: ', request_dict)
  584. lang = request_dict.get('lang', None)
  585. platform = request_dict.get('platform', None)
  586. recever = request_dict.get('app_list', None)
  587. msg = request_dict.get('msg', None)
  588. if not all([lang, platform, recever, msg]):
  589. return response.json(444)
  590. try:
  591. with transaction.atomic():
  592. # SysMassModel表创建群发消息记录数据
  593. nowTime = int(time.time())
  594. sender_id = Device_User.objects.filter(username='13800138001').values('userID').first()['userID']
  595. SysMassModel.objects.create(
  596. sender_id=sender_id,
  597. lang=lang,
  598. platform=platform,
  599. recever=recever,
  600. msg=msg,
  601. addTime=nowTime,
  602. updTime=nowTime,
  603. )
  604. # 根据UserExModel表的userID群发消息给用户
  605. region_list = lang.split(',')
  606. appBundleId_list = recever.split(',')
  607. userID_list = UserExModel.objects.filter(appBundleId__in=appBundleId_list, region__in=region_list) \
  608. .values_list('userID_id', flat=True)
  609. sys_msg_list = []
  610. for userID in userID_list:
  611. sys_msg_list.append(SysMsgModel(userID_id=userID, msg=msg, addTime=nowTime, updTime=nowTime))
  612. if len(sys_msg_list) > 5000:
  613. SysMsgModel.objects.bulk_create(sys_msg_list)
  614. sys_msg_list = []
  615. SysMsgModel.objects.bulk_create(sys_msg_list)
  616. return response.json(0)
  617. except Exception as e:
  618. print(e)
  619. return response.json(500, repr(e))
  620. def collectFeedBack(self, request_dict, response):
  621. feedBackID = request_dict.get('feedBackID', None)
  622. collect_status = request_dict.get('collectStatus', None)
  623. if not all([feedBackID, collect_status]):
  624. return response.json(444)
  625. feedBackID = int(feedBackID)
  626. collect_status = int(collect_status)
  627. feed_back_qs = FeedBackModel.objects.filter(id=feedBackID, collect_status=collect_status) # 判断是否存在
  628. if feed_back_qs.exists():
  629. feed_qs = feed_back_qs.first()
  630. feed_qs.collect_status = 0
  631. feed_qs.save()
  632. else:
  633. FeedBackModel.objects.filter(id=feedBackID).update(collect_status=collect_status)
  634. return response.json(0)