cwubs-shared 0.1.1__tar.gz

cwubs_shared-0.1.1/PKG-INFO

@@ -0,0 +1,15 @@
+Metadata-Version: 2.4
+Name: cwubs-shared
+Version: 0.1.1
+Summary: Shared utilities for cwubs services
+Author: Duncan Scuffell
+Author-email: duncan.scuffell@btinternet.com
+Requires-Python: >=3.12,<4.0
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
+Requires-Dist: sqlalchemy (>=2.0.46,<3.0.0)
+Description-Content-Type: text/markdown
+
+# cwubs-shared

cwubs_shared-0.1.1/README.md

@@ -0,0 +1 @@
+# cwubs-shared

cwubs_shared-0.1.1/pyproject.toml

@@ -0,0 +1,21 @@
+[tool.poetry]
+name = "cwubs-shared"
+version = "0.1.1" # placeholder
+description = "Shared utilities for cwubs services"
+authors = ["Duncan Scuffell <duncan.scuffell@btinternet.com>"]
+readme = "README.md"
+packages = [{ include = "shared" }]
+
+[tool.poetry.dependencies]
+python = ">=3.12,<4.0"
+sqlalchemy = "^2.0.46"
+
+[tool.poetry-dynamic-versioning]
+enable = false
+
+[build-system]
+requires = ["poetry-core>=1.0.0", "poetry-dynamic-versioning>=1.0.0,<2.0.0"]
+build-backend = "poetry_dynamic_versioning.backend"
+
+[tool.poetry.requires-plugins]
+poetry-dynamic-versioning = { version = ">=1.0.0,<2.0.0", extras = ["plugin"] }

cwubs_shared-0.1.1/shared/__init__.py

@@ -0,0 +1,3 @@
+from .base import BaseRepository
+
+__all__ = ["BaseRepository"]

cwubs_shared-0.1.1/shared/base.py

@@ -0,0 +1,39 @@
+from __future__ import annotations
+
+from typing import Generic, Optional, TypeVar
+
+from sqlalchemy import select
+from sqlalchemy.orm import Session
+
+T = TypeVar("T")
+ID = TypeVar("ID")
+
+
+class BaseRepository(Generic[T, ID]):
+    def __init__(self, db: Session, model: type[T]) -> None:
+        self._db = db
+        self._model = model
+
+    def create(self, obj: T) -> T:
+        self._db.add(obj)
+        self._db.commit()
+        self._db.refresh(obj)
+        return obj
+
+    def get(self, obj_id: ID) -> Optional[T]:
+        return self._db.get(self._model, obj_id)
+
+    def exists(self, obj_id: ID) -> bool:
+        return (
+            self._db.execute(select(self._model.id).where(self._model.id == obj_id)).first()
+            is not None
+        )
+
+    def save(self, obj: T) -> T:
+        self._db.commit()
+        self._db.refresh(obj)
+        return obj
+
+    def delete(self, obj: T) -> None:
+        self._db.delete(obj)
+        self._db.commit()