75 lines
No EOL
2.5 KiB
Python
75 lines
No EOL
2.5 KiB
Python
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) |