apexdevkit 1.15.4__tar.gz → 1.15.6__tar.gz

{apexdevkit-1.15.4 → apexdevkit-1.15.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: apexdevkit
-Version: 1.15.4
+Version: 1.15.6
 Summary: Apex Development Tools for python.
 Author: Apex Dev
 Author-email: dev@apex.ge
@@ -12,6 +12,7 @@ Classifier: Programming Language :: Python :: 3.13
 Requires-Dist: fastapi
 Requires-Dist: httpx
 Requires-Dist: pymongo
+Requires-Dist: pymssql (>=2.3.1,<3.0.0)
 Requires-Dist: python-dotenv
 Requires-Dist: sentry-sdk[fastapi]
 Requires-Dist: uvicorn

{apexdevkit-1.15.4 → apexdevkit-1.15.6}/apexdevkit/repository/__init__.py

@@ -12,6 +12,7 @@ from apexdevkit.repository.in_memory import (
     KeyValueStore,
 )
 from apexdevkit.repository.interface import Repository
+from apexdevkit.repository.mssql import MsSqlRepository
 
 __all__ = [
     "Connection",
@@ -24,4 +25,5 @@ __all__ = [
     "KeyValueStore",
     "Repository",
     "RepositoryBase",
+    "MsSqlRepository",
 ]

apexdevkit-1.15.6/apexdevkit/repository/connector.py

@@ -0,0 +1,112 @@
+from __future__ import annotations
+
+import sqlite3
+from contextlib import AbstractContextManager
+from dataclasses import dataclass
+from functools import cached_property
+from typing import Any, ContextManager
+
+import pymssql
+from pymongo import MongoClient
+from pymssql import Connection as _Connection
+from pymssql import Cursor
+
+from apexdevkit.repository import Connection
+
+
+@dataclass(frozen=True)
+class SqliteFileConnector:
+    dsn: str
+
+    def connect(self) -> ContextManager[Connection]:
+        connection = sqlite3.connect(self.dsn)
+        connection.row_factory = sqlite3.Row
+
+        return connection
+
+
+@dataclass(frozen=True)
+class SqliteInMemoryConnector:
+    dsn: str = ":memory:"
+
+    def connect(self) -> ContextManager[Connection]:
+        return self._connection
+
+    @cached_property
+    def _connection(self) -> ContextManager[Connection]:
+        connection = sqlite3.connect(self.dsn, check_same_thread=False)
+        connection.row_factory = sqlite3.Row
+
+        return connection
+
+
+@dataclass(frozen=True)
+class PyMongoConnector:
+    dsn: str
+
+    def connect(self) -> ContextManager[MongoClient[Any]]:
+        return MongoClient(self.dsn)
+
+
+@dataclass(frozen=True)
+class MsSqlConnector:
+    db_host: str
+    db_user: str
+    db_password: str
+    db_name: str
+    db_tds_version = "7.0"
+
+    def connect(self) -> ContextManager[Connection]:
+        return ConnectionContextManager(self._connection())
+
+    def _connection(self) -> Connection:
+        return MsSqlConnectionAdapter(
+            pymssql.connect(
+                tds_version=self.db_tds_version,
+                server=self.db_host,
+                user=self.db_user,
+                password=self.db_password,
+                database=self.db_name,
+                as_dict=True,
+                autocommit=True,
+            )
+        )
+
+
+class ConnectionContextManager(AbstractContextManager[Connection]):
+    def __init__(self, connection: Connection) -> None:
+        self.connection = connection
+
+    def __enter__(self) -> Connection:
+        return self.connection
+
+    def __exit__(self, exc_type: Any, exc_value: Any, traceback: Any) -> None:
+        self.connection.cursor().close()
+
+
+@dataclass
+class MsSqlConnectionAdapter:
+    connection: _Connection
+
+    def cursor(self) -> MsSqlCursorAdapter:
+        return MsSqlCursorAdapter(self.connection.cursor())
+
+
+@dataclass
+class MsSqlCursorAdapter:
+    cursor: Cursor
+
+    def execute(self, *args: Any, **kwargs: Any) -> Any:
+        self.cursor.execute(*args, **kwargs)  # type : ignore
+
+    def executemany(self, *args: Any, **kwargs: Any) -> Any:
+        self.cursor.executemany(*args, **kwargs)  # type : ignore
+
+    def fetchone(self, *args: Any, **kwargs: Any) -> Any:
+        return self.cursor.fetchone()  # type : ignore
+
+    def fetchall(self, *args: Any, **kwargs: Any) -> Any:
+        return self.cursor.fetchall()  # type : ignore
+
+    def close(self) -> None:
+        self.cursor.close()  # type : ignore

apexdevkit-1.15.6/apexdevkit/repository/mssql.py

@@ -0,0 +1,104 @@
+from __future__ import annotations
+
+from dataclasses import dataclass
+from typing import Any, Iterator, Protocol, TypeVar
+
+from pymssql.exceptions import DatabaseError
+
+from apexdevkit.error import DoesNotExistError, ExistsError
+from apexdevkit.repository import Database, DatabaseCommand, RepositoryBase
+
+ItemT = TypeVar("ItemT")
+
+
+@dataclass
+class MsSqlRepository(RepositoryBase[ItemT]):
+    db: Database
+    table: SqlTable[ItemT]
+
+    def __iter__(self) -> Iterator[ItemT]:
+        for raw in self.db.execute(self.table.select_all()).fetch_all():
+            yield self.table.load(raw)
+
+    def __len__(self) -> int:
+        raw = self.db.execute(self.table.count_all()).fetch_one()
+
+        try:
+            return int(raw["n_items"])
+        except KeyError:
+            raise UnknownError(raw)
+
+    def delete(self, item_id: str) -> None:
+        self.db.execute(self.table.delete(item_id)).fetch_none()
+
+    def delete_all(self) -> None:
+        self.db.execute(self.table.delete_all()).fetch_none()
+
+    def create(self, item: ItemT) -> ItemT:
+        try:
+            return self.table.load(self.db.execute(self.table.insert(item)).fetch_one())
+        except DatabaseError as e:
+            e = MssqlException(e)
+
+            if e.is_duplication():
+                raise self.table.exists(item)
+
+            raise UnknownError(e.message)
+
+    def read(self, item_id: str) -> ItemT:
+        raw = self.db.execute(self.table.select(item_id)).fetch_one()
+
+        if not raw:
+            raise DoesNotExistError(item_id)
+
+        return self.table.load(raw)
+
+    def update(self, item: ItemT) -> None:
+        self.db.execute(self.table.update(item)).fetch_none()
+
+
+class SqlTable(Protocol[ItemT]):  # pragma: no cover
+    def count_all(self) -> DatabaseCommand:
+        pass
+
+    def insert(self, item: ItemT) -> DatabaseCommand:
+        pass
+
+    def select(self, item_id: str) -> DatabaseCommand:
+        pass
+
+    def select_all(self) -> DatabaseCommand:
+        pass
+
+    def delete(self, item_id: str) -> DatabaseCommand:
+        pass
+
+    def delete_all(self) -> DatabaseCommand:
+        pass
+
+    def update(self, item: ItemT) -> DatabaseCommand:
+        pass
+
+    def load(self, data: dict[str, Any]) -> ItemT:
+        pass
+
+    def exists(self, duplicate: ItemT) -> ExistsError:
+        pass
+
+
+@dataclass
+class MssqlException:
+    code: int
+    message: str
+
+    def __init__(self, e: DatabaseError):
+        self.code = e.args[0]
+        self.message = e.args[1].decode()
+
+    def is_duplication(self) -> bool:
+        return self.code in [2627, 70003]
+
+
+@dataclass
+class UnknownError(Exception):
+    raw: Any

{apexdevkit-1.15.4 → apexdevkit-1.15.6}/apexdevkit/server.py

@@ -24,7 +24,7 @@ class UvicornServer:
     port: int = 8000
     path: str = ""
 
-    on_startup: Callable[[], None] = field(init=False, default=_do_nothing)
+    startup_handlers: list[Callable[[], None]] = field(default_factory=list)
 
     @classmethod
     def from_env(cls, path: str = ".env") -> UvicornServer:
@@ -49,7 +49,7 @@ class UvicornServer:
         return self
 
     def before_run(self, execute: Callable[[], None]) -> UvicornServer:
-        self.on_startup = execute
+        self.startup_handlers.append(execute)
 
         return self
 
@@ -64,6 +64,10 @@ class UvicornServer:
             log_config=self.logging_config,
         )
 
+    def on_startup(self) -> None:
+        for handler in self.startup_handlers:
+            handler()
+
 
 @dataclass
 class Sentry:

{apexdevkit-1.15.4 → apexdevkit-1.15.6}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "apexdevkit"
-version = "1.15.4"
+version = "1.15.6"
 description = "Apex Development Tools for python."
 authors = ["Apex Dev <dev@apex.ge>"]
 readme = "README.md"
@@ -13,6 +13,7 @@ pymongo = "*"
 uvicorn = "*"
 sentry-sdk = {extras = ["fastapi"], version = "*"}
 python-dotenv = "*"
+pymssql = "^2.3.1"
 
 [tool.poetry.group.dev.dependencies]
 pytest = "*"

apexdevkit-1.15.4/apexdevkit/repository/connector.py

@@ -1,42 +0,0 @@
-import sqlite3
-from dataclasses import dataclass
-from functools import cached_property
-from typing import Any, ContextManager
-
-from pymongo import MongoClient
-
-from apexdevkit.repository import Connection
-
-
-@dataclass(frozen=True)
-class SqliteFileConnector:
-    dsn: str
-
-    def connect(self) -> ContextManager[Connection]:
-        connection = sqlite3.connect(self.dsn)
-        connection.row_factory = sqlite3.Row
-
-        return connection
-
-
-@dataclass(frozen=True)
-class SqliteInMemoryConnector:
-    dsn: str = ":memory:"
-
-    def connect(self) -> ContextManager[Connection]:
-        return self._connection
-
-    @cached_property
-    def _connection(self) -> ContextManager[Connection]:
-        connection = sqlite3.connect(self.dsn, check_same_thread=False)
-        connection.row_factory = sqlite3.Row
-
-        return connection
-
-
-@dataclass(frozen=True)
-class PyMongoConnector:
-    dsn: str
-
-    def connect(self) -> ContextManager[MongoClient[Any]]:
-        return MongoClient(self.dsn)