I have no idea what I'm doing. But: Trust me, I'm an engineer.

This commit is contained in:
overflowerror 2014-05-28 21:06:17 +02:00
parent 9b79ba2719
commit ea11ee089b
2 changed files with 14 additions and 4 deletions

View file

@ -31,7 +31,11 @@ def index():
if not loggedin(): if not loggedin():
return redirect(url_for('login')) return redirect(url_for('login'))
diaries=db.query(Diary).filter_by(owner=session['id']).all() 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']) @app.route('/login',methods=['GET','POST'])
def login(): def login():
@ -49,12 +53,15 @@ def logout():
setuser('','') setuser('','')
return redirect(url_for('index')) return redirect(url_for('index'))
@app.route('/diary/?new',methods=['POST']) @app.route('/newdiary',methods=['POST'])
def newdiary(): def newdiary():
if not loggedin(): if not loggedin():
abort(403) abort(403)
d=Diary(session['id'], request.form['name']) d=db.query(Diary).filter_by(name=request.form['name'], owner=session['id']).first()
db.add(d) if d:
return redirect(url_for('index', error=1))
nd=Diary(session['id'], request.form['name'])
db.add(nd)
db.commit() db.commit()
return redirect(url_for('diary', name=request.form['name'])) return redirect(url_for('diary', name=request.form['name']))

View file

@ -7,6 +7,9 @@
<script src="{{ url_for('static',filename='script.js') }}"></script> <script src="{{ url_for('static',filename='script.js') }}"></script>
</head> </head>
<body> <body>
{% if error != '' %}
<div style="border: red 1px solid; color: red;">{{ error }}</div>
{% endif %}
<p> <p>
<a href="{{ url_for('logout') }}">Logout</a> <a href="{{ url_for('logout') }}">Logout</a>
<form method="post" action="{{ url_for('newdiary')}}"> <form method="post" action="{{ url_for('newdiary')}}">