fix: add electric amount

Author: edgarmoya

app/api/v1/endpoints/pdf_processing.py

@@ -92,7 +92,7 @@ async def sales(params: PdfContentRequest = Depends()):
         with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as temp_pdf:
             temp_pdf.write(await params.file.read())
 
-        total_amount, total_saldo, total_propia, total_movil, total_nauta, total_nauta_hogar, total_factura, profits, pdf_data = PdfAnalyzer(temp_pdf.name).sales(transaction_status=status)
+        total_amount, total_saldo, total_propia, total_movil, total_nauta, total_nauta_hogar, total_factura, total_electrica, profits, pdf_data = PdfAnalyzer(temp_pdf.name).sales(transaction_status=status)
 
         # If no exist `page` and `limit`, return all data
         if page is None or limit is None:
@@ -104,6 +104,7 @@ async def sales(params: PdfContentRequest = Depends()):
                 total_nauta=total_nauta,
                 total_nauta_hogar=total_nauta_hogar,
                 total_factura=total_factura,
+                total_electrica=total_electrica,
                 profits=profits, 
                 results=pdf_data
             )
@@ -122,6 +123,7 @@ async def sales(params: PdfContentRequest = Depends()):
             total_nauta=total_nauta,
             total_nauta_hogar=total_nauta_hogar,
             total_factura=total_factura,
+            total_electrica=total_electrica,
             profits=profits,
             results=paginated_data
         )

app/main.py

@@ -1,5 +1,6 @@
 from fastapi import FastAPI
 from app.api.v1.endpoints import pdf_processing
+from fastapi.responses import ORJSONResponse
 
 app = FastAPI(
     title="Estado de Cuenta API",
@@ -11,7 +12,8 @@
         "- Obtener el contenido completo del estado de cuenta en formato estructurado\n\n"
         "- Extraer información específica, como **depósitos**, **recargas** realizadas, **ganancias** y otros detalles relevantes\n\n"
     ),
-    version="1.0.0"
+    version="1.0.0",
+    default_response_class=ORJSONResponse
 )
 
 app.include_router(pdf_processing.router, prefix="/pdf", tags=['Procesar PDF'])

app/schemas/pdf.py

@@ -33,5 +33,6 @@ class PdfSalesResponse(BaseModel):
     total_nauta: float
     total_nauta_hogar: float
     total_factura: float
+    total_electrica: float
     profits: float
     results: List[TransactionData]

app/services/pdf_analyzer.py

@@ -66,7 +66,7 @@ def deposits(self, transaction_status: str = 'successful') -> List[Dict]:
     def sales(self, transaction_status: str = 'successful') -> List[Dict]:
         """Calculates the total amount deducted for recharges"""
         total_amount, total_saldo, total_movil, total_nauta = 0, 0, 0, 0
-        total_nauta_hogar, total_factura, total_propia = 0, 0, 0
+        total_nauta_hogar, total_factura, total_propia, total_electrica = 0, 0, 0, 0
         data = []
         for row in self.transactions(transaction_status):
             if row['transaction_type'] == 'Venta de Saldo AT':
@@ -81,8 +81,10 @@ def sales(self, transaction_status: str = 'successful') -> List[Dict]:
                 total_nauta_hogar += row['amount_paid']
             if row['transaction_type'] == 'Pago Factura AT':
                 total_factura += row['amount_paid']
+            if row['transaction_type'] == 'Factura Electrica':
+                total_electrica += row['amount_paid']
             if row['transaction_type'] not in ['Estado de Cuenta', 'Recarga Bolsa CUP']:
                 data.append(row)
                 total_amount += row['amount_paid']
         profits = total_amount / 0.9 - total_amount
-        return round(total_amount, 2), round(total_saldo, 2), round(total_propia, 2), round(total_movil, 2), round(total_nauta, 2), round(total_nauta_hogar, 2), round(total_factura, 2), round(profits, 2), data
+        return round(total_amount, 2), round(total_saldo, 2), round(total_propia, 2), round(total_movil, 2), round(total_nauta, 2), round(total_nauta_hogar, 2), round(total_factura, 2), round(total_electrica, 2), round(profits, 2), data

requirements.txt

@@ -1,14 +1,19 @@
 annotated-types==0.7.0
 anyio==4.8.0
+bcrypt==3.2.0
 certifi==2025.1.31
+cffi==1.17.1
 click==8.1.8
 colorama==0.4.6
+cryptography==44.0.3
 distro==1.9.0
 dnspython==2.7.0
+ecdsa==0.19.1
 email_validator==2.2.0
 et_xmlfile==2.0.0
 fastapi==0.115.8
 fastapi-cli==0.0.7
+greenlet==3.2.2
 h11==0.14.0
 httpcore==1.0.7
 httptools==0.6.4
@@ -20,20 +25,28 @@ MarkupSafe==3.0.2
 mdurl==0.1.2
 numpy==2.2.3
 openpyxl==3.1.5
+orjson==3.10.18
 pandas==2.2.3
+passlib==1.7.4
+pyasn1==0.4.8
+pycparser==2.22
 pydantic==2.10.6
 pydantic_core==2.27.2
 Pygments==2.19.1
 python-dateutil==2.9.0.post0
 python-dotenv==1.0.1
+python-jose==3.4.0
 python-multipart==0.0.20
 pytz==2025.1
 PyYAML==6.0.2
 rich==13.9.4
 rich-toolkit==0.13.2
+rsa==4.9.1
 shellingham==1.5.4
 six==1.17.0
 sniffio==1.3.1
+SQLAlchemy==2.0.41
+sqlmodel==0.0.24
 starlette==0.45.3
 tabula-py==2.10.0
 typer==0.15.1