PcInfo.py 30 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719
  1. import hashlib
  2. import logging
  3. import shutil
  4. import time
  5. import traceback
  6. import os
  7. from urllib import request, parse
  8. import requests
  9. from boto3 import Session
  10. from django.http import HttpResponse
  11. from django.views.generic.base import View
  12. from Controller.PctestController import TokenObject1
  13. from Model.models import Pc_Info, PctestlogModel, AVSSVersion
  14. from Object.ResponseObject import ResponseObject
  15. from Object.TokenObject import TokenObject
  16. from Service.CommonService import CommonService
  17. from Ansjer.config import AWS_ACCESS_KEY_ID, AWS_SECRET_ACCESS_KEY, AWS_ARN, BASE_DIR
  18. import boto3
  19. import botocore
  20. from botocore import client
  21. from wsgiref.util import FileWrapper
  22. from zlib import crc32
  23. from typing import Union
  24. class PcInfo(View):
  25. def dispatch(self, requset, *args, **kwargs):
  26. return super(PcInfo, self).dispatch(requset, *args, **kwargs)
  27. def get(self, request, *args, **kwargs):
  28. operation = kwargs.get('operation')
  29. request.encoding = 'utf-8'
  30. return self.validation(request.GET, request, operation)
  31. def post(self, request, *args, **kwargs):
  32. operation = kwargs.get('operation')
  33. request.encoding = 'utf-8'
  34. return self.validation(request.POST, request, operation)
  35. def validation(self, request_dict, request, operation):
  36. response = ResponseObject()
  37. if not operation:
  38. return response.json(444, 'operation')
  39. else:
  40. if operation == 'query': # pc端调用查询
  41. return self.query(request_dict, response)
  42. elif operation == 's3addandupload':
  43. return self.s3addandupload(request_dict, response, request)
  44. elif operation == 's3addanduploadlog': #上传日志文件
  45. return self.s3addanduploadlog(request_dict, response, request)
  46. elif operation == 's3download':
  47. return self.s3download(request_dict, response)
  48. elif operation == 's3downloadlog': #下载日志文件
  49. return self.s3downloadlog(request_dict, response)
  50. elif operation == 's3delete':
  51. return self.s3delete(request_dict, response)
  52. elif operation == 'edit':
  53. return self.edit(request_dict, response)
  54. elif operation == 'getnewversion': # 获取当前软件的最新版本
  55. return self.getnewversion(request_dict, response)
  56. elif operation == 'getnewversionV2': # 获取当前软件的最新版本
  57. return self.getnewversionV2(request_dict, response)
  58. elif operation == 'queryall': # 后台查询
  59. return self.queryall(request_dict, response)
  60. elif operation == 'addandupload': # 上传到服务器
  61. return self.addandupload(request_dict, response, request)
  62. elif operation == 'download': # 服务器下载
  63. return self.download(request_dict, response)
  64. elif operation == 'delete':
  65. return self.delete(request_dict, response)
  66. elif operation == 'AVSS/check-version':
  67. return self.check_version(request_dict, response)
  68. else:
  69. return response.json(414)
  70. def getnewversion(self, request_dict, response):
  71. pc_name = request_dict.get('pc_name', None)
  72. bundle_version = request_dict.get('bundle_version', None)
  73. pc_version = request_dict.get('pc_version', None)
  74. pc_test = request_dict.get('pc_test', None)
  75. param_flag = CommonService.get_param_flag(
  76. data=[pc_name, bundle_version, pc_version, pc_test])
  77. if param_flag is not True:
  78. return response.json(444)
  79. file = Pc_Info.objects.filter(pc_name=pc_name, bundle_version=bundle_version, pc_test=pc_test)
  80. if not file:
  81. return response.json(173)
  82. app_list = Pc_Info.objects.filter(pc_name=pc_name, bundle_version=bundle_version, pc_test=pc_test)
  83. all_version = app_list.values('pc_version')
  84. v = []
  85. for i in all_version:
  86. v.append(i['pc_version'])
  87. new_version = max(v)
  88. if pc_version == new_version:
  89. return response.json(10045)
  90. else:
  91. path = app_list.filter(pc_version=new_version).values('download_link')[0]['download_link']
  92. print('path', path)
  93. aws_s3_guonei = boto3.client(
  94. 's3',
  95. aws_access_key_id=AWS_ACCESS_KEY_ID[0],
  96. aws_secret_access_key=AWS_SECRET_ACCESS_KEY[0],
  97. config=botocore.client.Config(signature_version='s3v4'),
  98. region_name='cn-northwest-1')
  99. response_url = aws_s3_guonei.generate_presigned_url(
  100. ClientMethod='get_object',
  101. Params={'Bucket': 'pc-package', 'Key': path}, ExpiresIn=3600)
  102. res = {'pc_name': pc_name,
  103. 'new_version': new_version,
  104. 'path': path,
  105. 'response_url': response_url}
  106. return response.json(0, res)
  107. def getnewversionV2(self, request_dict, response):
  108. pc_name = request_dict.get('pc_name', None)
  109. bundle_version = request_dict.get('bundle_version', None)
  110. pc_version = request_dict.get('pc_version', None)
  111. pc_test = request_dict.get('pc_test', None)
  112. param_flag = CommonService.get_param_flag(
  113. data=[pc_name, bundle_version, pc_version, pc_test])
  114. if param_flag is not True:
  115. return response.json(444)
  116. file = Pc_Info.objects.filter(pc_name=pc_name, bundle_version=bundle_version, pc_test=pc_test)
  117. if not file.exists():
  118. return response.json(173)
  119. aws_s3_guonei = boto3.client(
  120. 's3',
  121. aws_access_key_id=AWS_ACCESS_KEY_ID[0],
  122. aws_secret_access_key=AWS_SECRET_ACCESS_KEY[0],
  123. config=botocore.client.Config(signature_version='s3v4'),
  124. region_name='cn-northwest-1')
  125. version_list = []
  126. for i in range(2):
  127. if pc_test == '1': # 测试版返回所有版本中的最新版本
  128. app_list = Pc_Info.objects.filter(pc_name=pc_name, bundle_version=bundle_version, is_update=i)
  129. else:
  130. app_list = Pc_Info.objects.filter(pc_name=pc_name, bundle_version=bundle_version, pc_test=pc_test, is_update=i)
  131. # 当前软件不存在强制更新(is_update=1)的版本返回空
  132. if not app_list.exists():
  133. res = {'pc_name': '',
  134. 'new_version': '',
  135. 'is_update': i,
  136. 'explain': '',
  137. 'path': '',
  138. 'response_url': ''}
  139. version_list.append(res)
  140. continue
  141. all_version = app_list.values('pc_version')
  142. v = []
  143. for i in all_version:
  144. v.append(i['pc_version'])
  145. new_version = max(v)
  146. # 当前软件版本为最新时返回空
  147. if pc_version >= new_version:
  148. res = {'pc_name': '',
  149. 'new_version': '',
  150. 'is_update': app_list.filter(pc_version=new_version).values('is_update')[0]['is_update'],
  151. 'explain': '',
  152. 'path': '',
  153. 'response_url': ''}
  154. version_list.append(res)
  155. continue
  156. path = app_list.filter(pc_version=new_version).values('download_link')[0]['download_link']
  157. print('path', path)
  158. response_url = aws_s3_guonei.generate_presigned_url(
  159. ClientMethod='get_object',
  160. Params={'Bucket': 'pc-package', 'Key': path}, ExpiresIn=3600)
  161. res = {'pc_name': pc_name,
  162. 'new_version': new_version,
  163. 'is_update': app_list.filter(pc_version=new_version).values('is_update')[0]['is_update'],
  164. 'explain': app_list.filter(pc_version=new_version).values('explain')[0]['explain'],
  165. 'path': path,
  166. 'response_url': response_url}
  167. version_list.append(res)
  168. return response.json(0, version_list)
  169. def query(self, request_dict, response):
  170. pc_name = request_dict.get('pc_name', None)
  171. bundle_version = request_dict.get('bundle_version', None)
  172. pc_version = request_dict.get('pc_version', None)
  173. pc_test = request_dict.get('pc_test', None)
  174. package = request_dict.get('package', None)
  175. file_type = request_dict.get('file_type', None)
  176. # 根据传的参数筛选,没传时返回全部
  177. queryset = Pc_Info.objects.all()
  178. if package:
  179. queryset = Pc_Info.objects.filter(package=package)
  180. if file_type:
  181. queryset = Pc_Info.objects.filter(file_type=file_type)
  182. if pc_name and file_type:
  183. queryset = Pc_Info.objects.filter(pc_name=pc_name, file_type=file_type)
  184. if file_type and package:
  185. queryset = Pc_Info.objects.filter(file_type=file_type, package=package)
  186. if pc_name and bundle_version and pc_version and pc_test:
  187. queryset = Pc_Info.objects.filter(pc_name=pc_name, bundle_version=bundle_version, pc_version=pc_version, pc_test=pc_test)
  188. count = queryset.count()
  189. res = queryset
  190. send_json = CommonService.qs_to_dict(res)
  191. send_json['count'] = count
  192. return response.json(0, send_json)
  193. def s3addandupload(self, request_dict, response, request):
  194. logger = logging.getLogger('info')
  195. logger.info('s3方式上传参数:')
  196. logger.info(request_dict)
  197. token = request_dict.get('token', None)
  198. tko = TokenObject(token)
  199. response.lang = tko.lang
  200. pc_name = request_dict.get('pc_name', None)
  201. bundle_version = request_dict.get('bundle_version', None)
  202. pc_version = request_dict.get('pc_version', None)
  203. pc_test = request_dict.get('pc_test', None)
  204. lang = request_dict.get('lang', None)
  205. file_name = request_dict.get('file_name', None)
  206. file_type = request_dict.get('file_type', None)
  207. package = request_dict.get('package', None)
  208. explain = request_dict.get('explain', '')
  209. content = request_dict.get('content', '')
  210. authority = request_dict.get('authority', '')
  211. is_update = request_dict.get('is_update', None)
  212. is_open = request_dict.get('is_open', None)
  213. # logger.info('文件名字:')
  214. # logger.info(file_name)
  215. param_flag = CommonService.get_param_flag(
  216. data=[pc_name, bundle_version, pc_version, pc_test, lang, file_name, file_type, package, is_update, is_open])
  217. if param_flag is not True:
  218. return response.json(444)
  219. else:
  220. file = Pc_Info.objects.filter(pc_name=pc_name, bundle_version=bundle_version,
  221. pc_version=pc_version, pc_test=pc_test)
  222. if file:
  223. return response.json(174)
  224. try:
  225. logger.info('开始上传')
  226. # 把安装包上传到s3
  227. aws_s3_guonei = boto3.client(
  228. 's3',
  229. aws_access_key_id=AWS_ACCESS_KEY_ID[0],
  230. aws_secret_access_key=AWS_SECRET_ACCESS_KEY[0],
  231. config=botocore.client.Config(signature_version='s3v4'),
  232. region_name='cn-northwest-1'
  233. )
  234. download_link = '{pc_name}/{pc_version}_{bundle_version}_{pc_test}_{file_name}'.format(
  235. pc_name=pc_name, pc_version=pc_version, bundle_version=bundle_version,
  236. pc_test=pc_test, file_name=file_name)
  237. response_url = aws_s3_guonei.generate_presigned_url(
  238. ClientMethod='put_object',
  239. Params={
  240. 'Bucket': 'pc-package',
  241. 'Key': download_link
  242. },
  243. ExpiresIn=3600
  244. )
  245. # 通过上传签名url对文件进行上传
  246. # requests.put(response_url, data=file_name)
  247. # logger.info('上传完成')
  248. add_time = time.time()
  249. create_dict = {
  250. 'pc_name': pc_name,
  251. 'bundle_version': bundle_version,
  252. 'pc_version': pc_version,
  253. 'pc_test': pc_test,
  254. 'lang': lang,
  255. 'download_link': download_link,
  256. 'add_time': add_time,
  257. 'update_time': add_time,
  258. 'file_type': file_type,
  259. 'package': package,
  260. 'explain': explain,
  261. 'content': content,
  262. 'authority': authority,
  263. 'is_update': is_update,
  264. 'is_open': is_open
  265. }
  266. pc_Info = Pc_Info(**create_dict)
  267. pc_Info.save()
  268. except Exception:
  269. errorInfo = traceback.format_exc()
  270. print(errorInfo)
  271. return response.json(500, {'details': errorInfo})
  272. else:
  273. if pc_Info.id:
  274. res = {'id': pc_Info.id,
  275. 'pc_name': pc_Info.pc_name,
  276. 'bundle_version': pc_Info.bundle_version,
  277. 'pc_version': pc_Info.pc_version,
  278. 'pc_test': pc_Info.pc_test,
  279. 'download_link': pc_Info.download_link,
  280. 'lang': pc_Info.lang,
  281. 'add_time': pc_Info.add_time,
  282. 'update_time': pc_Info.update_time,
  283. 'file_type': pc_Info.file_type,
  284. 'package': pc_Info.package,
  285. 'explain': pc_Info.explain,
  286. 'content': pc_Info.content,
  287. 'authority': pc_Info.authority,
  288. 'is_update': pc_Info.is_update,
  289. 'is_open': pc_Info.is_open,
  290. 'upload_url': response_url
  291. }
  292. return response.json(0, res)
  293. else:
  294. return response.json(500)
  295. def s3addanduploadlog(self, request_dict, response, request):
  296. logger = logging.getLogger('info')
  297. logger.info('s3方式上传参数:')
  298. logger.info(request_dict)
  299. token = request_dict.get('token', None)
  300. tko = TokenObject1(token)
  301. response.lang = tko.lang
  302. if tko.code != 0:
  303. return response.json(tko.code)
  304. # 获取访问者的id和岗位
  305. userID = tko.id
  306. datetime = request_dict.get('datetime', None)
  307. file_name = request_dict.get('file_name', None)
  308. pathtype = request_dict.get('pathtype', 'log')
  309. # logger.info('文件名字:')
  310. # logger.info(file_name)
  311. param_flag = CommonService.get_param_flag(
  312. data=[datetime, file_name, pathtype])
  313. if param_flag is not True:
  314. return response.json(444)
  315. logger.info('开始上传')
  316. # 把安装包上传到s3
  317. aws_s3_guonei = boto3.client(
  318. 's3',
  319. aws_access_key_id=AWS_ACCESS_KEY_ID[0],
  320. aws_secret_access_key=AWS_SECRET_ACCESS_KEY[0],
  321. config=botocore.client.Config(signature_version='s3v4'),
  322. region_name='cn-northwest-1'
  323. )
  324. download_link = '{pathtype}/{datetime}/{file_name}'.format(
  325. pathtype=pathtype, datetime=datetime, file_name=file_name)
  326. response_url = aws_s3_guonei.generate_presigned_url(
  327. ClientMethod='put_object',
  328. Params={
  329. 'Bucket': 'pc-package',
  330. 'Key': download_link
  331. },
  332. ExpiresIn=3600
  333. )
  334. add_time = time.time()
  335. PctestlogModel.objects.create(user_id=userID, content=download_link, addtime=add_time)
  336. return response.json(0, {'upload_url': response_url, 'datetime': datetime})
  337. def s3download(self, request_dict, response):
  338. pc_name = request_dict.get('pc_name', None)
  339. bundle_version = request_dict.get('bundle_version', None)
  340. pc_version = request_dict.get('pc_version', None)
  341. pc_test = request_dict.get('pc_test', None)
  342. param_flag = CommonService.get_param_flag(
  343. data=[pc_name, bundle_version, pc_version, pc_test])
  344. if param_flag is not True:
  345. return response.json(444)
  346. path = Pc_Info.objects.filter(pc_name=pc_name, bundle_version=bundle_version, pc_version=pc_version,
  347. pc_test=pc_test).values('download_link')
  348. if not path:
  349. return response.json(173)
  350. path = path[0]['download_link']
  351. aws_s3_guonei = boto3.client(
  352. 's3',
  353. aws_access_key_id=AWS_ACCESS_KEY_ID[0],
  354. aws_secret_access_key=AWS_SECRET_ACCESS_KEY[0],
  355. config=botocore.client.Config(signature_version='s3v4'),
  356. region_name='cn-northwest-1'
  357. )
  358. response_url = aws_s3_guonei.generate_presigned_url(
  359. ClientMethod='get_object',
  360. Params={
  361. 'Bucket': 'pc-package',
  362. 'Key': path
  363. },
  364. ExpiresIn=3600
  365. )
  366. res = {'path': path,
  367. 'response_url': response_url
  368. }
  369. return response.json(0, res)
  370. def s3downloadlog(self, request_dict, response):
  371. id = request_dict.get('id', None)
  372. pclog_qs = PctestlogModel.objects.filter(id=id)
  373. if not pclog_qs:
  374. return response.json(173)
  375. path = pclog_qs[0].content
  376. aws_s3_guonei = boto3.client(
  377. 's3',
  378. aws_access_key_id=AWS_ACCESS_KEY_ID[0],
  379. aws_secret_access_key=AWS_SECRET_ACCESS_KEY[0],
  380. config=botocore.client.Config(signature_version='s3v4'),
  381. region_name='cn-northwest-1'
  382. )
  383. response_url = aws_s3_guonei.generate_presigned_url(
  384. ClientMethod='get_object',
  385. Params={
  386. 'Bucket': 'pc-package',
  387. 'Key': path
  388. },
  389. ExpiresIn=3600
  390. )
  391. res = {'path': path,
  392. 'response_url': response_url
  393. }
  394. return response.json(0, res)
  395. def edit(self, request_dict, response):
  396. id = request_dict.get('id', None)
  397. explain = request_dict.get('explain', None)
  398. content = request_dict.get('content', None)
  399. authority = request_dict.get('authority', None)
  400. is_open = request_dict.get('is_open', None)
  401. param_flag = CommonService.get_param_flag(
  402. data=[id, explain, is_open])
  403. if param_flag is not True:
  404. return response.json(444)
  405. file = Pc_Info.objects.filter(id=id)
  406. if not file.exists():
  407. return response.json(173)
  408. file.update(explain=explain, content=content, authority=authority, is_open=is_open)
  409. return response.json(0)
  410. def s3delete(self, request_dict, response):
  411. global file
  412. token = request_dict.get('token', None)
  413. tko = TokenObject(token)
  414. response.lang = tko.lang
  415. if tko.code != 0:
  416. return response.json(tko.code)
  417. userID = tko.userID
  418. if not userID:
  419. return response.json(104)
  420. id = request_dict.get('id', None)
  421. package = request_dict.get('package', None)
  422. if id and package:
  423. return response.json(444)
  424. elif id and package is None:
  425. file = Pc_Info.objects.filter(id=id)
  426. elif package and id is None:
  427. try:
  428. package = int(package)
  429. file = Pc_Info.objects.filter(package=package)
  430. except Exception as e:
  431. return response.json(176)
  432. if not file.exists():
  433. return response.json(173)
  434. try:
  435. # 删除s3和数据库里的相应数据
  436. file_path = file[0].download_link
  437. aws_s3_guonei = boto3.client(
  438. 's3',
  439. aws_access_key_id=AWS_ACCESS_KEY_ID[0],
  440. aws_secret_access_key=AWS_SECRET_ACCESS_KEY[0],
  441. config=botocore.client.Config(signature_version='s3v4'),
  442. region_name='cn-northwest-1'
  443. )
  444. try:
  445. # 获取存储桶的对象,判断对象是否上传成功
  446. obj = aws_s3_guonei.get_object(Bucket='pc-package', Key=file_path)
  447. except Exception as e:
  448. print(e)
  449. file.delete()
  450. else:
  451. if obj:
  452. aws_s3_guonei.delete_object(Bucket='pc-package', Key=file_path)
  453. file.delete()
  454. except Exception as e:
  455. return response.json(176, repr(e))
  456. else:
  457. return response.json(0)
  458. def queryall(self, request_dict, response):
  459. token = request_dict.get('token', None)
  460. tko = TokenObject(token)
  461. response.lang = tko.lang
  462. if tko.code != 0:
  463. return response.json(tko.code)
  464. userID = tko.userID
  465. if not userID:
  466. return response.json(104)
  467. page = int(request_dict.get('page', None))
  468. line = int(request_dict.get('line', None))
  469. if page is None or line is None:
  470. return response.json(444, 'page,line')
  471. queryset = Pc_Info.objects.all()
  472. if queryset.exists():
  473. count = queryset.count()
  474. res = queryset[(page - 1) * line:page * line]
  475. send_json = CommonService.qs_to_dict(res)
  476. send_json['count'] = count
  477. return response.json(0, send_json)
  478. else:
  479. return response.json(173)
  480. def addandupload(self, request_dict, response, request):
  481. token = request_dict.get('token', None)
  482. tko = TokenObject(token)
  483. response.lang = tko.lang
  484. if tko.code != 0:
  485. return response.json(tko.code)
  486. userID = tko.userID
  487. if not userID:
  488. return response.json(104)
  489. pc_name = request_dict.get('pc_name', None)
  490. bundle_version = request_dict.get('bundle_version', None)
  491. pc_version = request_dict.get('pc_version', None)
  492. pc_test = request_dict.get('pc_test', None)
  493. lang = request_dict.get('lang', None)
  494. file_name = request.FILES.get('file_name', None)
  495. param_flag = CommonService.get_param_flag(
  496. data=[pc_name, bundle_version, pc_version, pc_test, lang, file_name])
  497. if param_flag is not True:
  498. return response.json(444)
  499. else:
  500. file = Pc_Info.objects.filter(pc_name=pc_name, bundle_version=bundle_version,
  501. pc_version=pc_version, pc_test=pc_test)
  502. if file:
  503. return response.json(174)
  504. try:
  505. # 安装包上传到服务器本地
  506. file_path = 'static/pc/' + pc_name
  507. if not os.path.exists(file_path):
  508. os.makedirs(os.path.join(BASE_DIR, file_path))
  509. a = os.path.splitext(str(file_name))[-1]
  510. if not a:
  511. return response.json(444, "文件无后缀")
  512. name = pc_version + '_' + bundle_version + '_' + pc_test + str(a)
  513. file_path = file_path + '/' + str(name)
  514. upload_path = os.path.join(BASE_DIR, file_path)
  515. print('upload_path:', upload_path)
  516. with open(upload_path, 'wb+') as destination:
  517. for chunk in file_name.chunks():
  518. destination.write(chunk)
  519. add_time = time.time()
  520. create_dict = {
  521. 'pc_name': pc_name,
  522. 'bundle_version': bundle_version,
  523. 'pc_version': pc_version,
  524. 'pc_test': pc_test,
  525. 'lang': lang,
  526. 'download_link': file_path,
  527. 'add_time': add_time,
  528. 'update_time': add_time
  529. }
  530. pc_Info = Pc_Info(**create_dict)
  531. pc_Info.save()
  532. except Exception:
  533. errorInfo = traceback.format_exc()
  534. print(errorInfo)
  535. return response.json(700, {'details': errorInfo})
  536. else:
  537. if pc_Info.id:
  538. res = {'pc_name': pc_Info.pc_name,
  539. 'bundle_version': pc_Info.bundle_version,
  540. 'pc_version': pc_Info.pc_version,
  541. 'pc_test': pc_Info.pc_test,
  542. 'download_link': pc_Info.download_link,
  543. 'lang': pc_Info.lang,
  544. 'add_time': pc_Info.add_time,
  545. 'update_time': pc_Info.update_time
  546. }
  547. return response.json(0, res)
  548. else:
  549. return response.json(500)
  550. def download(self, request_dict, response):
  551. pc_name = request_dict.get('pc_name', None)
  552. bundle_version = request_dict.get('bundle_version', None)
  553. pc_version = request_dict.get('pc_version', None)
  554. pc_test = request_dict.get('pc_test', None)
  555. param_flag = CommonService.get_param_flag(
  556. data=[pc_name, bundle_version, pc_version, pc_test])
  557. if param_flag is not True:
  558. return response.json(444)
  559. path = Pc_Info.objects.filter(pc_name=pc_name, bundle_version=bundle_version, pc_version=pc_version,
  560. pc_test=pc_test).values('download_link')
  561. if not path:
  562. return response.json(173)
  563. filepath = path[0]['download_link']
  564. fullPath = os.path.join(BASE_DIR, filepath)
  565. fullPath.replace('\\', '/')
  566. res = ResponseObject()
  567. print('fullPath:')
  568. print(fullPath)
  569. print(os.path.basename(fullPath))
  570. if fullPath:
  571. if os.path.isfile(fullPath):
  572. try:
  573. wrapper = FileWrapper(open(fullPath, 'rb'))
  574. response = HttpResponse(wrapper, content_type="application/octet-stream")
  575. response['Content-Length'] = os.path.getsize(fullPath)
  576. response['Content-Disposition'] = 'attachment; filename={}'.format(parse.quote_plus(os.path.basename(fullPath), encoding="utf-8"))
  577. response['Content-MD5'] = getMD5orSHA265(fullPath)
  578. # 校验文件md5值
  579. response['Content-SHA265'] = getMD5orSHA265(fullPath, 'SHA265')
  580. response['Content-CRC32'] = getMD5orSHA265(fullPath, 'CRC32')
  581. response['Content-Error'] = res.formal(0)
  582. return response
  583. except Exception as e:
  584. return res.json(906, repr(e))
  585. else:
  586. return res.json(907)
  587. else:
  588. return res.json(444, 'fullPath')
  589. def delete(self, request_dict, response):
  590. token = request_dict.get('token', None)
  591. tko = TokenObject(token)
  592. response.lang = tko.lang
  593. if tko.code != 0:
  594. return response.json(tko.code)
  595. userID = tko.userID
  596. if not userID:
  597. return response.json(104)
  598. id = request_dict.get('id', None)
  599. param_flag = CommonService.get_param_flag(data=[id])
  600. if param_flag is not True:
  601. return response.json(444)
  602. file = Pc_Info.objects.filter(id=id)
  603. if not file.exists():
  604. return response.json(173)
  605. try:
  606. # 删除文件,文件夹和数据库里的相应数据
  607. file_path = file[0].download_link
  608. file_path = os.path.join(BASE_DIR, file_path).replace('\\', '/')
  609. os.remove(file_path)
  610. # file_path = file_path.split("/")
  611. # file_path = [str(i) for i in file_path][:-1]
  612. # file_path = "/".join(file_path)
  613. # shutil.rmtree(file_path)
  614. file.delete()
  615. except Exception as e:
  616. return response.json(176, repr(e))
  617. else:
  618. return response.json(0)
  619. @staticmethod
  620. def check_version(request_dict, response):
  621. """
  622. AVSS检查版本
  623. @param request_dict: 请求数据
  624. @request_dict version: 当前版本
  625. @param response: 响应
  626. @return: response
  627. """
  628. version = request_dict.get('version', None)
  629. if version is None:
  630. return response.json(444)
  631. avss_version_qs = AVSSVersion.objects.filter().values('online_version')
  632. if not avss_version_qs.exists():
  633. return response.json(173)
  634. # 比较当前版本和线上版本
  635. online_version = avss_version_qs[0]['online_version']
  636. if version < online_version:
  637. return response.json(0, {'online_version': online_version})
  638. else:
  639. return response.json(0)
  640. def compareVersion(s1: str, s2: str) -> Union[str, int]:
  641. i, j = 0, 0
  642. while i < len(s1) or j < len(s2):
  643. k1 = i
  644. while k1 < len(s1) and s1[k1] != '.':
  645. k1 += 1
  646. k2 = j
  647. while k2 < len(s2) and s2[k2] != '.':
  648. k2 += 1
  649. a = int(s1[i:k1]) if i != k1 else 0
  650. b = int(s2[j:k2]) if j != k2 else 0
  651. if a > b:
  652. return s1
  653. if a < b:
  654. return s2
  655. i = k1 + 1
  656. j = k2 + 1
  657. return 0
  658. def getMD5orSHA265(fileName, encryptionType='MD5'):
  659. """
  660. :param filePath:
  661. :param encryptionType:
  662. :return:
  663. """
  664. if not os.path.isfile(fileName):
  665. return ''
  666. else:
  667. if encryptionType == 'MD5':
  668. encryption = hashlib.md5()
  669. elif encryptionType == 'SHA265':
  670. encryption = hashlib.sha256()
  671. elif encryptionType == 'CRC32':
  672. f = open(fileName, 'rb')
  673. chunk = f.read()
  674. return crc32(chunk)
  675. f = open(fileName, 'rb')
  676. block_size = 8192 # why is 8192 | 8192 is fast than 2048
  677. while True:
  678. chunk = f.read(block_size)
  679. if not chunk:
  680. break
  681. encryption.update(chunk)
  682. f.close()
  683. return encryption.hexdigest()