habulaj commited on
Commit
f3c5278
·
verified ·
1 Parent(s): 1b4f58e

Delete routers/db.py

Browse files
Files changed (1) hide show
  1. routers/db.py +0 -60
routers/db.py DELETED
@@ -1,60 +0,0 @@
1
- # db.py
2
- from fastapi import APIRouter, HTTPException
3
- from pydantic import BaseModel
4
- from typing import Optional
5
- import databases
6
- import sqlalchemy
7
- import os
8
-
9
- # Opção 1: Apenas variável de ambiente (mais seguro)
10
- DATABASE_URL = os.getenv("DATABASE_URL")
11
-
12
- # Verificação se a variável existe
13
- if not DATABASE_URL:
14
- raise ValueError("DATABASE_URL environment variable is not set")
15
-
16
- database = databases.Database(DATABASE_URL, min_size=5, max_size=20)
17
- metadata = sqlalchemy.MetaData()
18
-
19
- igbio = sqlalchemy.Table(
20
- "igbio",
21
- metadata,
22
- sqlalchemy.Column("url_instagram", sqlalchemy.Text),
23
- sqlalchemy.Column("url_web", sqlalchemy.Text),
24
- sqlalchemy.Column("image", sqlalchemy.Text),
25
- )
26
-
27
- router = APIRouter()
28
-
29
- class IGPost(BaseModel):
30
- url_instagram: Optional[str] = None
31
- url_web: Optional[str] = None
32
- image: Optional[str] = None
33
-
34
- async def connect_db():
35
- await database.connect()
36
-
37
- async def disconnect_db():
38
- await database.disconnect()
39
-
40
- @router.post("/igbio")
41
- async def add_igbio(item: IGPost):
42
- def clean_value(value: Optional[str]) -> Optional[str]:
43
- if value is None:
44
- return None
45
- val = value.strip()
46
- if val == "" or val.lower() == "null":
47
- return None
48
- return val
49
-
50
- query = igbio.insert().values(
51
- url_instagram=clean_value(item.url_instagram),
52
- url_web=clean_value(item.url_web),
53
- image=clean_value(item.image),
54
- )
55
-
56
- try:
57
- record_id = await database.execute(query)
58
- return {"success": True, "id": record_id}
59
- except Exception as e:
60
- raise HTTPException(status_code=500, detail=f"Erro ao inserir no banco: {e}")