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