forked from petewarden/open-speech-recording
-
Notifications
You must be signed in to change notification settings - Fork 10
/
Copy pathmain.py
61 lines (51 loc) · 1.66 KB
/
main.py
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
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
from flask import Flask
from flask import abort
from flask import make_response
from flask import redirect
from flask import render_template
from flask import request
from flask import session
from werkzeug.utils import secure_filename
import os
import uuid
app = Flask(__name__)
@app.route("/")
def welcome():
session_id = request.cookies.get('session_id')
if session_id:
return render_template("record.html")
else:
return render_template("welcome.html")
@app.route("/start")
def start():
response = make_response(redirect('/'))
session_id = uuid.uuid4().hex
response.set_cookie('session_id', session_id)
return response
@app.route('/upload', methods=['POST'])
def upload():
session_id = request.cookies.get('session_id')
if not session_id:
make_response('No session', 400)
word = request.args.get('word')
audio_data = request.data
filename = word + '_' + session_id + uuid.uuid4().hex + '.ogg'
secure_name = secure_filename(filename)
with open(secure_name, 'wb') as f:
f.write(audio_data)
return make_response('All good')
# CSRF protection, see http://flask.pocoo.org/snippets/3/.
@app.before_request
def csrf_protect():
if request.method == "POST":
token = session['_csrf_token']
if not token or token != request.args.get('_csrf_token'):
abort(403)
def generate_csrf_token():
if '_csrf_token' not in session:
session['_csrf_token'] = uuid.uuid4().hex
return session['_csrf_token']
app.jinja_env.globals['csrf_token'] = generate_csrf_token
app.secret_key = "DEPLOYING_LOCALLY_ONLY_PLACEHOLDER"
if __name__ == "__main__":
app.run(debug=True)