diff --git a/app.py b/app.py
index b76a24d..f196ca9 100755
--- a/app.py
+++ b/app.py
@@ -1,18 +1,24 @@
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import os, time, subprocess
-from flask import Flask
+from flask import Flask, flash, request, redirect, url_for
# HTTP auth
from flask_httpauth import HTTPBasicAuth
from werkzeug.security import generate_password_hash, check_password_hash
-#HTTPS
-
+# FILE UPLOAD
+from werkzeug.utils import secure_filename
+UPLOAD_FOLDER = os.path.expanduser('~/Videos')
+ALLOWED_EXTENSIONS = {'avi', 'mkv', 'mp4'}
+#HTTPS
ASSETS_DIR = os.path.dirname(os.path.abspath(__file__))
-
# HTTP Serve
from waitress import serve
app = Flask(__name__)
+#app.secret_key = b'flafoudi'
+app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
+# Max upload size 100M ( see nginx default also )
+app.config['MAX_CONTENT_LENGTH'] = 100 * 1000 * 1000
auth = HTTPBasicAuth()
users = {
@@ -29,6 +35,14 @@ signal = 0
DEBUG = 0
+# Check file extension is allowed
+def allowed_file(filename):
+ # Check for dot in filename
+ if "." in filename:
+ # Split from right at first dot to find ext and allow files with "." in name
+ if filename.rsplit(".",1)[-1] in ALLOWED_EXTENSIONS:
+ return True
+
# Get Wifi signal level
def getRSSI():
signal = subprocess.run( local_bin + "get_rssi.sh", capture_output=True)
@@ -76,7 +90,35 @@ def shutdown():
print(stdout)
return "Shuting down..."
+# File upload
+
+@app.route('/upload', methods=['GET', 'POST'])
+@auth.login_required
+def upload_file():
+ if request.method == 'POST':
+ # check if the post request has the file part
+ if 'file' not in request.files:
+ return "No file part: " + str(request.files)
+ file = request.files['file']
+ # If the user does not select a file, the browser submits an
+ # empty file without a filename.
+ if file.filename == '':
+ return 'No selected file'
+ if file and allowed_file(file.filename):
+ filename = secure_filename(file.filename)
+ file.save(os.path.join(app.config['UPLOAD_FOLDER'], filename))
+ return "File saved in " + UPLOAD_FOLDER
+ return "OK"
+# return '''
+#
+# Upload new File
+# Upload new File
+#
+# '''
+
if __name__ == '__main__':
# app.run()
-# serve(app, host='0.0.0.0', port=5000)
serve(app, host='127.0.0.1', port=5000, url_scheme='https')