/
app.py
80 lines (65 loc) · 2.42 KB
/
app.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
from flask import Flask, render_template, request, redirect, session, url_for, flash
import model
app = Flask(__name__)
app.secret_key = "shhhhthisisasecret"
@app.route("/")
def index():
if session.get('user_id'):
username = session["username"]
return redirect(url_for('view_user', username=username))
else:
return render_template("index.html")
@app.route("/", methods=["POST"])
def process_login():
model.connect_to_db()
username = request.form.get("username")
password = request.form.get("password")
user_id = model.authenticate(username, password)
if user_id:
flash("User authenticated")
session['user_id'] = user_id
session['username'] = username
else:
flash("Password incorrect, there may be a ferret stampede in progress!!!")
return redirect(url_for("index"))
@app.route("/register")
def register():
model.connect_to_db()
if session.get("user_id"):
return redirect(url_for('view_user', username=session.get("username")))
else:
return render_template("register.html")
@app.route('/logout')
def logout():
session.clear()
return redirect(url_for('index'))
@app.route('/user/<username>')
def view_user(username):
model.connect_to_db()
user_id = model.get_user_by_name(username)
wall_posts = model.get_wall_posts_for_user(user_id)
return render_template("wall_posts.html", username = username,
wall_posts = wall_posts,
user_id = session.get('user_id'))
@app.route('/user/<username>', methods=["POST"])
def post_to_wall(username):
model.connect_to_db()
wall_owner = model.get_user_by_name(username)
current_user = session.get('user_id')
wall_content = request.form.get("wall_content")
model.add_wall_post(current_user, wall_owner, wall_content)
return redirect(url_for('view_user', username=username))
@app.route('/create_account', methods=["POST"])
def create_account():
model.connect_to_db()
username = request.form.get('username')
password = request.form.get('password')
if model.get_user_by_name(username):
flash("That user already exists!")
return redirect(url_for("register"))
else:
model.add_new_user(username, password)
flash("New user created!")
return redirect(url_for("index"))
if __name__ == "__main__":
app.run(debug = True)