FileController.py 30 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804
  1. #!/usr/bin/env python3
  2. # -*- coding: utf-8 -*-
  3. import base64
  4. import json
  5. import logging
  6. import os
  7. import time
  8. import zipfile
  9. import xlwt
  10. from django.db.models import Count
  11. from django.http import StreamingHttpResponse, HttpResponse, QueryDict
  12. from django.utils.decorators import method_decorator
  13. from django.views import View
  14. from django.views.decorators.csrf import csrf_exempt
  15. from django.db import transaction
  16. from AnsjerUIDManage.config import BASE_DIR
  17. from Model.models import UIDModel, UserUIDModel, UserModel, LogModel, MacModel, OrderTaskModel, OrderUIDModel, \
  18. SerialNumberModel, OrderSerialNumberModel, UserSerialNumberModel, CompanySerialModel
  19. from Object.RedisObject import RedisObject
  20. from Object.TokenObject import TokenObject
  21. from Object.ResponseObject import ResponseObject
  22. from Service.CommonService import CommonService
  23. from bulk_update.helper import bulk_update
  24. from Object.S3Email import S3Email
  25. class UploadUIDFileView(View):
  26. @method_decorator(csrf_exempt)
  27. def dispatch(self, request, *args, **kwargs):
  28. return super(UploadUIDFileView, self).dispatch(request, *args, **kwargs)
  29. def get(self, request, *args, **kwargs):
  30. request.encoding = 'utf-8'
  31. request_dict = request.GET
  32. fileName = request.FILES.get('fileName', None)
  33. return self.validate(fileName, request_dict, request)
  34. def post(self, request, *args, **kwargs):
  35. request.encoding = 'utf-8'
  36. request_dict = request.POST
  37. fileName = request.FILES.get('fileName', None)
  38. return self.validate(fileName, request_dict, request)
  39. def validate(self, fileName, request_dict, request):
  40. token = request_dict.get('token', None)
  41. area = request_dict.get('area', None)
  42. isReset = request_dict.get('isReset', 0)
  43. content = request_dict.get('fileName', None)
  44. print('content')
  45. print(content)
  46. content = base64.b64decode(content).decode().strip()
  47. content = content[3:(len(content) - 3)]
  48. uids = content.split('\n')
  49. # print(uids)
  50. # print(len(uids))
  51. response = ResponseObject()
  52. tko = TokenObject(token)
  53. if tko.code != 0:
  54. return response.json(tko.code)
  55. # if not fileName and not area:
  56. # return response.json(444, 'fileName,area')
  57. data = {}
  58. duplicate = []
  59. for line in uids:
  60. if len(line) < 20:
  61. continue
  62. if data.__contains__(line):
  63. duplicate.append(line)
  64. else:
  65. data[line] = ''
  66. if isReset == 0:
  67. return self.do_insert(data, duplicate, area, response, request, tko)
  68. else:
  69. return self.do_reset(data, response, area, request, tko)
  70. @transaction.atomic
  71. def do_insert(self, data, duplicate, area, response, request, token):
  72. bulk = []
  73. count = 0
  74. add_time = int(time.time())
  75. update_time = int(time.time())
  76. keys = data.keys()
  77. # 获取最新的mac,判断分配到哪里
  78. mac = MacModel.objects.filter().values('id', 'value', 'is_active')[0]
  79. current_mac = mac['value']
  80. username = 'cspublic@ansjer.com'
  81. if current_mac[-8:] == '0F:42:40': # 一组一共1048576个,此mac是第100w个时
  82. sys_msg_text = "当前旧UID管理系统mac地址已分配到" + current_mac + ",此mac地址是当前组的第100w个,还剩下48576个可分配,mac地址即将用完。"
  83. S3Email().faEmail(sys_msg_text, username)
  84. elif current_mac[-8:] == '0F:90:60': # 一组一共1048576个,此mac是第102w个时
  85. sys_msg_text = "当前旧UID管理系统mac地址已分配到" + current_mac + ",此mac地址是当前组的第102w个,还剩下28576个可分配,mac地址即将用完。"
  86. S3Email().faEmail(sys_msg_text, username)
  87. elif not mac['is_active']:
  88. return response.json(175)
  89. elif current_mac[-8:] == '0F:FF:FF':
  90. MacModel.objects.filter().update(is_active=False) # 更改mac可使用的状态,当再此调用接口时使用上面条件进行阻止
  91. sys_msg_text = "当前旧UID管理系统mac地址已分配到" + current_mac + ",mac地址已分配使用完,请更换分组。"
  92. S3Email().faEmail(sys_msg_text, username)
  93. return response.json(175)
  94. if not mac['is_active']:
  95. return response.json(175)
  96. # redisObject = RedisObject(db=3)
  97. key = ''
  98. tmpMac = mac['value']
  99. savePoint = None
  100. for item in keys:
  101. key = item.strip()
  102. # value = redisObject.get_data(key)
  103. # if value is False:
  104. # # redisObject.set_data(key, '1', 600)
  105. # else:
  106. # duplicate.append(key)
  107. # continue
  108. bulk.append(UIDModel(
  109. uid=item.strip(),
  110. uid_extra='',
  111. status=0,
  112. add_time=add_time,
  113. update_time=update_time,
  114. area=area,
  115. mac=mac['value']
  116. ))
  117. try:
  118. try:
  119. if (count % 5000) == 0:
  120. savePoint = transaction.savepoint()
  121. UIDModel.objects.bulk_create(bulk)
  122. bulk.clear()
  123. data = {
  124. 'value': mac['value'],
  125. 'is_active': tmpMac is not None
  126. }
  127. MacModel.objects.filter().update(**data)
  128. except Exception as e:
  129. # print('--------------------------error 5000')
  130. # print(repr(e))
  131. if savePoint:
  132. transaction.rollback(savePoint)
  133. djangoLogger = logging.getLogger('django')
  134. djangoLogger.exception(repr(e))
  135. return response.json(174, str(e))
  136. else:
  137. savePoint = None
  138. except Exception as e:
  139. # print('--------------------------error 5001')
  140. # print(repr(e))
  141. return response.json(174, str(e))
  142. count += 1
  143. tmpMac = CommonService.updateMac(mac['value'])
  144. if tmpMac is None:
  145. # 能分配的mac已用完
  146. break
  147. else:
  148. mac['value'] = tmpMac
  149. # 当bulk不足5000时,还有数据要插入
  150. try:
  151. try:
  152. savePoint = transaction.savepoint() # 事务保存点
  153. if len(bulk) > 0:
  154. UIDModel.objects.bulk_create(bulk)
  155. bulk.clear()
  156. except Exception as e:
  157. # print('--------------------------error')
  158. # print(repr(e))
  159. if savePoint:
  160. transaction.rollback(savePoint)
  161. return response.json(174)
  162. else:
  163. del data
  164. del bulk
  165. data = {
  166. 'value': mac['value'],
  167. 'is_active': tmpMac is not None
  168. }
  169. MacModel.objects.filter().update(**data)
  170. except Exception as e:
  171. # print('--------------------------error 1111')
  172. # print(repr(e))
  173. return response.json(174)
  174. # print('重复:')
  175. # print(duplicate)
  176. operation = self.formatOperation(operation='上传', quantity=int(count), area=int(area))
  177. print(operation)
  178. self.add_log(request, operation, token)
  179. if tmpMac is None:
  180. return response.json(175, {'last_uid': key})
  181. return response.json(0, {'count': count, 'duplicate_count': len(duplicate), 'data': duplicate})
  182. def do_reset(self, data, response, area, request, token):
  183. keys = data.keys()
  184. uids = []
  185. count = 0
  186. for key in keys:
  187. uids.append(key.strip())
  188. if len(uids) % 5000 == 0:
  189. count += self.do_update_uid_status(uids, area)
  190. uids.clear()
  191. if len(uids) > 0:
  192. count += self.do_update_uid_status(uids, area)
  193. uids.clear()
  194. operation = self.formatOperation('重置', int(count), int(area))
  195. self.add_log(request, operation, token)
  196. return response.json(0)
  197. def do_update_uid_status(self, uids, area):
  198. uid_qs = UIDModel.objects.filter(uid__in=uids, area=area, status=2)
  199. if uid_qs.exists():
  200. uid_ids = []
  201. for uid in uid_qs:
  202. if uid.status == 2:
  203. uid.status = 1
  204. uid_ids.append(uid.id)
  205. UIDModel.objects.bulk_update(uid_qs, fields=['status'])
  206. try:
  207. OrderUIDModel.objects.filter(uid__id__in=tuple(uid_ids)).delete()
  208. except Exception as e:
  209. print(e)
  210. return uid_qs.count()
  211. return 0
  212. def add_log(self, request, operation, token):
  213. ip = CommonService.get_ip_address(request)
  214. now_time = time.time()
  215. content = json.loads(json.dumps(request.POST))
  216. user_qs = UserModel.objects.filter(id=token.userID)
  217. if content.__contains__('fileName'):
  218. del content['fileName']
  219. log = {
  220. 'status': 200,
  221. 'content': json.dumps(content),
  222. 'ip': ip,
  223. 'time': now_time,
  224. 'url': 'upload',
  225. 'operation': operation,
  226. 'user': user_qs[0]
  227. }
  228. try:
  229. LogModel.objects.create(**log)
  230. except Exception as e:
  231. print('出错')
  232. print(repr(e))
  233. def formatOperation(self, operation, quantity, area):
  234. str = '{operation}{quantity}个{area}UID'
  235. if area == 0:
  236. return str.format(operation=operation, quantity=quantity, area='国内')
  237. else:
  238. return str.format(operation=operation, quantity=quantity, area='国外')
  239. class UploadSerialNumberFileView(View):
  240. # 上传文件重置序列号使用状态
  241. @method_decorator(csrf_exempt)
  242. def dispatch(self, request, *args, **kwargs):
  243. return super(UploadSerialNumberFileView, self).dispatch(request, *args, **kwargs)
  244. def get(self, request, *args, **kwargs):
  245. request.encoding = 'utf-8'
  246. request_dict = request.GET
  247. return self.reset_serial_number(request_dict, request)
  248. def post(self, request, *args, **kwargs):
  249. request.encoding = 'utf-8'
  250. request_dict = request.POST
  251. return self.reset_serial_number(request_dict, request)
  252. def reset_serial_number(self, request_dict, request):
  253. token = request_dict.get('token', None)
  254. p2p = request_dict.get('area', None)
  255. isReset = request_dict.get('isReset', 0)
  256. content = request_dict.get('fileName', None)
  257. print('content: ', content)
  258. response = ResponseObject()
  259. tko = TokenObject(token)
  260. if tko.code != 0:
  261. return response.json(tko.code)
  262. if not p2p or not content:
  263. return response.json(444, 'p2p, content')
  264. try:
  265. content = base64.b64decode(content).decode().strip()
  266. content = content[3:(len(content)-3)]
  267. serial_number_list = content.split('\r\n')
  268. print('serial_number_list', serial_number_list)
  269. # 重置使用状态为已下载的序列号为已分配
  270. serial_number_qs = SerialNumberModel.objects.filter(serial_number__in=serial_number_list, p2p=p2p, use_status=2)
  271. count = serial_number_qs.count() # 需要重置的序列号个数
  272. if count:
  273. with transaction.atomic():
  274. serial_number_qs.update(use_status=1)
  275. OrderSerialNumberModel.objects.filter(serial_number__in=serial_number_qs).delete()
  276. operation = self.formatOperation('重置', int(count), p2p)
  277. self.add_log(request, operation, tko)
  278. return response.json(0)
  279. except Exception as e:
  280. # print(e)
  281. return response.json(500, repr(e))
  282. def add_log(self, request, operation, token):
  283. ip = CommonService.get_ip_address(request)
  284. now_time = time.time()
  285. content = json.loads(json.dumps(request.POST))
  286. user_qs = UserModel.objects.filter(id=token.userID)
  287. if content.__contains__('fileName'):
  288. del content['fileName']
  289. log = {
  290. 'status': 200,
  291. 'content': json.dumps(content),
  292. 'ip': ip,
  293. 'time': now_time,
  294. 'url': 'upload',
  295. 'operation': operation,
  296. 'user': user_qs[0]
  297. }
  298. LogModel.objects.create(**log)
  299. def formatOperation(self, operation, quantity, p2p):
  300. operate = '{operation}{quantity}个{p2p}序列号'
  301. p2p = '宸云' if p2p == '1' else 'TUTK'
  302. return operate.format(operation=operation, quantity=quantity, p2p=p2p)
  303. class DownloadUIDFileView(View):
  304. @method_decorator(csrf_exempt)
  305. def dispatch(self, request, *args, **kwargs):
  306. return super(DownloadUIDFileView, self).dispatch(request, *args, **kwargs)
  307. def get(self, request, *args, **kwargs):
  308. request.encoding = 'utf-8'
  309. request_dict = request.GET
  310. return self.validate(request_dict)
  311. def post(self, request, *args, **kwargs):
  312. request.encoding = 'utf-8'
  313. request_dict = request.POST
  314. return self.validate(request_dict)
  315. def validate(self, request_dict):
  316. token = request_dict.get('token', None)
  317. area = request_dict.get('area', None)
  318. quantity = int(request_dict.get('quantity', None))
  319. fileType = request_dict.get('fileType', None)
  320. order_number = request_dict.get('order_number', None)
  321. board = request_dict.get('board', None)
  322. plan = request_dict.get('plan', None)
  323. checksum = request_dict.get('checksum', None)
  324. ic_model = request_dict.get('ic_model', None)
  325. order_quantity = request_dict.get('order_quantity', None)
  326. response = ResponseObject()
  327. # print(area)
  328. # print(quantity)
  329. token = TokenObject(token)
  330. if token.code != 0:
  331. return response.json(token.code)
  332. if not area or not order_number or not board or not plan or not checksum or not ic_model or not order_quantity or not fileType:
  333. return response.json(444)
  334. area = int(area)
  335. if area >= 0 and quantity > 0:
  336. # 保存订单信息
  337. now_time = int(time.time())
  338. order = {
  339. 'order_number': order_number,
  340. 'board': board,
  341. 'plan': plan,
  342. 'checksum': checksum,
  343. 'ic_model': ic_model,
  344. 'quantity': order_quantity,
  345. 'add_time': now_time
  346. }
  347. tmp = OrderTaskModel.objects.create(**order)
  348. print(tmp)
  349. order = OrderTaskModel.objects.filter(order_number=order_number).order_by('-add_time')[0]
  350. uid_qs = UserUIDModel.objects.filter(user__id=token.userID, uid__status=1, uid__area=area)
  351. uid_values = uid_qs.values('uid__id', 'uid__uid', 'uid__mac', 'uid__uid_extra', 'uid__add_time', 'uid__update_time', 'uid__area')
  352. count = uid_values.count()
  353. if count < quantity:
  354. return response.json(444, '设备UID不足')
  355. if uid_values.exists():
  356. uid_values = uid_values[0:quantity]
  357. uid_qs = uid_qs[0: quantity]
  358. if fileType == 'txt':
  359. # return self.download_txt(uid_values, uid_qs, order)
  360. # return self.download_excel(uid_values, order)
  361. return self.download_zip(uid_values, order)
  362. elif fileType == 'excel':
  363. return self.download_excel(uid_values, order)
  364. else:
  365. return response.json(444, 'fileType')
  366. else:
  367. return response.json(444, '111')
  368. else:
  369. return response.json(444, '222')
  370. def download_txt(self, uid_values, uid_qs, order):
  371. updates = []
  372. updates_uid = []
  373. content = ''
  374. now_time = int(time.time())
  375. for i in range(len(uid_values)):
  376. # print(item)
  377. item = uid_values[i]
  378. mac: str = item['uid__mac']
  379. index = mac.rfind(':')
  380. tmp = mac[0:index] + '\t' + mac[index:]
  381. content += tmp + '\t'
  382. content += item['uid__uid'].strip()
  383. content += '\r\n'
  384. uidModel = UIDModel(
  385. id=item['uid__id'],
  386. uid=item['uid__uid'],
  387. mac=item['uid__mac'],
  388. uid_extra=item['uid__uid_extra'],
  389. status=2,
  390. add_time=item['uid__add_time'],
  391. update_time=now_time,
  392. area=item['uid__area']
  393. )
  394. updates.append(uidModel)
  395. order_uid = OrderUIDModel(uid=uidModel, order=order, add_time=now_time, update_time=now_time)
  396. updates_uid.append(order_uid)
  397. if len(updates) % 5000 == 0:
  398. UIDModel.objects.bulk_update(updates, fields=["status"])
  399. OrderUIDModel.objects.bulk_create(updates_uid)
  400. updates.clear()
  401. updates_uid.clear()
  402. # print(item['uid__uid'])
  403. if len(updates) > 0:
  404. UIDModel.objects.bulk_update(updates, fields=["status"])
  405. OrderUIDModel.objects.bulk_create(updates_uid)
  406. updates.clear()
  407. updates_uid.clear()
  408. del updates
  409. del updates_uid
  410. content = content[0:len(content) - 1]
  411. response = StreamingHttpResponse(content)
  412. response['Content-Type'] = 'application/octet-stream'
  413. response['Content-Disposition'] = 'attachment;filename=UID'+time.strftime('-%Y-%m-%d-%H-%M-%S', time.localtime()) + '.txt'
  414. return response
  415. def download_excel(self, uid_qs, order):
  416. response = HttpResponse(content_type='application/vnd.ms-excel')
  417. response['Content-Disposition'] = 'attachment; filename=UID' + time.strftime('-%Y-%m-%d-%H-%M-%S',
  418. time.localtime()) + '.xls'
  419. workbook = xlwt.Workbook(encoding='utf-8')
  420. sheet1 = workbook.add_sheet('UID')
  421. # row1 = [u'设备UID']
  422. # for i in range(0, len(row1)):
  423. # sheet1.write(0, i, row1[i])
  424. num = 1
  425. updates = []
  426. updates_uid = []
  427. now_time = int(time.time())
  428. for item in uid_qs:
  429. uid = item['uid__uid']
  430. mac = item['uid__mac']
  431. index = mac.rfind(':')
  432. sheet1.write(num, 0, mac[0:index])
  433. sheet1.write(num, 1, mac[index:])
  434. sheet1.write(num, 2, uid)
  435. num += 1
  436. uidModel = UIDModel(
  437. id=item['uid__id'],
  438. uid=item['uid__uid'],
  439. mac=item['uid__mac'],
  440. uid_extra=item['uid__uid_extra'],
  441. status=2,
  442. add_time=item['uid__add_time'],
  443. update_time=now_time,
  444. area=item['uid__area']
  445. )
  446. updates.append(uidModel)
  447. order_uid = OrderUIDModel(uid=uidModel, order=order, add_time=now_time, update_time=now_time)
  448. updates_uid.append(order_uid)
  449. if len(updates) % 5000 == 0:
  450. UIDModel.objects.bulk_update(updates, fields=["status"])
  451. OrderUIDModel.objects.bulk_create(updates_uid)
  452. updates.clear()
  453. updates_uid.clear()
  454. # print(item['uid__uid'])
  455. if len(updates) > 0:
  456. UIDModel.objects.bulk_update(updates, fields=["status"])
  457. OrderUIDModel.objects.bulk_create(updates_uid)
  458. updates.clear()
  459. updates_uid.clear()
  460. UIDModel.objects.bulk_update(updates, fields=["status"])
  461. workbook.save(response)
  462. return response
  463. def download_zip(self, uid_qs, order):
  464. dir_name = 'static/' + time.strftime('%Y_%m_%d_%H_%M_%S', time.localtime())
  465. path = '/'.join((BASE_DIR, dir_name)).replace('\\', '/') + '/'
  466. if not os.path.exists(path):
  467. os.mkdir(path)
  468. filename = time.strftime('_%Y_%m_%d_%H_%M_%S', time.localtime())
  469. txt_filename = 'UID' + filename + '.txt'
  470. excel_filename = 'UID' + filename + '.xls'
  471. txt_file = open(path + txt_filename, 'w+')
  472. workbook = xlwt.Workbook(encoding='utf-8')
  473. sheet1 = workbook.add_sheet('UID')
  474. num = 1
  475. updates = []
  476. updates_uid = []
  477. content = ''
  478. now_time = int(time.time())
  479. for i in range(len(uid_qs)):
  480. # print(item)
  481. item = uid_qs[i]
  482. uid = item['uid__uid']
  483. mac = item['uid__mac']
  484. index = mac.rfind(':')
  485. tmp = mac[0:index] + '\t' + mac[index:]
  486. content += tmp + '\t'
  487. content += item['uid__uid'].strip()
  488. content += '\r\n'
  489. sheet1.write(num, 0, mac[0:index])
  490. sheet1.write(num, 1, mac[index:])
  491. sheet1.write(num, 2, uid)
  492. num += 1
  493. uidModel = UIDModel(
  494. id=item['uid__id'],
  495. uid=item['uid__uid'],
  496. mac=item['uid__mac'],
  497. uid_extra=item['uid__uid_extra'],
  498. status=2,
  499. add_time=item['uid__add_time'],
  500. update_time=now_time,
  501. area=item['uid__area']
  502. )
  503. updates.append(uidModel)
  504. order_uid = OrderUIDModel(uid=uidModel, order=order, add_time=now_time, update_time=now_time)
  505. updates_uid.append(order_uid)
  506. if len(updates) % 5000 == 0:
  507. UIDModel.objects.bulk_update(updates, fields=["status"])
  508. OrderUIDModel.objects.bulk_create(updates_uid)
  509. updates.clear()
  510. updates_uid.clear()
  511. # print(item['uid__uid'])
  512. if len(updates) > 0:
  513. UIDModel.objects.bulk_update(updates, fields=["status"])
  514. OrderUIDModel.objects.bulk_create(updates_uid)
  515. updates.clear()
  516. updates_uid.clear()
  517. del updates
  518. del updates_uid
  519. content = content[0:len(content) - 1]
  520. txt_file.write(content)
  521. txt_file.close()
  522. workbook.save(path + excel_filename)
  523. zip_name = path[0:path.rfind('/')] + '.zip'
  524. return self.get_zip(path, os.listdir(path), zip_name)
  525. def get_zip(self, path, files, zip_name):
  526. zp = zipfile.ZipFile(zip_name, 'w', zipfile.ZIP_DEFLATED)
  527. for file in files:
  528. zp.write(filename=(path + file), arcname=str(file))
  529. zp.close()
  530. response = StreamingHttpResponse(open(zip_name, 'rb'))
  531. response['content_type'] = "application/octet-stream"
  532. response['Content-Disposition'] = 'attachment; filename=UID' + time.strftime('_%Y_%m_%d_%H_%M_%S',
  533. time.localtime()) + '.zip'
  534. return response
  535. class DownloadSerialNumberFileView(View):
  536. @method_decorator(csrf_exempt)
  537. def dispatch(self, request, *args, **kwargs):
  538. return super(DownloadSerialNumberFileView, self).dispatch(request, *args, **kwargs)
  539. def get(self, request, *args, **kwargs):
  540. request.encoding = 'utf-8'
  541. request_dict = request.GET
  542. return self.validate(request_dict)
  543. def post(self, request, *args, **kwargs):
  544. request.encoding = 'utf-8'
  545. request_dict = request.POST
  546. return self.validate(request_dict)
  547. def validate(self, request_dict):
  548. token = request_dict.get('token', None)
  549. area = request_dict.get('area', None)
  550. quantity = int(request_dict.get('quantity', None))
  551. fileType = request_dict.get('fileType', None)
  552. order_number = request_dict.get('order_number', None)
  553. board = request_dict.get('board', None)
  554. plan = request_dict.get('plan', None)
  555. checksum = request_dict.get('checksum', None)
  556. ic_model = request_dict.get('ic_model', None)
  557. order_quantity = request_dict.get('order_quantity', None)
  558. response = ResponseObject()
  559. # print(area)
  560. # print(quantity)
  561. token = TokenObject(token)
  562. if token.code != 0:
  563. return response.json(token.code)
  564. if not area or not order_number or not board or not plan or not checksum or not ic_model or not order_quantity or not fileType:
  565. return response.json(444)
  566. area = int(area)
  567. if area >= 0 and quantity > 0:
  568. # 保存订单信息
  569. now_time = int(time.time())
  570. order = {
  571. 'order_number': order_number,
  572. 'board': board,
  573. 'plan': plan,
  574. 'checksum': checksum,
  575. 'ic_model': ic_model,
  576. 'quantity': order_quantity,
  577. 'add_time': now_time
  578. }
  579. tmp = OrderTaskModel.objects.create(**order)
  580. print(tmp)
  581. order = OrderTaskModel.objects.filter(order_number=order_number).order_by('-add_time')[0]
  582. user_qs = UserModel.objects.filter(id=token.userID)
  583. if not user_qs.exists():
  584. return response.json(9)
  585. user = user_qs[0]
  586. result = UserSerialNumberModel.objects.filter(user__id=user.id, serial_number__p2p=area,
  587. serial_number__use_status=1).aggregate(
  588. num=Count('serial_number__use_status'))
  589. us_qs = UserSerialNumberModel.objects.filter(user__id=user.id, serial_number__p2p=area,
  590. serial_number__use_status=1).values('serial_number__serial_number')
  591. sns = []
  592. for us in us_qs:
  593. sns.append(us['serial_number__serial_number'])
  594. cs_qs = CompanySerialModel.objects.filter(serial_number__in=sns, status=1).values('serial_number')
  595. sns = []
  596. for cs in cs_qs:
  597. sns.append(cs['serial_number'])
  598. sn_qs = SerialNumberModel.objects.filter(serial_number__in=sns)
  599. count = result['num']
  600. if count < quantity:
  601. return response.json(444, '序列号不足')
  602. if sn_qs.exists():
  603. sn_qs = sn_qs[0: quantity]
  604. if fileType == 'txt':
  605. # return self.download_txt(uid_values, uid_qs, order)
  606. # return self.download_excel(uid_values, order)
  607. fileTypeName = 'chenyun'
  608. if area == 2:
  609. fileTypeName = 'tutk'
  610. return self.download_zip(sn_qs, order , fileTypeName )
  611. else:
  612. return response.json(444, 'fileType')
  613. else:
  614. return response.json(444, '111')
  615. else:
  616. return response.json(444, '222')
  617. def download_zip(self, sn_qs, order , fileTypeName):
  618. dir_name = 'static/' + time.strftime('%Y_%m_%d_%H_%M_%S', time.localtime())
  619. path = '/'.join((BASE_DIR, dir_name)).replace('\\', '/') + '/'
  620. if not os.path.exists(path):
  621. os.mkdir(path)
  622. filename = time.strftime('_%Y_%m_%d_%H_%M_%S', time.localtime())
  623. txt_filename = 'SerialNumber' + fileTypeName +'_'+ filename + '.txt'
  624. excel_filename = 'SerialNumber' + fileTypeName + '_' + filename + '.xls'
  625. txt_file = open(path + txt_filename, 'w+')
  626. workbook = xlwt.Workbook(encoding='utf-8')
  627. sheet1 = workbook.add_sheet('SerialNumber')
  628. num = 0
  629. updates = []
  630. updates_serial_number = []
  631. content = ''
  632. now_time = int(time.time())
  633. for i in range(len(sn_qs)):
  634. # print(item)
  635. item = sn_qs[i]
  636. serial_number = item.serial_number
  637. sheet1.write(num, 0, serial_number + '11A')
  638. num += 1
  639. content += serial_number.strip() + '11A'
  640. content += '\n'
  641. serialNumberModel = SerialNumberModel(
  642. id=item.id,
  643. serial_number = serial_number,
  644. status=item.status,
  645. use_status=2,
  646. p2p=item.p2p,
  647. add_time=item.add_time
  648. )
  649. updates.append(serialNumberModel)
  650. order_serial_number = OrderSerialNumberModel(serial_number=item, order=order, add_time=now_time, update_time=now_time)
  651. updates_serial_number.append(order_serial_number)
  652. if len(updates) % 5000 == 0:
  653. bulk_update(updates)
  654. OrderSerialNumberModel.objects.bulk_create(updates_serial_number)
  655. updates.clear()
  656. updates_serial_number.clear()
  657. # print(item['uid__uid'])
  658. if len(updates) > 0:
  659. bulk_update(updates)
  660. OrderSerialNumberModel.objects.bulk_create(updates_serial_number)
  661. updates.clear()
  662. updates_serial_number.clear()
  663. del updates
  664. del updates_serial_number
  665. content = content[0:len(content) - 1]
  666. txt_file.write(content)
  667. txt_file.close()
  668. workbook.save(path + excel_filename)
  669. zip_name = path[0:path.rfind('/')] + '.zip'
  670. return self.get_zip(path, os.listdir(path), zip_name , fileTypeName)
  671. def get_zip(self, path, files, zip_name, fileTypeName ):
  672. zp = zipfile.ZipFile(zip_name, 'w', zipfile.ZIP_DEFLATED)
  673. for file in files:
  674. zp.write(filename=(path + file), arcname=str(file))
  675. zp.close()
  676. response = StreamingHttpResponse(open(zip_name, 'rb'))
  677. response['content_type'] = "application/octet-stream"
  678. response['Content-Disposition'] = 'attachment; filename=SerialNumber_'+ fileTypeName + time.strftime('_%Y_%m_%d_%H_%M_%S',
  679. time.localtime()) + '.zip'
  680. return response