82 lines
3.1 KiB
Python
82 lines
3.1 KiB
Python
|
import aiohttp
|
|||
|
|
from django.http import HttpResponse, StreamingHttpResponse
|
|||
|
|
import os
|
|||
|
|
from datetime import datetime
|
|||
|
|
import requests
|
|||
|
|
from django.utils.decorators import method_decorator
|
|||
|
|
from django.views import View
|
|||
|
|
from django.views.decorators.csrf import csrf_exempt
|
|||
|
|
from urllib.parse import quote
|
|||
|
|
from django.http import HttpResponse
|
|||
|
|
from aiohttp import ClientSession
|
|||
|
|
from django.views.decorators.csrf import csrf_exempt
|
|||
|
|
|
|||
|
|
def acme_challenge(request, token):
|
|||
|
|
print(token)
|
|||
|
|
content = f'{token}.E4H2n2zqARGJeN2ulDerea5xYFCDXMy1y9v61hoQfms'
|
|||
|
|
return HttpResponse(content, content_type='text/plain')
|
|||
|
|
|
|||
|
|
|
|||
|
|
@csrf_exempt
|
|||
|
|
async def download_proxy(request):
|
|||
|
|
video_url = request.GET.get('url')
|
|||
|
|
if not video_url:
|
|||
|
|
return HttpResponse("URL is required", status=400)
|
|||
|
|
|
|||
|
|
headers = {
|
|||
|
|
'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/58.0.3029.110 Safari/537.3'
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
async with ClientSession() as session:
|
|||
|
|
try:
|
|||
|
|
async with session.get(video_url, headers=headers) as response:
|
|||
|
|
if response.status == 200:
|
|||
|
|
content = await response.read() # 读取内容到内存中，注意这不适用于大文件
|
|||
|
|
|
|||
|
|
# 创建一个 HttpResponse 对象并设置 Content-Type
|
|||
|
|
response = HttpResponse(content, content_type=response.headers['Content-Type'])
|
|||
|
|
|
|||
|
|
# 添加自定义响应头
|
|||
|
|
response['X-Content-Length'] = str(len(content))
|
|||
|
|
|
|||
|
|
return response
|
|||
|
|
else:
|
|||
|
|
return HttpResponse("Failed to download video", status=response.status)
|
|||
|
|
except aiohttp.ClientConnectionError:
|
|||
|
|
return HttpResponse("Connection closed", status=500)
|
|||
|
|
|
|||
|
|
|
|||
|
|
class ProxyView(View):
|
|||
|
|
@method_decorator(csrf_exempt)
|
|||
|
|
def dispatch(self, *args, **kwargs):
|
|||
|
|
return super().dispatch(*args, **kwargs)
|
|||
|
|
|
|||
|
|
def get(self, request, *args, **kwargs):
|
|||
|
|
# 获取目标URL
|
|||
|
|
target_url = request.GET.get('url')
|
|||
|
|
if not target_url:
|
|||
|
|
return HttpResponse("Missing 'url' parameter", status=400)
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
# 请求目标URL
|
|||
|
|
response = requests.get(target_url, stream=True)
|
|||
|
|
response.raise_for_status()
|
|||
|
|
except requests.exceptions.RequestException as e:
|
|||
|
|
return HttpResponse(f"Error fetching target URL: {e}", status=500)
|
|||
|
|
|
|||
|
|
# 获取文件类型并强制设置为 video/mp4
|
|||
|
|
content_type = 'video/mp4'
|
|||
|
|
|
|||
|
|
# 获取文件名并强制使用 .mp4 后缀
|
|||
|
|
filename = target_url.split("/")[-1].split('?')[0] # 移除查询参数
|
|||
|
|
if not filename.endswith('.mp4'):
|
|||
|
|
filename = f"{filename}.mp4"
|
|||
|
|
|
|||
|
|
# 设置Content-Disposition头部，确保文件在浏览器中直接播放而不是下载
|
|||
|
|
content_disposition = f'inline; filename="{filename}"'
|
|||
|
|
|
|||
|
|
# 创建响应对象并将内容传递给客户端
|
|||
|
|
proxy_response = HttpResponse(content=response.content, content_type=content_type)
|
|||
|
|
proxy_response['Content-Disposition'] = content_disposition
|
|||
|
|
|
|||
|
|
return proxy_response
|