Open Amandaraamos opened 3 months ago
esse código esta criando outro banco de dados para usar local? foi isso que entendi se poder me ajudar, depois com a conexão local faz conectar o banco e criar outra tabela para jogadores. Da bom usar 2 banco de dados em um app, ou só pra usar um método mais fácil e não ter que usar o postman?
DESAFIO
pip install fastapi sqlalchemy fastapi-pagination uvicorn
class Atleta(Base): tablename = "atletas"
from` sqlalchemy import create_engine from sqlalchemy.orm import sessionmaker from database import Base
DATABASE_URL = "postgresql://user:password@host:port/database"
engine = create_engine(DATABASE_URL) Base.metadata.create_all(engine)
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
`` from fastapi import FastAPI, Depends, HTTPException from fastapi_pagination import Page, add_pagination from sqlalchemy.orm import Session from models import Atleta from database import SessionLocal
app = FastAPI()
@app.on_event("startup") async def startup_event(): global session session = SessionLocal()
@app.on_event("shutdown") async def shutdown_event(): global session session.close()
def get_db(): db = SessionLocal() try: yield db finally: db.close()
@app.get("/atletas", response_model=Page[Atleta]) async def get_all_atletas(db: Session = Depends(get_db), limit: int | None = None, offset: int | None = None): atletas = db.query(Atleta).order_by(Atleta.id).limit(limit).offset(offset).all() return Page(atletas, len(atletas))
@app.get("/atletas/search", response_model=List[Atleta]) async def get_atleta_by_nome_cpf(nome: str | None = None, cpf: str | None = None, db: Session = Depends(get_db)): query = db.query(Atleta)
@app.post("/atletas") async def create_atleta(atleta: Atleta, db: Session = Depends(get_db)): db.add(atleta) db.commit() db.refresh(atleta) return atleta
add_pagination(app)
def get_db(session: Session = Depends(SessionLocal)): return session
@app.get("/atletas", response_model=Page[Atleta]) async def get_all_atletas(db: Session = Depends(get_db), limit: int | None = None, offset: int | None = None): atletas = db.query(Atleta).order_by(Atleta.id).limit(limit).offset(offset).all() return Page(atletas, len(atletas))
@app.get("/atletas", response_model=List[Atleta]) async def get_atleta_by_nome_cpf(nome: str | None = None, cpf: str | None = None, db: Session = Depends(get_db)): query = db.query(Atleta)
if nome: query = query.filter(Atleta.nome.ilike(f"%{nome}%"))
if cpf: query = query.filter(Atleta.cpf == cpf)
atletas = query.all()
if not atletas: raise HTTPException(status_code=404, detail="Atleta não encontrado")
return atletas
@app.post("/atletas") async def