45 lines
1.7 KiB
Python
45 lines
1.7 KiB
Python
from datetime import datetime, timezone
|
|
from sqlalchemy import create_engine, text
|
|
from sqlalchemy.exc import SQLAlchemyError
|
|
from sqlalchemy.orm import sessionmaker
|
|
|
|
from backend.ConexionSql.Base_conexion import ConexionBase
|
|
from backend.Credenciales.postgres_credencial import PostgresCredencial
|
|
|
|
|
|
class PostgresConexion(ConexionBase):
|
|
def __init__(self, *args, **kwargs):
|
|
self.estado = "pendiente"
|
|
self.timestamp = datetime.now(timezone.utc)
|
|
|
|
if args and isinstance(args[0], PostgresCredencial):
|
|
credencial = args[0]
|
|
self.host = credencial.host
|
|
self.port = credencial.port
|
|
self.dbname = credencial.dbname
|
|
self.user = credencial.user
|
|
self.password = credencial.password # se guarda la contraseña
|
|
uri = credencial.get_uri()
|
|
else:
|
|
self.user = kwargs.get("user")
|
|
self.password = kwargs.get("password") # se guarda la contraseña
|
|
self.host = kwargs.get("host")
|
|
self.port = kwargs.get("port", 5432)
|
|
self.dbname = kwargs.get("db") or kwargs.get("dbname")
|
|
uri = f"postgresql://{self.user}:{self.password}@{self.host}:{self.port}/{self.dbname}"
|
|
|
|
self.engine = create_engine(uri)
|
|
self.SessionLocal = sessionmaker(bind=self.engine)
|
|
|
|
def get_session(self):
|
|
return self.SessionLocal()
|
|
|
|
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 |