jararaca 0.3.12a6__py3-none-any.whl → 0.3.12a7__py3-none-any.whl

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/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"

{jararaca-0.3.12a6.dist-info → jararaca-0.3.12a7.dist-info}/METADATA

@@ -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.dist-info → jararaca-0.3.12a7.dist-info}/RECORD

@@ -1,6 +1,6 @@
 LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
 README.md,sha256=2qMM__t_MoLKZr4IY9tXjo-Jn6LKjuHMb1qbyXpgL08,3401
-pyproject.toml,sha256=3e1dOSeKg1pARRUHFxJ0d5wOPiGtoe4YiAqf3VngI6o,2040
+pyproject.toml,sha256=rfAelC2k9MV3eq2dge_J4UxWLHNXNRSyQy8_1W-YXos,2040
 jararaca/__init__.py,sha256=jN3paW2ujI97485DNZTcRe_8ORwO-OQbJUG5mmSI9LI,21226
 jararaca/__main__.py,sha256=-O3vsB5lHdqNFjUtoELDF81IYFtR-DSiiFMzRaiSsv4,67
 jararaca/broker_backend/__init__.py,sha256=GzEIuHR1xzgCJD4FE3harNjoaYzxHMHoEL0_clUaC-k,3528
@@ -67,13 +67,13 @@ jararaca/tools/app_config/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJW
 jararaca/tools/app_config/decorators.py,sha256=-ckkMZ1dswOmECdo1rFrZ15UAku--txaNXMp8fd1Ndk,941
 jararaca/tools/app_config/interceptor.py,sha256=HV8h4AxqUc_ACs5do4BSVlyxlRXzx7HqJtoVO9tfRnQ,2611
 jararaca/tools/typescript/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-jararaca/tools/typescript/decorators.py,sha256=F-4de0OKN4VrXCwtfsts1XaycGkXqSGuDKW6ElryMsY,1409
-jararaca/tools/typescript/interface_parser.py,sha256=dc-h74yaySqXklqIDXuIQUN9wwqEE7q4uGWQ4nByI84,33462
+jararaca/tools/typescript/decorators.py,sha256=DbwZADDzzo3cbMfNcY6ldILy_80N4x25ChyQoX1q56s,1956
+jararaca/tools/typescript/interface_parser.py,sha256=jgkIDtM5ifGGLSSSqR2D41CC9Xrz4kvKKcN5321JAVo,34161
 jararaca/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 jararaca/utils/rabbitmq_utils.py,sha256=ytdAFUyv-OBkaVnxezuJaJoLrmN7giZgtKeet_IsMBs,10918
 jararaca/utils/retry.py,sha256=DzPX_fXUvTqej6BQ8Mt2dvLo9nNlTBm7Kx2pFZ26P2Q,4668
-jararaca-0.3.12a6.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-jararaca-0.3.12a6.dist-info/METADATA,sha256=XPCPH0zrpKaicneY3XJSnwbgVvLlDlS4kfSWblCorWE,4995
-jararaca-0.3.12a6.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-jararaca-0.3.12a6.dist-info/entry_points.txt,sha256=WIh3aIvz8LwUJZIDfs4EeH3VoFyCGEk7cWJurW38q0I,45
-jararaca-0.3.12a6.dist-info/RECORD,,
+jararaca-0.3.12a7.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+jararaca-0.3.12a7.dist-info/METADATA,sha256=By0PdO0rXhXnTyH6feXx33JyvjpnzBqh9HqOFe3Ab70,4995
+jararaca-0.3.12a7.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+jararaca-0.3.12a7.dist-info/entry_points.txt,sha256=WIh3aIvz8LwUJZIDfs4EeH3VoFyCGEk7cWJurW38q0I,45
+jararaca-0.3.12a7.dist-info/RECORD,,

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"