Skip to content
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
14 changes: 6 additions & 8 deletions main.py
Original file line number Diff line number Diff line change
@@ -1,20 +1,18 @@
from contextlib import asynccontextmanager
from fastapi import FastAPI
from dotenv import dotenv_values
from pymongo import MongoClient
from routes import router as book_router

config = dotenv_values(".env")

app = FastAPI()

@app.on_event("startup")
def startup_db_client():
@asynccontextmanager
async def lifespan(app: FastAPI):
app.mongodb_client = MongoClient(config["ATLAS_URI"])
app.database = app.mongodb_client[config["DB_NAME"]]

@app.on_event("shutdown")
def shutdown_db_client():
yield
app.mongodb_client.close()

app.include_router(book_router, tags=["books"], prefix="/book")
app = FastAPI(lifespan=lifespan)

app.include_router(book_router, tags=["books"], prefix="/book")