Update main.py

main.py

@@ -1,32 +1,100 @@
-from fastapi import FastAPI, Request
-from fastapi.middleware.cors import CORSMiddleware  # Importa il middleware CORS
+from fastapi import FastAPI, Request, HTTPException
 from fastapi.responses import JSONResponse
-import requests
-import os
-import socket
-import time
-from enum import Enum
-import random
-import aiohttp
-import asyncio
 import json
+from datetime import datetime
+import pyodbc
+import os
+from dotenv import load_dotenv
 
-#--------------------------------------------------- Definizione Server FAST API ------------------------------------------------------
+load_dotenv()
 app = FastAPI()
 
-app.add_middleware(
-    CORSMiddleware,
-    allow_origins=["*"],
-    allow_credentials=True,
-    allow_methods=["*"],
-    allow_headers=["*"],
-)
+# --------------------------------------------| CONNESSIONE DB |-------------------------------------------------
+# Connessione al DB Web (Amazon RDS)
+def get_db_connection():
+    server = os.getenv('SERVER_DB')
+    username = os.getenv('USERNAME_DB')
+    password = os.getenv('PASSWORD_DB')
+    connection_string = f"DRIVER={{ODBC Driver 17 for SQL Server}};SERVER={server};UID={username};PWD={password}"
+    return pyodbc.connect(connection_string)
 
-@app.post("/Movimenti")
-async def generate_text(request: Request):
+# --------------------------------------------| CREAZIONE AZIENDA-DB |-------------------------------------------------
+# Controllo se presente il Database e la tabella aziendale. Se non ci sono le creo
+def crea_database_azienda(database, azienda):
+    tabella = f"{azienda.upper()}transazioni"
+    conn = get_db_connection()
+    cursor = conn.cursor()
+    # Controllo presenza DATABASE
+    conn.autocommit = True
+    check_db_query = f"""
+    IF NOT EXISTS (SELECT name FROM sys.databases WHERE name = '{database}')
+    BEGIN
+        CREATE DATABASE {database};
+    END
+    """
+    cursor.execute(check_db_query)
+    cursor.execute(f"USE {database}")
+    conn.autocommit = False
+    #Controllo presenza TABELLA 
+    check_table_query = f"""
+    IF NOT EXISTS (SELECT * FROM INFORMATION_SCHEMA.TABLES WHERE TABLE_SCHEMA = 'dbo' AND TABLE_NAME = '{tabella}')
+    BEGIN
+        EXEC('CREATE TABLE dbo.{tabella} (
+            id INT IDENTITY(1,1) PRIMARY KEY,
+            datreg DATETIME NOT NULL,
+            azione VARCHAR(255) NOT NULL,
+            id_transazione VARCHAR(50) NOT NULL,
+            jsonCompleto NVARCHAR(MAX) NOT NULL
+        )');
+    END
+    """
+    cursor.execute(check_table_query)
+    conn.commit()
+    cursor.close()
+    conn.close()
+
+# ------------------------------------------------| POST |--------------------------------------------------
+# Inizio transazione (questo endpoint viene chiamato dal RENTRI come CALLBACK per il PUSH Request)
+@app.post("/{azione}/{versione}/{database}/{azienda}/start")
+async def receive_post(request: Request, azione: str, versione :str, database: str, azienda: str):
     data = await request.json()
-    print(data)
-    return JSONResponse(content={"message": "Received", "data": data})    
+    print(azione + " " + database + " " + azienda + " " + json.dumps(data, indent=3))
+    conn = get_db_connection()
+    cursor = conn.cursor()
+    crea_database_azienda(database, azienda)
+    insert_query = f"INSERT INTO [{database}].dbo.{azienda}transazioni (datreg, azione, id_transazione, JsonCompleto) VALUES (?, ?, ?, ?)"
+    cursor.execute(insert_query, (datetime.now(), azione, data.get('transazione_id'), json.dumps(data)))
+    conn.commit()
+    cursor.close()
+    conn.close()
+    return JSONResponse(content={"message": "Received", "data": data})  
+
+# ------------------------------------------------| GET |-----------------------------------------------------
+# Result/Status transazione (Questo endpoint viene chiamato da AdHoc per leggere il risultato della transazione) /dati-registri/v1.0/id_transazione/STATUS/Azienda
+@app.get("/{azione}/{versione}/{database}/{azienda}/{id_transazione}/{tipo}")
+async def get_status(azione: str, versione: str, id_transazione: str, tipo: str, database: str, azienda: str):
+    conn = get_db_connection()
+    cursor = conn.cursor()
+    select_query = f"SELECT TOP 1 JsonCompleto FROM [{database}].dbo.{azienda}transazioni WHERE azione = ? AND id_transazione = ? order by id desc"
+    cursor.execute(select_query, (azione, id_transazione))
+    row = cursor.fetchone()
+    cursor.close()
+    conn.close()
+    # STATUS: Restituisce 303 se esiste la riga della transazione (cioè se per il rentri è terminata)
+    if tipo == "status": 
+        if not row:
+            raise HTTPException(status_code=200, detail="Transazione non terminata")
+        else:
+            raise HTTPException(status_code=303, detail="Transazione terminata")
+    # RESULT: Restituisce 303 se esiste la riga della transazione (cioè se per il rentri è terminata)
+    elif tipo == "result": 
+        if not row:
+            raise HTTPException(status_code=404, detail="Transazione non terminata")
+        else: 
+            result = json.loads(row.JsonCompleto)
+            return JSONResponse(content=result)
+    else:
+        raise HTTPException(status_code=404, detail="Endpoint non trovato")
 
 @app.get("/")
 def read_general():