u-toolkit 0.1.19__py3-none-any.whl → 0.1.20__py3-none-any.whl

u_toolkit/fastapi/pagination.py

@@ -1,79 +0,0 @@
-from collections.abc import Sequence
-from math import ceil
-from typing import Annotated, Generic, overload
-
-from fastapi import Depends, Query
-from pydantic import BaseModel, Field, NonNegativeInt, PositiveInt
-from sqlalchemy import MappingResult, ScalarResult
-
-from u_toolkit.pydantic.type_vars import BaseModelT
-from u_toolkit.sqlalchemy.type_vars import DeclarativeBaseT
-
-
-class Page(BaseModel, Generic[BaseModelT]):
-    page_size: PositiveInt = Field(description="page size")
-    page_no: PositiveInt = Field(description="page number")
-
-    page_count: NonNegativeInt = Field(description="page count")
-    count: NonNegativeInt = Field(description="result count")
-
-    results: list[BaseModelT] = Field(description="results")
-
-
-class PageParamsModel(BaseModel):
-    page_size: int = Query(
-        50,
-        ge=1,
-        le=100,
-        description="page size",
-    )
-    page_no: PositiveInt = Query(
-        1,
-        description="page number",
-    )
-
-
-PageParams = Annotated[PageParamsModel, Depends()]
-
-
-@overload
-def page(
-    model_class: type[BaseModelT],
-    pagination: PageParamsModel,
-    count: int,
-    results: ScalarResult[DeclarativeBaseT],
-) -> Page[BaseModelT]: ...
-
-
-@overload
-def page(
-    model_class: type[BaseModelT],
-    pagination: PageParamsModel,
-    count: int,
-    results: MappingResult,
-) -> Page[BaseModelT]: ...
-
-
-@overload
-def page(
-    model_class: type[BaseModelT],
-    pagination: PageParamsModel,
-    count: int,
-    results: Sequence[DeclarativeBaseT],
-) -> Page[BaseModelT]: ...
-
-
-def page(
-    model_class: type[BaseModelT],
-    pagination: PageParamsModel,
-    count: int,
-    results,
-) -> Page[BaseModelT]:
-    results_ = [model_class.model_validate(i) for i in results]
-    return Page(
-        page_size=pagination.page_size,
-        page_no=pagination.page_no,
-        page_count=ceil(count / pagination.page_size),
-        count=count,
-        results=results_,  # type: ignore
-    )

u_toolkit/fastapi/responses.py

@@ -1,67 +0,0 @@
-from pydantic import BaseModel
-
-from u_toolkit.merge import deep_merge_dict
-
-from .exception import HTTPErrorInterface
-
-
-def _merge_responses(
-    target: dict,
-    source: dict,
-):
-    for status, response in target.items():
-        model_class = response.get("model")
-        if status in source:
-            source_model_class = source[status].get("model")
-            if source_model_class and model_class:
-                target[status]["model"] = model_class | source_model_class
-
-    for status, response in source.items():
-        if status not in target:
-            target[status] = response
-
-
-def error_responses(*errors: type[HTTPErrorInterface]):
-    source = {}
-
-    for e in errors:
-        model_class = e.response_class()
-        if e.status in source:
-            current: type[BaseModel] = source[e.status]["model"]
-
-            source[e.status] = {"model": current | model_class}
-        else:
-            deep_merge_dict(source, {e.status: {"model": model_class}})
-    return source
-
-
-Response = (
-    tuple[int, type[BaseModel]]
-    | dict[int, type[BaseModel]]
-    | int
-    | type[HTTPErrorInterface]
-)
-
-
-def build_responses(*responses: Response):
-    result = {}
-    errors: list[type[HTTPErrorInterface]] = []
-
-    for arg in responses:
-        status = None
-        response = {}
-        if isinstance(arg, tuple):
-            status, response = arg
-        elif isinstance(arg, dict):
-            for status_, response_ in arg.items():
-                result[status_] = {"model": response_}
-        elif isinstance(arg, int):
-            status = arg
-        else:
-            errors.append(arg)
-            continue
-
-        result[status] = {"model": response}
-
-    _merge_responses(result, error_responses(*errors))
-    return result

u_toolkit/logger.py

@@ -1,11 +0,0 @@
-from u_toolkit.enum import NameEnum, auto
-
-
-class LogLevel(NameEnum):
-    TRACE = auto()
-    DEBUG = auto()
-    INFO = auto()
-    SUCCESS = auto()
-    WARNING = auto()
-    ERROR = auto()
-    CRITICAL = auto()

u_toolkit/pydantic/fields.py

@@ -1,24 +0,0 @@
-from datetime import datetime
-from typing import Annotated
-
-from pydantic import Field, PlainSerializer
-
-from u_toolkit.datetime import to_naive, to_utc, to_utc_naive
-
-
-DBSmallInt = Annotated[int, Field(ge=-32768, le=32767)]
-DBInt = Annotated[int, Field(ge=-2147483648, le=2147483647)]
-DBBigInt = Annotated[
-    int, Field(ge=-9223372036854775808, le=9223372036854775807)
-]
-DBSmallSerial = Annotated[int, Field(ge=1, le=32767)]
-DBIntSerial = Annotated[int, Field(ge=1, le=2147483647)]
-DBBigintSerial = Annotated[int, Field(ge=1, le=9223372036854775807)]
-
-
-# 去除时区信息
-NaiveDatetime = Annotated[datetime, PlainSerializer(to_naive)]
-# 将时区转成 UTC
-UTCDateTime = Annotated[datetime, PlainSerializer(to_utc)]
-# 将时间转成 UTC, 并且去除时区信息
-UTCNaiveDateTime = Annotated[datetime, PlainSerializer(to_utc_naive)]

u_toolkit/pydantic/models.py

@@ -1,41 +0,0 @@
-from datetime import datetime
-from typing import Annotated, Any
-
-from pydantic import BaseModel, ConfigDict, WrapSerializer
-
-from u_toolkit.alias_generators import to_camel
-from u_toolkit.datetime import to_utc
-
-from .type_vars import BaseModelT
-
-
-class Model(BaseModel):
-    model_config = ConfigDict(
-        populate_by_name=True,
-        from_attributes=True,
-    )
-
-
-class CamelModel(Model):
-    model_config = ConfigDict(
-        alias_generator=to_camel,
-        populate_by_name=True,
-        from_attributes=True,
-        field_title_generator=to_camel,
-    )
-
-
-def convert_to_utc(value: Any, handler, info) -> dict[str, datetime]:
-    # Note that `helper` can actually help serialize the `value` for
-    # further custom serialization in case it's a subclass.
-    partial_result = handler(value, info)
-    if info.mode == "json":
-        return {
-            k: to_utc(datetime.fromisoformat(v))
-            for k, v in partial_result.items()
-        }
-
-    return {k: to_utc(v) for k, v in partial_result.items()}
-
-
-ConvertToUTCModel = Annotated[BaseModelT, WrapSerializer(convert_to_utc)]

u_toolkit/pydantic/type_vars.py

@@ -1,6 +0,0 @@
-from typing import TypeVar
-
-from pydantic import BaseModel
-
-
-BaseModelT = TypeVar("BaseModelT", bound=BaseModel)

u_toolkit/sqlalchemy/function.py

@@ -1,12 +0,0 @@
-import sqlalchemy as sa
-from sqlalchemy.orm import DeclarativeBase, QueryableAttribute
-
-
-def json_object_build(
-    label: str,
-    table: type[DeclarativeBase],
-    *attrs: QueryableAttribute | sa.Column,
-):
-    return sa.func.json_build_object(
-        *[sa.text(f"'{i.key}', {table.__tablename__}.{i.key}") for i in attrs]
-    ).label(label)

u_toolkit/sqlalchemy/orm/fields.py

@@ -1,20 +0,0 @@
-from functools import partial
-from typing import Annotated
-from uuid import UUID, uuid4
-
-from sqlalchemy import BIGINT
-from sqlalchemy.orm import mapped_column
-
-
-primary_key_column = partial(mapped_column, primary_key=True, sort_order=-9999)
-
-IntPrimaryKey = Annotated[int, primary_key_column()]
-BigIntPrimaryKey = Annotated[int, primary_key_column(BIGINT)]
-
-_auto_pk = partial(primary_key_column, autoincrement=True)
-
-AutoIntPrimaryKey = Annotated[int, _auto_pk()]
-AutoBigIntPrimaryKey = Annotated[int, _auto_pk(BIGINT)]
-
-
-UUID4PrimaryKey = Annotated[UUID, primary_key_column(default=uuid4)]

u_toolkit/sqlalchemy/orm/models.py

@@ -1,23 +0,0 @@
-from datetime import datetime
-
-from pydantic.alias_generators import to_snake
-from sqlalchemy import func
-from sqlalchemy.orm import Mapped, declared_attr, mapped_column
-
-
-class TablenameMixin:
-    # 当继承该类时, 会自动将表名转为 snake 名称形式
-    @declared_attr.directive
-    @classmethod
-    def __tablename__(cls) -> str:
-        return f"{to_snake(cls.__name__)}_tb"
-
-
-class TimeStampMixin:
-    # 当继承该类时, 会给表添加创建时间和更新时间字段
-    created_at: Mapped[datetime] = mapped_column(
-        server_default=func.now(), sort_order=9998
-    )
-    updated_at: Mapped[datetime | None] = mapped_column(
-        onupdate=func.now(), sort_order=9999
-    )

u_toolkit/sqlalchemy/table_info.py

@@ -1,17 +0,0 @@
-import sqlalchemy as sa
-from pydantic import TypeAdapter
-
-from .type_vars import DeclarativeBaseT
-
-
-class TableInfo:
-    def __init__(self, table: type[DeclarativeBaseT]) -> None:
-        self.table = table
-        self.columns = sa.inspect(table).c
-        self.adapters = {
-            i.name: (
-                TypeAdapter(i.type.python_type),
-                i.type.python_type,
-            )
-            for i in self.columns
-        }

u_toolkit/sqlalchemy/type_vars.py

@@ -1,6 +0,0 @@
-from typing import TypeVar
-
-from sqlalchemy.orm import DeclarativeBase
-
-
-DeclarativeBaseT = TypeVar("DeclarativeBaseT", bound=DeclarativeBase)

u_toolkit-0.1.19.dist-info/METADATA

@@ -1,11 +0,0 @@
-Metadata-Version: 2.4
-Name: u-toolkit
-Version: 0.1.19
-Summary: Add your description here
-Requires-Python: >=3.11
-Requires-Dist: pydantic>=2.11.3
-Provides-Extra: fastapi
-Requires-Dist: fastapi>=0.115.12; extra == 'fastapi'
-Requires-Dist: pydantic-settings>=2.9.1; extra == 'fastapi'
-Provides-Extra: sqlalchemy
-Requires-Dist: sqlalchemy>=2.0.40; extra == 'sqlalchemy'

u_toolkit-0.1.19.dist-info/RECORD

@@ -1,35 +0,0 @@
-u_toolkit/__init__.py,sha256=7f_bFg1UERA4gyh3HjCAjOZo2cBUgElpwgX5x-Xk2WA,238
-u_toolkit/alias_generators.py,sha256=KmPL1ViGJjptONffZUAX4ENvkldrwDylm_Ly5RYE8b4,963
-u_toolkit/datetime.py,sha256=GOG0xa6yKeqvFXJkImK5Pt7wsPXnHMlNKju8deniGJ4,644
-u_toolkit/decorators.py,sha256=rBZfXtWVfja6YMDqF9izjAiabRVyp4pAWOlwnN0HkUw,2396
-u_toolkit/enum.py,sha256=nEdcrvAkLc32YioEZSZJM6DClLM2G__mlJFQC64AvBo,895
-u_toolkit/function.py,sha256=HgZcWd4IydNqUdUbS6A5pXVcAponGUZ2A8YKuiwHGjg,259
-u_toolkit/helpers.py,sha256=QSkDZH90FfuNB0hWI1GA2KHFk1ajZMzNBr7ysuBdFTg,161
-u_toolkit/logger.py,sha256=NOmdR24QfSuo1EMnetyFioa8pA8OYceYTlQ4qiQcBdE,209
-u_toolkit/merge.py,sha256=kHrloud-nPti5j48zkdvaiy4mIJYqOVguixAjWC46kE,924
-u_toolkit/path.py,sha256=IkyIHcU9hKBCOZfF30FrKf4CfL-MH91fjeYF9EY7eos,128
-u_toolkit/signature.py,sha256=-Q6n28PYBYYdd2OXBKESeVkL2rYpV6EaY3IVwQmzezQ,2161
-u_toolkit/fastapi/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-u_toolkit/fastapi/cbv.py,sha256=vRYGeopzduO7Gmf5O5oz76zGOYqFtD8aGAe1oCcK8R4,12910
-u_toolkit/fastapi/config.py,sha256=kGpokR9XXr1KxMA1GVKYkCdKwqIQAIwOJ-v6sGbqzAQ,267
-u_toolkit/fastapi/exception.py,sha256=iZy1kjNkhtkLw6nl5x0_g3uvgpkpwutEsjfB7VvX9M0,4640
-u_toolkit/fastapi/helpers.py,sha256=BCMMLxa1c6BMA_rKq-hCi0iyEjrR3Z5rPMeTvgaVJB0,447
-u_toolkit/fastapi/lifespan.py,sha256=W1TwWymW7xtmntx59QBC4LQ6xQr3L7yuMMGj4U8hhTQ,1813
-u_toolkit/fastapi/pagination.py,sha256=yOgEDUT04m_mZ0cPliuDbUHLFnmxGAmr5PyZlwfjT_s,1940
-u_toolkit/fastapi/responses.py,sha256=alXUuFK9g1xwkw7V_gpeJFyWuGTHSh-fuuWTuaG8of8,1765
-u_toolkit/pydantic/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-u_toolkit/pydantic/fields.py,sha256=9I8Jwek_oNn_niI98ZWaSPEUYIl5mw24YgbnpcEtgZk,839
-u_toolkit/pydantic/models.py,sha256=Dqp3HnPlUU7ZpfBbYbERfcrLUb2CBJKHySRvM1Rv3HE,1101
-u_toolkit/pydantic/type_vars.py,sha256=XBB3r2SipHOIeyCIuhaBwUDuIfm-M8aI3Gg8rx2uT_A,113
-u_toolkit/sqlalchemy/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-u_toolkit/sqlalchemy/fields.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-u_toolkit/sqlalchemy/function.py,sha256=DRkszvqDfl-BKBW9IcYeN665pgDYlwWiC3DOCjBnxGY,345
-u_toolkit/sqlalchemy/table_info.py,sha256=EjY2Vxbw2c6TUGBf3NeJl1BB2Tnn9NfaFikUx9I3BBk,441
-u_toolkit/sqlalchemy/type_vars.py,sha256=m2VeV41CBIK_1QX3w2kgz-n556sILAGZ-Kaz3TDDDIY,143
-u_toolkit/sqlalchemy/orm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-u_toolkit/sqlalchemy/orm/fields.py,sha256=3zoYil23I6YLtc_59aHDt9w5l1NBTkePT9AfXI3DMiY,593
-u_toolkit/sqlalchemy/orm/models.py,sha256=V8vf4ps3phAmwxyaFYK7pw8Igz7h097o4QBjKB0gwC8,705
-u_toolkit-0.1.19.dist-info/METADATA,sha256=cjvH0qajiIc2SdR5lpo49DBtGS1yyaePHdgPBphwb60,366
-u_toolkit-0.1.19.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-u_toolkit-0.1.19.dist-info/entry_points.txt,sha256=hTfAYCd5vvRiqgnJk2eBsoRIiIVB9pK8WZm3Q3jjKFU,45
-u_toolkit-0.1.19.dist-info/RECORD,,