django-cfg 1.1.56__py3-none-any.whl → 1.1.59__py3-none-any.whl

django_cfg/__init__.py

@@ -38,7 +38,7 @@ default_app_config = "django_cfg.apps.DjangoCfgConfig"
 from typing import TYPE_CHECKING
 
 # Version information
-__version__ = "1.1.56"
+__version__ = "1.1.59"
 __author__ = "Unrealos Team"
 __email__ = "info@unrealos.com"
 __license__ = "MIT"

django_cfg/management/commands/rundramatiq.py

@@ -143,9 +143,7 @@ class Command(BaseCommand):
         if options.get("pid_file"):
             args.extend(["--pid-file", options["pid_file"]])
         
-        # Add broker (required positional argument - must be first after options)
-        # Use our custom broker module
-        args.append("broker")
+        # Note: Using Python API instead of CLI, so no broker argument needed
         
         # Add discovered task modules
         discovered_modules = get_task_service().discover_tasks()
@@ -185,7 +183,7 @@ class Command(BaseCommand):
         self.stdout.write(f"  {' '.join(worker_args)}")
     
     def _start_workers(self, worker_args, options):
-        """Start Dramatiq workers."""
+        """Start Dramatiq workers using Python API instead of CLI."""
         self.stdout.write(
             self.style.SUCCESS("Starting Dramatiq workers...")
         )
@@ -194,9 +192,15 @@ class Command(BaseCommand):
         task_service = get_task_service()
         config = task_service.config
         
-        self.stdout.write(f"Processes: {config.get_effective_processes()}")
-        self.stdout.write(f"Threads: {config.dramatiq.threads}")
-        self.stdout.write(f"Queues: {', '.join(config.get_effective_queues())}")
+        processes = options.get("processes") or config.get_effective_processes()
+        threads = options.get("threads") or config.dramatiq.threads
+        queues = config.get_effective_queues()
+        if options.get("queues"):
+            queues = [q.strip() for q in options["queues"].split(",")]
+        
+        self.stdout.write(f"Processes: {processes}")
+        self.stdout.write(f"Threads: {threads}")
+        self.stdout.write(f"Queues: {', '.join(queues)}")
         
         # Write PID file if requested
         if options.get("pid_file"):
@@ -205,20 +209,33 @@ class Command(BaseCommand):
             self.stdout.write(f"PID written to: {options['pid_file']}")
         
         try:
-            # Import and run dramatiq
+            # Import dramatiq and get broker FIRST
             import dramatiq
-            from dramatiq.cli import main as dramatiq_main
+            from dramatiq.worker import Worker
+            from django_cfg.modules.django_tasks import get_dramatiq_broker
+            
+            # Get the configured broker
+            broker = get_dramatiq_broker()
+            
+            # CRITICAL: Set the broker as the global default BEFORE importing tasks
+            dramatiq.set_broker(broker)
+            
+            # Now discover and import task modules with the correct broker set
+            discovered_modules = get_task_service().discover_tasks()
+            for module_name in discovered_modules:
+                try:
+                    __import__(module_name)
+                    self.stdout.write(f"Loaded tasks from: {module_name}")
+                except ImportError as e:
+                    self.stdout.write(f"Warning: Could not import {module_name}: {e}")
             
-            # Replace sys.argv with our arguments
-            original_argv = sys.argv[:]
-            sys.argv = worker_args
+            # Verify broker is set correctly
+            current_broker = dramatiq.get_broker()
+            self.stdout.write(f"Using broker: {current_broker}")
             
-            try:
-                # Run dramatiq CLI
-                dramatiq_main()
-            finally:
-                # Restore original argv
-                sys.argv = original_argv
+            # Create and start worker
+            worker = Worker(broker, worker_timeout=600000)
+            worker.start()
                 
         except KeyboardInterrupt:
             self.stdout.write("\nShutting down workers...")

django_cfg/modules/django_tasks.py

@@ -681,6 +681,16 @@ def initialize_task_system():
         
         if config and config.enabled:
             logger.info("🔧 Initializing Django-CFG task system...")
+            
+            # CRITICAL: Set up global broker for task sending
+            try:
+                import dramatiq
+                broker = get_dramatiq_broker()
+                dramatiq.set_broker(broker)
+                logger.debug(f"Set global Dramatiq broker: {broker}")
+            except Exception as e:
+                logger.warning(f"Failed to set global broker: {e}")
+            
             logger.info("✅ Task system initialized successfully")
             logger.info("💡 To start workers, run: python manage.py rundramatiq")
         else:
@@ -712,6 +722,72 @@ def extend_constance_config_with_tasks():
         return []
 
 
+# === Broker Creation ===
+
+def create_dramatiq_broker():
+    """
+    Create and configure Dramatiq broker from Django settings.
+    
+    This function creates a broker instance that can be used directly
+    by the Dramatiq CLI without requiring a separate broker.py file.
+    
+    Returns:
+        dramatiq.Broker: Configured Dramatiq broker instance
+    """
+    try:
+        from django.conf import settings
+        
+        if not hasattr(settings, 'DRAMATIQ_BROKER'):
+            raise RuntimeError("DRAMATIQ_BROKER not configured in Django settings")
+        
+        broker_config = settings.DRAMATIQ_BROKER
+        
+        # Create broker from Django settings
+        broker = RedisBroker(**broker_config['OPTIONS'])
+        
+        # Add middleware (only if not already present)
+        existing_middleware_types = {type(mw).__name__ for mw in broker.middleware}
+        
+        for middleware_path in broker_config['MIDDLEWARE']:
+            try:
+                module_path, class_name = middleware_path.rsplit('.', 1)
+                
+                # Skip if middleware of this type already exists
+                if class_name in existing_middleware_types:
+                    continue
+                    
+                module = __import__(module_path, fromlist=[class_name])
+                middleware_class = getattr(module, class_name)
+                broker.add_middleware(middleware_class())
+            except Exception as e:
+                logger.warning(f"Failed to add middleware {middleware_path}: {e}")
+        
+        return broker
+        
+    except Exception as e:
+        logger.error(f"Failed to create Dramatiq broker: {e}")
+        raise
+
+
+# Global broker instance (lazy-loaded)
+_broker_instance = None
+
+
+def get_dramatiq_broker():
+    """
+    Get the global Dramatiq broker instance.
+    
+    Returns:
+        dramatiq.Broker: The singleton broker instance
+    """
+    global _broker_instance
+    
+    if _broker_instance is None:
+        _broker_instance = create_dramatiq_broker()
+    
+    return _broker_instance
+
+
 # === Exports ===
 
 __all__ = [
@@ -725,4 +801,6 @@ __all__ = [
     "extend_constance_config_with_tasks",
     "initialize_task_system",
     "clear_dramatiq_queues",
+    "create_dramatiq_broker",
+    "get_dramatiq_broker",
 ]

{django_cfg-1.1.56.dist-info → django_cfg-1.1.59.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: django-cfg
-Version: 1.1.56
+Version: 1.1.59
 Summary: 🚀 Production-ready Django configuration framework with type-safe settings, smart automation, and modern developer experience
 Project-URL: Homepage, https://github.com/markolofsen/django-cfg
 Project-URL: Documentation, https://django-cfg.readthedocs.io

{django_cfg-1.1.56.dist-info → django_cfg-1.1.59.dist-info}/RECORD

@@ -1,4 +1,4 @@
-django_cfg/__init__.py,sha256=QfEcj-OA9Y8t400zz0Bdm2xffpmYAB0qeg8B5_pemUI,14288
+django_cfg/__init__.py,sha256=9_vseCVwoJ7CXG8_EjpsdPmqrw021qouoH-wvSSGNwA,14288
 django_cfg/apps.py,sha256=k84brkeXJI7EgKZLEpTkM9YFZofKI4PzhFOn1cl9Msc,1656
 django_cfg/exceptions.py,sha256=RTQEoU3PfR8lqqNNv5ayd_HY2yJLs3eioqUy8VM6AG4,10378
 django_cfg/integration.py,sha256=-7hvd-4ohLdzH4eufCZTOe3yTzPoQyB_HCfvsSm9AAw,5218
@@ -156,7 +156,7 @@ django_cfg/management/commands/create_token.py,sha256=beHtUTuyFZhG97F9vSkaX-u7ti
 django_cfg/management/commands/generate.py,sha256=w0BF7IMftxNjxTxFuY8cw5pNKGW-LmTScJ8kZpxHu_8,4248
 django_cfg/management/commands/list_urls.py,sha256=D8ikInA3uE1LbQGLWmfdLnEqPg7wqrI3caQA6iTe_-0,11009
 django_cfg/management/commands/migrator.py,sha256=mhMM63uv_Jp9zHVVM7TMwCB90uv3iFZn1vOG-rXyi3s,16191
-django_cfg/management/commands/rundramatiq.py,sha256=NbzpwvqimrxuCbXVfNb3DJq4v0QPpYQNcRNVOfAyTOI,8146
+django_cfg/management/commands/rundramatiq.py,sha256=y9LYzko8ppZw_nQZPIIQ3v5-gqxFz6U0Yx35eQzc5Ag,9117
 django_cfg/management/commands/runserver_ngrok.py,sha256=mcTioDIzHgha6sGo5eazlJhdKr8y5-uEQIc3qG3AvCI,5237
 django_cfg/management/commands/script.py,sha256=I6zOEQEGaED0HoLxl2EqKz39HwbKg9HhdxnGKybfH5s,16974
 django_cfg/management/commands/show_config.py,sha256=0YJ99P1XvymT3fWritaNmn_HJ-PVb0I-yBy757M_bn8,8337
@@ -189,7 +189,7 @@ django_cfg/modules/base.py,sha256=X90X-0iBfnaUSaC7S8_ULa_rdT41tqTVJnT05_RuyK4,47
 django_cfg/modules/django_email.py,sha256=uBvvqRVe1DG73Qq2d2IBYTjhFRdvHgsIbkVw3ge9OW8,16586
 django_cfg/modules/django_logger.py,sha256=VfcPCurTdU3iI593EJNs3wUoWQowu7-ykJGuHNkE79M,6325
 django_cfg/modules/django_ngrok.py,sha256=OAvir2pBFHfko-XaVgZTjeJwyZw-NSEILaKNlqQziqA,10476
-django_cfg/modules/django_tasks.py,sha256=Zn69ewpj_3xZZxeI6sVrQMrAKf4fxtI4ME6bQo5KEmY,26531
+django_cfg/modules/django_tasks.py,sha256=qulUuE76sgTykHLaBp6k9A29ULHnp-lUNdsZfNBFZC0,29048
 django_cfg/modules/django_telegram.py,sha256=Mun2tAm0P2cUyQlAs8FaPe-FVgcrv7L_-FPTXQQEUT0,16356
 django_cfg/modules/logger.py,sha256=4_zeasNehr8LGz8r_ckv15-fQS63zCodiqD4CYIEyFI,10546
 django_cfg/modules/django_currency/README.md,sha256=Ox3jgRtsbOIaMuYDkIhrs9ijLGLbn-2R7mD9n2tjAVE,8512
@@ -251,8 +251,8 @@ django_cfg/templates/emails/base_email.html,sha256=TWcvYa2IHShlF_E8jf1bWZStRO0v8
 django_cfg/utils/__init__.py,sha256=64wwXJuXytvwt8Ze_erSR2HmV07nGWJ6DV5wloRBvYE,435
 django_cfg/utils/path_resolution.py,sha256=eML-6-RIGTs5TePktIQN8nxfDUEFJ3JA0AzWBcihAbs,13894
 django_cfg/utils/smart_defaults.py,sha256=iL6_n3jGDW5812whylWAwXik0xBSYihdLp4IJ26T5eA,20547
-django_cfg-1.1.56.dist-info/METADATA,sha256=0QKfOqPQDlVxMFbHyQu0zG1KvDiGFAuug4CmJKF3smg,38953
-django_cfg-1.1.56.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-django_cfg-1.1.56.dist-info/entry_points.txt,sha256=Ucmde4Z2wEzgb4AggxxZ0zaYDb9HpyE5blM3uJ0_VNg,56
-django_cfg-1.1.56.dist-info/licenses/LICENSE,sha256=xHuytiUkSZCRG3N11nk1X6q1_EGQtv6aL5O9cqNRhKE,1071
-django_cfg-1.1.56.dist-info/RECORD,,
+django_cfg-1.1.59.dist-info/METADATA,sha256=fhstTU5UCjrxQMKM692CCMivv-FNyiZleZBBPkQTyKw,38953
+django_cfg-1.1.59.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+django_cfg-1.1.59.dist-info/entry_points.txt,sha256=Ucmde4Z2wEzgb4AggxxZ0zaYDb9HpyE5blM3uJ0_VNg,56
+django_cfg-1.1.59.dist-info/licenses/LICENSE,sha256=xHuytiUkSZCRG3N11nk1X6q1_EGQtv6aL5O9cqNRhKE,1071
+django_cfg-1.1.59.dist-info/RECORD,,