from flask import Flask, render_template, request, redirect, url_for import uuid import csv import os from wtforms import Form, StringField, SelectField, BooleanField, validators from wtforms.fields.html5 import EmailField app = Flask(__name__) app.jinja_env.trim_blocks = True app.jinja_env.lstrip_blocks = True datafile = "data.csv" def check_input(form, field): if field.data == 'default': raise validators.ValidationError('Bitte wähle eine Option aus') class QuestionForm(Form): computers = [('default', 'Bitte auswählen'), ('hal', 'HAL 9000 (2001: Odyssee im Weltraum)'), ('deepthought', 'Deep Thought (Hitchhiker\'s Guide)'), ('neuromancer', 'Neuromancer (Neuromancer)'), ('samantha', 'Samatha (Her)'), ('tars', 'TARS und CASE (Interstellar)')] computer = SelectField(u'Fiktionaler Lieblingscomputer', choices=computers, validators=[validators.InputRequired(), check_input]) statuses = [('default', 'Bitte auswählen'), ('200', '200 OK'), ('300', '300 Multiple Choices'), ('301', '301 Moved Permanently'), ('402', '402 Payment Required'), ('404', '404 (Sleep) Not Found'), ('408', '408 Request Timeout'), ('418', '418 I am a teapot'), ('450', '450 Blocked by Windows Parental Controls'), ('451', '451 Unavailable For Legal Reasons'), ('502', '502 Bad Gateway (Internetkurort)')] status = SelectField(u'Welcher HTTP-Statuscode trifft am häufigsten auf dich zu?', choices=statuses, validators=[validators.InputRequired(), check_input]) vegetables = [('default', 'Bitte auswählen'), ('karotte', '🥕'), ('broccoli','🥦'), ('aubergine', '🍆'), ('kartoffel', '🥔'), ('bretzel', '🥨'), ('tomate', '🍅'), ('chili', '🌶️')] vegetable = SelectField(u'Lieblingsgemüse', choices=vegetables, validators=[validators.InputRequired(), check_input]) spirit_animals = [('default', 'Bitte auswählen'), ('feuer', 'Feuerfuchs'), ('wasser', 'Wasserhahn'), ('erde', 'Erdferkel'), ('luft', 'Luftschlange')] spirit_animal = SelectField(u'Spirit Animal', choices=spirit_animals, validators=[validators.InputRequired(), check_input]) operating_systems = [('default', 'Bitte auswählen'), ('windows', 'Windows'), ('mac', 'Mac'), ('linux', 'Linux'), ('kaffee', 'Kaffee'), ('glados', 'GLaDOS')] operating_system = SelectField(u'Bevorzugtes Betriebssystem', choices=operating_systems, validators=[validators.InputRequired(), check_input]) check = BooleanField('Bitte nicht ankreuzen') email = EmailField('Email-Adresse') def create_used_id_list(): data = list() if not os.path.exists(datafile): open(datafile, "a").close() with open(datafile, "r") as f: reader = csv.reader(f) for line in reader: data.append(line[0]) return data class AnswerForm(Form): user_id = StringField('Deine Id', [validators.InputRequired(), validators.AnyOf(values=create_used_id_list(), message='Leider ist das keine gültige ID')]) @app.route('/') def index(): return render_template('index.html', title='Sortierhut') @app.route('/questions', methods=['GET', 'POST']) def questions(): form = QuestionForm(request.form) if request.method == 'POST' and form.validate(): with open(datafile, "a", newline='') as f: new_row = list() new_id = generate_id() new_row.append(new_id) for item in request.form: new_row.append(request.form[item]) writer = csv.writer(f) writer.writerow(new_row) # user = User() # user.name = form.username # spirit = request.form['spirit_animal'] # user.save() return redirect(url_for('result', user_id=new_id)) return render_template('questions.html', form=form) # @app.route('/form') # def form(): # id = generate_id() # return render_template('form.html', title='Sortierhut', text='Hallo, Fooboar', your_id=id) @app.route('/result') def result(user_id): return render_template('result.html', title='Sortierhut', hausname='Knuth', haustext='Mitglieder dieses Hauses mögen LaTeX-Anzüge und Eisbären.', id=user_id) @app.route('/admin') def admin(): return render_template('admin.html', title='Sortierhut', text='Hallo, Admin Fooboar') @app.route('/answers', methods=['GET', 'POST']) def answers(): form = AnswerForm(request.form) if request.method == 'POST' and form.validate(): ident = request.form['user_id'] return redirect(url_for('view_answers', user_id=ident)) return render_template('answers.html', form=form, title='Sortierhut', text='Hallo, Admin Fooboar') @app.route('/view_answers') def view_answers(user_id): user_row = list() with open(datafile, "r", newline='') as f: reader = csv.reader(f) for line in reader: if line[0] == user_id: user_row = line break return render_template('view.html', title='Sortierhut', user_id=user_id, user_row=user_row) @app.route('/solution') def solution(): return render_template('solution.html', title='Sortierhut', text='Hallo, Lösung') @app.route('/patrons') def patrons(): return render_template('patrons.html', title='Sortierhut') def generate_id(): data = create_used_id_list() # best loop ever while True: new_id = str(uuid.uuid1()).split("-")[0] if new_id in data: break else: return new_id