howler-api 4.0.7.dev1428__tar.gz → 4.0.8.dev1451__tar.gz

{howler_api-4.0.7.dev1428 → howler_api-4.0.8.dev1451}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: howler-api
-Version: 4.0.7.dev1428
+Version: 4.0.8.dev1451
 Summary: Howler - API server
 License: MIT
 Keywords: howler,alerting,gc,canada,cse-cst,cse,cst,cyber,cccs

{howler_api-4.0.7.dev1428 → howler_api-4.0.8.dev1451}/howler/cronjobs/action_queue_worker.py

@@ -13,7 +13,7 @@ from apscheduler.schedulers.base import BaseScheduler
 from howler.common.logging import get_logger
 from howler.config import config
 from howler.odm.models.action import VALID_TRIGGERS
-from howler.services.action_service import _get_action_queue, process_action_batch
+from howler.services.action_service import TriggeredAction, get_action_queue, process_action_batch
 
 logger = get_logger(__file__)
 
@@ -33,7 +33,7 @@ def run_worker(trigger: str) -> None:  # pragma: no cover – long-running loop,
         logger.info("Action queue worker disabled by configuration, not starting for trigger=%s", trigger)
         return
 
-    queue = _get_action_queue(trigger)
+    queue = get_action_queue(trigger)
     logger.info(
         "Action queue worker started for trigger=%s (batch_size=%d, timeout=%ds)",
         trigger,
@@ -41,7 +41,7 @@ def run_worker(trigger: str) -> None:  # pragma: no cover – long-running loop,
         BATCH_TIMEOUT,
     )
 
-    batch: list[dict] = []
+    batch: list[TriggeredAction] = []
 
     while True:
         try:

{howler_api-4.0.7.dev1428 → howler_api-4.0.8.dev1451}/howler/services/action_service.py

@@ -1,5 +1,6 @@
 import json
-from typing import Any, Optional
+from collections import defaultdict
+from typing import Any, Optional, TypedDict
 
 from flask import Response
 
@@ -18,11 +19,19 @@ from howler.utils.str_utils import sanitize_lucene_query
 
 logger = get_logger(__file__)
 
+
+class TriggeredAction(TypedDict):
+    """Type for triggered actions"""
+
+    hit_ids: list[str]
+    uname: str | None
+
+
 # Per-trigger persistent queues for buffering action execution requests.
-_action_queues: dict[str, NamedQueue[dict]] = {}
+_action_queues: dict[str, NamedQueue[TriggeredAction]] = {}
 
 
-def _get_action_queue(trigger: str) -> NamedQueue[dict]:
+def get_action_queue(trigger: str) -> NamedQueue[TriggeredAction]:
     """Return the action queue for *trigger*, creating it on first use.
 
     Raises:
@@ -65,10 +74,10 @@ def enqueue_action_execution(hit_ids: list[str], trigger: str = "create", user:
         return
 
     try:
-        _get_action_queue(trigger).push(
+        get_action_queue(trigger).push(
             {
                 "hit_ids": hit_ids,
-                "user": user.as_primitives() if user is not None and hasattr(user, "as_primitives") else user,
+                "uname": user.uname if user else None,
             }
         )
     except Exception:
@@ -77,7 +86,7 @@ def enqueue_action_execution(hit_ids: list[str], trigger: str = "create", user:
         bulk_execute_on_query(query, trigger=trigger, user=user)
 
 
-def process_action_batch(trigger: str, items: list[dict]) -> None:
+def process_action_batch(trigger: str, items: list[TriggeredAction]) -> None:
     """Process a batch of queued action execution requests for a single trigger.
 
     Groups items by user and issues a single coalesced
@@ -90,26 +99,23 @@ def process_action_batch(trigger: str, items: list[dict]) -> None:
     if not items:
         return
 
-    groups: dict[str | None, tuple[list[str], Any]] = {}
+    groups: dict[str | None, list[str]] = defaultdict(list)
 
     for item in items:
-        user_data = item.get("user")
-        user_key = user_data["uname"] if isinstance(user_data, dict) and "uname" in user_data else None
-
-        if user_key not in groups:
-            groups[user_key] = ([], user_data)
+        uname = item.get("uname")
 
-        groups[user_key][0].extend(item["hit_ids"])
+        groups[uname].extend(item["hit_ids"])
 
-    for user_key, (hit_ids, user_data) in groups.items():
+    for uname, hit_ids in groups.items():
         # Deduplicate IDs within a batch group
         unique_ids = list(dict.fromkeys(hit_ids))
         query = f"howler.id:({' OR '.join(sanitize_lucene_query(h) for h in unique_ids)})"
 
         try:
-            bulk_execute_on_query(query, trigger=trigger, user=user_data)
+            user = datastore().user.get(uname)
+            bulk_execute_on_query(query, trigger=trigger, user=user)
         except Exception:
-            logger.exception("Error processing action batch for trigger=%s user=%s", trigger, user_key)
+            logger.exception("Error processing action batch for trigger=%s user=%s", trigger, uname)
 
 
 def validate_action(new_action: Any) -> Optional[Response]:  # noqa: C901

{howler_api-4.0.7.dev1428 → howler_api-4.0.8.dev1451}/pyproject.toml

@@ -152,7 +152,7 @@ suppress-none-returning = true
 [tool.poetry]
 package-mode = true
 name = "howler-api"
-version = "4.0.7.dev1428"
+version = "4.0.8.dev1451"
 description = "Howler - API server"
 authors = [
     "Canadian Centre for Cyber Security <howler@cyber.gc.ca>",