Add subscription functionality

new-registration-app/main.py

@@ -10,7 +10,8 @@ from fastapi.staticfiles import StaticFiles
 from fastapi.templating import Jinja2Templates
 from sqlmodel import Session, SQLModel, create_engine, select
 
-from models import Event, Household, Registration, TeamRegistration
+from models import (Event, Household, Registration, Subscription,
+                    TeamRegistration)
 
 sqlite_file_name = "database.db"
 sqlite_url = f"sqlite:///{sqlite_file_name}"
@@ -79,6 +80,75 @@ async def past_events(request: Request, session: SessionDep):
     )
 
 
+@app.get("/subscribe")
+async def subscribe(request: Request, session: SessionDep):
+    statement = select(Household)
+    households = session.exec(statement).all()
+
+    # filter out households with existing registrations
+    # households = [
+    #     h
+    #     for h in households
+    #     if h.id not in [reg.household_id for reg in event.registrations]
+    # ]
+
+    subscriptions = session.exec(select(Subscription)).all()
+
+    return templates.TemplateResponse(
+        request=request,
+        name="subscribe.html",
+        context={"households": households, "subscriptions": subscriptions},
+    )
+
+
+@app.post("/subscribe")
+async def add_subscribe(request: Request, session: SessionDep):
+    form_data = await request.form()
+    # TODO: Make this return a nicer error message
+    try:
+        num_adult_meals = int(form_data["numAdults"]) if form_data["numAdults"] else 0
+        num_children_meals = int(form_data["numKids"]) if form_data["numKids"] else 0
+        num_small_children_meals = (
+            int(form_data["numSmallKids"]) if form_data["numSmallKids"] else 0
+        )
+    except ValueError:
+        raise ValueError("All number fields must be integers")
+
+    subscription = Subscription(
+        household_id=form_data["household"],
+        num_adult_meals=num_adult_meals,
+        num_children_meals=num_children_meals,
+        num_small_children_meals=num_small_children_meals,
+    )
+
+    selected_days = form_data.getlist("days")
+    if selected_days:
+        subscription.monday = "1" in selected_days
+        subscription.tuesday = "2" in selected_days
+        subscription.wednesday = "3" in selected_days
+        subscription.thursday = "4" in selected_days
+        subscription.friday = "5" in selected_days
+        subscription.saturday = "6" in selected_days
+        subscription.sunday = "7" in selected_days
+
+    session.add(subscription)
+    session.commit()
+
+    return RedirectResponse(url="/subscribe", status_code=status.HTTP_302_FOUND)
+
+
+@app.get("/subscribe/{household_id}/delete")
+async def delete_subscription(request: Request, session: SessionDep, household_id: int):
+
+    statement = select(Subscription).where(Subscription.household_id == household_id)
+    sub = session.exec(statement).one()
+
+    session.delete(sub)
+    session.commit()
+
+    return RedirectResponse(url="/subscribe", status_code=status.HTTP_302_FOUND)
+
+
 @app.get("/event/add")
 async def add_event_form(request: Request, session: SessionDep):
     return templates.TemplateResponse(request=request, name="add_event.html")