summaryrefslogtreecommitdiff
path: root/yaas.py
blob: b98d97aafb7c2207bc1f71ecd9cf4a44ae7f6f02 (plain)
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
import os
import youtube_dl

from flask import Flask, redirect, request, render_template

app = Flask(__name__)

@app.route('/')
def index():
    vids = os.listdir('static/vids')
    return render_template('index.html', vids=vids)


@app.route('/upload', methods=['POST'])
def upload():
    url = request.form['url']
    if url:
        ydl_opts = {
            'outtmpl': 'static/vids/%(id)s.%(ext)s'
        }
        with youtube_dl.YoutubeDL(ydl_opts) as ydl:
            info_dict = ydl.extract_info(url, download=False)
            ydl.download([url])

    return redirect('/')

if __name__ == '__main__':
    app.run()