Basic POC working

1 files changed, 22 insertions, 3 deletions

diff --git a/yaas.py b/yaas.py
index 7fad1a7..b98d97a 100644
--- a/yaas.py
+++ b/yaas.py
@@ -1,9 +1,28 @@
-from flask import Flask
+import os
+import youtube_dl
+
+from flask import Flask, redirect, request, render_template
+
 app = Flask(__name__)
 
 @app.route('/')
-def hello():
-    return 'Hello World!'
+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()