fastapi-extra 0.2.0__tar.gz → 0.2.2__tar.gz

{fastapi_extra-0.2.0 → fastapi_extra-0.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fastapi-extra
-Version: 0.2.0
+Version: 0.2.2
 Summary: extra package for fastapi.
 Author-email: Ziyan Yin <408856732@qq.com>
 License: BSD-3-Clause

{fastapi_extra-0.2.0 → fastapi_extra-0.2.2}/fastapi_extra/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "0.2.0"
+__version__ = "0.2.2"
 
 
 from fastapi import FastAPI

fastapi_extra-0.2.2/fastapi_extra/cursor.pyi

@@ -0,0 +1,8 @@
+__author__ = "ziyan.yin"
+__describe__ = ""
+
+
+class Cursor:
+    
+    def next_val(self) -> int:
+        ...

{fastapi_extra-0.2.0 → fastapi_extra-0.2.2}/fastapi_extra/database/service.py

@@ -5,7 +5,7 @@ from contextvars import ContextVar
 from typing import Any, Generic, Self, TypeVar
 
 from fastapi_extra.database.model import SQLModel
-from fastapi_extra.database.session import DefaultSession
+from fastapi_extra.database.session import AsyncSession, DefaultSession
 from fastapi_extra.dependency import AbstractService
 
 Model = TypeVar("Model", bound=SQLModel)
@@ -32,7 +32,7 @@ class ModelService(AbstractService, Generic[Model], abstract=True):
         self.__session_container__.set(session)
     
     @property
-    def session(self):
+    def session(self) -> AsyncSession:
         _session = self.__session_container__.get()
         assert _session is not None, "Session is not initialized"
         return _session

{fastapi_extra-0.2.0 → fastapi_extra-0.2.2}/fastapi_extra/native/cursor.pyx

@@ -37,8 +37,8 @@ cdef class Cursor:
         self.cursor = count
         return (point << (_sequence_length + 4)) + (self.seed << _sequence_length) + count
 
-    def next_val(self) -> str:
+    def next_val(self) -> int:
         index = self.fetch()
         while index == 0:
             index = self.fetch()
-        return str(index)
+        return index

fastapi_extra-0.2.2/fastapi_extra/native/routing.pyx

@@ -0,0 +1,191 @@
+__author__ = "ziyan.yin"
+__describe__ = ""
+
+cimport cython
+
+from starlette import _utils as starlette_utils
+from starlette.datastructures import URL
+from starlette.responses import RedirectResponse
+
+
+cdef int find_params(unicode path):
+    for i, ch in enumerate(path):
+        if ch == "{":
+            return i
+    return -1
+
+
+cdef int get_longest_common_prefix(unicode path, unicode node_path):
+    cdef int i
+    cdef int max_len = min(len(path), len(node_path))
+    for i in range(max_len):
+        if path[i] != node_path[i]:
+            return i
+    return max_len
+
+
+@cython.no_gc
+cdef class RouteNode:
+
+    cdef readonly:
+        unicode prefix
+        list params_routes
+        list static_routes
+        dict children
+    
+    cdef public object parent
+
+    def __cinit__(self, prefix: str):
+        self.prefix = prefix
+        self.params_routes = []
+        self.static_routes = []
+        self.children = {}
+        self.parent = None
+    
+    def add_route(self, fullpath: str, handler: object):
+        wild_child = False
+        if (index := find_params(fullpath)) >= 0:
+            wild_child = True
+            path = fullpath[:index]
+        else:
+            path = fullpath
+        insert_route(self, path, wild_child, handler)
+
+
+cdef void insert_route(RouteNode node, unicode path, bint wild_child, object handler):
+    if node.prefix == path:
+        add_node(node, wild_child, handler)
+        return
+
+    cdef Py_UCS4 key = path.removeprefix(node.prefix)[0]
+    if key not in node.children:
+        add_child_node(node, key, path, wild_child, handler)
+        return
+
+    child_node = node.children[key]
+    i = get_longest_common_prefix(child_node.prefix, path)
+    longest_prefix = child_node.prefix[0: i]
+    if i == len(child_node.prefix):
+        insert_route(node.children[key], path, wild_child, handler)
+        return
+    next_node = RouteNode.__new__(RouteNode, longest_prefix)
+    next_node.parent = node
+    node.children[key] = next_node
+    next_node.children[child_node.prefix[i]] = child_node
+    child_node.parent = next_node
+    insert_route(next_node, path, wild_child, handler)
+
+
+cdef inline void add_child_node(RouteNode node, Py_UCS4 key, unicode path, bint wild_child, object handler):
+    child = RouteNode.__new__(RouteNode, path)
+    child.parent = node
+    add_node(child, wild_child, handler)
+    node.children[key] = child
+
+
+cdef inline void add_node(RouteNode node, bint wild_child, object handler):
+    if wild_child:
+        node.params_routes.append(handler)
+    else:
+        node.static_routes.append(handler)
+
+
+root_node = RouteNode.__new__(RouteNode, "")
+
+
+cdef RouteNode search_node(unicode url):
+    cdef RouteNode current_node = root_node
+    cdef int n = len(url)
+    cdef int i = get_longest_common_prefix(url, current_node.prefix)
+    
+    while i < n:
+        key = url[i]
+        if key not in current_node.children:
+            break
+        current_node = current_node.children[key]
+        i = get_longest_common_prefix(url, current_node.prefix)
+
+    return current_node
+
+
+async def handle(scope, receive, send):
+    router = scope["app"].router
+    assert scope["type"] in ("http", "websocket", "lifespan")
+
+    if "router" not in scope:
+        scope["router"] = router
+
+    if scope["type"] == "lifespan":
+        await router.lifespan(scope, receive, send)
+        return
+
+    partial = None
+
+    scope["path"] = route_path = starlette_utils.get_route_path(scope)
+    leaf_node = search_node(route_path)
+
+    if leaf_node.prefix == route_path:
+        for route in leaf_node.static_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
+    else:
+        current_node = leaf_node
+        routes = current_node.params_routes
+        while current_node.parent:
+            for route in 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
+            current_node = current_node.parent
+
+    if partial is not None:
+        scope.update(partial_scope)
+        await partial.handle(scope, receive, send)
+        return
+
+    if scope["type"] == "http" and router.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"] + "/"
+        
+        if leaf_node.prefix == redirect_scope["path"]:
+            for route in leaf_node.static_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
+        else:
+            current_node = leaf_node
+            routes = current_node.params_routes
+            while current_node.parent:
+                for route in routes:
+                    if match.value != 0:
+                        redirect_url = URL(scope=redirect_scope)
+                        response = RedirectResponse(url=str(redirect_url))
+                        await response(scope, receive, send)
+                        return
+                current_node = current_node.parent
+
+    await router.default(scope, receive, send)
+
+
+def install(app):
+    for route in app.routes:
+        root_node.add_route(route.path, route)
+
+    app.router.middleware_stack = handle

fastapi_extra-0.2.2/fastapi_extra/routing.pyi

@@ -0,0 +1,16 @@
+__author__ = "ziyan.yin"
+__describe__ = ""
+
+
+from typing import Any
+from fastapi import FastAPI
+
+
+class RouteNode:
+
+    def add_route(self, fullpath: str, handler: Any):
+        ...
+
+
+def install(app: FastAPI) -> None:
+    ...

{fastapi_extra-0.2.0 → fastapi_extra-0.2.2}/fastapi_extra.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fastapi-extra
-Version: 0.2.0
+Version: 0.2.2
 Summary: extra package for fastapi.
 Author-email: Ziyan Yin <408856732@qq.com>
 License: BSD-3-Clause

{fastapi_extra-0.2.0 → fastapi_extra-0.2.2}/fastapi_extra.egg-info/SOURCES.txt

@@ -2,10 +2,12 @@ LICENSE
 README.rst
 pyproject.toml
 fastapi_extra/__init__.py
+fastapi_extra/cursor.pyi
 fastapi_extra/dependency.py
 fastapi_extra/form.py
 fastapi_extra/py.typed
 fastapi_extra/response.py
+fastapi_extra/routing.pyi
 fastapi_extra/settings.py
 fastapi_extra/types.py
 fastapi_extra/utils.py

fastapi_extra-0.2.0/fastapi_extra/native/routing.pyx

@@ -1,135 +0,0 @@
-__author__ = "ziyan.yin"
-__describe__ = ""
-
-cimport cython
-
-from typing import MutableMapping
-
-from starlette import _utils as starlette_utils
-from starlette.datastructures import URL
-from starlette.responses import RedirectResponse
-
-
-@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, RouteNode 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 = []
-    if current_node.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]
-            if current_node.routes:
-                routes += current_node.routes
-            continue
-        break
-    return routes
-
-
-root_node = RouteNode.__new__(RouteNode, "")
-
-
-async def handle(router, scope, receive, send):
-    assert scope["type"] in ("http", "websocket", "lifespan")
-
-    if "router" not in scope:
-        scope["router"] = router
-
-    if scope["type"] == "lifespan":
-        await router.lifespan(scope, receive, send)
-        return
-
-    partial = None
-
-    scope["path"] = route_path = starlette_utils.get_route_path(scope)
-    scope["root_path"] = ""
-    matched_routes = find_routes(route_path, root_node)
-    n = len(matched_routes)
-
-    for i in range(n):
-        route = matched_routes[n - i - 1]
-        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 router.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 i in range(n):
-            route = matched_routes[n - i - 1]
-            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 router.default(scope, receive, send)
-
-
-def install(app):
-    for route in app.routes:
-        add_route(route.path, root_node, route)
-    app.router.app = handle