jimmy/events (#6)

Co-authored-by: Jimmy Vargo <james@ayo.tokyo>
Reviewed-on: ayo/website#6

events/app.py

@@ -0,0 +1,75 @@
+from flask import Flask, request, render_template
+import json
+from pathlib import Path
+
+from form_content import FormContent
+
+app = Flask(__name__)
+save_path = Path('data.txt')
+all_keys = set(FormContent.times | FormContent.topics | FormContent.languages) | { 'name', 'email', 'topic-other' }
+
+def get_data_stats() -> dict:
+    data = {}
+    summed = {
+        key: {
+            'sum': 0,
+            'total': 0,
+            'percent': 0
+        } for key in all_keys
+    }
+    summed['topic-other'] = set()
+
+    if save_path.exists():
+        for raw_line in save_path.read_text().split('\n'):
+            if raw_line:
+                line = json.loads(raw_line)
+                data[line['email']] = line
+        for line in data.values():
+            for key, value in dict.items(line):
+                if key == 'topic-other' and value is not None:
+                    summed['topic-other'] = summed['topic-other'] | { value }
+                if type(value) is bool:
+                    sum = summed[key]['sum'] if key in summed else 0
+                    sum = sum + 1 if value is True else sum
+                    total = summed[key]['total'] + 1 if key in summed else 1
+                    summed[key] = {
+                        'sum': sum,
+                        'total': total,
+                        'percent': f'{((sum / total) * 100):.1f}'
+                    }
+    summed['topic-other'] = list(summed['topic-other'])
+    return summed
+
+stats_cache = (0, get_data_stats())
+
+@app.route('/')
+def form():
+    return render_template('form.html', times=FormContent.times, topics=FormContent.topics, languages=FormContent.languages)
+
+@app.route('/submit', methods=['POST'])
+def submit():
+    if save_path.exists() and save_path.stat().st_size > 1_000_000:
+        return { 'error': 'Database full.' }, 500
+    if len(request.data) > 1000:
+        return { 'error': 'Request too long.' }, 400
+
+    data = request.get_json()
+
+    if set(data) != all_keys:
+        return { 'error': 'Invalid request format.' }, 400
+
+    with save_path.open(mode='a') as file:
+        file.write(f'{json.dumps(data)}\n')
+    return {}, 200
+
+@app.route('/stats')
+def stats():
+    global stats_cache
+    if save_path.exists() and save_path.stat().st_mtime > stats_cache[0]:
+        stats_cache = (save_path.stat().st_mtime, get_data_stats())
+    return render_template('stats.html', times=FormContent.times, topics=FormContent.topics,
+                           languages=FormContent.languages, data=stats_cache[1])
+
+
+if __name__ == '__main__':
+    app.run(host='0.0.0.0', port=9700, debug=False)