diff --git a/app.py b/app.py index 23acef8..a39aa81 100644 --- a/app.py +++ b/app.py @@ -33,6 +33,15 @@ def bookmark(id): csr.execute("insert into bookmarks(book, page) values(%s, %s)", (id, page,)) return redirect(url_for("book", id=id), code=303) +@app.route("/book/new", methods=["POST"]) +def new_book(): + title = request.form["title"] + csr = get_cursor() + csr.execute("insert into books(title, current) values(%s, true) returning *", (title,)) + r = csr.fetchone() + return redirect(url_for("book", id=r.id), code=303) + + def get_cursor(): db = get_db() csr = db.cursor(cursor_factory=psycopg2.extras.NamedTupleCursor) diff --git a/templates/home.html b/templates/home.html index 335f5ae..802bcb1 100644 --- a/templates/home.html +++ b/templates/home.html @@ -14,5 +14,9 @@ {% endfor %} +