File size: 2,478 Bytes
2259d2e
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
from library.librerias import *
from models.gastos import Gastos

router = APIRouter(
    prefix="/gastos",
    tags=["Gastos"],
    responses={404: {"description": "No encontrado"}},
)

"""CREATE TABLE gastos (
  ID_Gasto INTEGER PRIMARY KEY AUTOINCREMENT,
  Tipo TEXT,
  Monto REAL,
  Fecha DATE
);
"""


@router.get("/")
def get_gastos():
    try:
        with DatabaseConnection().get_connection() as conn:
            cursor = conn.cursor()
            cursor.execute("SELECT * FROM gastos")
            gastos = cursor.fetchall()
            return gastos
    except Exception as e:
        print(e)
        return []


# @router.post("/")
def post_gasto(gasto: Gastos):
    try:
        with DatabaseConnection().get_connection() as conn:
            cursor = conn.cursor()
            cursor.execute(
                "INSERT INTO gastos (Tipo, Monto, Fecha) VALUES (?, ?, ?)",
                (gasto.Tipo, gasto.Monto, gasto.Fecha),
            )
            conn.commit()
            return {"message": "Gasto creado"}
    except Exception as e:
        print(e)
        return []


# @router.put("/")
def put_gasto(gasto: Gastos):
    try:
        with DatabaseConnection().get_connection() as conn:
            cursor = conn.cursor()
            cursor.execute(
                "UPDATE gastos SET Tipo = ?, Monto = ?, Fecha = ? WHERE ID_Gasto = ?",
                (gasto.Tipo, gasto.Monto, gasto.Fecha, gasto.ID_Gasto),
            )
            conn.commit()
            return {"message": "Gasto actualizado"}
    except Exception as e:
        print(e)
        return []


# @router.delete("/{ID_Gasto}")
def delete_gasto(ID_Gasto: int):
    try:
        with DatabaseConnection().get_connection() as conn:
            cursor = conn.cursor()
            cursor.execute("DELETE FROM gastos WHERE ID_Gasto = ?", (ID_Gasto,))
            conn.commit()
            return {"message": "Gasto eliminado"}
    except Exception as e:
        print(e)
        return []


# listar gastos de una fecha en especifico con formato YYYY-MM-DD
@router.get("/fecha/{Fecha}")
def get_gastos_by_date(Fecha: str):
    try:
        fecha = datetime.strptime(Fecha, "%Y-%m-%d").date()
        with DatabaseConnection().get_connection() as conn:
            cursor = conn.cursor()
            cursor.execute("SELECT * FROM gastos WHERE Fecha = ?", (fecha,))
            gastos = cursor.fetchall()
            return gastos
    except Exception as e:
        print(e)
        return []