Update main.py

main.py

@@ -1,131 +1,131 @@
-from fastapi import FastAPI, Depends, HTTPException, status, File, UploadFile, Form
-from fastapi.responses import JSONResponse
-from sqlalchemy.ext.asyncio import AsyncSession
-from sqlalchemy import select
-from database import async_session, engine
-import models
-import schemas
-import os
-from fastapi.middleware.cors import CORSMiddleware
-
-app = FastAPI()
-
-from fastapi.staticfiles import StaticFiles
-
-# Monta o diretório "uploads" para servir arquivos estáticos
-app.mount("/uploads", StaticFiles(directory="uploads"), name="uploads")
-
-
-UPLOAD_DIRECTORY = "uploads"
-if not os.path.exists(UPLOAD_DIRECTORY):
-    os.makedirs(UPLOAD_DIRECTORY)
-
-app.add_middleware(
-    CORSMiddleware,
-    allow_origins=["*"],
-    allow_credentials=True,
-    allow_methods=["*"],
-    allow_headers=["*"],
-)
-
-async def get_db():
-    async with async_session() as session:
-        yield session
-
-@app.on_event("startup")
-async def on_startup():
-    async with engine.begin() as conn:
-        await conn.run_sync(models.Base.metadata.create_all)
-
-from sqlalchemy import or_
-
-# Nova rota para buscar promoções por palavra-chave
-@app.get("/promocoes/search/", response_model=list[schemas.Promocao])
-async def buscar_promocoes(q: str, db: AsyncSession = Depends(get_db)):
-    # Realiza a busca no título ou descrição
-    result = await db.execute(
-        select(models.Promocao)
-        .where(or_(models.Promocao.titulo.contains(q), models.Promocao.descricao.contains(q)))
-    )
-    promocoes = result.scalars().all()
-    return promocoes
-
-
-# 1. Criar Promoção
-@app.post("/empresas/{empresa_id}/promocoes/", response_model=schemas.Promocao)
-async def criar_promocao(
-    empresa_id: int,
-    titulo: str = Form(...),
-    descricao: str = Form(...),
-    imagem: UploadFile = File(...),
-    db: AsyncSession = Depends(get_db)
-):
-    # Validações de campos obrigatórios
-    if not titulo or not descricao:
-        raise HTTPException(status_code=400, detail="Campos de título e descrição são obrigatórios")
-
-    if not imagem:
-        raise HTTPException(status_code=400, detail="O arquivo de imagem é obrigatório")
-
-    # Salvar a imagem no diretório de uploads
-    image_location = os.path.join(UPLOAD_DIRECTORY, imagem.filename)
-    with open(image_location, "wb") as file:
-        content = await imagem.read()
-        file.write(content)
-
-    image_url = f"http://127.0.0.1:8000/uploads/{imagem.filename}"
-
-    # Criação da promoção
-    nova_promocao = models.Promocao(
-        titulo=titulo,
-        descricao=descricao,
-        imagem=image_url,
-        empresa_id=empresa_id
-    )
-
-    db.add(nova_promocao)
-    await db.commit()
-    await db.refresh(nova_promocao)
-    return nova_promocao
-
-# 2. Listar Promoções de uma Empresa
-@app.get("/empresas/{empresa_id}/promocoes/", response_model=list[schemas.Promocao])
-async def listar_promocoes_empresa(empresa_id: int, db: AsyncSession = Depends(get_db)):
-    result = await db.execute(select(models.Promocao).where(models.Promocao.empresa_id == empresa_id))
-    promocoes = result.scalars().all()
-    return promocoes
-
-# 3. Atualizar Promoção
-@app.put("/empresas/{empresa_id}/promocoes/{promocao_id}", response_model=schemas.Promocao)
-async def atualizar_promocao(
-    empresa_id: int,
-    promocao_id: int,
-    promocao: schemas.PromocaoCreate,
-    db: AsyncSession = Depends(get_db)
-):
-    result = await db.execute(select(models.Promocao).where(models.Promocao.id == promocao_id, models.Promocao.empresa_id == empresa_id))
-    promocao_db = result.scalars().first()
-
-    if not promocao_db:
-        raise HTTPException(status_code=404, detail="Promoção não encontrada.")
-
-    promocao_db.titulo = promocao.titulo
-    promocao_db.descricao = promocao.descricao
-
-    await db.commit()
-    await db.refresh(promocao_db)
-    return promocao_db
-
-# 4. Deletar Promoção
-@app.delete("/empresas/{empresa_id}/promocoes/{promocao_id}", status_code=status.HTTP_204_NO_CONTENT)
-async def deletar_promocao(empresa_id: int, promocao_id: int, db: AsyncSession = Depends(get_db)):
-    result = await db.execute(select(models.Promocao).where(models.Promocao.id == promocao_id, models.Promocao.empresa_id == empresa_id))
-    promocao_db = result.scalars().first()
-
-    if not promocao_db:
-        raise HTTPException(status_code=404, detail="Promoção não encontrada.")
-
-    await db.delete(promocao_db)
-    await db.commit()
-    return JSONResponse(status_code=status.HTTP_204_NO_CONTENT, content={"message": "Promoção deletada com sucesso"})
-
+from fastapi import FastAPI, Depends, HTTPException, status, File, UploadFile, Form
+from fastapi.responses import JSONResponse
+from sqlalchemy.ext.asyncio import AsyncSession
+from sqlalchemy import select
+from database import async_session, engine
+import models
+import schemas
+import os
+from fastapi.middleware.cors import CORSMiddleware
+
+app = FastAPI()
+
+from fastapi.staticfiles import StaticFiles
+
+# Monta o diretório "uploads" para servir arquivos estáticos
+app.mount("/uploads", StaticFiles(directory="uploads"), name="uploads")
+
+
+UPLOAD_DIRECTORY = "uploads"
+if not os.path.exists(UPLOAD_DIRECTORY):
+    os.makedirs(UPLOAD_DIRECTORY)
+
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+async def get_db():
+    async with async_session() as session:
+        yield session
+
+@app.on_event("startup")
+async def on_startup():
+    async with engine.begin() as conn:
+        await conn.run_sync(models.Base.metadata.create_all)
+
+from sqlalchemy import or_
+
+# Nova rota para buscar promoções por palavra-chave
+@app.get("/promocoes/search/", response_model=list[schemas.Promocao])
+async def buscar_promocoes(q: str, db: AsyncSession = Depends(get_db)):
+    # Realiza a busca no título ou descrição
+    result = await db.execute(
+        select(models.Promocao)
+        .where(or_(models.Promocao.titulo.contains(q), models.Promocao.descricao.contains(q)))
+    )
+    promocoes = result.scalars().all()
+    return promocoes
+
+
+# 1. Criar Promoção
+@app.post("/empresas/{empresa_id}/promocoes/", response_model=schemas.Promocao)
+async def criar_promocao(
+    empresa_id: int,
+    titulo: str = Form(...),
+    descricao: str = Form(...),
+    imagem: UploadFile = File(...),
+    db: AsyncSession = Depends(get_db)
+):
+    # Validações de campos obrigatórios
+    if not titulo or not descricao:
+        raise HTTPException(status_code=400, detail="Campos de título e descrição são obrigatórios")
+
+    if not imagem:
+        raise HTTPException(status_code=400, detail="O arquivo de imagem é obrigatório")
+
+    # Salvar a imagem no diretório de uploads
+    image_location = os.path.join(UPLOAD_DIRECTORY, imagem.filename)
+    with open(image_location, "wb") as file:
+        content = await imagem.read()
+        file.write(content)
+
+    image_url = f"https://fabiosantos-buscaofertaapi.hf.space/uploads/{imagem.filename}"
+
+    # Criação da promoção
+    nova_promocao = models.Promocao(
+        titulo=titulo,
+        descricao=descricao,
+        imagem=image_url,
+        empresa_id=empresa_id
+    )
+
+    db.add(nova_promocao)
+    await db.commit()
+    await db.refresh(nova_promocao)
+    return nova_promocao
+
+# 2. Listar Promoções de uma Empresa
+@app.get("/empresas/{empresa_id}/promocoes/", response_model=list[schemas.Promocao])
+async def listar_promocoes_empresa(empresa_id: int, db: AsyncSession = Depends(get_db)):
+    result = await db.execute(select(models.Promocao).where(models.Promocao.empresa_id == empresa_id))
+    promocoes = result.scalars().all()
+    return promocoes
+
+# 3. Atualizar Promoção
+@app.put("/empresas/{empresa_id}/promocoes/{promocao_id}", response_model=schemas.Promocao)
+async def atualizar_promocao(
+    empresa_id: int,
+    promocao_id: int,
+    promocao: schemas.PromocaoCreate,
+    db: AsyncSession = Depends(get_db)
+):
+    result = await db.execute(select(models.Promocao).where(models.Promocao.id == promocao_id, models.Promocao.empresa_id == empresa_id))
+    promocao_db = result.scalars().first()
+
+    if not promocao_db:
+        raise HTTPException(status_code=404, detail="Promoção não encontrada.")
+
+    promocao_db.titulo = promocao.titulo
+    promocao_db.descricao = promocao.descricao
+
+    await db.commit()
+    await db.refresh(promocao_db)
+    return promocao_db
+
+# 4. Deletar Promoção
+@app.delete("/empresas/{empresa_id}/promocoes/{promocao_id}", status_code=status.HTTP_204_NO_CONTENT)
+async def deletar_promocao(empresa_id: int, promocao_id: int, db: AsyncSession = Depends(get_db)):
+    result = await db.execute(select(models.Promocao).where(models.Promocao.id == promocao_id, models.Promocao.empresa_id == empresa_id))
+    promocao_db = result.scalars().first()
+
+    if not promocao_db:
+        raise HTTPException(status_code=404, detail="Promoção não encontrada.")
+
+    await db.delete(promocao_db)
+    await db.commit()
+    return JSONResponse(status_code=status.HTTP_204_NO_CONTENT, content={"message": "Promoção deletada com sucesso"})
+