jararaca 0.3.12a22__py3-none-any.whl → 0.3.13__py3-none-any.whl

jararaca/__init__.py

@@ -11,7 +11,12 @@ if TYPE_CHECKING:
         retry_later,
         use_bus_message_controller,
     )
-    from jararaca.microservice import AppContext, AppInterceptor, AppTransactionContext
+    from jararaca.microservice import (
+        AppContext,
+        AppInterceptor,
+        AppTransactionContext,
+        use_app_type,
+    )
     from jararaca.observability.interceptor import ObservabilityInterceptor
     from jararaca.observability.providers.otel import OtelObservabilityProvider
     from jararaca.persistence.sort_filter import (
@@ -308,6 +313,7 @@ if TYPE_CHECKING:
         # Exception classes
         "TimeoutException",
         "AppTransactionContext",
+        "use_app_type",
         "AppContext",
         "ControllerMemberReflect",
         "ControllerReflect",
@@ -544,6 +550,7 @@ _dynamic_imports: "dict[str, tuple[str, str, str | None]]" = {
     "AppContext": (__SPEC_PARENT__, "microservice", None),
     "AppInterceptor": (__SPEC_PARENT__, "microservice", None),
     "AppTransactionContext": (__SPEC_PARENT__, "microservice", None),
+    "use_app_type": (__SPEC_PARENT__, "microservice", None),
     "ControllerMemberReflect": (__SPEC_PARENT__, "reflect.controller_inspect", None),
     "ControllerReflect": (__SPEC_PARENT__, "reflect.controller_inspect", None),
 }

jararaca/messagebus/worker.py

@@ -58,6 +58,7 @@ from jararaca.microservice import (
     SchedulerTransactionData,
     ShutdownState,
     provide_shutdown_state,
+    providing_app_type,
 )
 from jararaca.scheduler.decorators import ScheduledActionData
 from jararaca.utils.rabbitmq_utils import RabbitmqUtils
@@ -1725,61 +1726,64 @@ class MessageBusWorker:
     async def start_async(self) -> None:
         all_message_handlers_set: MESSAGE_HANDLER_DATA_SET = set()
         all_scheduled_actions_set: SCHEDULED_ACTION_DATA_SET = set()
-        async with self.lifecycle():
-            for instance_class in self.app.controllers:
-                controller = MessageBusController.get_messagebus(instance_class)
-
-                if controller is None:
-                    continue
-
-                instance: Any = self.container.get_by_type(instance_class)
-
-                factory = controller.get_messagebus_factory()
-                handlers, schedulers = factory(instance)
-
-                message_handler_data_map: dict[str, MessageHandlerData] = {}
-                all_scheduled_actions_set.update(schedulers)
-                for handler_data in handlers:
-                    message_type = handler_data.spec.message_type
-                    topic = message_type.MESSAGE_TOPIC
-
-                    # Filter handlers by name if specified
-                    if (
-                        self.handler_names is not None
-                        and handler_data.spec.name is not None
-                    ):
-                        if handler_data.spec.name not in self.handler_names:
-                            continue
-                    elif (
-                        self.handler_names is not None
-                        and handler_data.spec.name is None
-                    ):
-                        # Skip handlers without names when filtering is requested
-                        continue
+        with providing_app_type("worker"):
+            async with self.lifecycle():
+                for instance_class in self.app.controllers:
+                    controller = MessageBusController.get_messagebus(instance_class)
 
-                    if (
-                        topic in message_handler_data_map
-                        and message_type.MESSAGE_TYPE == "task"
-                    ):
-                        logger.warning(
-                            "Task handler for topic '%s' already registered. Skipping"
-                            % topic
-                        )
+                    if controller is None:
                         continue
-                    message_handler_data_map[topic] = handler_data
-                    all_message_handlers_set.add(handler_data)
 
-            broker_backend = get_message_broker_backend_from_url(url=self.backend_url)
+                    instance: Any = self.container.get_by_type(instance_class)
 
-            consumer = self._consumer = create_message_bus(
-                broker_url=self.broker_url,
-                broker_backend=broker_backend,
-                scheduled_actions=all_scheduled_actions_set,
-                message_handler_set=all_message_handlers_set,
-                uow_context_provider=self.uow_context_provider,
-            )
+                    factory = controller.get_messagebus_factory()
+                    handlers, schedulers = factory(instance)
+
+                    message_handler_data_map: dict[str, MessageHandlerData] = {}
+                    all_scheduled_actions_set.update(schedulers)
+                    for handler_data in handlers:
+                        message_type = handler_data.spec.message_type
+                        topic = message_type.MESSAGE_TOPIC
+
+                        # Filter handlers by name if specified
+                        if (
+                            self.handler_names is not None
+                            and handler_data.spec.name is not None
+                        ):
+                            if handler_data.spec.name not in self.handler_names:
+                                continue
+                        elif (
+                            self.handler_names is not None
+                            and handler_data.spec.name is None
+                        ):
+                            # Skip handlers without names when filtering is requested
+                            continue
+
+                        if (
+                            topic in message_handler_data_map
+                            and message_type.MESSAGE_TYPE == "task"
+                        ):
+                            logger.warning(
+                                "Task handler for topic '%s' already registered. Skipping"
+                                % topic
+                            )
+                            continue
+                        message_handler_data_map[topic] = handler_data
+                        all_message_handlers_set.add(handler_data)
+
+                broker_backend = get_message_broker_backend_from_url(
+                    url=self.backend_url
+                )
+
+                consumer = self._consumer = create_message_bus(
+                    broker_url=self.broker_url,
+                    broker_backend=broker_backend,
+                    scheduled_actions=all_scheduled_actions_set,
+                    message_handler_set=all_message_handlers_set,
+                    uow_context_provider=self.uow_context_provider,
+                )
 
-            await consumer.consume()
+                await consumer.consume()
 
     def start_sync(self) -> None:
 

jararaca/microservice.py

@@ -59,6 +59,8 @@ class WebSocketTransactionData:
     context_type: Literal["websocket"] = "websocket"
 
 
+APP_TYPE = Literal["http", "worker", "beat"]
+
 TransactionData = (
     MessageBusTransactionData
     | HttpTransactionData
@@ -367,6 +369,32 @@ def provide_shutdown_state(
             shutdown_state_ctx.reset(token)
 
 
+app_type_ctx = ContextVar[APP_TYPE]("app_type")
+
+
+def use_app_type() -> APP_TYPE:
+    """
+    Returns the current application type.
+    This function is used to access the application type in the context of an application transaction.
+    If no context is set, it raises a LookupError.
+    """
+    return app_type_ctx.get()
+
+
+@contextmanager
+def providing_app_type(app_type: APP_TYPE) -> Generator[None, None, None]:
+    """
+    Context manager to provide the application type.
+    This is used to set the application type for the current transaction.
+    """
+    token = app_type_ctx.set(app_type)
+    try:
+        yield
+    finally:
+        with suppress(ValueError):
+            app_type_ctx.reset(token)
+
+
 __all__ = [
     "AppTransactionContext",
     "AppInterceptor",

jararaca/presentation/server.py

@@ -17,6 +17,7 @@ from jararaca.microservice import (
     ShutdownState,
     WebSocketTransactionData,
     provide_shutdown_state,
+    providing_app_type,
 )
 from jararaca.presentation.decorators import RestController
 from jararaca.presentation.http_microservice import HttpMicroservice
@@ -37,49 +38,30 @@ class HttpAppLifecycle:
 
     @asynccontextmanager
     async def __call__(self, api: FastAPI) -> AsyncGenerator[None, None]:
-        async with self.lifecycle():
+        with providing_app_type("http"):
+            async with self.lifecycle():
 
-            # websocket_interceptors = [
-            #     interceptor
-            #     for interceptor in self.lifecycle.initialized_interceptors
-            #     if isinstance(interceptor, WebSocketInterceptor)
-            # ]
+                for controller_t in self.lifecycle.app.controllers:
+                    controller = RestController.get_controller(controller_t)
 
-            # for interceptor in websocket_interceptors:
-            #     router = interceptor.get_ws_router(
-            #         self.lifecycle.app, self.lifecycle.container, self.uow_provider
-            #     )
+                    if controller is None:
+                        continue
 
-            #     api.include_router(router)
+                    instance: Any = self.lifecycle.container.get_by_type(controller_t)
 
-            for controller_t in self.lifecycle.app.controllers:
-                controller = RestController.get_controller(controller_t)
+                    router = controller.get_router_factory()(self.lifecycle, instance)
 
-                if controller is None:
-                    continue
+                    api.include_router(router)
 
-                instance: Any = self.lifecycle.container.get_by_type(controller_t)
+                    for middleware in self.http_app.middlewares:
+                        middleware_instance = self.lifecycle.container.get_by_type(
+                            middleware
+                        )
+                        api.router.dependencies.append(
+                            Depends(middleware_instance.intercept)
+                        )
 
-                # dependencies: list[DependsCls] = []
-                # for middleware in controller.middlewares:
-                #     middleware_instance = self.lifecycle.container.get_by_type(
-                #         middleware
-                #     )
-                #     dependencies.append(Depends(middleware_instance.intercept))
-
-                router = controller.get_router_factory()(self.lifecycle, instance)
-
-                api.include_router(router)
-
-                for middleware in self.http_app.middlewares:
-                    middleware_instance = self.lifecycle.container.get_by_type(
-                        middleware
-                    )
-                    api.router.dependencies.append(
-                        Depends(middleware_instance.intercept)
-                    )
-
-            yield
+                yield
 
 
 class HttpShutdownState(ShutdownState):

jararaca/scheduler/beat_worker.py

@@ -31,7 +31,7 @@ from jararaca.broker_backend.mapper import get_message_broker_backend_from_url
 from jararaca.core.uow import UnitOfWorkContextProvider
 from jararaca.di import Container
 from jararaca.lifecycle import AppLifecycle
-from jararaca.microservice import Microservice
+from jararaca.microservice import Microservice, providing_app_type
 from jararaca.scheduler.decorators import (
     ScheduledAction,
     ScheduledActionData,
@@ -572,22 +572,23 @@ class BeatWorker:
         Declares the scheduled actions and starts the scheduler.
         This is the main entry point for the scheduler.
         """
-        async with self.lifecycle():
+        with providing_app_type("beat"):
+            async with self.lifecycle():
 
-            scheduled_actions = _extract_scheduled_actions(
-                self.app, self.container, self.scheduler_names
-            )
+                scheduled_actions = _extract_scheduled_actions(
+                    self.app, self.container, self.scheduler_names
+                )
 
-            # Initialize and wait for connection to be established
-            logger.info("Initializing broker connection...")
-            await self.broker.initialize(scheduled_actions)
+                # Initialize and wait for connection to be established
+                logger.info("Initializing broker connection...")
+                await self.broker.initialize(scheduled_actions)
 
-            # Wait for connection to be healthy before starting scheduler
-            logger.info("Waiting for connection to be established...")
-            await self._wait_for_broker_connection()
+                # Wait for connection to be healthy before starting scheduler
+                logger.info("Waiting for connection to be established...")
+                await self._wait_for_broker_connection()
 
-            logger.info("Connection established, starting scheduler...")
-            await self.run_scheduled_actions(scheduled_actions)
+                logger.info("Connection established, starting scheduler...")
+                await self.run_scheduled_actions(scheduled_actions)
 
     async def run_scheduled_actions(
         self, scheduled_actions: list[ScheduledActionData]

{jararaca-0.3.12a22.dist-info → jararaca-0.3.13.dist-info}/METADATA

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

{jararaca-0.3.12a22.dist-info → jararaca-0.3.13.dist-info}/RECORD

@@ -1,7 +1,7 @@
 LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
 README.md,sha256=2qMM__t_MoLKZr4IY9tXjo-Jn6LKjuHMb1qbyXpgL08,3401
-pyproject.toml,sha256=oFFxCzmbhLtjUJxmR15XetlFISif5aYSACnRXpD4tdQ,2060
-jararaca/__init__.py,sha256=vK3zyIVLckwZgj1FPX6jzSbzaSWmSy3wQ2KMwmpJnmg,22046
+pyproject.toml,sha256=2-YSpMDgBFSW37MDYlZYmX9p5hTXQaydhNsqs0H0POk,2057
+jararaca/__init__.py,sha256=qWLfavFIGRC3dfcTmiNP3kDh1Iih9wPMTTUOtLjtT_M,22186
 jararaca/__main__.py,sha256=-O3vsB5lHdqNFjUtoELDF81IYFtR-DSiiFMzRaiSsv4,67
 jararaca/broker_backend/__init__.py,sha256=GzEIuHR1xzgCJD4FE3harNjoaYzxHMHoEL0_clUaC-k,3528
 jararaca/broker_backend/mapper.py,sha256=vTsi7sWpNvlga1PWPFg0rCJ5joJ0cdzykkIc2Tuvenc,696
@@ -23,8 +23,8 @@ jararaca/messagebus/interceptors/aiopika_publisher_interceptor.py,sha256=_DEHwIH
 jararaca/messagebus/interceptors/publisher_interceptor.py,sha256=ojy1bRhqMgrkQljcGGS8cd8-8pUjL8ZHjIUkdmaAnNM,1325
 jararaca/messagebus/message.py,sha256=U6cyd2XknX8mtm0333slz5fanky2PFLWCmokAO56vvU,819
 jararaca/messagebus/publisher.py,sha256=JTkxdKbvxvDWT8nK8PVEyyX061vYYbKQMxRHXrZtcEY,2173
-jararaca/messagebus/worker.py,sha256=6uKAShB9vgxL_D1dpUonHuRYChx9Rb015SZUHcAsCmU,73679
-jararaca/microservice.py,sha256=4uQWPH1Ytxl2tg4IK-M_6T-28X89ku4ObJXJyg9CDr4,10663
+jararaca/messagebus/worker.py,sha256=TdWdKHFUc3AAsqhLZI08Y3uc5iqs0GtDHmEhyqYWObI,73968
+jararaca/microservice.py,sha256=OYCw5C4797X_tVnM_9sEz8BdjbICPHSVsCixsA_FwE4,11419
 jararaca/observability/decorators.py,sha256=MOIr2PttPYYvRwEdfQZEwD5RxKHOTv8UEy9n1YQVoKw,2281
 jararaca/observability/interceptor.py,sha256=U4ZLM0f8j6Q7gMUKKnA85bnvD-Qa0ii79Qa_X8KsXAQ,1498
 jararaca/observability/providers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -42,7 +42,7 @@ jararaca/presentation/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3h
 jararaca/presentation/decorators.py,sha256=dNcK1xYWhdHVBeWDa8dhqh4z8yQcPkJZTQkfgLXz6RI,11655
 jararaca/presentation/hooks.py,sha256=WBbU5DG3-MAm2Ro2YraQyYG_HENfizYfyShL2ktHi6k,1980
 jararaca/presentation/http_microservice.py,sha256=g771JosV6jTY3hQtG-HkLOo-T0e-r3b3rp1ddt99Qf0,533
-jararaca/presentation/server.py,sha256=Im-tei9EynKm1DO-63UIGeTJK5ZvxIhpAxbsYGn7ZW0,6244
+jararaca/presentation/server.py,sha256=YzTcG39xEmEstrtB6nmVUSf2cMP8cSguppKdz_jA_tE,5537
 jararaca/presentation/websocket/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 jararaca/presentation/websocket/base_types.py,sha256=AvUeeZ1TFhSiRMcYqZU1HaQNqSrcgTkC5R0ArP5dGmA,146
 jararaca/presentation/websocket/context.py,sha256=A6K5W3kqo9Hgeh1m6JiI7Cdz5SfbXcaICSVX7u1ARZo,1903
@@ -62,7 +62,7 @@ jararaca/rpc/http/backends/otel.py,sha256=Uc6CjHSCZ5hvnK1fNFv3ota5xzUFnvIl1JOpG3
 jararaca/rpc/http/decorators.py,sha256=0BCcmP_jolIAB73VeXm3TKyVr3AWbzkXzUVMGqYtqSQ,22145
 jararaca/rpc/http/httpx.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 jararaca/scheduler/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-jararaca/scheduler/beat_worker.py,sha256=f5gh8TFl0aScuriADCx21EN34GjcJ_eI-BqHgVcNZyk,29430
+jararaca/scheduler/beat_worker.py,sha256=5vbvIBDugtHtKLXUtmhOeEOQFPEJ4R7YbemzAkYDTzU,29539
 jararaca/scheduler/decorators.py,sha256=iyWFvPLCRh9c0YReQRemI2mLuaUv7r929So-xuKIWUs,4605
 jararaca/scheduler/types.py,sha256=4HEQOmVIDp-BYLSzqmqSFIio1bd51WFmgFPIzPpVu04,135
 jararaca/tools/app_config/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -74,8 +74,8 @@ jararaca/tools/typescript/interface_parser.py,sha256=hJzP-XCwVQ7kjJHBj9VGq8mrSdN
 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.12a22.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-jararaca-0.3.12a22.dist-info/METADATA,sha256=nsMOR_wNRfilnsinCu3kl-9IG6ePmLvBjxP7jCGRCYk,5036
-jararaca-0.3.12a22.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-jararaca-0.3.12a22.dist-info/entry_points.txt,sha256=WIh3aIvz8LwUJZIDfs4EeH3VoFyCGEk7cWJurW38q0I,45
-jararaca-0.3.12a22.dist-info/RECORD,,
+jararaca-0.3.13.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+jararaca-0.3.13.dist-info/METADATA,sha256=oLyzVkVe28eyKh5Ro_nRruvfgzY2J5k9JAtUGp-z8zQ,5033
+jararaca-0.3.13.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+jararaca-0.3.13.dist-info/entry_points.txt,sha256=WIh3aIvz8LwUJZIDfs4EeH3VoFyCGEk7cWJurW38q0I,45
+jararaca-0.3.13.dist-info/RECORD,,

pyproject.toml

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