CronTaskController.py 48 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888
  1. #!/usr/bin/python3.6
  2. # -*- coding: utf-8 -*-
  3. #
  4. # Copyright (C) 2022 #
  5. # @Time : 2022/4/1 11:27
  6. # @Author : ming
  7. # @Email : zhangdongming@asj6.wecom.work
  8. # @File : CronTaskController.py
  9. # @Software: PyCharm
  10. import datetime
  11. import logging
  12. import threading
  13. import time
  14. import requests
  15. from django.db import connection, connections, transaction
  16. from django.db.models import Q, Sum, Count
  17. from django.views import View
  18. from Ansjer.config import USED_SERIAL_REDIS_LIST, UNUSED_SERIAL_REDIS_LIST
  19. from Model.models import Device_User, Device_Info, UidSetModel, UID_Bucket, Unused_Uid_Meal, Order_Model, StsCrdModel, \
  20. VodHlsModel, ExperienceContextModel, AiService, VodHlsSummary, VideoPlaybackTimeModel, DeviceUserSummary, \
  21. CountryModel, DeviceTypeModel, OrdersSummary, DeviceInfoSummary, CompanySerialModel, \
  22. CloudLogModel, UidCloudStorageCount, UserExModel
  23. from Object.RedisObject import RedisObject
  24. from Object.ResponseObject import ResponseObject
  25. from Object.utils import LocalDateTimeUtil
  26. from Service.CommonService import CommonService
  27. LOGGER = logging.getLogger('info')
  28. class CronDelDataView(View):
  29. def get(self, request, *args, **kwargs):
  30. request.encoding = 'utf-8'
  31. operation = kwargs.get('operation')
  32. return self.validation(request.GET, request, operation)
  33. def post(self, request, *args, **kwargs):
  34. request.encoding = 'utf-8'
  35. operation = kwargs.get('operation')
  36. return self.validation(request.POST, request, operation)
  37. def validation(self, request_dict, request, operation):
  38. response = ResponseObject()
  39. if operation == 'delAccessLog': # 定时删除访问接口数据
  40. return self.delAccessLog(response)
  41. elif operation == 'delPushInfo': # 定时删除推送数据
  42. return self.delPushInfo(response)
  43. elif operation == 'delVodHls': # 定时删除云存播放列表
  44. return self.delVodHls(response)
  45. elif operation == 'delCloudLog': # 定时删除云存接口数据
  46. return self.delCloudLog(response)
  47. elif operation == 'delTesterDevice': # 定时删除测试账号下的设备数据
  48. return self.delTesterDevice(response)
  49. elif operation == 'cloud-log':
  50. return self.uid_cloud_storage_upload_count(response)
  51. else:
  52. return response.json(404)
  53. @staticmethod
  54. def uid_cloud_storage_upload_count(response):
  55. try:
  56. now_time = int(time.time())
  57. local_time = LocalDateTimeUtil.get_before_days_timestamp(now_time)
  58. format_str = '%Y-%m-%d'
  59. date_str = LocalDateTimeUtil.time_stamp_to_time(local_time, format_str)
  60. start_time, end_time = LocalDateTimeUtil.get_start_and_end_time(date_str, format_str)
  61. cs_uid_qs = UID_Bucket.objects.filter(addTime__gte=int(1669824000)).values('uid')
  62. if not cs_uid_qs.exists():
  63. return response.json(0)
  64. for item in cs_uid_qs:
  65. uid = item['uid']
  66. cloud_log_qs = CloudLogModel.objects.filter(uid=uid, operation=r'cloudstorage/storeplaylist',
  67. time__gte=start_time, time__lte=end_time)
  68. cloud_log_qs = cloud_log_qs.values('uid')[0:1]
  69. if not cloud_log_qs.exists():
  70. continue
  71. count_data = {'uid': uid, 'count': cloud_log_qs.count(), 'created_time': end_time,
  72. 'updated_time': end_time}
  73. UidCloudStorageCount.objects.create(**count_data)
  74. return response.json(0)
  75. except Exception as e:
  76. LOGGER.info('异常详情,errLine:{}, errMsg:{}'.format(e.__traceback__.tb_lineno, repr(e)))
  77. return response.json(500)
  78. @staticmethod
  79. def delAccessLog(response):
  80. try:
  81. cursor = connection.cursor()
  82. # 删除7天前的数据
  83. last_week = LocalDateTimeUtil.get_last_week()
  84. sql = 'DELETE FROM access_log WHERE time < %s limit %s'
  85. cursor.execute(sql, [last_week, 10000])
  86. # 关闭游标
  87. cursor.close()
  88. connection.close()
  89. return response.json(0)
  90. except Exception as e:
  91. return response.json(500, repr(e))
  92. @staticmethod
  93. def delPushInfo(response):
  94. now_time = int(time.time())
  95. cursor = connections['mysql02'].cursor()
  96. try:
  97. # 当前时间转日期
  98. local_date_now = str(datetime.datetime.fromtimestamp(int(now_time)).date())
  99. # 根据日期获取周几
  100. week_val = LocalDateTimeUtil.date_to_week(local_date_now)
  101. # 根据当前时间获取7天前时间戳
  102. expiration_time = LocalDateTimeUtil.get_before_days_timestamp(now_time, 7)
  103. # 每次删除条数
  104. size = 10000
  105. # 删除7天前的数据
  106. sql = "DELETE FROM equipment_info WHERE addTime<= %s LIMIT %s "
  107. for i in range(6):
  108. cursor.execute(sql, [expiration_time, size])
  109. if week_val == 1:
  110. sql = "DELETE FROM equipment_info_sunday WHERE add_time<= %s LIMIT %s "
  111. if week_val == 2:
  112. sql = "DELETE FROM equipment_info_monday WHERE add_time<= %s LIMIT %s "
  113. if week_val == 3:
  114. sql = "DELETE FROM equipment_info_tuesday WHERE add_time<= %s LIMIT %s "
  115. if week_val == 4:
  116. sql = "DELETE FROM equipment_info_wednesday WHERE add_time<= %s LIMIT %s "
  117. if week_val == 5:
  118. sql = "DELETE FROM equipment_info_thursday WHERE add_time<= %s LIMIT %s "
  119. if week_val == 6:
  120. sql = "DELETE FROM equipment_info_friday WHERE add_time<= %s LIMIT %s "
  121. if week_val == 7:
  122. sql = "DELETE FROM equipment_info_saturday WHERE add_time<= %s LIMIT %s "
  123. for i in range(5):
  124. cursor.execute(sql, [expiration_time, size])
  125. # 关闭游标
  126. cursor.close()
  127. return response.json(0)
  128. except Exception as e:
  129. return response.json(500, repr(e))
  130. @staticmethod
  131. def delVodHls(response):
  132. nowTime = int(time.time())
  133. try:
  134. cursor = connection.cursor()
  135. month_ago_time = nowTime - 30 * 24 * 60 * 60 # 删除1个月前的数据
  136. sql = 'DELETE FROM `vod_hls` WHERE endTime<{} LIMIT 50000'.format(month_ago_time)
  137. cursor.execute(sql)
  138. cursor.close()
  139. return response.json(0)
  140. except Exception as e:
  141. return response.json(500, repr(e))
  142. @staticmethod
  143. def delCloudLog(response):
  144. nowTime = int(time.time())
  145. cursor = connection.cursor()
  146. try:
  147. # 删除3个月前的数据
  148. sql = "DELETE FROM `cloud_log` WHERE time<={} LIMIT 50000".format(
  149. nowTime - 3 * 30 * 24 * 60 * 60)
  150. cursor.execute(sql)
  151. # 关闭游标
  152. cursor.close()
  153. return response.json(0)
  154. except Exception as e:
  155. return response.json(500, repr(e))
  156. @staticmethod
  157. def delTesterDevice(response):
  158. try:
  159. userID_list = [
  160. 'tech01@ansjer.com',
  161. 'tech02@ansjer.com',
  162. 'tech03@ansjer.com',
  163. 'tech04@ansjer.com',
  164. 'tech05@ansjer.com',
  165. 'tech06@ansjer.com',
  166. 'tech07@ansjer.com',
  167. 'tech08@ansjer.com',
  168. 'tech09@ansjer.com',
  169. 'tech10@ansjer.com',
  170. 'fix01@ansjer.com',
  171. 'fix02@ansjer.com',
  172. 'fix03@ansjer.com',
  173. 'fix04@ansjer.com',
  174. 'fix05@ansjer.com']
  175. device_user = Device_User.objects.filter(username__in=userID_list)
  176. device_info_qs = Device_Info.objects.filter(
  177. userID__in=device_user).values('UID')
  178. uid_list = []
  179. for device_info in device_info_qs:
  180. uid_list.append(device_info['UID'])
  181. with transaction.atomic():
  182. # 删除设备云存相关数据
  183. UidSetModel.objects.filter(uid__in=uid_list).delete()
  184. UID_Bucket.objects.filter(uid__in=uid_list).delete()
  185. Unused_Uid_Meal.objects.filter(uid__in=uid_list).delete()
  186. Order_Model.objects.filter(UID__in=uid_list).delete()
  187. StsCrdModel.objects.filter(uid__in=uid_list).delete()
  188. VodHlsModel.objects.filter(uid__in=uid_list).delete()
  189. ExperienceContextModel.objects.filter(
  190. uid__in=uid_list).delete()
  191. Device_Info.objects.filter(userID__in=device_user).delete()
  192. return response.json(0)
  193. except Exception as e:
  194. return response.json(500, repr(e))
  195. class CronUpdateDataView(View):
  196. def get(self, request, *args, **kwargs):
  197. request.encoding = 'utf-8'
  198. operation = kwargs.get('operation')
  199. return self.validation(request.GET, request, operation)
  200. def post(self, request, *args, **kwargs):
  201. request.encoding = 'utf-8'
  202. operation = kwargs.get('operation')
  203. return self.validation(request.POST, request, operation)
  204. def validation(self, request_dict, request, operation):
  205. response = ResponseObject()
  206. if operation == 'updateUnusedUidBucket': # 定时更新过期云存关联的未使用套餐状态
  207. return self.updateUnusedUidBucket(response)
  208. elif operation == 'updateUnusedAiService': # 定时更新过期ai关联的未使用套餐状态
  209. return self.updateUnusedAiService(response)
  210. elif operation == 'reqUpdateSerialStatus': # 定时请求更新序列号状态
  211. return self.reqUpdateSerialStatus(response)
  212. elif operation == 'updateSerialStatus': # 更新序列号状态
  213. return self.updateSerialStatus(request_dict, response)
  214. else:
  215. return response.json(404)
  216. @staticmethod
  217. def updateUnusedUidBucket(response):
  218. """
  219. 监控云存套餐过期修改状态
  220. @param response:
  221. @return:
  222. """
  223. # 定时更新已过期套餐修改状态为2
  224. now_time = int(time.time())
  225. expired_uid_bucket = UID_Bucket.objects.filter(endTime__lte=now_time)
  226. expired_uid_bucket = expired_uid_bucket.filter(~Q(use_status=2)).values('id')
  227. if expired_uid_bucket.exists():
  228. expired_uid_bucket.update(use_status=2)
  229. # 监控有未使用套餐则自动生效
  230. expired_uid_buckets = \
  231. UID_Bucket.objects.filter(endTime__lte=now_time, has_unused=1).values("id", "uid")[0:1000]
  232. for expired_uid_bucket in expired_uid_buckets:
  233. unuseds = Unused_Uid_Meal.objects.filter(
  234. uid=expired_uid_bucket['uid']).values(
  235. "id",
  236. "uid",
  237. "channel",
  238. "addTime",
  239. "expire",
  240. "num",
  241. "bucket_id").order_by('addTime')[0:1]
  242. if not unuseds.exists():
  243. continue
  244. unused = unuseds[0]
  245. try:
  246. with transaction.atomic():
  247. count_unused = Unused_Uid_Meal.objects.filter(
  248. uid=expired_uid_bucket['uid']).count()
  249. has_unused = 1 if count_unused > 1 else 0
  250. endTime = CommonService.calcMonthLater(
  251. unused['expire'] * unused['num'])
  252. UID_Bucket.objects.filter(
  253. uid=expired_uid_bucket['uid']).update(
  254. channel=unused['channel'],
  255. endTime=endTime,
  256. bucket_id=unused['bucket_id'],
  257. updateTime=now_time,
  258. use_status=1,
  259. has_unused=has_unused)
  260. Unused_Uid_Meal.objects.filter(id=unused['id']).delete()
  261. StsCrdModel.objects.filter(
  262. uid=expired_uid_bucket['uid']).delete() # 删除sts记录
  263. except Exception as e:
  264. print(repr(e))
  265. continue
  266. return response.json(0)
  267. @staticmethod
  268. def updateUnusedAiService(response):
  269. now_time = int(time.time())
  270. ai_service_qs = AiService.objects.filter(
  271. endTime__lte=now_time,
  272. use_status=1).values(
  273. 'id',
  274. 'uid')[
  275. 0:200]
  276. for ai_service in ai_service_qs:
  277. try:
  278. with transaction.atomic():
  279. AiService.objects.filter(
  280. id=ai_service['id']).update(
  281. use_status=2) # 更新过期ai订单状态
  282. # 如果存在未使用套餐,更新为使用
  283. unused_ai_service = AiService.objects.filter(
  284. uid=ai_service['uid'],
  285. use_status=0).order_by('addTime')[
  286. :1].values(
  287. 'id',
  288. 'endTime')
  289. if unused_ai_service.exists():
  290. # 未使用套餐的endTime在购买的时候保存为有效时间
  291. effective_day = unused_ai_service[0]['endTime']
  292. endTime = now_time + effective_day
  293. AiService.objects.filter(
  294. id=unused_ai_service[0]['id']).update(
  295. use_status=1, endTime=endTime, updTime=now_time)
  296. except Exception:
  297. continue
  298. return response.json(0)
  299. @classmethod
  300. def reqUpdateSerialStatus(cls, response):
  301. redis_obj = RedisObject()
  302. # 更新已使用序列号其他服务器的状态
  303. logger = logging.getLogger('info')
  304. used_serial_redis_list = redis_obj.lrange(USED_SERIAL_REDIS_LIST, 0, -1)
  305. logger.info('---请求更新序列号接口---used_serial_redis_list:{}'.format(used_serial_redis_list))
  306. if used_serial_redis_list:
  307. used_serial_redis_list = [str(i, 'utf-8') for i in used_serial_redis_list]
  308. cls.do_request_thread(used_serial_redis_list, 3)
  309. # 更新未使用序列号其他服务器的状态
  310. unused_serial_redis_list = redis_obj.lrange(UNUSED_SERIAL_REDIS_LIST, 0, -1)
  311. logger.info('---请求更新序列号接口---unused_serial_redis_list:{}'.format(unused_serial_redis_list))
  312. if unused_serial_redis_list:
  313. unused_serial_redis_list = [str(i, 'utf-8') for i in unused_serial_redis_list]
  314. cls.do_request_thread(unused_serial_redis_list, 1)
  315. return response.json(0)
  316. @staticmethod
  317. def do_request_thread(serial_redis_list, status):
  318. """
  319. 请求更新序列号线程
  320. @param serial_redis_list: 序列号redis列表
  321. @param status: 状态, 1: 未使用, 3: 已占用
  322. """
  323. data = {
  324. 'serial_redis_list': str(serial_redis_list),
  325. 'status': status
  326. }
  327. # 确认域名列表
  328. orders_domain_name_list = CommonService.get_orders_domain_name_list()
  329. redis_obj = RedisObject()
  330. logger = logging.getLogger('info')
  331. logger.info('---请求更新序列号线程---data:{},orders_domain_name_list:{}'.format(data, orders_domain_name_list))
  332. try:
  333. flag = 0
  334. for domain_name in orders_domain_name_list:
  335. url = '{}cron/update/updateSerialStatus'.format(domain_name)
  336. response = requests.post(url=url, data=data, timeout=2)
  337. result = response.json()
  338. if result['result_code'] != 0:
  339. flag = 1
  340. break
  341. if flag == 0:
  342. if status == 1:
  343. for i in serial_redis_list:
  344. redis_obj.lrem(UNUSED_SERIAL_REDIS_LIST, 0, i)
  345. elif status == 3:
  346. for i in serial_redis_list:
  347. redis_obj.lrem(USED_SERIAL_REDIS_LIST, 0, i)
  348. except Exception as e:
  349. logger.info('---更新序列号状态异常---:{}'.format(repr(e)))
  350. @staticmethod
  351. def updateSerialStatus(request_dict, response):
  352. """
  353. 更新序列号状态
  354. @param request_dict: 请求参数
  355. @request_dict serial_redis_list: 序列号redis列表
  356. @request_dict status: 状态, 1: 未使用, 3: 已占用
  357. @param response: 响应对象
  358. """
  359. serial_redis_list = request_dict.get('serial_redis_list', None)
  360. status = request_dict.get('status', None)
  361. logger = logging.getLogger('info')
  362. logger.info('---更新序列号状态参数---serial_redis_list:{},status:{}'.format(serial_redis_list, status))
  363. if not all([serial_redis_list, status]):
  364. return response.json(444)
  365. now_time = int(time.time())
  366. try:
  367. serial_redis_list = eval(serial_redis_list)
  368. CompanySerialModel.objects.filter(serial_number__in=serial_redis_list).update(status=int(status),
  369. update_time=now_time)
  370. return response.json(0)
  371. except Exception as e:
  372. logger.info('---更新序列号状态异常---:{}'.format(repr(e)))
  373. return response.json(500)
  374. class CronCollectDataView(View):
  375. def get(self, request, *args, **kwargs):
  376. request.encoding = 'utf-8'
  377. operation = kwargs.get('operation')
  378. return self.validation(request.GET, request, operation)
  379. def post(self, request, *args, **kwargs):
  380. request.encoding = 'utf-8'
  381. operation = kwargs.get('operation')
  382. return self.validation(request.POST, request, operation)
  383. def validation(self, request_dict, request, operation):
  384. response = ResponseObject()
  385. if operation == 'collectPlayBack': # 定时保存云存视频回放
  386. return self.collect_play_back(response)
  387. elif operation == 'collectDeviceUser': # 定时保存用户数据
  388. return self.collect_device_user(response)
  389. elif operation == 'collectOrder': # 定时保存订单数据
  390. return self.collect_order(response)
  391. elif operation == 'collectDeviceInfo': # 定时保存设备数据
  392. return self.collect_device_info(response)
  393. else:
  394. return response.json(404)
  395. @staticmethod
  396. def collect_play_back(response):
  397. try:
  398. now_time = int(time.time())
  399. today = datetime.datetime.today()
  400. start_time = datetime.datetime(today.year, today.month, today.day)
  401. end_time = start_time + datetime.timedelta(days=1)
  402. start_time = CommonService.str_to_timestamp(start_time.strftime('%Y-%m-%d %H:%M:%S'))
  403. end_time = CommonService.str_to_timestamp(end_time.strftime('%Y-%m-%d %H:%M:%S'))
  404. this_month_str = datetime.datetime(today.year, today.month, 1)
  405. this_month_stamp = CommonService.str_to_timestamp(this_month_str.strftime('%Y-%m-%d %H:%M:%S'))
  406. video_play_back_time_qs = VideoPlaybackTimeModel.objects.filter(startTime__gte=start_time,
  407. startTime__lt=end_time,
  408. playMode='cloud').values('uid').annotate(
  409. play_duration=Sum('duration'), play_frequency=Count('uid'))
  410. with transaction.atomic():
  411. for item in video_play_back_time_qs:
  412. vod_hls_summary_qs = VodHlsSummary.objects.filter(uid=item['uid'], time=this_month_stamp)
  413. if vod_hls_summary_qs.exists():
  414. vod_hls_summary = vod_hls_summary_qs.first()
  415. vod_hls_summary.play_duration += item['play_duration']
  416. vod_hls_summary.play_frequency += 1
  417. vod_hls_summary.updated_time = now_time
  418. vod_hls_summary.save()
  419. else:
  420. VodHlsSummary.objects.create(uid=item['uid'], time=this_month_stamp, created_time=now_time,
  421. play_duration=item['play_duration'], play_frequency=1,
  422. updated_time=now_time)
  423. return response.json(0)
  424. except Exception as e:
  425. return response.json(500, repr(e))
  426. @staticmethod
  427. def collect_device_user(response):
  428. try:
  429. created_time = int(time.time())
  430. today = datetime.datetime.today()
  431. start_time = datetime.datetime(today.year, today.month, today.day)
  432. end_time = start_time + datetime.timedelta(days=1)
  433. increase_user_qs = Device_User.objects.filter(data_joined__year=today.year, data_joined__month=today.month,
  434. data_joined__day=today.day).values('region_country')
  435. start_time = CommonService.str_to_timestamp(start_time.strftime('%Y-%m-%d %H:%M:%S'))
  436. end_time = CommonService.str_to_timestamp(end_time.strftime('%Y-%m-%d %H:%M:%S'))
  437. active_user_qs = UserExModel.objects.filter(updTime__gte=start_time, updTime__lt=end_time).values(
  438. 'userID__region_country')
  439. country_qs = CountryModel.objects.all().values('id', 'region__name', 'country_name')
  440. country_dict = {}
  441. continent_dict = {}
  442. for item in country_qs:
  443. country_dict[item['id']] = item['country_name']
  444. continent_dict[item['country_name']] = item['region__name']
  445. with transaction.atomic():
  446. if increase_user_qs.exists():
  447. increase_user_count = increase_user_qs.count()
  448. increase_user_country_list = increase_user_qs.values('region_country').annotate(
  449. count=Count('region_country')).order_by('count')
  450. increase_user_country_dict = {}
  451. increase_user_continent_dict = {}
  452. for item in increase_user_country_list:
  453. country_name = country_dict.get(item['region_country'], '未知国家')
  454. continent_name = continent_dict.get(country_name, '未知大洲')
  455. increase_user_country_dict[country_name] = item['count']
  456. if continent_name not in increase_user_continent_dict:
  457. increase_user_continent_dict[continent_name] = 0
  458. increase_user_continent_dict[continent_name] += item['count']
  459. DeviceUserSummary.objects.create(time=start_time, count=increase_user_count,
  460. country=increase_user_country_dict, created_time=created_time,
  461. continent=increase_user_continent_dict)
  462. if active_user_qs.exists():
  463. active_user_count = active_user_qs.count()
  464. active_user_country_list = active_user_qs.values('userID__region_country').annotate(
  465. count=Count('userID__region_country')).order_by('count')
  466. active_user_country_dict = {}
  467. active_user_continent_dict = {}
  468. for item in active_user_country_list:
  469. country_name = country_dict.get(item['userID__region_country'], '未知国家')
  470. continent_name = continent_dict.get(country_name, '未知大洲')
  471. active_user_country_dict[country_name] = item['count']
  472. if continent_name not in active_user_continent_dict:
  473. active_user_continent_dict[continent_name] = 0
  474. active_user_continent_dict[continent_name] += item['count']
  475. DeviceUserSummary.objects.create(time=start_time, query_type=1, count=active_user_count,
  476. country=active_user_country_dict, created_time=created_time,
  477. continent=active_user_continent_dict)
  478. return response.json(0)
  479. except Exception as e:
  480. return response.json(500, repr(e))
  481. @staticmethod
  482. def collect_order(response):
  483. try:
  484. created_time = int(time.time())
  485. today = datetime.datetime.today()
  486. start_time = datetime.datetime(today.year, today.month, today.day)
  487. end_time = start_time + datetime.timedelta(days=1)
  488. start_time = CommonService.str_to_timestamp(start_time.strftime('%Y-%m-%d %H:%M:%S'))
  489. end_time = CommonService.str_to_timestamp(end_time.strftime('%Y-%m-%d %H:%M:%S'))
  490. order_qs = Order_Model.objects.filter(addTime__gte=start_time, addTime__lt=end_time,
  491. status=1).values('UID', 'order_type',
  492. 'store_meal_name', 'price',
  493. 'addTime', 'currency').order_by(
  494. 'addTime')
  495. uid_list = []
  496. all_order_qs = Order_Model.objects.filter(addTime__lt=start_time, status=1).values('UID')
  497. for item in all_order_qs:
  498. if item['UID'] not in uid_list:
  499. uid_list.append(item['UID'])
  500. # 国家表数据
  501. country_qs = CountryModel.objects.values('id', 'country_name')
  502. country_dict = {}
  503. for item in country_qs:
  504. country_dict[item['id']] = item['country_name']
  505. # 设备类型数据
  506. device_type_qs = DeviceTypeModel.objects.values('name', 'type')
  507. device_type_dict = {}
  508. for item in device_type_qs:
  509. device_type_dict[item['type']] = item['name']
  510. with transaction.atomic():
  511. for item in order_qs:
  512. is_pay = 0
  513. price = float(item['price'])
  514. currency = item['currency']
  515. uid_set_qs = UidSetModel.objects.filter(uid=item['UID']).values('tb_country')
  516. country_id = uid_set_qs[0]['tb_country'] if uid_set_qs.exists() else 0
  517. country_name = country_dict.get(country_id, '未知国家')
  518. order_type = item['order_type']
  519. device_info_qs = Device_Info.objects.filter(UID=item['UID']).values('Type')
  520. device_type_id = device_info_qs[0]['Type'] if device_info_qs.exists() else 0
  521. device_type_name = device_type_dict.get(device_type_id, '未知设备')
  522. store_meal_name = item['store_meal_name']
  523. add_time_stamp = item['addTime']
  524. add_time_str = datetime.datetime.fromtimestamp(int(add_time_stamp))
  525. add_time_str = datetime.datetime(add_time_str.year, add_time_str.month, add_time_str.day)
  526. add_time_stamp = CommonService.str_to_timestamp(add_time_str.strftime('%Y-%m-%d %H:%M:%S'))
  527. if price == 0:
  528. is_pay = 1
  529. order_summary_qs = OrdersSummary.objects.filter(time=add_time_stamp, query_type=1,
  530. service_type=order_type)
  531. else:
  532. order_summary_qs = OrdersSummary.objects.filter(time=add_time_stamp, query_type=0,
  533. service_type=order_type)
  534. if item['UID'] not in uid_list:
  535. pay_order_summary_qs = OrdersSummary.objects.filter(time=add_time_stamp, query_type=2,
  536. service_type=order_type)
  537. query_type = 2
  538. else:
  539. pay_order_summary_qs = OrdersSummary.objects.filter(time=add_time_stamp, query_type=3,
  540. service_type=order_type)
  541. query_type = 3
  542. if pay_order_summary_qs.exists():
  543. pay_order_summary = pay_order_summary_qs.first()
  544. pay_order_summary.count += 1
  545. temp_total = eval(pay_order_summary.total)
  546. if currency not in temp_total:
  547. temp_total[currency] = price
  548. else:
  549. temp_total[currency] = round(temp_total[currency] + price, 2)
  550. pay_order_summary.total = temp_total
  551. country_temp_dict = eval(pay_order_summary.country)
  552. if country_name in country_temp_dict:
  553. country_temp_dict[country_name]['数量'] += 1
  554. if currency not in country_temp_dict[country_name]:
  555. country_temp_dict[country_name][currency] = price
  556. else:
  557. country_temp_dict[country_name][currency] = round(
  558. country_temp_dict[country_name][currency] + price, 2)
  559. else:
  560. country_temp_dict[country_name] = {'数量': 1, currency: price}
  561. pay_order_summary.country = country_temp_dict
  562. device_type_temp_dict = eval(pay_order_summary.device_type)
  563. if device_type_name in device_type_temp_dict:
  564. device_type_temp_dict[device_type_name]['数量'] += 1
  565. if currency not in device_type_temp_dict[device_type_name]:
  566. device_type_temp_dict[device_type_name][currency] = price
  567. else:
  568. device_type_temp_dict[device_type_name][currency] = round(
  569. device_type_temp_dict[device_type_name][currency] + price, 2)
  570. else:
  571. device_type_temp_dict[device_type_name] = {'数量': 1, currency: price}
  572. pay_order_summary.device_type = device_type_temp_dict
  573. store_meal_temp_dict = eval(pay_order_summary.store_meal)
  574. if store_meal_name in store_meal_temp_dict:
  575. store_meal_temp_dict[store_meal_name]['数量'] += 1
  576. if currency not in store_meal_temp_dict[store_meal_name]:
  577. store_meal_temp_dict[store_meal_name][currency] = price
  578. else:
  579. store_meal_temp_dict[store_meal_name][currency] = round(
  580. store_meal_temp_dict[store_meal_name][currency] + price, 2)
  581. else:
  582. store_meal_temp_dict[store_meal_name] = {'数量': 1, currency: price}
  583. pay_order_summary.store_meal = store_meal_temp_dict
  584. pay_order_summary.save()
  585. else:
  586. final_total = {currency: price}
  587. country_temp_dict = {
  588. country_name: {
  589. '数量': 1,
  590. currency: price
  591. }
  592. }
  593. device_type_temp_dict = {
  594. device_type_name: {
  595. '数量': 1,
  596. currency: price
  597. }
  598. }
  599. store_meal_temp_dict = {
  600. store_meal_name: {
  601. '数量': 1,
  602. currency: price
  603. }
  604. }
  605. OrdersSummary.objects.create(time=add_time_stamp, count=1, query_type=query_type,
  606. service_type=order_type, total=final_total,
  607. country=country_temp_dict, created_time=created_time,
  608. device_type=device_type_temp_dict,
  609. store_meal=store_meal_temp_dict)
  610. if order_summary_qs.exists():
  611. order_summary = order_summary_qs.first()
  612. order_summary.count += 1
  613. temp_total = eval(order_summary.total)
  614. if currency not in temp_total:
  615. temp_total[currency] = price
  616. else:
  617. temp_total[currency] = round(temp_total[currency] + price, 2)
  618. order_summary.total = temp_total
  619. country_temp_dict = eval(order_summary.country)
  620. if country_name in country_temp_dict:
  621. if is_pay == 0:
  622. country_temp_dict[country_name]['数量'] += 1
  623. if currency not in country_temp_dict[country_name]:
  624. country_temp_dict[country_name][currency] = price
  625. else:
  626. country_temp_dict[country_name][currency] = round(
  627. country_temp_dict[country_name][currency] + price, 2)
  628. else:
  629. country_temp_dict[country_name] += 1
  630. else:
  631. if is_pay == 0:
  632. country_temp_dict[country_name] = {'数量': 1, currency: price}
  633. else:
  634. country_temp_dict[country_name] = 1
  635. order_summary.country = country_temp_dict
  636. device_type_temp_dict = eval(order_summary.device_type)
  637. if device_type_name in device_type_temp_dict:
  638. if is_pay == 0:
  639. device_type_temp_dict[device_type_name]['数量'] += 1
  640. if currency not in device_type_temp_dict[device_type_name]:
  641. device_type_temp_dict[device_type_name][currency] = price
  642. else:
  643. device_type_temp_dict[device_type_name][currency] = round(
  644. device_type_temp_dict[device_type_name][currency] + price, 2)
  645. else:
  646. device_type_temp_dict[device_type_name] += 1
  647. else:
  648. if is_pay == 0:
  649. device_type_temp_dict[device_type_name] = {'数量': 1, currency: price}
  650. else:
  651. device_type_temp_dict[device_type_name] = 1
  652. order_summary.device_type = device_type_temp_dict
  653. store_meal_temp_dict = eval(order_summary.store_meal)
  654. if store_meal_name in store_meal_temp_dict:
  655. if is_pay == 0:
  656. store_meal_temp_dict[store_meal_name]['数量'] += 1
  657. if currency not in store_meal_temp_dict[store_meal_name]:
  658. store_meal_temp_dict[store_meal_name][currency] = price
  659. else:
  660. store_meal_temp_dict[store_meal_name][currency] = round(
  661. store_meal_temp_dict[store_meal_name][currency] + price, 2)
  662. else:
  663. store_meal_temp_dict[store_meal_name] += 1
  664. else:
  665. if is_pay == 0:
  666. store_meal_temp_dict[store_meal_name] = {'数量': 1, currency: price}
  667. else:
  668. store_meal_temp_dict[store_meal_name] = 1
  669. order_summary.store_meal = store_meal_temp_dict
  670. order_summary.save()
  671. else:
  672. final_total = {currency: price}
  673. if is_pay == 0:
  674. country_temp_dict = {
  675. country_name: {
  676. '数量': 1,
  677. currency: price
  678. }
  679. }
  680. device_type_temp_dict = {
  681. device_type_name: {
  682. '数量': 1,
  683. currency: price
  684. }
  685. }
  686. store_meal_temp_dict = {
  687. store_meal_name: {
  688. '数量': 1,
  689. currency: price
  690. }
  691. }
  692. else:
  693. device_type_temp_dict = {
  694. device_type_name: 1
  695. }
  696. store_meal_temp_dict = {
  697. store_meal_name: 1
  698. }
  699. country_temp_dict = {
  700. country_name: 1
  701. }
  702. OrdersSummary.objects.create(time=add_time_stamp, count=1, query_type=is_pay,
  703. service_type=order_type, total=final_total,
  704. country=country_temp_dict, created_time=created_time,
  705. device_type=device_type_temp_dict, store_meal=store_meal_temp_dict)
  706. return response.json(0)
  707. except Exception as e:
  708. return response.json(500, repr(e))
  709. @staticmethod
  710. def collect_device_info(response):
  711. try:
  712. created_time = int(time.time())
  713. today = datetime.datetime.today()
  714. start_time = datetime.datetime(today.year, today.month, today.day)
  715. end_time = start_time + datetime.timedelta(days=1)
  716. start_time = CommonService.str_to_timestamp(start_time.strftime('%Y-%m-%d %H:%M:%S'))
  717. end_time = CommonService.str_to_timestamp(end_time.strftime('%Y-%m-%d %H:%M:%S'))
  718. increase_device_qs = UidSetModel.objects.filter(addTime__gte=start_time, addTime__lt=end_time).values(
  719. 'tb_country',
  720. 'uid',
  721. 'device_type',
  722. 'cloud_vod',
  723. 'is_ai',
  724. 'mobile_4g',
  725. 'addTime')
  726. video_play_back_time_qs = VideoPlaybackTimeModel.objects.filter(startTime__gte=start_time,
  727. startTime__lt=end_time).values('uid')
  728. active_device_qs = UidSetModel.objects.filter(uid__in=video_play_back_time_qs).values('tb_country',
  729. 'addTime',
  730. 'device_type',
  731. 'cloud_vod',
  732. 'is_ai',
  733. 'mobile_4g',
  734. 'uid')
  735. increase_device_count = increase_device_qs.count()
  736. active_device_count = active_device_qs.count()
  737. # 国家表数据
  738. country_qs = CountryModel.objects.values('id', 'country_name', 'region__name')
  739. country_dict = {}
  740. continent_dict = {}
  741. for item in country_qs:
  742. country_dict[item['id']] = item['country_name']
  743. continent_dict[item['country_name']] = item['region__name']
  744. # 设备类型数据
  745. device_type_qs = DeviceTypeModel.objects.values('name', 'type')
  746. device_type_dict = {}
  747. for item in device_type_qs:
  748. device_type_dict[item['type']] = item['name']
  749. with transaction.atomic():
  750. if increase_device_qs.exists():
  751. # 国家大洲设备数据
  752. increase_device_country_list = increase_device_qs.values('tb_country').annotate(
  753. count=Count('tb_country')).order_by('count')
  754. increase_device_country_dict = {}
  755. increase_device_continent_dict = {}
  756. for item in increase_device_country_list:
  757. country_name = country_dict.get(item['tb_country'], '未知国家')
  758. continent_name = continent_dict.get(country_name, '未知大洲')
  759. increase_device_country_dict[country_name] = item['count']
  760. if continent_name not in increase_device_continent_dict:
  761. increase_device_continent_dict[continent_name] = 0
  762. increase_device_continent_dict[continent_name] += item['count']
  763. # 设备类型数据
  764. increase_device_type_list = increase_device_qs.values('device_type').annotate(
  765. count=Count('device_type')).order_by('count')
  766. increase_device_type_dict = {}
  767. for item in increase_device_type_list:
  768. type_name = device_type_dict.get(item['device_type'], '未知设备类型')
  769. increase_device_type_dict[type_name] = item['count']
  770. # 云存设备类型数据
  771. increase_device_vod_list = increase_device_qs.filter(~Q(cloud_vod=2)).values(
  772. 'device_type').annotate(
  773. count=Count('device_type')).order_by('count')
  774. increase_device_vod_dict = {}
  775. for item in increase_device_vod_list:
  776. type_name = device_type_dict.get(item['device_type'], '未知设备类型')
  777. increase_device_vod_dict[type_name] = item['count']
  778. # AI设备类型数据
  779. increase_device_ai_list = increase_device_qs.filter(~Q(is_ai=2)).values('device_type').annotate(
  780. count=Count('device_type')).order_by('count')
  781. increase_device_ai_dict = {}
  782. for item in increase_device_ai_list:
  783. type_name = device_type_dict.get(item['device_type'], '未知设备类型')
  784. increase_device_ai_dict[type_name] = item['count']
  785. # 联通设备类型数据
  786. increase_device_unicom_list = increase_device_qs.filter(~Q(mobile_4g=2)).values(
  787. 'device_type').annotate(
  788. count=Count('device_type')).order_by('count')
  789. increase_device_unicom_dict = {}
  790. for item in increase_device_unicom_list:
  791. type_name = device_type_dict.get(item['device_type'], '未知设备类型')
  792. increase_device_unicom_dict[type_name] = item['count']
  793. DeviceInfoSummary.objects.create(time=start_time, count=increase_device_count,
  794. query_type=0, created_time=created_time,
  795. country=increase_device_country_dict,
  796. continent=increase_device_continent_dict,
  797. vod_service=increase_device_vod_dict,
  798. ai_service=increase_device_ai_dict,
  799. unicom_service=increase_device_unicom_dict,
  800. device_type=increase_device_type_dict)
  801. if active_device_qs.exists():
  802. # 国家大洲设备数据
  803. active_device_country_list = active_device_qs.values('tb_country').annotate(
  804. count=Count('tb_country')).order_by('count')
  805. active_device_country_dict = {}
  806. active_device_continent_dict = {}
  807. for item in active_device_country_list:
  808. country_name = country_dict.get(item['tb_country'], '未知国家')
  809. continent_name = continent_dict.get(country_name, '未知大洲')
  810. active_device_country_dict[country_name] = item['count']
  811. if continent_name not in active_device_continent_dict:
  812. active_device_continent_dict[continent_name] = 0
  813. active_device_continent_dict[continent_name] += item['count']
  814. # 设备类型数据
  815. active_device_type_list = active_device_qs.values('device_type').annotate(
  816. count=Count('device_type')).order_by('count')
  817. active_device_type_dict = {}
  818. for item in active_device_type_list:
  819. type_name = device_type_dict.get(item['device_type'], '未知设备类型')
  820. active_device_type_dict[type_name] = item['count']
  821. # 云存设备类型数据
  822. active_device_vod_list = active_device_qs.filter(~Q(cloud_vod=2)).values('device_type').annotate(
  823. count=Count('device_type')).order_by('count')
  824. active_device_vod_dict = {}
  825. for item in active_device_vod_list:
  826. type_name = device_type_dict.get(item['device_type'], '未知设备类型')
  827. active_device_vod_dict[type_name] = item['count']
  828. # AI设备类型数据
  829. active_device_ai_list = active_device_qs.filter(~Q(is_ai=2)).values('device_type').annotate(
  830. count=Count('device_type')).order_by('count')
  831. active_device_ai_dict = {}
  832. for item in active_device_ai_list:
  833. type_name = device_type_dict.get(item['device_type'], '未知设备类型')
  834. active_device_ai_dict[type_name] = item['count']
  835. # 联通设备类型数据
  836. active_device_unicom_list = active_device_qs.filter(~Q(mobile_4g=2)).values('device_type').annotate(
  837. count=Count('device_type')).order_by('count')
  838. active_device_unicom_dict = {}
  839. for item in active_device_unicom_list:
  840. type_name = device_type_dict.get(item['device_type'], '未知设备类型')
  841. active_device_unicom_dict[type_name] = item['count']
  842. DeviceInfoSummary.objects.create(time=start_time, count=active_device_count,
  843. query_type=1, created_time=created_time,
  844. country=active_device_country_dict,
  845. continent=active_device_continent_dict,
  846. vod_service=active_device_vod_dict,
  847. ai_service=active_device_ai_dict,
  848. unicom_service=active_device_unicom_dict,
  849. device_type=active_device_type_dict)
  850. return response.json(0)
  851. except Exception as e:
  852. return response.json(500, repr(e))