fastapi-extra 0.1.0__tar.gz

fastapi_extra-0.1.0/LICENSE

@@ -0,0 +1,27 @@
+Copyright (c) 2024 NoLongerCoding
+All rights reserved.
+
+Redistribution and use in source and binary forms, with or without
+modification, are permitted provided that the following conditions are met:
+
+* Redistributions of source code must retain the above copyright notice, this
+  list of conditions and the following disclaimer.
+
+* Redistributions in binary form must reproduce the above copyright notice,
+  this list of conditions and the following disclaimer in the documentation
+  and/or other materials provided with the distribution.
+
+* Neither the name of the copyright holder nor the names of its
+  contributors may be used to endorse or promote products derived from
+  this software without specific prior written permission.
+
+THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
+FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
+SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
+CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
+OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

fastapi_extra-0.1.0/PKG-INFO

@@ -0,0 +1,24 @@
+Metadata-Version: 2.1
+Name: fastapi-extra
+Version: 0.1.0
+Summary: extra package for fastapi.
+Author-email: Ziyan Yin <408856732@qq.com>
+License: BSD-3-Clause
+Classifier: Development Status :: 4 - Beta
+Classifier: Environment :: Web Environment
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: BSD License
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3 :: Only
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: Implementation :: CPython
+Classifier: Topic :: Internet
+Classifier: Topic :: Software Development :: Libraries :: Application Frameworks
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Requires-Python: >=3.12
+Description-Content-Type: text/x-rst
+License-File: LICENSE
+Requires-Dist: fastapi<0.116.0,>=0.115.0
+Requires-Dist: httpx<0.29.0,>=0.28.0
+Requires-Dist: pydantic-settings>=2.7.0
+Requires-Dist: sqlmodel>=0.0.22

fastapi_extra-0.1.0/fastapi_extra/__init__.py

@@ -0,0 +1,13 @@
+__version__ = "0.1.0"
+
+
+def install():
+    try:
+        from fastapi import routing
+
+        from fastapi_extra import routing as native_routing  # type: ignore
+        
+        routing.APIRouter = native_routing.BaseRouter  # type: ignore
+
+    except ImportError:  # pragma: nocover
+        pass

fastapi_extra-0.1.0/fastapi_extra/databases.py

@@ -0,0 +1,96 @@
+__author__ = "ziyan.yin"
+__date__ = "2024-12-26"
+
+
+from typing import Annotated, Any, Literal
+
+from fastapi.params import Depends
+from pydantic import AnyUrl, BaseModel
+from sqlalchemy import Engine, NullPool
+from sqlalchemy.ext.asyncio import AsyncEngine
+from sqlalchemy.ext.asyncio import AsyncSession as _AsyncSession
+from sqlalchemy.orm import Session as _Session
+from sqlalchemy.util import _concurrency_py3k
+from sqlmodel import create_engine
+
+from fastapi_extra.settings import Settings
+
+
+class DatabaseConfig(BaseModel):
+    url: AnyUrl
+    echo: bool = False
+    echo_pool: bool = False
+    isolation_level: Literal[
+        "SERIALIZABLE",
+        "REPEATABLE READ",
+        "READ COMMITTED",
+        "READ UNCOMMITTED",
+        "AUTOCOMMIT",
+    ] | None = None
+    max_overflow: int = 10
+    pool_pre_ping: bool = False
+    pool_size: int = 5
+    pool_recycle: int = -1
+    pool_timeout: int = 30
+    pool_use_lifo: bool = False
+    query_cache_size: int = 500
+
+
+class DatabaseSettings(Settings):
+    datasources: dict[str, DatabaseConfig]
+    
+
+_settings = DatabaseSettings()  # type: ignore
+_engines: dict[str, Engine] = {}
+
+
+def load_engine(name: str = "default", **kw: Any) -> Engine:
+    if name in _engines:
+        return _engines[name]
+    if name in _settings.datasources:
+        config = _settings.datasources[name]
+        _engines[name] = create_engine(
+            url=str(config.url),
+            **config.model_dump(exclude_defaults=True, exclude={"url"}), 
+            **kw
+        )
+            
+        return _engines[name]
+        
+    raise KeyError(f"cannot find datasources.{name}")
+
+
+async def shutdown() -> None:
+    for engine in _engines.values():
+        await _concurrency_py3k.greenlet_spawn(engine.dispose)
+
+
+class SessionFactory(Depends):
+    __slots__ = ("engine", )
+    datasource: str = "default"
+    
+    def __init__(self):
+        super().__init__()
+        if _settings.mode == "test":
+            self.engine = load_engine(self.datasource, poolclass=NullPool)
+        else:
+            self.engine = load_engine(self.datasource)
+        self.dependency = self
+
+    def __call__(self):
+        with _Session(self.engine) as session:
+            yield session
+
+
+class AsyncSessionFactory(SessionFactory):
+    
+    def __init__(self):
+        super().__init__()
+        self.engine = AsyncEngine(self.engine)
+    
+    async def __call__(self):
+        async with _AsyncSession(self.engine) as session:
+            yield session
+
+
+AsyncSession = Annotated[_AsyncSession, AsyncSessionFactory()]

fastapi_extra-0.1.0/fastapi_extra/form.py

@@ -0,0 +1,30 @@
+__author__ = "ziyan.yin"
+__date__ = "2024-12-24"
+
+
+from typing import Generic, Literal
+
+from pydantic import BaseModel, Field, model_validator
+
+from fastapi_extra.types import C, S
+
+
+class DataRange(BaseModel, Generic[C]):
+    start: C | None = Field(default=None, title="起始")
+    end: C | None = Field(default=None, title="终止")
+
+
+class ColumnExpression(BaseModel, Generic[S]):
+    column_name: str = Field(title="列名")
+    option: Literal["eq", "ne", "gt", "lt", "ge", "le"] = Field(default="eq", title="逻辑值")
+    value: S = Field(title="参考值")
+    
+    @model_validator(mode="after")
+    def validate_value(self):
+        if self.value is None and self.option not in ("eq", "ne"):
+            raise ValueError("NoneType is not comparable")
+
+
+class WhereClause(BaseModel):
+    option: Literal["and", "or"] = Field(default="and", title="关系")
+    column_clauses: list[ColumnExpression | "WhereClause"]

fastapi_extra-0.1.0/fastapi_extra/model.py

@@ -0,0 +1,73 @@
+__author__ = "ziyan.yin"
+__date__ = "2024-12-25"
+
+
+import datetime
+from typing import Annotated
+
+from pydantic import PlainSerializer
+from sqlalchemy import BigInteger, DateTime, Integer, SmallInteger, func
+from sqlalchemy.ext.declarative import declared_attr
+from sqlmodel import Field, SQLModel
+
+from fastapi_extra.cursor import Cursor as _Cursor  # type: ignore
+from fastapi_extra.utils import get_machine_seed
+
+Cursor = Annotated[
+    int, PlainSerializer(lambda x: str(x), return_type=str)
+]
+LocalDateTime = Annotated[
+    datetime.datetime, PlainSerializer(lambda x: x.strftime("%Y-%m-%d %H:%M:%S"), return_type=str)
+]
+
+
+class SQLBase(SQLModel):
+    id: int | None = Field(
+        default_factory=lambda: None, 
+        title="ID", 
+        primary_key=True
+    )
+    create_at: LocalDateTime = Field(
+        default_factory=datetime.datetime.now, 
+        title="CREATE_AT",
+        sa_type=DateTime,
+        sa_column_kwargs={"default": func.now(), "nullable": False, "comment": "CREATE_AT"},
+        schema_extra={"json_schema_extra": {"readOnly": True}},
+    )
+    update_at: LocalDateTime = Field(
+        default_factory=datetime.datetime.now, 
+        title="UPDATE_AT",
+        sa_type=DateTime,
+        sa_column_kwargs={"default": func.now(), "onupdate": func.now(), "nullable": False, "comment": "UPDATE_AT"},
+        schema_extra={"json_schema_extra": {"readOnly": True}},
+    )
+
+
+class LocalPK(SQLBase):
+    id: Cursor | None = Field(
+        default_factory=_Cursor(get_machine_seed()).next_val, title="ID", primary_key=True, sa_type=BigInteger
+    )
+
+
+class Deleted(SQLBase):
+    deleted: int = Field(
+        default=0,
+        title="DELETED",
+        sa_type=SmallInteger,
+        sa_column_kwargs={"nullable": False, "comment": "DELETED"},
+        schema_extra={"json_schema_extra": {"readOnly": True}},
+    )
+
+
+class Versioned(SQLBase):
+    version_id: int = Field(
+        default=0,
+        title="VERSION_ID",
+        sa_type=Integer,
+        sa_column_kwargs={"nullable": False, "comment": "VERSION_ID"},
+        schema_extra={"json_schema_extra": {"readOnly": True}},
+    )
+    
+    @declared_attr  # type: ignore
+    def __mapper_args__(cls) -> dict:
+        return {"version_id_col": "version_id"}

fastapi_extra-0.1.0/fastapi_extra/native/cursor.pyx

@@ -0,0 +1,44 @@
+cimport cython
+from cpython cimport time
+
+import datetime
+
+
+cdef int _sequence_length = 10
+cdef double _start_point = datetime.datetime(2020, 1, 1).timestamp()
+
+
+@cython.no_gc
+cdef class Cursor:
+    __slots__ = "cursor", "last_point"
+    cdef:
+        int cursor
+        int seed
+        long long last_point
+
+    def __init__(self, seed: int):
+        self.seed = seed
+        if self.seed > 16:
+            self.seed %= 16
+        self.cursor = 0
+        self.last_point = 0
+
+    cdef inline long long fetch(self) nogil:
+        cdef:
+            long long count = 0
+            long long point = int((time.time() - _start_point) * 100)
+
+        if self.last_point == point:
+            count = self.cursor + 1
+            if count >= (1 << _sequence_length):
+                return 0
+        else:
+            self.last_point = point
+        self.cursor = count
+        return (point << (_sequence_length + 4)) + (self.seed << _sequence_length) + count
+
+    def next_val(self) -> str:
+        index = self.fetch()
+        while index == 0:
+            index = self.fetch()
+        return str(index)

fastapi_extra-0.1.0/fastapi_extra/native/routing.pyx

@@ -0,0 +1,163 @@
+__author__ = "ziyan.yin"
+__describe__ = ""
+
+cimport cython
+
+from typing import MutableMapping
+
+from basex.common import strings
+from fastapi import APIRouter
+from starlette import _utils as starlette_utils
+from starlette.datastructures import URL
+from starlette.responses import RedirectResponse
+
+
+def get_route_path(scope: MutableMapping) -> str:
+    root_path = scope.get("root_path", "")
+    route_path = scope["path"].removeprefix(root_path)
+    return route_path
+
+starlette_utils.get_route_path = get_route_path
+
+
+@cython.no_gc
+cdef class RouteNode:
+    cdef readonly:
+        list routes
+        dict leaves
+        unicode prefix
+
+    def __cinit__(self, prefix):
+        self.prefix = prefix
+        self.routes = []
+        self.leaves = {}
+
+    def add_route(self, route):
+        self.routes.append(route)
+
+    def add_leaf(self, node):
+        if node.prefix in self.leaves:
+            raise KeyError(node.prefix)
+        else:
+            self.leaves[node.prefix] = node
+
+
+cdef list change_path_to_ranks(unicode path):
+    ranks = path.lstrip('/').split('/')
+    return ranks
+
+
+cdef void add_route(unicode path, object root, object route):
+    current_node = root
+    ranks = change_path_to_ranks(path)
+    for r in ranks:
+        if r.find('{') >= 0 and r.find('}') > 0:
+            break
+        if not r:
+            continue
+        if r in current_node.leaves:
+            current_node = current_node.leaves[r]
+        else:
+            next_node = RouteNode.__new__(RouteNode, r)
+            current_node.add_leaf(next_node)
+            current_node = next_node
+    current_node.add_route(route)
+
+
+cdef list find_routes(unicode path, RouteNode root):
+    current_node = root
+    ranks = change_path_to_ranks(path)
+
+    routes = []
+    routes += current_node.routes
+    for r in ranks:
+        if not r:
+            continue
+        if r in current_node.leaves:
+            current_node = current_node.leaves[r]
+            routes += current_node.routes
+            continue
+        break
+
+    routes.reverse()
+    return routes
+
+
+_super_router = APIRouter
+
+
+class BaseRouter(_super_router):
+
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        self.root_node = RouteNode.__new__(RouteNode, '')
+        for route in self.routes:
+            add_route(route.path, self.root_node, route)
+
+    def add_route(self, path, endpoint, **kwargs):
+        super().add_route(path, endpoint, **kwargs)
+        add_route(self.routes[-1].path, self.root_node, self.routes[-1])
+
+    def add_api_route(self, path, endpoint, **kwargs):
+        super().add_api_route(path)
+        add_route(self.routes[-1].path, self.root_node, self.routes[-1])
+
+    def add_websocket_route(self, path, endpoint, **kwargs):
+        super().add_websocket_route(path, endpoint, **kwargs)
+        add_route(self.routes[-1].path, self.root_node, self.routes[-1])
+
+    def add_api_websocket_route(self, path, endpoint, **kwargs):
+        super().add_api_websocket_route(path, endpoint, **kwargs)
+        add_route(self.routes[-1].path, self.root_node, self.routes[-1])
+
+    def mount(self, path, app, **kwargs):
+        super().mount(path, app, **kwargs)
+        add_route(self.routes[-1].path, self.root_node, self.routes[-1])
+
+    async def __call__(self, scope, receive, send):
+        assert scope["type"] in ("http", "websocket", "lifespan")
+
+        if "router" not in scope:
+            scope["router"] = self
+
+        if scope["type"] == "lifespan":
+            await self.lifespan(scope, receive, send)
+            return
+
+        partial = None
+
+        route_path = get_route_path(scope)
+        matched_routes = find_routes(route_path, self.root_node)
+
+        for route in matched_routes:
+            match, child_scope = route.matches(scope)
+            if match.value == 2:
+                scope.update(child_scope)
+                await route.handle(scope, receive, send)
+                return
+            elif match.value == 1 and partial is None:
+                partial = route
+                partial_scope = child_scope
+
+        if partial is not None:
+            scope.update(partial_scope)
+            await partial.handle(scope, receive, send)
+            return
+
+
+        if scope["type"] == "http" and self.redirect_slashes and route_path != "/":
+            redirect_scope = dict(scope)
+            if route_path.endswith("/"):
+                redirect_scope["path"] = redirect_scope["path"].rstrip("/")
+            else:
+                redirect_scope["path"] = redirect_scope["path"] + "/"
+
+            for route in matched_routes:
+                match, child_scope = route.matches(redirect_scope)
+                if match.value != 0:
+                    redirect_url = URL(scope=redirect_scope)
+                    response = RedirectResponse(url=str(redirect_url))
+                    await response(scope, receive, send)
+                    return
+
+        await self.default(scope, receive, send)

fastapi_extra-0.1.0/fastapi_extra/response.py

@@ -0,0 +1,217 @@
+__author__ = "ziyan.yin"
+__date__ = "2024-12-24"
+
+
+from enum import Enum
+from typing import Generic
+
+from pydantic import BaseModel, Field
+
+from fastapi_extra.types import T
+
+
+# alibaba result enum
+class ResultEnum(Enum):
+    SUCCESS = ("00000", "OK")
+    FAILED = ("99999", "系统异常")
+    A0001 = ("A0001", "用户端错误")
+    A0100 = ("A0100", "用户注册错误")
+    A0101 = ("A0101", "用户未同意隐私协议")
+    A0102 = ("A0102", "注册国家或地区受限")
+    A0110 = ("A0110", "用户名校验失败")
+    A0111 = ("A0111", "用户名已存在")
+    A0112 = ("A0112", "用户名包含敏感词")
+    A0113 = ("A0113", "用户名包含特殊字符")
+    A0120 = ("A0120", "密码校验失败")
+    A0121 = ("A0121", "密码长度不够")
+    A0122 = ("A0122", "密码强度不够")
+    A0130 = ("A0130", "校验码输入错误")
+    A0131 = ("A0131", "短信校验码输入错误")
+    A0132 = ("A0132", "邮件校验码输入错误")
+    A0133 = ("A0133", "语音校验码输入错误")
+    A0140 = ("A0140", "用户证件异常")
+    A0141 = ("A0141", "用户证件类型未选择")
+    A0142 = ("A0142", "大陆身份证编号校验非法")
+    A0143 = ("A0143", "护照编号校验非法")
+    A0144 = ("A0144", "军官证编号校验非法")
+    A0150 = ("A0150", "用户基本信息校验失败")
+    A0151 = ("A0151", "手机格式校验失败")
+    A0152 = ("A0152", "地址格式校验失败")
+    A0153 = ("A0153", "邮箱格式校验失败")
+    A0200 = ("A0200", "用户登录异常")
+    A0201 = ("A0201", "用户账户不存在")
+    A0202 = ("A0202", "用户账户被冻结")
+    A0203 = ("A0203", "用户账户已作废")
+    A0210 = ("A0210", "用户密码错误")
+    A0211 = ("A0211", "用户输入密码错误次数超限")
+    A0220 = ("A0220", "用户身份校验失败")
+    A0221 = ("A0221", "用户指纹识别失败")
+    A0222 = ("A0222", "用户面容识别失败")
+    A0223 = ("A0223", "用户未获得第三方登录授权")
+    A0230 = ("A0230", "用户登录已过期")
+    A0240 = ("A0240", "用户验证码错误")
+    A0241 = ("A0241", "用户验证码尝试次数超限")
+    A0300 = ("A0300", "访问权限异常")
+    A0301 = ("A0301", "访问未授权")
+    A0302 = ("A0302", "正在授权中")
+    A0303 = ("A0303", "用户授权申请被拒绝")
+    A0310 = ("A0310", "因访问对象隐私设置被拦截")
+    A0311 = ("A0311", "授权已过期")
+    A0312 = ("A0312", "无权限使用 API")
+    A0320 = ("A0320", "用户访问被拦截")
+    A0321 = ("A0321", "黑名单用户")
+    A0322 = ("A0322", "账号被冻结")
+    A0323 = ("A0323", "非法 IP 地址")
+    A0324 = ("A0324", "网关访问受限")
+    A0325 = ("A0325", "地域黑名单")
+    A0330 = ("A0330", "服务已欠费")
+    A0340 = ("A0340", "用户签名异常")
+    A0341 = ("A0341", "RSA 签名错误")
+    A0400 = ("A0400", "用户请求参数错误")
+    A0401 = ("A0401", "包含非法恶意跳转链接")
+    A0402 = ("A0402", "无效的用户输入")
+    A0410 = ("A0410", "请求必填参数为空")
+    A0411 = ("A0411", "用户订单号为空")
+    A0412 = ("A0412", "订购数量为空")
+    A0413 = ("A0413", "缺少时间戳参数")
+    A0414 = ("A0414", "非法的时间戳参数")
+    A0420 = ("A0420", "请求参数值超出允许的范围")
+    A0421 = ("A0421", "参数格式不匹配")
+    A0422 = ("A0422", "地址不在服务范围")
+    A0423 = ("A0423", "时间不在服务范围")
+    A0424 = ("A0424", "金额超出限制")
+    A0425 = ("A0425", "数量超出限制")
+    A0426 = ("A0426", "请求批量处理总个数超出限制")
+    A0427 = ("A0427", "请求 JSON 解析失败")
+    A0430 = ("A0430", "用户输入内容非法")
+    A0431 = ("A0431", "包含违禁敏感词")
+    A0432 = ("A0432", "图片包含违禁信息")
+    A0433 = ("A0433", "文件侵犯版权")
+    A0440 = ("A0440", "用户操作异常")
+    A0441 = ("A0441", "用户支付超时")
+    A0442 = ("A0442", "确认订单超时")
+    A0443 = ("A0443", "订单已关闭")
+    A0500 = ("A0500", "用户请求服务异常")
+    A0501 = ("A0501", "请求次数超出限制")
+    A0502 = ("A0502", "请求并发数超出限制")
+    A0503 = ("A0503", "用户操作请等待")
+    A0504 = ("A0504", "WebSocket 连接异常")
+    A0505 = ("A0505", "WebSocket 连接断开")
+    A0506 = ("A0506", "用户重复请求")
+    A0600 = ("A0600", "用户资源异常")
+    A0601 = ("A0601", "账户余额不足")
+    A0602 = ("A0602", "用户磁盘空间不足")
+    A0603 = ("A0603", "用户内存空间不足")
+    A0604 = ("A0604", "用户 OSS 容量不足")
+    A0605 = ("A0605", "用户配额已用光")
+    A0700 = ("A0700", "用户上传文件异常")
+    A0701 = ("A0701", "用户上传文件类型不匹配")
+    A0702 = ("A0702", "用户上传文件太大")
+    A0703 = ("A0703", "用户上传图片太大")
+    A0704 = ("A0704", "用户上传视频太大")
+    A0705 = ("A0705", "用户上传压缩文件太大")
+    A0800 = ("A0800", "用户当前版本异常")
+    A0801 = ("A0801", "用户安装版本与系统不匹配")
+    A0802 = ("A0802", "用户安装版本过低")
+    A0803 = ("A0803", "用户安装版本过高")
+    A0804 = ("A0804", "用户安装版本已过期")
+    A0805 = ("A0805", "用户 API 请求版本不匹配")
+    A0806 = ("A0806", "用户 API 请求版本过高")
+    A0807 = ("A0807", "用户 API 请求版本过低")
+    A0900 = ("A0900", "用户隐私未授权")
+    A0901 = ("A0901", "用户隐私未签署")
+    A0902 = ("A0902", "用户摄像头未授权")
+    A0903 = ("A0903", "用户相机未授权")
+    A0904 = ("A0904", "用户图片库未授权")
+    A0905 = ("A0905", "用户文件未授权")
+    A0906 = ("A0906", "用户位置信息未授权")
+    A0907 = ("A0907", "用户通讯录未授权")
+    A1000 = ("A1000", "用户设备异常")
+    A1001 = ("A1001", "用户相机异常")
+    A1002 = ("A1002", "用户麦克风异常")
+    A1003 = ("A1003", "用户听筒异常")
+    A1004 = ("A1004", "用户扬声器异常")
+    A1005 = ("A1005", "用户 GPS 定位异常")
+    B0001 = ("B0001", "系统执行出错")
+    B0100 = ("B0100", "系统执行超时")
+    B0101 = ("B0101", "系统订单处理超时")
+    B0200 = ("B0200", "系统容灾功能被触发")
+    B0210 = ("B0210", "系统限流")
+    B0220 = ("B0220", "系统功能降级")
+    B0300 = ("B0300", "系统资源异常")
+    B0310 = ("B0310", "系统资源耗尽")
+    B0311 = ("B0311", "系统磁盘空间耗尽")
+    B0312 = ("B0312", "系统内存耗尽")
+    B0313 = ("B0313", "文件句柄耗尽")
+    B0314 = ("B0314", "系统连接池耗尽")
+    B0315 = ("B0315", "系统线程池耗尽")
+    B0320 = ("B0320", "系统资源访问异常")
+    B0321 = ("B0321", "系统读取磁盘文件失败")
+    C0001 = ("C0001", "调用第三方服务出错")
+    C0100 = ("C0100", "中间件服务出错")
+    C0110 = ("C0110", "RPC 服务出错")
+    C0111 = ("C0111", "RPC 服务未找到")
+    C0112 = ("C0112", "RPC 服务未注册")
+    C0113 = ("C0113", "接口不存在")
+    C0120 = ("C0120", "消息服务出错")
+    C0121 = ("C0121", "消息投递出错")
+    C0122 = ("C0122", "消息消费出错")
+    C0123 = ("C0123", "消息订阅出错")
+    C0124 = ("C0124", "消息分组未查到")
+    C0130 = ("C0130", "缓存服务出错")
+    C0131 = ("C0131", "key 长度超过限制")
+    C0132 = ("C0132", "value 长度超过限制")
+    C0133 = ("C0133", "存储容量已满")
+    C0134 = ("C0134", "不支持的数据格式")
+    C0140 = ("C0140", "配置服务出错")
+    C0150 = ("C0150", "网络资源服务出错")
+    C0151 = ("C0151", "VPN 服务出错")
+    C0152 = ("C0152", "CDN 服务出错")
+    C0153 = ("C0153", "域名解析服务出错")
+    C0154 = ("C0154", "网关服务出错")
+    C0200 = ("C0200", "第三方系统执行超时")
+    C0210 = ("C0210", "RPC 执行超时")
+    C0220 = ("C0220", "消息投递超时")
+    C0230 = ("C0230", "缓存服务超时")
+    C0240 = ("C0240", "配置服务超时")
+    C0250 = ("C0250", "数据库服务超时")
+    C0300 = ("C0300", "数据库服务出错")
+    C0311 = ("C0311", "表不存在")
+    C0312 = ("C0312", "列不存在")
+    C0321 = ("C0321", "多表关联中存在多个相同名称的列")
+    C0331 = ("C0331", "数据库死锁")
+    C0341 = ("C0341", "主键冲突")
+    C0400 = ("C0400", "第三方容灾系统被触发")
+    C0401 = ("C0401", "第三方系统限流")
+    C0402 = ("C0402", "第三方功能降级")
+    C0500 = ("C0500", "通知服务出错")
+    C0501 = ("C0501", "短信提醒服务失败")
+    C0502 = ("C0502", "语音提醒服务失败")
+    C0503 = ("C0503", "邮件提醒服务失败")
+
+
+class APIResult(BaseModel, Generic[T]):
+    data: T | None = Field(default=None, title="返回数据")
+
+    @classmethod
+    def ok(cls, data: T | None = None) -> "APIResult[T]":
+        return APIResult(data=data)
+
+
+class APIError(Exception):
+    __slots__ = ("code", "message")
+    
+    def __init__(self, result: ResultEnum | None = None, code: str = "00000", message: str = "") -> None:
+        if result:
+            self.code = result.value[0]
+            self.message = message or result.value[1]
+        else:
+            self.code = code
+            self.message = message
+        super().__init__(self)
+    
+    def __str__(self) -> str:
+        return self.message
+
+    def __repr__(self) -> str:
+        return f"[{self.code}]{self}"

fastapi_extra-0.1.0/fastapi_extra/settings.py

@@ -0,0 +1,36 @@
+__author__ = "ziyan.yin"
+__date__ = "2024-12-26"
+
+
+from typing import Literal, Tuple, Type
+
+from pydantic_settings import (BaseSettings, PydanticBaseSettingsSource,
+                               SettingsConfigDict, TomlConfigSettingsSource)
+
+
+class Settings(BaseSettings):
+    model_config = SettingsConfigDict(
+        toml_file=["config.default.toml", "config.custom.toml"], 
+        validate_default=False
+    )
+    
+    title: str = ""
+    version: str = "0.1.0"
+    debug: bool = False
+    mode: Literal["dev", "test", "prod"] = "dev"
+    
+    @classmethod
+    def settings_customise_sources(
+        cls,
+        settings_cls: Type[BaseSettings],
+        init_settings: PydanticBaseSettingsSource,
+        env_settings: PydanticBaseSettingsSource,
+        dotenv_settings: PydanticBaseSettingsSource,
+        file_secret_settings: PydanticBaseSettingsSource,
+    ) -> Tuple[PydanticBaseSettingsSource, ...]:
+        return (
+            TomlConfigSettingsSource(settings_cls), 
+            env_settings, 
+            init_settings,
+            file_secret_settings
+        )

fastapi_extra-0.1.0/fastapi_extra/types.py

@@ -0,0 +1,16 @@
+__author__ = "ziyan.yin"
+__date__ = "2024-12-25"
+
+
+import datetime
+import decimal
+from typing import Any, TypeVar, Union
+
+Comparable = Union[int, float, decimal.Decimal, datetime.datetime, datetime.date, datetime.time]
+Serializable = Union[Comparable, bool, str, None]
+
+
+T = TypeVar("T", bound=Any)
+E = TypeVar("E", bound=Exception)
+C = TypeVar("C", bound=Comparable)
+S = TypeVar("S", bound=Serializable)

fastapi_extra-0.1.0/fastapi_extra/utils.py

@@ -0,0 +1,18 @@
+__author__ = "ziyan.yin"
+__date__ = "2024-12-25"
+
+
+import os
+
+
+def get_machine_seed() -> int:
+    """获取生成id的机器码, 可以覆盖该方法替换
+
+    Returns:
+        int: 机器码
+    """
+    ppid, pid = os.getppid(), os.getpid()
+    try:
+        return (pid - ppid) % 0x10
+    except ValueError:
+        return 0

fastapi_extra-0.1.0/fastapi_extra.egg-info/PKG-INFO

@@ -0,0 +1,24 @@
+Metadata-Version: 2.1
+Name: fastapi-extra
+Version: 0.1.0
+Summary: extra package for fastapi.
+Author-email: Ziyan Yin <408856732@qq.com>
+License: BSD-3-Clause
+Classifier: Development Status :: 4 - Beta
+Classifier: Environment :: Web Environment
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: BSD License
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3 :: Only
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: Implementation :: CPython
+Classifier: Topic :: Internet
+Classifier: Topic :: Software Development :: Libraries :: Application Frameworks
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Requires-Python: >=3.12
+Description-Content-Type: text/x-rst
+License-File: LICENSE
+Requires-Dist: fastapi<0.116.0,>=0.115.0
+Requires-Dist: httpx<0.29.0,>=0.28.0
+Requires-Dist: pydantic-settings>=2.7.0
+Requires-Dist: sqlmodel>=0.0.22

fastapi_extra-0.1.0/fastapi_extra.egg-info/SOURCES.txt

@@ -0,0 +1,20 @@
+LICENSE
+README.rst
+pyproject.toml
+fastapi_extra/__init__.py
+fastapi_extra/databases.py
+fastapi_extra/form.py
+fastapi_extra/model.py
+fastapi_extra/response.py
+fastapi_extra/settings.py
+fastapi_extra/types.py
+fastapi_extra/utils.py
+fastapi_extra.egg-info/PKG-INFO
+fastapi_extra.egg-info/SOURCES.txt
+fastapi_extra.egg-info/dependency_links.txt
+fastapi_extra.egg-info/requires.txt
+fastapi_extra.egg-info/top_level.txt
+fastapi_extra/native/cursor.pyx
+fastapi_extra/native/routing.pyx
+fastapi_extra/native/cursor.pyx
+fastapi_extra/native/routing.pyx

fastapi_extra-0.1.0/fastapi_extra.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

fastapi_extra-0.1.0/fastapi_extra.egg-info/requires.txt

@@ -0,0 +1,4 @@
+fastapi<0.116.0,>=0.115.0
+httpx<0.29.0,>=0.28.0
+pydantic-settings>=2.7.0
+sqlmodel>=0.0.22

fastapi_extra-0.1.0/fastapi_extra.egg-info/top_level.txt

@@ -0,0 +1 @@
+fastapi_extra

fastapi_extra-0.1.0/pyproject.toml

@@ -0,0 +1,47 @@
+[build-system]
+requires = [
+    "setuptools>=74.1.0",
+    "wheel",
+    "cython>=3.0"
+]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "fastapi-extra"
+dynamic = ["version"]
+description = "extra package for fastapi."
+readme = "README.rst"
+license = { text = "BSD-3-Clause" }
+requires-python = ">=3.12"
+authors = [
+    { name = "Ziyan Yin", email = "408856732@qq.com" }
+]
+classifiers = [
+    "Development Status :: 4 - Beta",
+    "Environment :: Web Environment",
+    "Intended Audience :: Developers",
+    "License :: OSI Approved :: BSD License",
+    "Operating System :: OS Independent",
+    "Programming Language :: Python :: 3 :: Only",
+    "Programming Language :: Python :: 3.12",
+    "Programming Language :: Python :: Implementation :: CPython",
+    "Topic :: Internet",
+    "Topic :: Software Development :: Libraries :: Application Frameworks",
+    "Topic :: Software Development :: Libraries :: Python Modules"
+]
+dependencies = [
+    "fastapi>=0.115.0,<0.116.0",
+    "httpx>=0.28.0,<0.29.0",
+    "pydantic-settings>=2.7.0",
+    "sqlmodel>=0.0.22",
+]
+
+[tool.setuptools]
+packages = ["fastapi_extra"]
+ext-modules = [
+    { name = "fastapi_extra.cursor", sources = ["fastapi_extra/native/cursor.pyx"] },
+    { name = "fastapi_extra.routing", sources = ["fastapi_extra/native/routing.pyx"] }
+]
+
+[tool.setuptools.dynamic]
+version = {attr = "fastapi_extra.__version__"}  # any module attribute compatible with ast.literal_eval

fastapi_extra-0.1.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+