endpoint_biblioteca_funcionando
This commit is contained in:
@@ -1,12 +1,39 @@
|
||||
from abc import ABC, abstractmethod
|
||||
from sqlalchemy.orm import Session
|
||||
from datetime import datetime, timezone
|
||||
from sqlalchemy.orm import Session, sessionmaker
|
||||
from sqlalchemy.engine import Engine
|
||||
from sqlalchemy import create_engine, text
|
||||
from sqlalchemy.exc import SQLAlchemyError
|
||||
|
||||
class ConexionBase(ABC):
|
||||
def __init__(self, uri: str):
|
||||
self.estado = "pendiente"
|
||||
self.timestamp = datetime.now(timezone.utc)
|
||||
self._engine: Engine = create_engine(uri)
|
||||
self._Session = sessionmaker(bind=self._engine)
|
||||
self._session_instance: Session | None = None
|
||||
|
||||
@abstractmethod
|
||||
def get_session(self) -> Session:
|
||||
pass
|
||||
if self._session_instance is None:
|
||||
self._session_instance = self._Session()
|
||||
return self._session_instance
|
||||
|
||||
@abstractmethod
|
||||
def get_engine(self) -> Engine:
|
||||
pass
|
||||
return self._engine
|
||||
|
||||
def probar_conexion(self) -> bool:
|
||||
try:
|
||||
with self._engine.connect() as connection:
|
||||
connection.execute(text("SELECT 1"))
|
||||
self.estado = "exito"
|
||||
return True
|
||||
except SQLAlchemyError:
|
||||
self.estado = "fallo"
|
||||
return False
|
||||
|
||||
def close(self):
|
||||
if self._session_instance is not None:
|
||||
self._session_instance.close()
|
||||
self._session_instance = None
|
||||
@@ -4,7 +4,6 @@ from sqlalchemy.exc import SQLAlchemyError
|
||||
from sqlalchemy.orm import sessionmaker, Session
|
||||
from sqlalchemy.engine import Engine
|
||||
|
||||
|
||||
from src.ConexionSql.Base_conexion import ConexionBase
|
||||
from src.Credenciales.postgres_credencial import PostgresCredencial
|
||||
|
||||
@@ -32,10 +31,14 @@ class PostgresConexion(ConexionBase):
|
||||
self._engine: Engine = create_engine(uri)
|
||||
self._Session = sessionmaker(bind=self._engine)
|
||||
|
||||
# ✅ INICIALIZAR LA SESIÓN AQUÍ
|
||||
self._session_instance: Session | None = None
|
||||
|
||||
def get_session(self) -> Session:
|
||||
return self._Session()
|
||||
|
||||
if self._session_instance is None:
|
||||
self._session_instance = self._Session()
|
||||
return self._session_instance
|
||||
|
||||
def get_engine(self) -> Engine:
|
||||
return self._engine
|
||||
|
||||
@@ -47,4 +50,9 @@ class PostgresConexion(ConexionBase):
|
||||
return True
|
||||
except SQLAlchemyError:
|
||||
self.estado = "fallo"
|
||||
return False
|
||||
return False
|
||||
|
||||
def close(self):
|
||||
if self._session_instance is not None:
|
||||
self._session_instance.close()
|
||||
self._session_instance = None
|
||||
|
||||
Reference in New Issue
Block a user