xn-model 1.0.3__tar.gz → 1.0.5__tar.gz

{xn_model-1.0.3/xn_model.egg-info → xn_model-1.0.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: xn-model
-Version: 1.0.3
+Version: 1.0.5
 Summary: Base model for xn-api
 Author-email: Mike Artemiev <mixartemev@gmail.com>
 License: MIT

xn_model-1.0.5/x_model/__init__.py

@@ -0,0 +1,12 @@
+from types import ModuleType
+
+from tortoise import Tortoise, connections
+from tortoise.backends.asyncpg import AsyncpgDBClient
+
+
+async def init_db(dsn: str, models: ModuleType, create_tables: bool = True) -> AsyncpgDBClient | str:
+    await Tortoise.init(db_url=dsn, modules={"models": [models]})
+    if create_tables:
+        await Tortoise.generate_schemas()
+    cn: AsyncpgDBClient = connections.get("default")
+    return cn

{xn_model-1.0.3 → xn_model-1.0.5}/x_model/field.py

@@ -1,7 +1,7 @@
 from enum import IntEnum
 from typing import Any
 
-from asyncpg import Point, Polygon, Range, Box
+from asyncpg import Range  # Box, Point, Polygon,
 from tortoise import Model
 from tortoise.contrib.postgres.fields import ArrayField
 from tortoise.fields import Field, SmallIntField, IntField, FloatField, DatetimeField
@@ -51,23 +51,23 @@ class RangeField(CollectionField[Range]):
         return value
 
 
-class PointField(CollectionField[Point]):
-    SQL_TYPE = "POINT"
-    field_type = Point
-    base_field = FloatField
-    labels = ("lat", "lon")
-
-
-class PolygonField(ListField[Polygon]):
-    SQL_TYPE = "POLYGON"
-    field_type = Polygon
-    base_field = PointField
-
-
-class BoxField(ListField[Box]):
-    SQL_TYPE = "BOX"
-    field_type = Box
-    base_field = PointField
+# class PointField(CollectionField[Point]):
+#     SQL_TYPE = "POINT"
+#     field_type = Point
+#     base_field = FloatField
+#     labels = ("lat", "lon")
+#
+#
+# class PolygonField(ListField[Polygon]):
+#     SQL_TYPE = "POLYGON"
+#     field_type = Polygon
+#     base_field = PointField
+#
+#
+# class BoxField(ListField[Box]):
+#     SQL_TYPE = "BOX"
+#     field_type = Box
+#     base_field = PointField
 
 
 class DatetimeSecField(DatetimeField):

{xn_model-1.0.3 → xn_model-1.0.5}/x_model/models.py

@@ -1,16 +1,10 @@
 from datetime import datetime
-
-from pydantic import ConfigDict
-from tortoise import Model as BaseModel
+from pydantic import ConfigDict, BaseModel
+from tortoise import Model as TrtModel
 from tortoise.contrib.pydantic import pydantic_model_creator, PydanticModel
-from tortoise.fields import DatetimeField, IntField
-
-from x_model import HTTPException, FailReason
+from tortoise.fields import IntField
 
-
-class DatetimeSecField(DatetimeField):
-    class _db_postgres:
-        SQL_TYPE = "TIMESTAMPTZ(0)"
+from x_model.field import DatetimeSecField
 
 
 class TsTrait:
@@ -18,10 +12,19 @@ class TsTrait:
     updated_at: datetime | None = DatetimeSecField(auto_now=True)
 
 
-class Model(BaseModel):
+class PydIn(BaseModel):
+    _unq: list[str] = ["id"]
+
+    def df_unq(self) -> dict:
+        d = self.model_dump(exclude_none=True)
+        return {**{k: d.pop(k) for k in self._unq}, "defaults": d}
+
+
+class Model(TrtModel):
     id: int = IntField(True)
 
-    _pyd: type[PydanticModel] = None
+    _pyd: type[PydanticModel] = None  # overridable
+    _pydIn: type[PydIn | PydanticModel] = None  # overridable
     _name: tuple[str] = ("name",)
     _sorts: tuple[str] = ("-id",)
 
@@ -34,12 +37,16 @@ class Model(BaseModel):
             cls._pyd = pydantic_model_creator(cls, name=cls.__name__)
         return cls._pyd
 
-    # # # CRUD Methods # # #
     @classmethod
-    async def get_one(cls, id_: int, **filters) -> PydanticModel:
-        if obj := await cls.get_or_none(id=id_, **filters):
-            return await cls.pyd().from_tortoise_orm(obj)
-        raise HTTPException(reason=FailReason.path, status_=404, parent=f"{cls.__name__}#{id_} not found")
+    def pyd_in(cls):
+        if not cls._pydIn:
+            cls._pydIn = pydantic_model_creator(
+                cls, name=cls.__name__ + "In", exclude_readonly=True, meta_override=cls.PydanticMetaIn
+            )
+        return cls._pydIn
+
+    async def one(self) -> PydanticModel:
+        return await self.pyd().from_tortoise_orm(self)
 
     @classmethod
     async def get_or_create_by_name(cls, name: str, attr_name: str = None, def_dict: dict = None) -> "Model":
@@ -55,11 +62,16 @@ class Model(BaseModel):
         # exclude: tuple[str, ...] = ("Meta",)
         # computed: tuple[str, ...] = ()
         # backward_relations: bool = True
-        max_recursion: int = 0  # default: 3
+        max_recursion: int = 1  # default: 3
         # allow_cycles: bool = False
         # exclude_raw_fields: bool = True
         # sort_alphabetically: bool = False
         # model_config: ConfigDict | None = None
 
+    class PydanticMetaIn:
+        backward_relations: bool = False
+        max_recursion: int = 0
+        exclude_raw_fields: bool = False
+
     class Meta:
         abstract = True

{xn_model-1.0.3 → xn_model-1.0.5/xn_model.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: xn-model
-Version: 1.0.3
+Version: 1.0.5
 Summary: Base model for xn-api
 Author-email: Mike Artemiev <mixartemev@gmail.com>
 License: MIT

{xn_model-1.0.3 → xn_model-1.0.5}/xn_model.egg-info/SOURCES.txt

@@ -10,7 +10,6 @@ x_model/__init__.py
 x_model/field.py
 x_model/func.py
 x_model/models.py
-x_model/pydantic.py
 xn_model.egg-info/PKG-INFO
 xn_model.egg-info/SOURCES.txt
 xn_model.egg-info/dependency_links.txt

xn_model-1.0.3/x_model/__init__.py

@@ -1,36 +0,0 @@
-import logging
-from enum import IntEnum
-from types import ModuleType
-
-from tortoise import Tortoise, connections
-from tortoise.backends.asyncpg import AsyncpgDBClient
-
-
-async def init_db(dsn: str, models: ModuleType, create_tables: bool = False) -> AsyncpgDBClient | str:
-    await Tortoise.init(db_url=dsn, modules={"models": [models]})
-    if create_tables:
-        await Tortoise.generate_schemas()
-    cn: AsyncpgDBClient = connections.get("default")
-    return cn
-
-
-class FailReason(IntEnum):
-    body = 8
-    query = 9
-    path = 10
-    host = 11
-    protocol = 12
-    method = 13
-
-
-class HTTPException(Exception):
-    def __init__(
-        self,
-        reason: IntEnum,
-        parent: Exception | str = None,
-        status_: int = 400,
-        hdrs: dict = None,
-    ) -> None:
-        detail = f"{reason.name}{f': {parent}' if parent else ''}"
-        logging.error(detail)
-        super().__init__(status_, detail, hdrs)

xn_model-1.0.3/x_model/pydantic.py

@@ -1,16 +0,0 @@
-from pydantic import BaseModel
-
-
-class Names(BaseModel):
-    # models for name endpoint for select2 inputs
-    class Name(BaseModel):
-        id: int
-        text: str
-        logo: str | None = None
-        selected: bool | None = None
-
-    class Pagination(BaseModel):
-        more: bool
-
-    results: list[Name]
-    pagination: Pagination