diff options
Diffstat (limited to 'Web/Server/Server.py')
-rw-r--r-- | Web/Server/Server.py | 82 |
1 files changed, 82 insertions, 0 deletions
diff --git a/Web/Server/Server.py b/Web/Server/Server.py new file mode 100644 index 0000000..a936b96 --- /dev/null +++ b/Web/Server/Server.py @@ -0,0 +1,82 @@ +import os +try: #python3 + from urllib.request import urlopen +except: #python2 + from urllib2 import urlopen +from flask import Flask, flash, request, redirect, render_template, Response +from werkzeug.utils import secure_filename +import requests +import json + +app = Flask(__name__) +app.secret_key = "ski u mah" +app.config['MAX_CONTENT_LENGTH'] = 16 * 1024 * 1024 + +ALLOWED_EXTENSIONS = set(['py']) + +PI_URL = 'http://localhost:8000' + +# UNCOMMENT IF RUNNING PI.PY ON PI +#PI_URL = 'http://192.168.1.10:8000' + +RESULTS_DESTINATION = "Results/results.csv" + +def allowed_file(filename): + return '.' in filename and filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS + +@app.route('/') +def home(): + return render_template('index.html') + +@app.route('/upload', methods=['POST']) +def upload_file(): + if request.method == 'POST': + # Check if the post request has the file part + if 'file' not in request.files: + flash('No file part') + return redirect(request.url) + + # Grab the file + file = request.files['file'] + if file.filename == '': + flash('No file selected for uploading') + return redirect(request.url) + + # Send the file content as a post to the PI + if file and allowed_file(file.filename): + + dictToSend = {'filename':file.filename, 'file_content':file.read()} + file.close + print('Running test') + response = requests.post(PI_URL + '/tests/endpoint', json=dictToSend) + + results_filename = json.loads(response.text)[u'results_filename'] + results_content = json.loads(response.text)[u'results_content'] + flash('Results file:' + results_filename) + flash('Response from server:' + results_content) + + results = open(RESULTS_DESTINATION, "w") + results.write(results_content) + results.close() + + return render_template('index.html', results = 'True') + else: + flash('Allowed file types are .py') + return redirect(request.url) + +@app.route('/results', methods=['GET']) +def download(): + # Grab content from results file + with open(RESULTS_DESTINATION, 'r') as results: + results_content = results.read() + results.close() + + # Put content as a download file + return Response( + results_content, + mimetype="text/csv", + headers={"Content-disposition": + "attachment; filename=results.csv"}) + +if __name__ == "__main__": + app.run(host="localhost", port=5000)
\ No newline at end of file |