| from flask import Flask, request, send_file, render_template_string, abort |
| import yt_dlp |
| import os |
| import tempfile |
|
|
| app = Flask(__name__) |
|
|
| |
| DEMO_HTML = """ |
| <!DOCTYPE html> |
| <html> |
| <head><title>yt-dlp Flask Demo</title></head> |
| <body> |
| <h1>YouTube動画ダウンロードデモ</h1> |
| <form action="/download" method="get"> |
| URL: <input type="text" name="url" size="60" required><br><br> |
| フォーマット (例: mp4, webm): <input type="text" name="format" placeholder="mp4"><br><br> |
| 画質 (例: 720p, 1080p, best): <input type="text" name="quality" placeholder="best"><br><br> |
| <button type="submit">ダウンロード</button> |
| </form> |
| </body> |
| </html> |
| """ |
|
|
| @app.route('/') |
| def index(): |
| return render_template_string(DEMO_HTML) |
|
|
| @app.route('/download') |
| def download_video(): |
| url = request.args.get('url') |
| format_ext = request.args.get('format') |
| quality = request.args.get('quality', 'best') |
|
|
| if not url: |
| return abort(400, 'URLパラメーターが必要です') |
|
|
| |
| temp_dir = tempfile.mkdtemp() |
| |
| output_template = os.path.join(temp_dir, '%(title)s.%(ext)s') |
|
|
| |
| ydl_opts = { |
| 'outtmpl': output_template, |
| 'format': 'bestvideo[ext={}][height<={}] + bestaudio/best'.format(format_ext if format_ext else '', quality if quality != 'best' else '9999'), |
| 'merge_output_format': format_ext if format_ext else 'mp4', |
| 'noplaylist': True, |
| 'quiet': True, |
| 'no_warnings': True, |
| } |
|
|
| |
| if quality == 'best': |
| if format_ext: |
| ydl_opts['format'] = f'bestvideo[ext={format_ext}]+bestaudio/best' |
| else: |
| ydl_opts['format'] = 'bestvideo+bestaudio/best' |
|
|
| |
| try: |
| with yt_dlp.YoutubeDL(ydl_opts) as ydl: |
| info = ydl.extract_info(url) |
| |
| filename = ydl.prepare_filename(info) |
| if not os.path.exists(filename): |
| return abort(500, '動画ファイルが見つかりませんでした') |
| except Exception as e: |
| return abort(500, f'動画ダウンロード中にエラーが発生しました: {str(e)}') |
|
|
| |
| return send_file(filename, as_attachment=True) |
|
|
| if __name__ == '__main__': |
| app.run(debug=True, port=7860, host="0.0.0.0") |
|
|
