Ben 11 kuukautta sitten
vanhempi
commit
1225d1a148
1 muutettua tiedostoa jossa 8 lisäystä ja 3 poistoa
  1. 8 3
      app.py

+ 8 - 3
app.py

@@ -3,7 +3,7 @@ import logging
 import boto3
 import requests
 import yt_dlp
-from flask import Flask, request
+from flask import Flask, request, Response
 
 logging.basicConfig(format='%(asctime)s [%(levelname)s] %(filename)s:%(lineno)d %(message)s',
                     datefmt='%Y-%m-%d %H:%M:%S', level=logging.INFO)
@@ -35,8 +35,13 @@ def p():
             for item in info.get("formats", []):
                 if item.get("format_id") == format_id:
                     url = item.get("url")
-                    response = requests.get(url)
-                    return response.content, response.status_code, response.headers.items()
+                    response = requests.get(url, stream=True)
+
+                    def generate():
+                        for chunk in response.iter_content(chunk_size=1024):
+                            yield chunk
+
+                    return Response(generate(), status=response.status_code, headers=dict(response.headers))
     return {"status": 0}