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 != '' %} +