[email protected]

from fastapi import FastAPI
from pymongo import MongoClient
from pydantic import BaseModel

class Reservation(BaseModel):
    name : str
    time: int
    table_number: int
    
client = MongoClient('mongodb://localhost', 27017)

# TODO fill in database name
db = client["afternoon-assign"]

# TODO fill in collection name
collection = db["table"]

app = FastAPI()


# TODO complete all endpoint.
@app.get("/reservation/by-name/{name}")
def get_reservation_by_name(name:str):
    result = collection.find({"name" : name})
    for r in result:
        print(r)
        print("\n")

@app.get("reservation/by-table/{table}")
def get_reservation_by_table(table: int):
    result = collection.find({"table" : table},{"_id": 0, "name": 1,"table":0,"hour":1})
    for r in result:
        print(r)
        print("\n")


@app.post("/reservation")
def reserve(reservation : Reservation):
    pass

@app.put("/reservation/update/")
def update_reservation(reservation: Reservation):
    pass

@app.delete("/reservation/delete/{name}/{table_number}")
def cancel_reservation(name: str, table_number : int):
    pass