taskboy 0.1.1__py3-none-any.whl

taskboy/slack_users.py

@@ -0,0 +1,32 @@
+"""shared Slack user-profile cache and refresh logic."""
+
+from datetime import datetime, timezone
+
+from taskboy.store import Store
+
+USER_CACHE_SECONDS = 24 * 60 * 60
+
+
+async def cached_user_profile(store: Store, client, user_id: str) -> dict | None:
+    cached = store.get_slack_user(user_id)
+    if cached:
+        try:
+            age = datetime.now(timezone.utc) - datetime.fromisoformat(cached["updated_at"])
+            if age.total_seconds() < USER_CACHE_SECONDS:
+                return cached
+        except (KeyError, TypeError, ValueError):
+            pass
+    response = await client.users_info(user=user_id)
+    user = response.get("user") or {}
+    profile = user.get("profile") or {}
+    return store.upsert_slack_user(
+        user_id,
+        team_id=user.get("team_id"),
+        username=user.get("name"),
+        real_name=user.get("real_name") or profile.get("real_name"),
+        display_name=profile.get("display_name"),
+        email=profile.get("email"),
+        title=profile.get("title"),
+        tz=user.get("tz"),
+        is_bot=int(bool(user.get("is_bot"))),
+    )

taskboy/started_messages.py

@@ -0,0 +1,33 @@
+"""token-free task-started message selection from agent and reviewer yaml pools."""
+
+import random
+from pathlib import Path
+
+import yaml
+
+
+def validate_content(content: str) -> dict[str, list[str]]:
+    try:
+        pools = yaml.safe_load(content)
+    except yaml.YAMLError as e:
+        raise ValueError(f"task-started messages are not valid yaml: {e}") from e
+    if not isinstance(pools, dict) or set(pools) != {"agent", "reviewer"}:
+        raise ValueError("task-started messages must be a yaml mapping with 'agent' and 'reviewer' lists")
+    for pool, choices in pools.items():
+        if not isinstance(choices, list) or not choices or not all(isinstance(item, str) and item.strip() for item in choices):
+            raise ValueError(f"task-started messages for {pool!r} must be a non-empty list of non-empty strings")
+    return pools
+
+
+def pick(path: str | None, pool: str | None) -> str | None:
+    if not path or not pool:
+        return None
+    try:
+        pools = yaml.safe_load(Path(path).read_text())
+        choices = pools.get(pool) if isinstance(pools, dict) else None
+        if not isinstance(choices, list):
+            return None
+        valid = [item.strip() for item in choices if isinstance(item, str) and item.strip()]
+        return random.choice(valid) if valid else None
+    except (OSError, UnicodeError, yaml.YAMLError):
+        return None