Cambios a las 3 bases Model mapper repo para que funcionen a partir de las clases heredando todos los metodos comunes

src/ApiKeys/openai_apikey_mmr.py

@@ -8,6 +8,13 @@ from src.base import Base
 from src.ApiKeys.openai_apikey import OpenAICredencial
 from src.Security.Encriptar import Encriptar_fernet
 from entrypoint import ENV_PATH
+from src.ArquitectureLayer.Mapper import Mapper_base
+
+
+from sqlalchemy import Column, String
+from src.ArquitectureLayer.Model import Model_base
+from src.ArquitectureLayer.Repo import Repo_base
+
 
 # ----------------------
 # Cargar clave maestra
@@ -21,7 +28,7 @@ if pssword is None:
 # MODELO (SQLAlchemy)
 # ----------------------
 
-class OpenAICredencialModel(Base):
+class OpenAICredencialModel(Base, Model_base):
     __tablename__ = 'openai_credenciales'
 
     id = Column(String, primary_key=True)
@@ -33,7 +40,30 @@ class OpenAICredencialModel(Base):
 # MAPPER
 # ----------------------
 
-class OpenAICredencialMapper:
+class OpenAICredencialMapper(Mapper_base[OpenAICredencial, OpenAICredencialModel]):
+
+    @staticmethod
+    def to_model(obj: OpenAICredencial) -> OpenAICredencialModel:
+        return OpenAICredencialModel(
+            id=obj.id,
+            titulo=obj.titulo,
+            api_key=base64.b64encode(
+                Encriptar_fernet.encriptar(obj.api_key, pssword)
+            ).decode("utf-8"),
+            organizacion=obj.organizacion
+        )
+
+    @staticmethod
+    def from_model(model: OpenAICredencialModel) -> OpenAICredencial:
+        return OpenAICredencial(
+            id=model.id,
+            titulo=model.titulo,
+            api_key=Encriptar_fernet.desencriptar(
+                base64.b64decode(model.api_key), pssword
+            ),
+            organizacion=model.organizacion
+        )
+
     @staticmethod
     def to_dict(obj: OpenAICredencial) -> dict:
         return {
@@ -41,7 +71,7 @@ class OpenAICredencialMapper:
             "titulo": obj.titulo,
             "api_key": base64.b64encode(
                 Encriptar_fernet.encriptar(obj.api_key, pssword)
-            ).decode('utf-8'),
+            ).decode("utf-8"),
             "organizacion": obj.organizacion
         }
 
@@ -56,40 +86,22 @@ class OpenAICredencialMapper:
             organizacion=data.get("organizacion")
         )
 
-    @staticmethod
-    def from_model(model: OpenAICredencialModel) -> OpenAICredencial:
-        return OpenAICredencial(
-            id=model.id,
-            titulo=model.titulo,
-            api_key=Encriptar_fernet.desencriptar(
-                base64.b64decode(model.api_key), pssword
-            ),
-            organizacion=model.organizacion
-        )
-
 # ----------------------
 # REPO
 # ----------------------
 
-class OpenAICredencialRepo:
+class OpenAICredencialRepo(Repo_base[OpenAICredencialModel, OpenAICredencial]):
     def __init__(self, conexion: ConexionBase):
-        self.session = conexion.get_session()
-
-    def add(self, credencial: OpenAICredencial) -> str:
-        data = OpenAICredencialMapper.to_dict(credencial)
-        model = OpenAICredencialModel(**data)
-        self.session.add(model)
-        self.session.commit()
-        return model.id
-
-    def get_all(self) -> list[OpenAICredencial]:
-        models = self.session.query(OpenAICredencialModel).all()
-        return [OpenAICredencialMapper.from_model(m) for m in models]
+        super().__init__(
+            session=conexion.get_session(),
+            modelo=OpenAICredencialModel,
+            mapper=OpenAICredencialMapper
+        )
 
     def get_by_titulo(self, titulo: str) -> OpenAICredencial | None:
-        model = self.session.query(OpenAICredencialModel).filter_by(titulo=titulo).first()
-        return OpenAICredencialMapper.from_model(model) if model else None
-
-    def get_by_id(self, id_: str) -> OpenAICredencial | None:
-        model = self.session.get(OpenAICredencialModel, id_)
-        return OpenAICredencialMapper.from_model(model) if model else None
+        model = (
+            self.session.query(self.Modelo)
+            .filter_by(titulo=titulo, sys_deleted_at=None)
+            .first()
+        )
+        return self.Mapper.from_model(model) if model else None