-
Notifications
You must be signed in to change notification settings - Fork 4
/
main.py
180 lines (144 loc) · 5.83 KB
/
main.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
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
import os
from datetime import date
from cs50 import SQL
from flask import Flask, flash, redirect, render_template, request, session
from flask_session import Session
from tempfile import mkdtemp
from werkzeug.exceptions import default_exceptions, HTTPException, InternalServerError
from werkzeug.security import check_password_hash, generate_password_hash
from threading import Thread
from helpers import apology, login_required
# Configure application
app = Flask(__name__)
# Ensure templates are auto-reloaded
app.config["TEMPLATES_AUTO_RELOAD"] = True
# Ensure responses aren't cached
@app.after_request
def after_request(response):
response.headers["Cache-Control"] = "no-cache, no-store, must-revalidate"
response.headers["Expires"] = 0
response.headers["Pragma"] = "no-cache"
return response
# Configure session to use filesystem (instead of signed cookies)
app.config["SESSION_FILE_DIR"] = mkdtemp()
app.config["SESSION_PERMANENT"] = False
app.config["SESSION_TYPE"] = "filesystem"
Session(app)
# Configure CS50 Library to use SQLite database
db = SQL("sqlite:///todolist.db")
@app.route("/")
def index():
if not session:
return render_template("index.html")
else:
return redirect("/main")
@app.route("/main", methods=["GET", "POST"])
@login_required
def main():
if request.method == "GET":
duetasks = db.execute("UPDATE tasks SET type = ? WHERE date('now') > date(dt)", "Overdue")
tasks = db.execute("SELECT * FROM tasks WHERE user_id = ? AND type = ?" ,
session["user_id"], "Due");
return render_template("main.html", tasks=tasks)
else:
id = request.form.get("id")
print(id)
db.execute("UPDATE tasks SET type=? WHERE user_id=? AND id = ?", "Done", session["user_id"], id)
db.execute("UPDATE tasks SET cdt=? WHERE user_id=? AND id = ?", date.today(), session["user_id"], id)
return redirect("/main")
@app.route("/addtask", methods=["GET", "POST"])
@login_required
def addT():
if request.method == "GET":
return render_template("add.html")
else:
deadline = request.form.get("deadline")
topic = request.form.get("topic")
priority = request.form.get("priority")
if not deadline or not topic or not priority or isinstance(deadline, int):
flash("Fill all the details!")
return redirect("/addtask")
y, m, d = [int(x) for x in deadline.split('-')]
if date(y,m,d) < date.today():
flash("Your task is past the deadline!")
return redirect("/addtask")
if priority == "HIGH":
priority = 3
elif priority == "MEDIUM":
priority = 2
else:
priority = 1
db.execute("INSERT INTO tasks (user_id, priority, topic, type, dt) VALUES(?, ?, ?, ?, ?)", session["user_id"], priority, topic, "Due", deadline)
return redirect("/main")
@app.route("/history")
@login_required
def history():
tasks = db.execute("SELECT * FROM tasks WHERE user_id = ?", session["user_id"])
return render_template("history.html", tasks=tasks)
@app.route("/login", methods=["GET", "POST"])
def login():
"""Log user in"""
# Forget any user_id
session.clear()
# User reached route via POST (as by submitting a form via POST)
if request.method == "POST":
# Ensure username was submitted
if not request.form.get("username"):
return apology("must provide username", 403)
# Ensure password was submitted
elif not request.form.get("password"):
return apology("must provide password", 403)
# Query database for username
rows = db.execute("SELECT * FROM users WHERE username = ?", request.form.get("username"))
# Ensure username exists and password is correct
if len(rows) != 1 or not check_password_hash(rows[0]["hash"], request.form.get("password")):
return apology("invalid Username and/or Password", 403)
# Remember which user has logged in
session["user_id"] = rows[0]["id"]
# Redirect user to home page
return redirect("/main")
# User reached route via GET (as by clicking a link or via redirect)
else:
return render_template("login.html")
@app.route("/register", methods=["GET", "POST"])
def register():
"""Register user"""
if request.method == "GET":
return render_template("register.html")
else:
# Store name in name and password in db and check if previously exist
name = request.form.get("username")
prev = db.execute("SELECT EXISTS(SELECT * FROM users WHERE username = ?) ", name)
prev = [i for i in prev[0].items()]
if prev[0][1] or not name:
return apology("Username invalid or not available")
# Check if passwords match
password = request.form.get("password")
confirm = request.form.get("confirmation")
if not password or password != confirm:
return apology("Password invalid")
# Store the details in the database
else:
hash = generate_password_hash(password, method='pbkdf2:sha256', salt_length=8)
db.execute("INSERT INTO users (username, hash) VALUES(?, ?)", name, hash)
flash("User had been registered successfully!")
return redirect("/login")
@app.route("/logout")
def logout():
"""Log user out"""
# Forget any user_id
session.clear()
# Redirect user to login form
return redirect("/")
def errorhandler(e):
"""Handle error"""
if not isinstance(e, HTTPException):
e = InternalServerError()
return apology(e.name, e.code)
# Listen for errors
for code in default_exceptions:
app.errorhandler(code)(errorhandler)
if __name__ == '__main__':
# Run the Flask app
t = Thread(target=app.run(host='0.0.0.0', port=8080, debug=True))
t.start()