architectonics 0.0.25__tar.gz → 0.0.26__tar.gz

{architectonics-0.0.25 → architectonics-0.0.26}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: architectonics
-Version: 0.0.25
+Version: 0.0.26
 Summary: 
 Author: Your Name
 Author-email: you@example.com

architectonics-0.0.26/architectonics/core/models/base_model.py

@@ -0,0 +1,9 @@
+from dataclasses import dataclass
+from datetime import datetime
+
+
+@dataclass
+class BaseModel:
+    id: str
+    created_at: datetime
+    updated_at: datetime

{architectonics-0.0.25 → architectonics-0.0.26}/architectonics/infrastructure/entities/base_entity.py

@@ -4,6 +4,7 @@ from sqlalchemy import Column, DateTime, String
 from sqlalchemy.orm import as_declarative
 
 from architectonics.common.utils.utils import get_current_datetime
+from architectonics.core.models.base_model import BaseModel
 
 
 @as_declarative()
@@ -28,3 +29,6 @@ class BaseEntity:
     )
 
     PK_FIELD = "id"
+
+    def to_model(self) -> BaseModel:
+        raise NotImplementedError()

architectonics-0.0.26/architectonics/infrastructure/repositories/base_exceptions.py

@@ -0,0 +1,10 @@
+class ObjectNotFoundException(Exception):
+    """Exception raised when can't find a row."""
+
+
+class ObjectAlreadyExistsException(Exception):
+    """Exception raised when unique constraint failes."""
+
+
+class IntegrityErrorException(Exception):
+    """Exception raised when integrity constraint fails."""

architectonics-0.0.26/architectonics/infrastructure/repositories/base_repository.py

@@ -0,0 +1,149 @@
+from abc import ABC
+from typing import Callable
+
+from asyncpg.exceptions import ForeignKeyViolationError
+from sqlalchemy import delete, select, update
+from sqlalchemy.exc import IntegrityError
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from architectonics.core.models.base_model import BaseModel
+from infrastructure.entities.base_entity import BaseEntity
+from infrastructure.repositories.base_exceptions import (
+    IntegrityErrorException,
+    ObjectAlreadyExistsException,
+    ObjectNotFoundException,
+)
+
+
+class AbstractBaseRepository(ABC):
+    _entity: type[BaseEntity] = NotImplemented
+    _session: Callable = NotImplemented
+    _integrity_error: type[IntegrityError] = IntegrityError
+
+    def get_session(self) -> AsyncSession:
+        return self._session()
+
+    @property
+    def model_fields(self):
+        return self._entity.__table__.columns
+
+    async def create_model(
+        self,
+        values: dict[str, any],
+    ) -> BaseModel:
+
+        entity = self._entity(**values)
+
+        async with self.get_session() as session:
+            session.add(entity)
+
+            try:
+                await session.commit()
+            except self._integrity_error as e:
+                raise ObjectAlreadyExistsException(e)
+
+            return entity.to_model()
+
+    async def get_model_by_id(
+        self,
+        model_id: str,
+    ) -> BaseModel:
+
+        statement = select(
+            self._entity,
+        ).where(
+            self._entity.id == model_id,
+        )
+
+        async with self.get_session() as session:
+
+            result = await session.execute(
+                statement=statement,
+            )
+
+            entity = result.scalars().first()
+
+            if entity is None:
+                raise ObjectNotFoundException()
+
+            return entity.to_model()
+
+    async def update_model(
+        self,
+        model_id: str,
+        values: dict[str, any],
+    ) -> BaseModel:
+
+        filtered_values = {k: v for k, v in values.items() if v is not None}
+
+        update_statement = (
+            update(
+                self._entity,
+            )
+            .where(
+                self._entity.id == model_id,
+            )
+            .values(
+                **filtered_values,
+            )
+        )
+
+        get_statement = select(
+            self._entity,
+        ).where(
+            self._entity.id == model_id,
+        )
+
+        async with self.get_session() as session:
+            try:
+                result = await session.execute(update_statement)
+
+                await session.commit()
+            except self._integrity_error as e:
+                orig = getattr(e.orig, "__cause__", None)
+
+                if isinstance(orig, ForeignKeyViolationError):
+                    raise ObjectNotFoundException()
+
+                raise IntegrityErrorException(e)
+
+            result = await session.execute(get_statement)
+            model = result.scalars().first()
+
+            if model is None:
+                raise ObjectNotFoundException()
+
+            return model.to_model()
+
+    async def delete_model(
+        self,
+        model_id: str,
+    ) -> None:
+
+        statement = delete(
+            self._entity,
+        ).where(
+            self._entity.id == model_id,
+        )
+
+        async with self.get_session() as session:
+            await session.execute(statement)
+            await session.commit()
+
+    async def get_models_list(
+        self,
+    ) -> list[BaseModel]:
+
+        statement = select(
+            self._entity,
+        )
+
+        async with self.get_session() as session:
+
+            result = await session.execute(
+                statement=statement,
+            )
+
+            models = result.scalars().all()
+
+            return [model.to_model() for model in models]

{architectonics-0.0.25 → architectonics-0.0.26}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "architectonics"
-version = "0.0.25"
+version = "0.0.26"
 description = ""
 authors = ["Your Name <you@example.com>"]
 

{architectonics-0.0.25 → architectonics-0.0.26}/setup.py

@@ -8,9 +8,11 @@ packages = \
  'architectonics.core',
  'architectonics.core.config',
  'architectonics.core.factory',
+ 'architectonics.core.models',
  'architectonics.infrastructure',
  'architectonics.infrastructure.config',
- 'architectonics.infrastructure.entities']
+ 'architectonics.infrastructure.entities',
+ 'architectonics.infrastructure.repositories']
 
 package_data = \
 {'': ['*']}
@@ -27,7 +29,7 @@ install_requires = \
 
 setup_kwargs = {
     'name': 'architectonics',
-    'version': '0.0.25',
+    'version': '0.0.26',
     'description': '',
     'long_description': None,
     'author': 'Your Name',