sex

app/services/base_service.py

@@ -0,0 +1,45 @@
+from typing import Type, TypeVar, Generic, List, Optional, Any
+from sqlalchemy.orm import Session
+from app.db.database import Base
+from app.services.service_registry import ServiceRegistry
+
+T = TypeVar('T')
+
+class BaseService(Generic[T]):
+    def __init__(self, model: Type[T]):
+        self.model = model
+        # Register the service instance
+        ServiceRegistry.register(self.__class__.__name__, self)
+
+    def get(self, db: Session, id: int) -> Optional[T]:
+        return db.query(self.model).filter(self.model.id == id).first()
+
+    def get_all(self, db: Session, skip: int = 0, limit: int = 100) -> List[T]:
+        return db.query(self.model).offset(skip).limit(limit).all()
+
+    def create(self, db: Session, obj_in: dict) -> T:
+        db_obj = self.model(**obj_in)
+        db.add(db_obj)
+        db.commit()
+        db.refresh(db_obj)
+        return db_obj
+
+    def update(self, db: Session, db_obj: T, obj_in: dict) -> T:
+        for field in obj_in:
+            if hasattr(db_obj, field):
+                setattr(db_obj, field, obj_in[field])
+        db.add(db_obj)
+        db.commit()
+        db.refresh(db_obj)
+        return db_obj
+
+    def delete(self, db: Session, id: int) -> bool:
+        obj = db.query(self.model).filter(self.model.id == id).first()
+        if obj:
+            db.delete(obj)
+            db.commit()
+            return True
+        return False
+
+    def get_service(self, service_name: str) -> Any:
+        return ServiceRegistry.get(service_name)