-
Notifications
You must be signed in to change notification settings - Fork 0
/
app.py
46 lines (31 loc) · 1.34 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
import os
from cs50 import SQL
from flask import Flask, flash, jsonify, redirect, render_template, request, session
# Configure application
app = Flask(__name__)
# Ensure templates are auto-reloaded
app.config["TEMPLATES_AUTO_RELOAD"] = True
# Configure CS50 Library to use SQLite database
db = SQL("sqlite:///birthdays.db")
@app.after_request
def after_request(response):
"""Ensure responses aren't cached"""
response.headers["Cache-Control"] = "no-cache, no-store, must-revalidate"
response.headers["Expires"] = 0
response.headers["Pragma"] = "no-cache"
return response
@app.route("/", methods=["GET", "POST"])
def index():
if request.method == "POST":
# TODO: Add the user's entry into the database
name = request.form.get("name")
month = request.form.get("month")
day = request.form.get("day")
if (name and month and day and name.isalpha() and month.isdigit()
and day.isdigit() and 1 <= int(month) <= 12 and 1 <= int(day) <= 31):
db.execute("INSERT INTO birthdays (name, month, day) VALUES (?, ?, ?);", name, month, day)
return redirect("/")
else:
# TODO: Display the entries in the database on index.html
birthdays = db.execute("SELECT name, day, month FROM birthdays;")
return render_template("index.html", birthdays=birthdays)