diff --git a/app/main.py b/app/main.py new file mode 100644 index 0000000..7411d22 --- /dev/null +++ b/app/main.py @@ -0,0 +1,27 @@ +from fastapi import FastAPI +from app import crud, schemas +from app.database import SessionLocal, engine +from app import models + +# Create tables if not exist +models.Base.metadata.create_all(bind=engine) + +app = FastAPI(title="Data Lab API") + +@app.get("/") +def root(): + return {"message": "Welcome to Data Lab API"} + +@app.get("/customers") +def get_customers(): + db = SessionLocal() + customers = crud.get_customers(db) + db.close() + return customers + +@app.get("/accounts") +def get_accounts(): + db = SessionLocal() + accounts = crud.get_accounts(db) + db.close() + return accounts