You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
169 lines
6.9 KiB
169 lines
6.9 KiB
#!/usr/bin/env python
|
|
# -*- coding:utf-8 -*-
|
|
# project: 3月
|
|
# author: liuyu
|
|
# date: 2020/3/6
|
|
from rest_framework.views import APIView
|
|
from api.utils.response import BaseResponse
|
|
from rest_framework.response import Response
|
|
from fir_ser import settings
|
|
from api.utils.TokenManager import DownloadToken
|
|
from api.utils.app.randomstrings import make_random_uuid
|
|
from api.utils.app.apputils import make_resigned
|
|
from api.utils.storage.storage import Storage,LocalStorage
|
|
import os,time
|
|
from django.core.cache import cache
|
|
from rest_framework_extensions.cache.decorators import cache_response
|
|
|
|
from api.utils.serializer import AppsSerializer
|
|
from api.models import Apps,AppReleaseInfo,UserInfo
|
|
from django.db.models import F
|
|
from django.http import FileResponse
|
|
class DownloadView(APIView):
|
|
'''
|
|
文件下载接口,适用于本地存储和所有plist文件下载
|
|
'''
|
|
# authentication_classes = [ExpiringTokenAuthentication, ]
|
|
# parser_classes = (MultiPartParser,)
|
|
|
|
def get(self,request,filename):
|
|
res = BaseResponse()
|
|
downtoken = request.query_params.get("token", None)
|
|
ftype = request.query_params.get("ftype",None)
|
|
if not downtoken:
|
|
res.code=1004
|
|
res.msg="缺失token"
|
|
return Response(res.dict)
|
|
|
|
dtoken = DownloadToken()
|
|
if dtoken.verify_token(downtoken,filename):
|
|
if not ftype:
|
|
file_path = os.path.join(settings.MEDIA_ROOT, filename)
|
|
try:
|
|
response = FileResponse(open(file_path, 'rb'))
|
|
except Exception as e:
|
|
print(e)
|
|
response = FileResponse()
|
|
response['content_type'] = "application/octet-stream"
|
|
response['Content-Disposition'] = 'attachment; filename=' + filename
|
|
return response
|
|
else:
|
|
if ftype == 'plist':
|
|
release_obj = AppReleaseInfo.objects.filter(release_id=filename.split('.')[0]).first()
|
|
if release_obj:
|
|
storage = Storage(release_obj.app_id.user_id)
|
|
bundle_id = release_obj.app_id.bundle_id
|
|
app_version = release_obj.app_version
|
|
name = release_obj.app_id.name
|
|
ios_plist_bytes = make_resigned(storage.get_download_url(filename),storage.get_download_url(release_obj.icon_url),bundle_id,app_version,name)
|
|
response = FileResponse(ios_plist_bytes)
|
|
response['content_type'] = "application/x-plist"
|
|
response['Content-Disposition'] = 'attachment; filename=' + make_random_uuid()
|
|
return response
|
|
|
|
res.code=1004
|
|
res.msg="token校验失败"
|
|
return Response(res.dict)
|
|
|
|
|
|
|
|
class ShortDownloadView(APIView):
|
|
'''
|
|
根据下载短链接，获取应用信息
|
|
'''
|
|
|
|
@cache_response(timeout=300-60, cache="default",key_func='calculate_cache_key',cache_errors=False)
|
|
def get(self,request,short):
|
|
res = BaseResponse()
|
|
release_id = request.query_params.get("release_id", None)
|
|
app_obj = Apps.objects.filter(short=short).first()
|
|
if not app_obj:
|
|
res.code=1003
|
|
res.msg="该应用不存在"
|
|
return Response(res.dict)
|
|
|
|
app_serializer = AppsSerializer(app_obj,context={"release_id":release_id,"storage":Storage(app_obj.user_id)})
|
|
res.data = app_serializer.data
|
|
return Response(res.dict)
|
|
|
|
#key的设置
|
|
def calculate_cache_key(self, view_instance, view_method,
|
|
request, args, kwargs):
|
|
id = "download"
|
|
rtn = '_'.join([
|
|
id,
|
|
"short",
|
|
kwargs.get("short",None)
|
|
])
|
|
# print( request.META)
|
|
return rtn
|
|
|
|
class InstallView(APIView):
|
|
'''
|
|
安装操作，前端通过token 认证，认证成功之后，返回 下载连接，并且 让下载次数加一
|
|
'''
|
|
|
|
def get(self,request,app_id):
|
|
res = BaseResponse()
|
|
downtoken = request.query_params.get("token", None)
|
|
short = request.query_params.get("short",None)
|
|
release_id = request.query_params.get("release_id",None)
|
|
isdownload = request.query_params.get("isdownload",None)
|
|
|
|
if not downtoken or not short or not release_id:
|
|
res.code=1004
|
|
res.msg="参数丢失"
|
|
return Response(res.dict)
|
|
|
|
dtoken = DownloadToken()
|
|
if dtoken.verify_token(downtoken,release_id):
|
|
# app_obj = Apps.objects.filter(app_id=app_id).values("pk",'user_id','type','short').first()
|
|
app_obj = self.get_app_instance_by_cache(app_id,900)
|
|
if app_obj:
|
|
|
|
Apps.objects.filter(app_id=app_id).update(count_hits=F('count_hits') + 1)
|
|
UserInfo.objects.filter(pk=app_obj.get("user_id")).update(all_download_times=F('all_download_times') + 1)
|
|
|
|
# release_obj = AppReleaseInfo.objects.filter(app_id=app_obj.get('pk')).values("release_id")
|
|
# if release_obj:
|
|
if app_obj.get("type") == 0:
|
|
apptype = '.apk'
|
|
download_url = self.get_download_url_by_cache(app_obj,release_id + apptype,600)
|
|
else:
|
|
apptype = '.ipa'
|
|
if isdownload :
|
|
download_url = self.get_download_url_by_cache(app_obj,release_id + apptype, 600)
|
|
else:
|
|
download_url = self.get_download_url_by_cache(app_obj,release_id + apptype,600,isdownload)
|
|
|
|
res.data={"download_url":download_url}
|
|
return Response(res.dict)
|
|
else:
|
|
res.code = 1004
|
|
res.msg = "token校验失败"
|
|
return Response(res.dict)
|
|
|
|
res.code = 1006
|
|
res.msg = "该应用不存在"
|
|
return Response(res.dict)
|
|
|
|
def get_download_url_by_cache(self,app_obj,filename,limit,isdownload=False):
|
|
now = time.time()
|
|
download_val = cache.get("%s_%s" % ('download_url', filename))
|
|
if download_val:
|
|
if download_val.get("time") > now - 60:
|
|
return download_val.get("download_url")
|
|
else:
|
|
if isdownload:
|
|
local_storage = LocalStorage('localhost', False)
|
|
return local_storage.get_download_url(filename, limit, 'plist')
|
|
user_obj = UserInfo.objects.filter(pk=app_obj.get("user_id")).first()
|
|
storage = Storage(user_obj)
|
|
return storage.get_download_url(filename, limit)
|
|
|
|
def get_app_instance_by_cache(self,app_id,limit):
|
|
app_obj_cache = cache.get("%s_%s" % ('app_instance', app_id))
|
|
if not app_obj_cache:
|
|
app_obj_cache = Apps.objects.filter(app_id=app_id).values("pk",'user_id','type').first()
|
|
cache.set("%s_%s" % ('app_instance', app_id),app_obj_cache,limit)
|
|
return app_obj_cache
|
|
|