diff --git a/flash_cards.py b/flash_cards.py index e2ad4da..f2197c9 100644 --- a/flash_cards.py +++ b/flash_cards.py @@ -362,6 +362,9 @@ def init_tag(): db.execute('INSERT INTO tags (tagName) VALUES (?)', ["code"]) db.commit() + db.execute('INSERT INTO tags (tagName) VALUES (?)', + ["bookmark"]) + db.commit() @app.route('/show') def show(): @@ -391,5 +394,15 @@ def getTag(tag_id): tag = cur.fetchone() return tag +@app.route('/bookmark//') +def bookmark(card_type, card_id): + if not session.get('logged_in'): + return redirect(url_for('login')) + db = get_db() + db.execute('UPDATE cards SET type = ? WHERE id = ?',[card_type,card_id]) + db.commit() + flash('Card saved.') + return redirect(url_for('mem', card_type=card_type)) + if __name__ == '__main__': app.run(host='0.0.0.0')