scalable-pypeline 2.0.6__tar.gz → 2.0.8__tar.gz

{scalable-pypeline-2.0.6/scalable_pypeline.egg-info → scalable-pypeline-2.0.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: scalable-pypeline
-Version: 2.0.6
+Version: 2.0.8
 Summary: PypeLine - Python pipelines for the Real World
 Home-page: https://gitlab.com/bravos2/pypeline
 Author: Bravos Power Corporation

scalable-pypeline-2.0.8/pypeline/__init__.py

@@ -0,0 +1 @@
+__version__ = "2.0.8"

{scalable-pypeline-2.0.6 → scalable-pypeline-2.0.8}/pypeline/dramatiq.py

@@ -4,9 +4,10 @@ import sys
 import typing
 import pika
 import logging
-+from urllib.parse import urlparse
 
 import click
+from urllib.parse import urlparse
+
 from pypeline.extensions import pypeline_config
 from warnings import warn
 from functools import wraps
@@ -178,8 +179,7 @@ def register_actors_for_workers(broker: Broker):
                 pipeline_meta["store_results"] = True
                 _ = register_lazy_actor(broker, tmp_handler, pipeline_meta)
         except Exception as e:
-            print(e)
-            print(f"Unable to add a task to dramatiq: {e}")
+            logger.exception(f"Unable to add a task to dramatiq: {e}")
 
 
 class Dramatiq:
@@ -259,7 +259,7 @@ class Dramatiq:
 
         # Callable function is expected to setBroker()
         if callable(broker_or_callable):
-            print(f"Configuring broker via {DEFAULT_BROKER_CALLABLE}")
+            logger.info(f"Configuring broker via {DEFAULT_BROKER_CALLABLE}")
             broker_or_callable()
         else:
             raise TypeError("DEFAULT_BROKER_CALLABLE must point to a callable function")
@@ -367,7 +367,15 @@ def list_managed_actors(broker, queues):
 def cron_scheduler():  # pragma: no cover
     # Configure our broker that we will schedule registered tasks for
     scheduler = BlockingScheduler()
-    configure_default_broker()
+    module_name, broker_or_callable = import_object(DEFAULT_BROKER_CALLABLE)
+
+    # Callable function is expected to setBroker()
+    if callable(broker_or_callable):
+        logger.info(f"Configuring broker via {DEFAULT_BROKER_CALLABLE}")
+        broker_or_callable()
+    else:
+        raise TypeError("DEFAULT_BROKER_CALLABLE must point to a callable function")
+
     broker = get_broker()
     jobs = retrieve_latest_schedule_config()
 

{scalable-pypeline-2.0.6 → scalable-pypeline-2.0.8/scalable_pypeline.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: scalable-pypeline
-Version: 2.0.6
+Version: 2.0.8
 Summary: PypeLine - Python pipelines for the Real World
 Home-page: https://gitlab.com/bravos2/pypeline
 Author: Bravos Power Corporation

scalable-pypeline-2.0.6/pypeline/__init__.py

@@ -1 +0,0 @@
-__version__ = "2.0.6"