jararaca 0.3.12a6__tar.gz → 0.3.12a7__tar.gz

{jararaca-0.3.12a6 → jararaca-0.3.12a7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: jararaca
-Version: 0.3.12a6
+Version: 0.3.12a7
 Summary: A simple and fast API framework for Python
 Home-page: https://github.com/LuscasLeo/jararaca
 Author: Lucas S

{jararaca-0.3.12a6 → jararaca-0.3.12a7}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "jararaca"
-version = "0.3.12a6"
+version = "0.3.12a7"
 description = "A simple and fast API framework for Python"
 authors = ["Lucas S <me@luscasleo.dev>"]
 readme = "README.md"

{jararaca-0.3.12a6 → jararaca-0.3.12a7}/src/jararaca/tools/typescript/decorators.py

@@ -1,4 +1,4 @@
-from typing import Any, Callable, TypeVar
+from typing import Any, Callable, TypeVar, cast
 
 DECORATED_FUNC = TypeVar("DECORATED_FUNC", bound=Callable[..., Any])
 
@@ -10,21 +10,32 @@ class QueryEndpoint:
 
     METADATA_KEY = "__jararaca_query_endpoint__"
 
-    def __init__(self) -> None: ...
+    def __init__(self, has_infinite_query: bool = False) -> None:
+        """
+        Initialize the QueryEndpoint decorator.
+
+        Args:
+            has_infinite_query: Whether the query endpoint supports infinite queries.
+            Important:
+            - Make sure a PaginatedQuery child instance is on the first argument
+            - Make sure the endpoint is a Patch (recommended) or Put method
+            - Make sure the endpoint returns a Paginated[T]
+        """
+        self.has_infinite_query = has_infinite_query
 
     def __call__(self, func: DECORATED_FUNC) -> DECORATED_FUNC:
         """
         Decorate the function to mark it as a query endpoint.
         """
-        setattr(func, self.METADATA_KEY, True)
+        setattr(func, self.METADATA_KEY, self)
         return func
 
     @staticmethod
-    def is_query(func: Any) -> bool:
+    def extract_query_endpoint(func: Any) -> "QueryEndpoint":
         """
         Check if the function is marked as a query endpoint.
         """
-        return getattr(func, QueryEndpoint.METADATA_KEY, False)
+        return cast(QueryEndpoint, getattr(func, QueryEndpoint.METADATA_KEY, None))
 
 
 class MutationEndpoint:

{jararaca-0.3.12a6 → jararaca-0.3.12a7}/src/jararaca/tools/typescript/interface_parser.py

@@ -389,7 +389,7 @@ def write_microservice_to_typescript_interface(
 
     final_buffer.write(
         """
-import { createClassQueryHooks , createClassMutationHooks, createClassInfiniteQueryHooks } from "@jararaca/core";
+import { createClassQueryHooks , createClassMutationHooks, createClassInfiniteQueryHooks, paginationModelByFirstArgPaginationFilter } from "@jararaca/core";
 export type WebSocketMessageMap = {
 %s
 }
@@ -512,7 +512,7 @@ def write_rest_controller_to_typescript_interface(
 
     class_name = controller.__name__
 
-    decorated_queries: list[tuple[str, FunctionType]] = []
+    decorated_queries: list[tuple[str, FunctionType, QueryEndpoint]] = []
     decorated_mutations: list[tuple[str, FunctionType]] = []
 
     class_buffer = StringIO()
@@ -533,8 +533,8 @@ def write_rest_controller_to_typescript_interface(
             if return_type is None:
                 return_type = NoneType
 
-            if QueryEndpoint.is_query(member):
-                decorated_queries.append((name, member))
+            if query_endpoint := QueryEndpoint.extract_query_endpoint(member):
+                decorated_queries.append((name, member, query_endpoint))
             if MutationEndpoint.is_mutation(member):
                 decorated_mutations.append((name, member))
 
@@ -624,9 +624,22 @@ def write_rest_controller_to_typescript_interface(
             controller_hooks_builder.write(
                 f"\t...createClassQueryHooks({class_name},\n"
             )
-            for name, member in decorated_queries:
+            for name, member, _ in decorated_queries:
                 controller_hooks_builder.write(f'\t\t"{snake_to_camel(name)}",\n')
             controller_hooks_builder.write("\t),\n")
+
+        if decorated_queries and any(
+            query.has_infinite_query for _, _, query in decorated_queries
+        ):
+            controller_hooks_builder.write(
+                f"\t...createClassInfiniteQueryHooks({class_name}, {{\n"
+            )
+            for name, member, query in decorated_queries:
+                if query.has_infinite_query:
+                    controller_hooks_builder.write(
+                        f'\t\t"{snake_to_camel(name)}": paginationModelByFirstArgPaginationFilter(),\n'
+                    )
+            controller_hooks_builder.write("\t}),\n")
         if decorated_mutations:
             controller_hooks_builder.write(
                 f"\t...createClassMutationHooks({class_name},\n"