From ea11ee089b91bfb7d41dab467dc1128f74fbfa6e Mon Sep 17 00:00:00 2001 From: overflowerror Date: Wed, 28 May 2014 21:06:17 +0200 Subject: [PATCH] I have no idea what I'm doing. But: Trust me, I'm an engineer. --- server.py | 15 +++++++++++---- templates/index.html | 3 +++ 2 files changed, 14 insertions(+), 4 deletions(-) diff --git a/server.py b/server.py index 14977c9..0ec854f 100755 --- a/server.py +++ b/server.py @@ -31,7 +31,11 @@ def index(): if not loggedin(): return redirect(url_for('login')) diaries=db.query(Diary).filter_by(owner=session['id']).all() - return render_template('index.html',diaries=diaries) + if request.args.get('error') == '1': + error='you already used that name' + else: + error='' + return render_template('index.html',diaries=diaries, error=error) @app.route('/login',methods=['GET','POST']) def login(): @@ -49,12 +53,15 @@ def logout(): setuser('','') return redirect(url_for('index')) -@app.route('/diary/?new',methods=['POST']) +@app.route('/newdiary',methods=['POST']) def newdiary(): if not loggedin(): abort(403) - d=Diary(session['id'], request.form['name']) - db.add(d) + d=db.query(Diary).filter_by(name=request.form['name'], owner=session['id']).first() + if d: + return redirect(url_for('index', error=1)) + nd=Diary(session['id'], request.form['name']) + db.add(nd) db.commit() return redirect(url_for('diary', name=request.form['name'])) diff --git a/templates/index.html b/templates/index.html index 6c3870f..b8b0bf7 100644 --- a/templates/index.html +++ b/templates/index.html @@ -7,6 +7,9 @@ + {% if error != '' %} +
{{ error }}
+ {% endif %}

Logout