apexdevkit 1.17.5__tar.gz → 1.17.6__tar.gz

{apexdevkit-1.17.5 → apexdevkit-1.17.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: apexdevkit
-Version: 1.17.5
+Version: 1.17.6
 Summary: Apex Development Tools for python.
 Author: Apex Dev
 Author-email: dev@apex.ge

apexdevkit-1.17.6/apexdevkit/repository/mssql.py

@@ -0,0 +1,467 @@
+from __future__ import annotations
+
+from dataclasses import dataclass
+from typing import Any, Generic, Iterable, Iterator, TypeVar
+
+from pymssql.exceptions import DatabaseError
+
+from apexdevkit.error import DoesNotExistError, ExistsError
+from apexdevkit.formatter import Formatter
+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(Generic[ItemT]):  # pragma: no cover
+    def count_all(self) -> DatabaseCommand:
+        raise NotImplementedError
+
+    def insert(self, item: ItemT) -> DatabaseCommand:
+        raise NotImplementedError
+
+    def select(self, item_id: str) -> DatabaseCommand:
+        raise NotImplementedError
+
+    def select_all(self) -> DatabaseCommand:
+        raise NotImplementedError
+
+    def delete(self, item_id: str) -> DatabaseCommand:
+        raise NotImplementedError
+
+    def delete_all(self) -> DatabaseCommand:
+        raise NotImplementedError
+
+    def update(self, item: ItemT) -> DatabaseCommand:
+        raise NotImplementedError
+
+    def load(self, data: dict[str, Any]) -> ItemT:
+        raise NotImplementedError
+
+    def exists(self, duplicate: ItemT) -> ExistsError:
+        raise NotImplementedError
+
+
+@dataclass
+class SqlTableDecorator(Generic[ItemT]):
+    table: DefaultSqlTable[ItemT]
+
+    def count_all(self) -> DatabaseCommand:
+        return self.table.count_all()
+
+    def insert(self, item: ItemT) -> DatabaseCommand:
+        return self.table.insert(item)
+
+    def select(self, item_id: str) -> DatabaseCommand:
+        return self.table.select(item_id)
+
+    def select_all(self) -> DatabaseCommand:
+        return self.table.select_all()
+
+    def delete(self, item_id: str) -> DatabaseCommand:
+        return self.table.delete(item_id)
+
+    def delete_all(self) -> DatabaseCommand:
+        return self.table.delete_all()
+
+    def update(self, item: ItemT) -> DatabaseCommand:
+        return self.table.update(item)
+
+    def load(self, data: dict[str, Any]) -> ItemT:
+        return self.table.load(data)
+
+    def exists(self, duplicate: ItemT) -> ExistsError:
+        return self.table.exists(duplicate)
+
+
+@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
+
+
+@dataclass(frozen=True)
+class MsSqlTableBuilder(Generic[ItemT]):
+    username: str | None = None
+    schema: str | None = None
+    table: str | None = None
+    formatter: Formatter[dict[str, Any], ItemT] | None = None
+    fields: list[MsSqlField] | None = None
+    ordering: list[str] | None = None
+    parent_field: str | None = None
+    parent_value: Any | None = None
+
+    def with_username(self, value: str) -> MsSqlTableBuilder[ItemT]:
+        return MsSqlTableBuilder[ItemT](
+            value,
+            self.schema,
+            self.table,
+            self.formatter,
+            self.fields,
+            self.ordering,
+            self.parent_field,
+            self.parent_value,
+        )
+
+    def with_schema(self, value: str) -> MsSqlTableBuilder[ItemT]:
+        return MsSqlTableBuilder[ItemT](
+            self.username,
+            value,
+            self.table,
+            self.formatter,
+            self.fields,
+            self.ordering,
+            self.parent_field,
+            self.parent_value,
+        )
+
+    def with_table(self, value: str) -> MsSqlTableBuilder[ItemT]:
+        return MsSqlTableBuilder[ItemT](
+            self.username,
+            self.schema,
+            value,
+            self.formatter,
+            self.fields,
+            self.ordering,
+            self.parent_field,
+            self.parent_value,
+        )
+
+    def with_formatter(
+        self, value: Formatter[dict[str, Any], ItemT]
+    ) -> MsSqlTableBuilder[ItemT]:
+        return MsSqlTableBuilder[ItemT](
+            self.username,
+            self.schema,
+            self.table,
+            value,
+            self.fields,
+            self.ordering,
+            self.parent_field,
+            self.parent_value,
+        )
+
+    def with_fields(self, fields: Iterable[str]) -> MsSqlTableBuilder[ItemT]:
+        return MsSqlTableBuilder[ItemT](
+            self.username,
+            self.schema,
+            self.table,
+            self.formatter,
+            [MsSqlField(field, field == "id") for field in list(fields)],
+            self.ordering,
+            self.parent_field,
+            self.parent_value,
+        )
+
+    def with_order_fields(self, ordering: Iterable[str]) -> MsSqlTableBuilder[ItemT]:
+        ordered = list(ordering)
+        assert self.fields is not None, "Set fields first."
+
+        for key in ordered:
+            if key not in [field.name for field in self.fields]:
+                raise ValueError("Missing fields in the table.")
+        return MsSqlTableBuilder[ItemT](
+            self.username,
+            self.schema,
+            self.table,
+            self.formatter,
+            self.fields,
+            ordered,
+            self.parent_field,
+            self.parent_value,
+        )
+
+    def with_id(self, identifier: str) -> MsSqlTableBuilder[ItemT]:
+        assert self.fields is not None, "Set fields first."
+        if identifier not in [field.name for field in self.fields]:
+            raise ValueError("Missing fields in the table.")
+
+        return MsSqlTableBuilder[ItemT](
+            self.username,
+            self.schema,
+            self.table,
+            self.formatter,
+            [MsSqlField(field.name, field.name == identifier) for field in self.fields],
+            self.ordering,
+            self.parent_field,
+            self.parent_value,
+        )
+
+    def with_parent(
+        self, parent_field: str, parent_value: Any
+    ) -> MsSqlTableBuilder[ItemT]:
+        assert self.fields is not None, "Set fields first."
+        if parent_field not in [field.name for field in self.fields]:
+            raise ValueError("Missing fields in the table.")
+
+        return MsSqlTableBuilder[ItemT](
+            self.username,
+            self.schema,
+            self.table,
+            self.formatter,
+            self.fields,
+            self.ordering,
+            parent_field,
+            parent_value,
+        )
+
+    def build(self) -> SqlTable[ItemT]:
+        if not self.schema or not self.table or not self.formatter or not self.fields:
+            raise ValueError("Cannot build sql table.")
+
+        return DefaultSqlTable(
+            self.username,
+            self.schema,
+            self.table,
+            self.formatter,
+            self.fields,
+            self.ordering,
+            self.parent_field,
+            self.parent_value,
+        )
+
+
+@dataclass(frozen=True)
+class DefaultSqlTable(SqlTable[ItemT]):
+    username: str | None
+    schema: str
+    table: str
+    formatter: Formatter[dict[str, Any], ItemT]
+    fields: list[MsSqlField]
+    ordering: list[str] | None
+    parent_key: str | None
+    parent_value: Any | None
+
+    def count_all(self) -> DatabaseCommand:
+        where_statement = ""
+        if self.parent_key is not None:
+            where_statement += (
+                "WHERE [" + self.parent_key + "] = " + self._parent_value_sql
+            )
+
+        return DatabaseCommand(f"""
+            {self._user_check}
+            SELECT count(*) AS n_items
+            FROM [{self.schema}].[{self.table}]
+            {where_statement}
+            REVERT
+        """)
+
+    def insert(self, item: ItemT) -> DatabaseCommand:
+        dumped = self.formatter.dump(item)
+        if self.parent_key is not None:
+            dumped[self.parent_key] = self.parent_value
+
+        columns = ", ".join(["[" + field.name + "]" for field in self.fields])
+        placeholders = ", ".join([f"%({key.name})s" for key in self.fields])
+        output = ", ".join(["INSERTED." + field.name for field in self.fields])
+
+        return DatabaseCommand(f"""
+            {self._user_check}
+            INSERT INTO [{self.schema}].[{self.table}] (
+                {columns}
+            ) OUTPUT
+                {output}
+            VALUES (
+                {placeholders}
+            )
+            REVERT
+        """).with_data(dumped)
+
+    def select(self, item_id: str) -> DatabaseCommand:
+        raw: dict[str, Any] = {self._id: item_id}
+        where_statement = f"WHERE [{self._id}] = %({self._id})s"
+        if self.parent_key is not None:
+            raw[self.parent_key] = self.parent_value
+            where_statement += (
+                ", [" + self.parent_key + "] = %(" + self.parent_key + ")s"
+            )
+
+        columns = ", ".join(["[" + field.name + "]" for field in self.fields])
+
+        return DatabaseCommand(f"""
+            {self._user_check}
+            SELECT
+                {columns} 
+            FROM [{self.schema}].[{self.table}]
+            {where_statement}
+            REVERT
+        """).with_data(raw)
+
+    def select_all(self) -> DatabaseCommand:
+        columns = ", ".join(["[" + field.name + "]" for field in self.fields])
+        where_statement = ""
+        if self.parent_key is not None:
+            where_statement += (
+                "WHERE [" + self.parent_key + "] = " + self._parent_value_sql
+            )
+
+        return DatabaseCommand(f"""
+            {self._user_check}
+            SELECT
+                {columns}
+            FROM [{self.schema}].[{self.table}]
+            {where_statement}
+            {self._order}
+            REVERT
+        """)
+
+    def update(self, item: ItemT) -> DatabaseCommand:
+        dumped = self.formatter.dump(item)
+        if self.parent_key is not None:
+            dumped[self.parent_key] = self.parent_value
+
+        where_statement = f"WHERE [{self._id}] = %({self._id})s"
+        if self.parent_key is not None:
+            where_statement += (
+                ", [" + self.parent_key + "] = %(" + self.parent_key + ")s"
+            )
+
+        updates = ", ".join(
+            [
+                f"{field.name} = %({field.name})s"
+                for field in self.fields
+                if not field.is_id and field.name != self.parent_key
+            ]
+        )
+
+        return DatabaseCommand(f"""
+            {self._user_check}
+            UPDATE [{self.schema}].[{self.table}]
+            SET
+                {updates}
+            {where_statement}
+            REVERT
+        """).with_data(dumped)
+
+    def delete(self, item_id: str) -> DatabaseCommand:
+        raw: dict[str, Any] = {self._id: item_id}
+        where_statement = f"WHERE [{self._id}] = %({self._id})s"
+        if self.parent_key is not None:
+            raw[self.parent_key] = self.parent_value
+            where_statement += (
+                ", [" + self.parent_key + "] = %(" + self.parent_key + ")s"
+            )
+
+        return DatabaseCommand(f"""
+            {self._user_check}
+            DELETE
+            FROM [{self.schema}].[{self.table}]
+            {where_statement}
+            REVERT
+        """).with_data(raw)
+
+    def delete_all(self) -> DatabaseCommand:
+        where_statement = ""
+        if self.parent_key is not None:
+            where_statement += (
+                "WHERE [" + self.parent_key + "] = " + self._parent_value_sql
+            )
+
+        return DatabaseCommand(f"""
+            {self._user_check}
+            DELETE
+            FROM [{self.schema}].[{self.table}]
+            {where_statement}
+            REVERT
+        """)
+
+    def load(self, data: dict[str, Any]) -> ItemT:
+        return self.formatter.load(data)
+
+    def exists(self, duplicate: ItemT) -> ExistsError:
+        raw = self.formatter.dump(duplicate)
+        return ExistsError(duplicate).with_duplicate(
+            lambda i: f"{self._id}<{raw[self._id]}>"
+        )
+
+    @property
+    def _id(self) -> str:
+        result = next((field for field in self.fields if field.is_id), None)
+        if result is None:
+            raise ValueError("Id field is required.")
+        return result.name
+
+    @property
+    def _user_check(self) -> str:
+        if self.username is not None:
+            return f"EXECUTE AS USER = '{self.username}'"
+        else:
+            return ""
+
+    @property
+    def _order(self) -> str:
+        if self.ordering is not None and len(self.ordering) > 0:
+            return "ORDER BY " + ", ".join(self.ordering)
+        else:
+            return ""
+
+    @property
+    def _parent_value_sql(self) -> str:
+        if isinstance(self.parent_value, str):
+            return "'" + self.parent_value + "'"
+        else:
+            return str(self.parent_value)
+
+
+@dataclass(frozen=True)
+class MsSqlField:
+    name: str
+    is_id: bool

{apexdevkit-1.17.5 → apexdevkit-1.17.6}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "apexdevkit"
-version = "1.17.5"
+version = "1.17.6"
 description = "Apex Development Tools for python."
 authors = ["Apex Dev <dev@apex.ge>"]
 readme = "README.md"

apexdevkit-1.17.5/apexdevkit/repository/mssql.py

@@ -1,104 +0,0 @@
-from __future__ import annotations
-
-from dataclasses import dataclass
-from typing import Any, Generic, Iterator, 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(Generic[ItemT]):  # pragma: no cover
-    def count_all(self) -> DatabaseCommand:
-        raise NotImplementedError
-
-    def insert(self, item: ItemT) -> DatabaseCommand:
-        raise NotImplementedError
-
-    def select(self, item_id: str) -> DatabaseCommand:
-        raise NotImplementedError
-
-    def select_all(self) -> DatabaseCommand:
-        raise NotImplementedError
-
-    def delete(self, item_id: str) -> DatabaseCommand:
-        raise NotImplementedError
-
-    def delete_all(self) -> DatabaseCommand:
-        raise NotImplementedError
-
-    def update(self, item: ItemT) -> DatabaseCommand:
-        raise NotImplementedError
-
-    def load(self, data: dict[str, Any]) -> ItemT:
-        raise NotImplementedError
-
-    def exists(self, duplicate: ItemT) -> ExistsError:
-        raise NotImplementedError
-
-
-@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