def complete(id):
get_habit(id)
db = get_db()
db.execute('INSERT INTO completion (habit_id, timestamp)'
' VALUES (?, ?)', (id, to_timestamp(datetime.now())))
db.commit()
return redirect(url_for('habits.index'))
def completed_today(habit_id):
h = get_habit(id)
db = get_db()
ts = db.execute('SELECT timestamp FROM completions '
' WHERE habit_id = ?'
' ORDER BY timestamp DESC').fetchone()['timestamp']
return is_today(ts)
def load_logged_in_user():
user_id = session.get('user_id')
if user_id is None:
g.user = None
else:
g.user = get_db().execute('SELECT * FROM user WHERE id = ?',
(user_id, )).fetchone()
def get_habit(id):
habit = get_db().execute(
'SELECT h.id, name, user_id, u.username '
' FROM habit h JOIN user u ON h.user_id = u.id'
' WHERE h.id = ?', (id, )).fetchone()
if habit is None:
abort(404, "Habit id {0} doesn't exist".format(id))
if habit['user_id'] != g.user['id']:
abort(403)
return habit
def create():
if request.method == 'POST':
name = request.form['name']
error = None
if not name:
error = 'Name is required'
if error is not None:
flash(error)
else:
db = get_db()
db.execute('INSERT INTO habit (name, user_id) '
'VALUES (?, ?)', (name, g.user['id']))
db.commit()
return redirect(url_for('habits.index'))
return render_template('habits/create.html')
def update(id):
habit = get_habit(id)
if request.method == 'POST':
name = request.form['name']
error = None
if not name:
error = 'Name is required'
if error is not None:
flash(error)
else:
db = get_db()
db.execute('UPDATE habit SET name = ?' ' WHERE id = ?', (name, id))
db.commit()
return redirect(url_for('habits.index'))
return render_template('habits/update.html', habit=habit)
def login():
if request.method == 'POST':
username = request.form['username']
password = request.form['password']
db = get_db()
error = None
user = db.execute('SELECT * FROM user WHERE username = ?',
(username, )).fetchone()
if user is None:
error = 'Incorrect username'
elif not check_password_hash(user['password'], password):
error = 'Incorrect password'
if error is None:
session.clear()
session['user_id'] = user['id']
return redirect(url_for('index'))
flash(error)
return render_template('auth/login.html')
def register():
if request.method == 'POST':
username = request.form['username']
password = request.form['password']
db = get_db()
error = None
if not username:
error = 'Username required'
elif not password:
error = 'Password required'
elif db.execute('SELECT id FROM user WHERE username = ?',
(username, )).fetchone() is not None:
error = 'User {} is already registered'.format(username)
if error is None:
db.execute('INSERT INTO user (username, password) VALUES (?, ?)',
(username, generate_password_hash(password)))
db.commit()
return redirect(url_for('auth.login'))
flash(error)
return render_template('auth/register.html')
def index():
db = get_db()
habits = db.execute('SELECT id, name '
'FROM habit WHERE user_id = ?',
(g.user['id'], )).fetchall()
completions = [
db.execute(
'SELECT * FROM completion WHERE '
' habit_id = ?'
' ORDER BY timestamp DESC', (h['id'], )).fetchall() for h in habits
]
periods = [completed_periods(10, c) for c in completions]
to_complete = [not p[0] for p in periods]
# Do the conversion logic here to turn timestamps into completion data -
# probably will need a separate module in which I do these conversions based
# on the current date, then pass to the templating engine etc
return render_template('habits/index.html',
data=zip(habits, periods, to_complete))
def delete(id):
get_habit(id)
db = get_db()
db.execute('DELETE FROM habit WHERE id = ?', (id, ))
db.commit()
return redirect(url_for('habits.index'))