#!/usr/bin/env python3 # -*- coding: utf-8 -*- """ @Copyright (C) ansjer cop Video Technology Co.,Ltd.All rights reserved. @AUTHOR: ASJRD018 @NAME: AnsjerFormal @software: PyCharm @DATE: 2018/12/5 9:30 @Version: python3.6 @MODIFY DECORD:ansjer dev @file: cloudstorage.py @Contact: chanjunkai@163.com """ import base64 import json import os import time import urllib from urllib.parse import quote, parse_qs, unquote import apns2 import boto3 import jpush import oss2 import paypalrestsdk import threading import calendar import datetime import logging import sys from aliyunsdkcore import client from aliyunsdksts.request.v20150401 import AssumeRoleRequest from boto3.session import Session from django.http import JsonResponse, HttpResponseRedirect, HttpResponse from django.db import transaction from django.views.generic.base import View import jwt from Object.ETkObject import ETkObject from pyfcm import FCMNotification from Ansjer.config import OSS_STS_ACCESS_KEY, OSS_STS_ACCESS_SECRET, OSS_ROLE_ARN, SERVER_DOMAIN, PAYPAL_CRD, \ SERVER_DOMAIN_SSL, AWS_ACCESS_KEY_ID, AWS_SECRET_ACCESS_KEY, AWS_ARN, APNS_MODE, APNS_CONFIG, BASE_DIR, \ JPUSH_CONFIG, FCM_CONFIG, OAUTH_ACCESS_TOKEN_SECRET from Controller.CheckUserData import DataValid from Model.models import Device_Info, Order_Model, Store_Meal, VodHlsModel, OssCrdModel, UID_Bucket, StsCrdModel, \ ExperienceContextModel, Pay_Type, CDKcontextModel, Device_User, SysMassModel, SysMsgModel, UidPushModel, \ Unused_Uid_Meal, UIDMainUser, UserModel, PromotionRuleModel, VideoPlaybackTimeModel, CloudLogModel, CouponModel, \ AiStoreMeal, AiService, UidSetModel from Object.AWS.S3Email import S3Email from Object.AliPayObject import AliPayObject from Object.AliSmsObject import AliSmsObject from Object.RedisObject import RedisObject from Object.ResponseObject import ResponseObject from Object.TokenObject import TokenObject from Object.UidTokenObject import UidTokenObject from Service.CommonService import CommonService from Object.m3u8generate import PlaylistGenerator from Object.WechatPayObject import WechatPayObject from django.db.models import Q, F, Count from Controller.PaymentCycle import Paypal from decimal import Decimal from Ansjer.config import SERVER_TYPE from Service.ModelService import ModelService # AI服务 class AiView(View): def get(self, request, *args, **kwargs): request.encoding = 'utf-8' operation = kwargs.get('operation') return self.validation(request.GET, request, operation) def post(self, request, *args, **kwargs): request.encoding = 'utf-8' operation = kwargs.get('operation') return self.validation(request.POST, request, operation) def validation(self, request_dict, request, operation): response = ResponseObject() if operation is None: return response.json(444, 'error path') elif operation == 'identification': # ai识别 return self.do_ai_identification(request_dict, response) else: token = request_dict.get('token', None) # 设备主键uid tko = TokenObject(token) response.lang = tko.lang if tko.code != 0: return response.json(tko.code) userID = tko.userID if operation == 'createpayorder': # 创建支付订单 ip = CommonService.get_ip_address(request) return self.do_create_pay_order(request_dict, userID, ip, response) elif operation == 'changeaistatus': # 修改云存状态，传送两个url，即getsignsts接口和storeplaylist接口 return self.do_change_ai_status(userID, request_dict, response) elif operation == 'commoditylist': # 修改云存状态，传送两个url，即getsignsts接口和storeplaylist接口 return self.do_commodity_list(userID, request_dict, response) else: return response.json(414) def do_change_ai_status(self, userID, request_dict, response): token_val = request_dict.get('token_val', None) appBundleId = request_dict.get('appBundleId', None) app_type = request_dict.get('app_type', None) push_type = request_dict.get('push_type', None) status = request_dict.get('status', None) m_code = request_dict.get('m_code', None) uid = request_dict.get('uid', None) # 设备语言 lang = request_dict.get('lang', 'en') tz = request_dict.get('tz', '0') # 消息提醒功能新增 # 如果传空上来，就默认为0 if tz == '': tz = 0 else: tz = tz.replace("GMT", "") detect_group = request_dict.get('detect_group', None) interval = request_dict.get('interval', None) if not status: return response.json(444, 'status') # 关闭推送 if not all([appBundleId, app_type, token_val, uid, m_code]): return response.json(444, 'appBundleId,app_type,token_val,uid,m_code') # 判断推送类型对应key是否存在 print('push_type:', push_type) if push_type == '0': if appBundleId not in APNS_CONFIG.keys(): return response.json(904) elif push_type == '1': if appBundleId not in FCM_CONFIG.keys(): return response.json(904) elif push_type == '2': if appBundleId not in JPUSH_CONFIG.keys(): return response.json(904) else: return response.json(173) hasAiService = AiService.objects.filter(uid=uid,use_status=1) if not hasAiService.exists(): return response.json(10053) nowTime = int(time.time()) endTime = hasAiService.values('endTime')[0]['endTime'] if nowTime >endTime: return response.json(10054) dvqs = Device_Info.objects.filter(userID_id=userID, UID=uid) status = int(status) nowTime = int(time.time()) if dvqs.exists(): # 修改状态 # dvqs.update(NotificationMode=status) uid_set_qs = UidSetModel.objects.filter(uid=uid) # uid配置信息是否存在 if uid_set_qs.exists(): uid_set_id = uid_set_qs[0].id qs_data = { 'detect_status': status, 'updTime': nowTime, } if interval: qs_data['detect_interval'] = int(interval) if detect_group: qs_data['detect_group'] = detect_group uid_set_qs.update(**qs_data) else: qs_data = { 'uid': uid, 'addTime': nowTime, 'updTime': nowTime, 'detect_status': status, } if interval: qs_data['detect_interval'] = int(interval) if detect_group: qs_data['detect_group'] = detect_group # 添加设备配置 uid_set_qs = UidSetModel.objects.create(**qs_data) uid_set_id = uid_set_qs.id if status == 0: hasAiService.update(**qs_data) # UidPushModel.objects.filter(uid_set__uid=uid).delete() # 状态为0的时候删除redis缓存数据 # self.do_delete_redis(uid) return response.json(0) elif status == 1: hasAiService.update(**qs_data) uid_push_qs = UidPushModel.objects.filter(userID_id=userID, m_code=m_code, uid_set__uid=uid) if uid_push_qs.exists(): uid_push_update_dict = { 'appBundleId': appBundleId, 'app_type': app_type, 'push_type': push_type, 'token_val': token_val, 'updTime': nowTime, 'lang': lang, 'tz': tz } uid_push_qs.update(**uid_push_update_dict) else: # uid_set_id = uid_set_qs[0].id uid_push_create_dict = { 'uid_set_id': uid_set_id, 'userID_id': userID, 'appBundleId': appBundleId, 'app_type': app_type, 'push_type': push_type, 'token_val': token_val, 'm_code': m_code, 'addTime': nowTime, 'updTime': nowTime, 'lang': lang, 'tz': tz } # 绑定设备推送 UidPushModel.objects.create(**uid_push_create_dict) # if interval: # self.do_delete_redis(uid, int(interval)) # else: # self.do_delete_redis(uid) # utko = UidTokenObject() # # right # utko.generate(data={'uid': uid}) etkObj = ETkObject(etk='') etk = etkObj.encrypt(uid) #只返回一个接口就行 # detectUrl = "{DETECT_PUSH_DOMAIN}AiService/push?etk={etk}&endTime={endTime}". \ # format(etk=etk, DETECT_PUSH_DOMAIN=SERVER_DOMAIN_SSL, endTime=endTime) aiIdentificationUrl = "{DETECT_PUSH_DOMAIN}AiService/identification".format(DETECT_PUSH_DOMAIN=SERVER_DOMAIN_SSL) return response.json(0, {'aiIdentificationUrl':aiIdentificationUrl, 'endTime':endTime, 'etk':etk}) else: return response.json(14) def do_commodity_list(self, userID, request_dict, response): # 查询套餐列表 uid = request_dict.get('uid', None) lang = request_dict.get('lang', 'en') nowTime = int(time.time()) # DVR/NVR设备暂不返回云存套餐列表 device_info_qs = Device_Info.objects.filter(Q(UID=uid), Q(Type__lte=4) | Q(Type=10001)) if device_info_qs.exists(): return response.json(0) qs = AiStoreMeal.objects qs = qs.filter(is_show=1) #过滤隐藏套餐 qs = qs.annotate(ai_meal_id=F('id')) qs = qs.values("ai_meal_id", "title", "content", "price", "effective_day", "currency", "virtual_price", "symbol", "pay_type") if qs.exists(): res = list(qs) for key, val in enumerate(res): pay_types = Pay_Type.objects.filter(aistoremeal=res[key]['ai_meal_id']).values("id", "payment") res[key]['pay_type'] = list(pay_types) result = { 'meals': res, } return response.json(0, result) else: return response.json(0) def do_create_pay_order(self, request_dict, userID, ip, response): uid = request_dict.get('uid', None) channel = request_dict.get('channel', None) pay_type = int(request_dict.get('pay_type', 1)) ai_meal_id = request_dict.get('ai_meal_id', None) lang = request_dict.get('lang', 'en') if not uid or not channel or not pay_type or not ai_meal_id: return response.json(444) # dv_qs = Device_Info.objects.filter(userID_id=userID, UID=uid, isShare=False, isExist=1).values( # 'vodPrimaryUserID', # 'vodPrimaryMaster') # if not dv_qs.exists(): # return response.json(12) # dvq = Device_Info.objects.filter(UID=uid) # dvq = dvq.filter(~Q(vodPrimaryUserID='')).values('vodPrimaryUserID') # if dvq.exists(): # if dvq[0]['vodPrimaryUserID'] != userID: # return response.json(10033) nowTime = int(time.time()) smqs = AiStoreMeal.objects.filter(id=ai_meal_id, pay_type=pay_type, is_show=1). \ values('currency', 'price', 'content', 'effective_day', 'title') if not smqs.exists(): return response.json(173) currency = smqs[0]['currency'] price = smqs[0]['price'] content = smqs[0]['content'] day = smqs[0]['effective_day'] orderID = CommonService.createOrderID() price = float(price) price = round(price, 2) if pay_type == 1: #正常扣款 cal_url = "{SERVER_DOMAIN_SSL}web/paid2/fail.html".format(SERVER_DOMAIN_SSL=SERVER_DOMAIN_SSL) if lang != 'cn': cal_url = "{SERVER_DOMAIN_SSL}web/paid2/en_fail.html".format(SERVER_DOMAIN_SSL=SERVER_DOMAIN_SSL) call_sub_url = "{SERVER_DOMAIN_SSL}cloudstorage/dopaypalcallback?orderID={orderID}&lang={lang}". \ format(SERVER_DOMAIN_SSL=SERVER_DOMAIN_SSL, orderID=orderID, lang=lang) # call_sub_url = "http://binbin.uicp.vip/cloudstorage/dopaypalcallback?orderID={orderID}".format( # SERVER_DOMAIN_SSL=SERVER_DOMAIN_SSL, orderID=orderID) Order_Model.objects.create(orderID=orderID, UID=uid, channel=channel, userID_id=userID, desc=content, payType=pay_type, payTime=nowTime, price=price, currency=currency, addTime=nowTime, updTime=nowTime, pay_url='', commodity_code='', commodity_type=1, ai_rank_id=ai_meal_id, rank_id=1, order_type=1) has = AiService.objects.filter(uid=uid, channel=channel, use_status=1) if has.exists(): use_status = 0 else: use_status = 1 # return response.json(0) AiService.objects.create(orders_id=orderID, uid=uid, channel=channel, detect_status=0, endTime=nowTime+(day*86400), addTime=nowTime, updTime=nowTime, use_status=use_status, detect_group='1') return response.json(0, {"orderID": orderID}) return response.json(10, 'generate_order_false') def do_ai_identification(self, request_dict): etk = request_dict.get('etk', None) if etk: eto = ETkObject(etk) uid = eto.uid file_post_one = request_dict.get('file_one', None) file_post = file_post_one.replace(' ', '+') # file_decode = base64.b64decode(file_post) file_post_two = request_dict.get('file_two', None) file_post_two = file_post_two.replace(' ', '+') # file_decode_two = base64.b64decode(file_post_two) now_time = int(time.time()) dir_path = os.path.join(BASE_DIR, 'static/ai/') if not os.path.exists(dir_path): os.makedirs(dir_path) file_path_one = dir_path + 'one' + str(now_time) + '.jpeg' file_path_two = dir_path + 'two' + str(now_time) + '.jpeg' file_list = ['one', 'two'] for index in file_list: file_path = dir_path + index + str(now_time) + '.jpeg' with open(file_path, 'wb') as f: # file_byte = file_post.encode('utf-8') f.write(file_decode) return HttpResponse("seccess") file2 = request.FILES.get('file_one', None) file3 = request.FILES.get('file_two', None) file4 = request.FILES.get('file_three', None) print('--------------------------file') print(file) print('===========================post_file') print(file_post) post_file_list = [file, file2, file3, file4] file_list = [] for index in range(len(post_file_list)): if post_file_list[index]: file_list.append(post_file_list[index]) del post_file_list if len(file_list) > 1: merge = [] now_time = int(time.time()) dir_path = os.path.join(BASE_DIR, 'static/', str(now_time)) if not os.path.exists(dir_path): os.makedirs(dir_path) for item in file_list: if hasattr(item, 'name'): file_path = os.path.join(dir_path, item.name) with open(file_path, 'wb') as f: for c in item.chunks(): f.write(c) merge.append(file_path) image_size = 500 # 每张小图片的大小 image_colnum = 2 # 合并成一张图后，一行有几个小图 MergePic.merge_images(dir_path, image_size, image_colnum) # return HttpResponse(dir_path + '.jpg') photo = open(dir_path + '.jpg', 'rb') else: photo = file_list[0] else: return HttpResponse("fail")