Spaces:
Runtime error
Runtime error
File size: 2,229 Bytes
6309782 65a4590 6309782 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 |
import os
import time
from flask import Flask, jsonify, request, send_from_directory, send_file
from beta.api.task_manager import TaskManager # Assuming the TaskManager class is in task_manager.py
from mainLogic.main import Main
from mainLogic.startup.checkup import CheckState
from mainLogic.utils.glv import Global
from mainLogic.utils.basicUtils import BasicUtils
app = Flask(__name__)
task_manager = TaskManager()
OUT_DIR = Global.api_webdl_directory
try:
if not os.path.exists(OUT_DIR): os.makedirs(OUT_DIR)
except Exception as e:
Global.errprint(f"Could not create output directory {OUT_DIR}")
Global.sprint(f"Defaulting to './' ")
Global.errprint(f"Error: {e}")
OUT_DIR = './'
def setup_directory():
pass
def download_pw_video(task_id, id, name, out_dir, progress_callback):
print(f"Downloading {name} with id {id} to {out_dir}")
ch = CheckState()
prefs = ch.checkup(Global.EXECUTABLES, directory="./", verbose=True)
nm3 = prefs['nm3']
ffmpeg = prefs['ffmpeg']
mp4d = prefs['mp4decrypt']
verbose = True
Main(id=id, name=f"{name}-{task_id}", directory=out_dir, tmpDir="/*auto*/", nm3Path=nm3, ffmpeg=ffmpeg, mp4d=mp4d, verbose=verbose, progress_callback=progress_callback).process()
@app.route('/create_task', methods=['POST'])
def create_task():
data = request.json
id = data.get('id')
name = data.get('name')
if not id or not name:
return jsonify({'error': 'id and name are required'}), 400
task_id = task_manager.create_task(download_pw_video, id, name, OUT_DIR)
return jsonify({'task_id': task_id}), 202
@app.route('/progress/<task_id>', methods=['GET'])
def get_progress(task_id):
progress = task_manager.get_progress(task_id)
return jsonify(progress), 200
@app.route('/get-file/<task_id>/<name>', methods=['GET'])
def get_file(task_id,name):
file = BasicUtils.abspath(f"{OUT_DIR}/{name}-{task_id}.mp4")
if not os.path.exists(file):
return jsonify({'error': 'file not found'}), 404
return send_file( f"{file}", as_attachment=True), 200
@app.route('/', methods=['GET'])
def index():
return jsonify({'message': 'Hello, World!'}), 200
if __name__ == '__main__':
app.run(debug=True)
|