from zlib import crc32 from django.core import serializers from wsgiref.util import FileWrapper import traceback, hashlib from Ansjer.config import SERVER_DOMAIN from django.views.decorators.csrf import csrf_exempt from django.views.generic import TemplateView from django.utils.decorators import method_decorator from Model.models import Equipment_Version from Model.models import Device_User from Service.ModelService import ModelService from Service.CommonService import CommonService import time, os, simplejson as json from Object.ResponseObject import ResponseObject from Object.TokenObject import TokenObject from django.http import HttpResponse from Ansjer.config import BASE_DIR def addNewEquipmentVersion(deviceContent, token, response): """ :param deviceContent: :return: """ if token is not None: tko = TokenObject(token) # 设置语言 response.lang = tko.lang if tko.code == 0: userID = tko.userID own_permission = ModelService.check_perm(userID=userID, permID=220) if own_permission is True: try: print(deviceContent) print(type(deviceContent)) deviceData = json.loads(deviceContent) except Exception as e: return response.json(444, repr(e)) else: version = deviceData.get('version', None) if version is not None: filePath = deviceData.get('filePath', None) if filePath is not None: try: deviceData['filePath'] = ','.join(filePath) equipmentVersion = Equipment_Version( eid=CommonService.getUserID(getUser=False, setOTAID=True), **deviceData) equipmentVersion.save() except Exception as e: errorInfo = traceback.format_exc() print('添加设备错误: %s ' % errorInfo) return response.json(444, repr(e)) else: res = CommonService.qs_to_dict([equipmentVersion]) return response.json(0, res) else: return response.json(806) else: return response.json(806) else: return response.json(404) else: return response.json(309) def downloadUrl(fileType, fileCode, fileVersion, fileName): fullPath = os.path.join(BASE_DIR, "static/Upgrade").replace('\\', '/') if fileType == 'IPC': Path = '/'.join((fullPath, 'IPC', fileCode, fileVersion, fileName)).replace('\\', '/') elif fileType == 'DVR': Path = '/'.join((fullPath, 'DVR', fileCode, fileVersion, fileName)).replace('\\', '/') elif fileType == 'NVR': Path = '/'.join((fullPath, 'NVR', fileCode, fileVersion, fileName)).replace('\\', '/') elif fileType == 'XVR': Path = '/'.join((fullPath, 'IPC', fileCode, fileVersion, fileName)).replace('\\', '/') else: if fileType == 'CHM': Path = fileName else: Path = '/'.join((fullPath, 'Other', fileName)).replace('\\', '/') if os.path.isfile(Path): try: JSON = json.dumps( { "result_code": 0, "reason": 'Success', "result": {}, "error_code": 0, }, ensure_ascii=False ) wrapper = FileWrapper(open(Path, 'rb')) response = HttpResponse(wrapper, content_type="application/octet-stream") response['Content-Length'] = os.path.getsize(Path) response['Content-Disposition'] = 'attachment; filename=%s' % os.path.basename(Path) response['Content-MD5'] = getMD5orSHA265(Path) response['Content-SHA265'] = getMD5orSHA265(Path, 'SHA265') response['Content-CRC32'] = getMD5orSHA265(Path, 'CRC32') response['Content-Error'] = JSON return response except Exception as e: res = ResponseObject() errorJSON = res.formal(906, {'details': repr(e)}) response = HttpResponse(errorJSON, content_type='text/plain', charset='utf-8') response['Content-Error'] = errorJSON return response else: res = ResponseObject() errorJSON = res.formal(907) response = HttpResponse(errorJSON, content_type='text/plain', charset='utf-8') response['Content-Error'] = errorJSON return response def getMD5orSHA265(fileName, encryptionType='MD5'): """ :param filePath: :param encryptionType: :return: """ if not os.path.isfile(fileName): return '' else: if encryptionType == 'MD5': encryption = hashlib.md5() elif encryptionType == 'SHA265': encryption = hashlib.sha256() elif encryptionType == 'CRC32': f = open(fileName, 'rb') chunk = f.read() return crc32(chunk) f = open(fileName, 'rb') block_size = 8192 # why is 8192 | 8192 is fast than 2048 while True: chunk = f.read(block_size) if not chunk: break encryption.update(chunk) f.close() return encryption.hexdigest() @csrf_exempt def downloadUpdataFileUrl(request): response = ResponseObject() if request.method == 'GET': request_dict = request.GET elif request.method == 'POST': request_dict = request.POST else: errorJSON = response.formal(444) response = HttpResponse(errorJSON, content_type='text/plain', charset='utf-8') response['Content-Error'] = errorJSON return response fileType = request_dict.get('fileType', None) fileCode = request_dict.get('fileCode', None) fileVersion = request_dict.get('fileVersion', None) fileName = request_dict.get('fileName', None) if fileType != None and fileCode != None and fileVersion != \ None and fileName != None: return downloadUrl(fileType, fileCode, fileVersion, fileName, response) else: errorJSON = response.formal(444) response = HttpResponse(errorJSON, content_type='text/plain', charset='utf-8') response['Content-Error'] = errorJSON return response ''' http://192.168.136.40:8077/OTA/getEquipmentVersion?31AX162001A ''' @csrf_exempt def getEquipmentVersionInterface(request): response = ResponseObject() if request.method == "POST": request.encoding = 'utf-8' request_dict = request.POST elif request.method == "GET": request.encoding = 'gb2312' request_dict = request.GET else: return response.json(444) code = request_dict.get('code', None) ov = request_dict.get('ov', None) if not code: return response.json(444, 'code') eqs = Equipment_Version.objects.filter(code=code, status=1, lang='en').order_by('-data_joined') if not eqs.exists(): return response.json(900) if ov is not None: # 判断大小 nv = eqs[0].softwareVersion if ov > nv: return response.json(900) equipment = eqs[0] return response.json(0, {'softwareVersion': equipment.softwareVersion}) @csrf_exempt def getUpdataFileUrlInterface(request): response = ResponseObject() if request.method == "POST": request.encoding = 'utf-8' code = request.POST.get('code', None) elif request.method == "GET": request.encoding = 'utf-8' code = request.GET.get('code', None) else: return response.json(444) if not code: return response.json(444, 'code') eq = Equipment_Version.objects.filter(code=code, status=1, lang='en').order_by('-data_joined') # 判断是否有版本存在 if not eq.exists(): return response.json(902) file_path = eq[0].filePath urls = [] if file_path: if file_path.find('static/Upgrade/') != -1: path = file_path.replace('static/Upgrade/', '').replace('\\', '/') url = SERVER_DOMAIN + 'OTA/downloads/' + path + '?time=' + str(time.time()) urls.append(url) if file_path.find('static/otapack') != -1: url = SERVER_DOMAIN + 'OTA/downloadsPack/' + file_path + '?time=' + str(time.time()) urls.append(url) if len(urls) > 0: res = { "urlCount": len(urls), "url": urls, "fileSize": eq[0].fileSize, "Description": eq[0].Description, } return response.json(0, res) else: return response.json(901) @csrf_exempt def downloadUpdataFileUrlInterface(request, fileType, fileName, *callback_args, **callback_kwargs): res = ResponseObject() if fileType is not None and fileName is not None: fullPath = os.path.join(BASE_DIR, "static/Upgrade/").replace('\\', '/') if fileType == 'IPC': fullPath += 'IPC/' + fileName elif fileType == 'DVR': fullPath += 'DVR/' + fileName elif fileType == 'NVR': fullPath += 'NVR/' + fileName elif fileType == 'XVR': fullPath += 'XVR/' + fileName elif fileType == 'User': fullPath = os.path.join(BASE_DIR, "static/").replace('\\', '/') fullPath += 'User/' + fileName elif fileType == 'ADCloud': fullPath = os.path.join(BASE_DIR, "static/APK/").replace('\\', '/') fullPath += 'ADCloud/' + fileName elif fileType == 'ACCloud': fullPath = os.path.join(BASE_DIR, "static/APK/").replace('\\', '/') fullPath += 'ACCloud/' + fileName else: fullPath += 'Other/' + fileName print(fullPath) if os.path.isfile(fullPath): try: JSON = res.formal(0) if fileType != 'User': wrapper = FileWrapper(open(fullPath, 'rb')) response = HttpResponse(wrapper, content_type="application/octet-stream") response['Content-Length'] = os.path.getsize(fullPath) response['Content-Disposition'] = 'attachment; filename=%s' % os.path.basename(fullPath) response['Content-MD5'] = getMD5orSHA265(fullPath) # 校验文件md5值 response['Content-SHA265'] = getMD5orSHA265(fullPath, 'SHA265') response['Content-CRC32'] = getMD5orSHA265(fullPath, 'CRC32') response['Content-Error'] = JSON return response else: Imagedata = open(fullPath, 'rb').read() response = HttpResponse(Imagedata, content_type="image/jpeg") return response except Exception as e: errorJSON = res.formal(906) response = HttpResponse(errorJSON, content_type='text/plain', charset='utf-8') response['Content-Error'] = errorJSON return response else: errorJSON = res.formal(907) response = HttpResponse(errorJSON, content_type='text/plain', charset='utf-8') response['Content-Error'] = errorJSON return response else: errorJSON = res.formal(444) response = HttpResponse(errorJSON, content_type='text/plain', charset='utf-8') response['Content-Error'] = errorJSON return response class getUploadFiletoDirView(TemplateView): @method_decorator(csrf_exempt) def dispatch(self, *args, **kwargs): return super(getUploadFiletoDirView, self).dispatch(*args, **kwargs) def post(self, request, *args, **kwargs): request.encoding = 'utf-8' request_dict = request.POST fileName = request.FILES.get('fileName', None) return self.ValidationError(request_dict, fileName) def get(self, request, *args, **kwargs): request.encoding = 'gb2312' request_dict = request.GET fileName = request.FILES.get('fileName', None) return self.ValidationError(request_dict, fileName) def ValidationError(self, request_dict, fileName): response = ResponseObject() token = request_dict.get('token', None) fileType = request_dict.get('fileType', None) fileCode = request_dict.get('fileCode', None) fileVersion = request_dict.get('fileVersion', None) if not fileName or not fileType: return response.json(444, 'fileName,fileType') tko = TokenObject(token) response.lang = tko.lang if tko.code != 0: return response.json(tko.code) userID = tko.userID if not userID: return response.json(104) own_perm = ModelService.check_perm(userID=userID, permID=210) if not own_perm: return response.json(404) if fileVersion != None and fileCode != None: return self.getUploadFiletoDir(userID, fileType, fileName, response, fileCode, fileVersion=fileVersion) else: return self.getUploadFiletoDir(userID, fileType, fileName, response) def getUploadFiletoDir(self, userID, fileType, fileName, response, *args, **kwargs): """ 将上传的文件写入服务器对应目录下 :param Type: equipment type :param fileName: File name of upgrade file. :return: filePath """ User = Device_User.objects.filter(userID=userID) if not User: return response.json(104) own_perm = ModelService.check_perm(userID, permID=210) if own_perm is not True: return response.json(404) updataFile = fileName.name updataFile = updataFile.replace(' ', '+') versionIndex = updataFile.find('.', updataFile.find('.', updataFile.find('.') + 1) + 1) codeIndex = versionIndex + 12 if codeIndex != -1 and versionIndex != -1: fileVersion = len(updataFile[1: versionIndex]) > 0 and updataFile[1: versionIndex] or None fileCode = len(updataFile[versionIndex + 1: codeIndex]) > 0 and \ updataFile[versionIndex + 1: codeIndex] or None if fileCode is not None and fileVersion is not None: return self.getDir(fileType, fileName, fileCode, fileVersion, response) else: fileCode = kwargs.get('fileCode', None) fileVersion = kwargs.get('fileVersion', None) print(fileCode, fileVersion) if fileCode != None and fileVersion != None: return self.getDir(fileType, fileName, fileCode, fileVersion) else: return response.json(903) else: fileCode = kwargs.get('fileCode', None) fileVersion = kwargs.get('fileVersion', None) if fileCode != None and fileVersion != None: return self.getDir(fileType, fileName, fileCode, fileVersion, response) else: return response.json(903) def getDir(self, fileType, fileName, fileCode, fileVersion, response): try: if fileCode != None and fileVersion != None: path = '/'.join((BASE_DIR, 'static/Upgrade', fileType, fileCode, fileVersion)).replace('\\', '/') + '/' else: if fileType != 'IPC' and fileType != 'DVR' and fileType != 'NVR' and fileType != 'XVR': path = '/'.join((BASE_DIR, "static/Upgrade", 'Other')).replace('\\', '/') + '/' if not os.path.exists(path): os.makedirs(path) file_name = path + str(fileName) if os.path.exists(file_name): os.remove(file_name) destination = open(file_name, 'wb+') for chunk in fileName.chunks(): destination.write(chunk) destination.close() else: file_name = path + str(fileName) if os.path.exists(file_name): os.remove(file_name) destination = open(file_name, 'wb+') for chunk in fileName.chunks(): destination.write(chunk) destination.close() except Exception as e: errorInfo = traceback.format_exc() print('上传文件错误: %s' % errorInfo) return response.json(700, {'details': repr(e)}) else: index = file_name.find('static/') filePath = file_name[index:] return response.json(0, {'filePath': filePath}) @csrf_exempt def addNewEquipmentVersionInterface(request): request.encoding = 'utf-8' response = ResponseObject() if request.method == "POST": request_dict = request.POST elif request.method == "GET": request_dict = request.GET else: return response.json(444) deviceContent = request_dict.get('content', None).encode('utf-8') token = request_dict.get('token', None) deviceContent = str(deviceContent, encoding='utf-8') deviceContent = deviceContent.replace(' ', ' ').replace('\'', '\"') if deviceContent is not None: return addNewEquipmentVersion(deviceContent, token, response) else: return response.json(800) def showAllEquipmentVersion(userID, response): userValid = Device_User.objects.filter(userID=userID).order_by('-data_joined') if not userValid.exists(): return response.json(104) own_perm = ModelService.check_perm(userID=userID, permID=240) if not own_perm: return response.json(404) qs = Equipment_Version.objects.all() res = CommonService.qs_to_dict(qs) return response.json(0, res) # 检测ota更新包 @csrf_exempt def getNewVerInterface(request): response = ResponseObject() if request.method == "POST": request.encoding = 'utf-8' request_dict = request.POST elif request.method == "GET": request.encoding = 'utf-8' request_dict = request.GET code = request_dict.get('code', None) token = request_dict.get('token', None) lang = request_dict.get('lang', None) if not code: return response.json(444,'code') tko = TokenObject(token) response.lang = tko.lang if tko.code != 0: return response.json(tko.code) if lang == 'zh-Hans': equipmentValid = Equipment_Version.objects.filter(code=code, status=1, lang=lang).order_by( '-data_joined') else: equipmentValid = Equipment_Version.objects.filter(code=code, status=1, lang='en').order_by( '-data_joined') if equipmentValid.exists(): ver = equipmentValid[0].softwareVersion equipment = equipmentValid[0] file_path = equipment.filePath # 检测ip进行替换 client_ip = CommonService.get_ip_address(request) url = SERVER_DOMAIN + 'OTA/downloadsPack/' + file_path + '?time=' + str(time.time()) return response.json(0, { 'ver': ver, 'url': url, "Description": equipment.Description, }) else: return response.json(902) # ota包上传 class uploadOTAInterfaceView(TemplateView): @method_decorator(csrf_exempt) def dispatch(self, *args, **kwargs): return super(uploadOTAInterfaceView, self).dispatch(*args, **kwargs) def post(self, request, *args, **kwargs): request.encoding = 'utf-8' token = request.POST.get('token', None) fileName = request.FILES.get('fileName', None) fileType = request.POST.get('fileType', None) language = request.POST.get('language', None) return self.validate(token, fileName, fileType, language) def get(self, request, *args, **kwargs): request.encoding = 'gb2312' token = request.GET.get('token', None) fileName = request.FILES.get('fileName', None) fileType = request.GET.get('fileType', None) language = request.GET.get('language', None) return self.validate(token, fileName, fileType, language) def validate(self, token, fileName, fileType, language): response = ResponseObject() if not fileName or not fileType: return response.json(444,'fileName,fileType') tko = TokenObject(token) response.lang = tko.lang if tko.code != 0: return response.json(tko.code) userID = tko.userID if not userID: return response.json(104) own_permission = ModelService.check_perm(userID=userID, permID=210) if own_permission is not True: return response.json(404) return self.upload_ota_file(fileName, response, fileType, language) def upload_ota_file(self, fileName, response, fileType, language): try: if not language: language = 0 path = '/'.join((BASE_DIR, 'static/otapack', fileType, str(language))).replace('\\', '/') + '/' if not os.path.exists(path): os.makedirs(path) file_name = path + str(fileName) if os.path.exists(file_name): os.remove(file_name) destination = open(file_name, 'wb+') for chunk in fileName.chunks(): destination.write(chunk) destination.close() else: file_name = path + str(fileName) if os.path.exists(file_name): os.remove(file_name) destination = open(file_name, 'wb+') for chunk in fileName.chunks(): destination.write(chunk) destination.close() except Exception as e: errorInfo = traceback.format_exc() print('上传文件错误: %s' % errorInfo) return response.json(700, {'details': repr(e)}) else: index = file_name.find('static/') filePath = file_name[index:] return response.json(0, {'filePath': filePath}) # ota包下载 # class downloadOTAInterfaceView(TemplateView): @csrf_exempt def downloadOTAInterface(request, fullPath, *callback_args, **callback_kwargs): res = ResponseObject() print('fullPath:') print(fullPath) if fullPath: if os.path.isfile(fullPath): try: wrapper = FileWrapper(open(fullPath, 'rb')) response = HttpResponse(wrapper, content_type="application/octet-stream") response['Content-Length'] = os.path.getsize(fullPath) response['Content-Disposition'] = 'attachment; filename=%s' % os.path.basename(fullPath) response['Content-MD5'] = getMD5orSHA265(fullPath) # 校验文件md5值 response['Content-SHA265'] = getMD5orSHA265(fullPath, 'SHA265') response['Content-CRC32'] = getMD5orSHA265(fullPath, 'CRC32') response['Content-Error'] = res.formal(0) return response except Exception as e: return res.json(906, repr(e)) else: return res.json(907) else: return res.json(444,'fullPath')