watchman-sdk 0.1.0__tar.gz

watchman_sdk-0.1.0/PKG-INFO

@@ -0,0 +1,5 @@
+Metadata-Version: 2.4
+Name: watchman-sdk
+Version: 0.1.0
+Requires-Python: >=3.9
+Requires-Dist: httpx>=0.27.0

watchman_sdk-0.1.0/pyproject.toml

@@ -0,0 +1,9 @@
+[build-system]
+requires = ["setuptools>=68"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "watchman-sdk"
+version = "0.1.0"
+requires-python = ">=3.9"
+dependencies = ["httpx>=0.27.0"]

watchman_sdk-0.1.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

watchman_sdk-0.1.0/tests/test_client.py

@@ -0,0 +1,28 @@
+import sys, os
+sys.path.insert(0, os.path.join(os.path.dirname(__file__), ".."))
+from unittest.mock import patch, MagicMock
+from watchman_sdk.client import WatchmanClient
+
+def test_send_heartbeat_success():
+    client = WatchmanClient("test-svc", "token123", "http://localhost:8000")
+    mock_resp = MagicMock()
+    mock_resp.status_code = 200
+    with patch("httpx.post", return_value=mock_resp):
+        result = client.send_heartbeat()
+    assert result is True
+
+def test_send_heartbeat_failure():
+    client = WatchmanClient("test-svc", "token123", "http://localhost:8000")
+    with patch("httpx.post", side_effect=Exception("Connection refused")):
+        result = client.send_heartbeat()
+    assert result is False
+
+def test_send_event():
+    client = WatchmanClient("test-svc", "token123", "http://localhost:8000")
+    mock_resp = MagicMock()
+    mock_resp.status_code = 200
+    with patch("httpx.post", return_value=mock_resp) as mock_post:
+        result = client.send_event("error", "critical", "Something failed")
+    assert result is True
+    call_args = mock_post.call_args
+    assert call_args[1]["json"]["type"] == "error"

watchman_sdk-0.1.0/watchman_sdk/__init__.py

@@ -0,0 +1,39 @@
+import functools
+from watchman_sdk.client import WatchmanClient
+from watchman_sdk.heartbeat import HeartbeatThread
+from watchman_sdk.error_capture import setup_error_capture
+from typing import Optional
+
+_client: Optional[WatchmanClient] = None
+_heartbeat: Optional[HeartbeatThread] = None
+
+def init(token: str, core_url: str, service: str = None, heartbeat_interval: int = 60) -> None:
+    global _client, _heartbeat
+    _client = WatchmanClient(service=service, token=token, core=core_url)
+    setup_error_capture(_client)
+    _heartbeat = HeartbeatThread(_client, interval_seconds=heartbeat_interval)
+    _heartbeat.start()
+
+def send_event(type: str, severity: str, message: str, meta: dict = None) -> bool:
+    if _client is None:
+        raise RuntimeError("watchman.init() must be called first")
+    return _client.send_event(type=type, severity=severity, message=message, meta=meta)
+
+def check(name: str):
+    """Decorator for check functions."""
+    def decorator(func):
+        @functools.wraps(func)
+        def wrapper(*args, **kwargs):
+            if _client is None:
+                raise RuntimeError("watchman.init() must be called first")
+            result = func(*args, **kwargs)
+            ok = result.get("ok", False)
+            _client.send_event(
+                type="check",
+                severity="critical" if not ok else "info",
+                message=f"Check {name}: {'ok' if ok else 'failed'}",
+                meta={**result, "check_name": name},
+            )
+            return result
+        return wrapper
+    return decorator

watchman_sdk-0.1.0/watchman_sdk/client.py

@@ -0,0 +1,47 @@
+import httpx
+from typing import Optional, Any
+
+class WatchmanClient:
+    def __init__(self, service: str, token: str, core: str):
+        self.service = service
+        self.token = token
+        self.core = core.rstrip("/")
+
+    def _headers(self) -> dict:
+        return {"Authorization": f"Bearer {self.token}", "Content-Type": "application/json"}
+
+    def send_heartbeat(self, version: Optional[str] = None) -> bool:
+        try:
+            resp = httpx.post(
+                f"{self.core}/api/heartbeat",
+                json={"service": self.service, "status": "alive", "version": version},
+                headers=self._headers(),
+                timeout=5.0,
+            )
+            return resp.status_code == 200
+        except Exception:
+            return False
+
+    def send_event(
+        self,
+        type: str,
+        severity: str,
+        message: str,
+        meta: Optional[dict[str, Any]] = None,
+    ) -> bool:
+        try:
+            resp = httpx.post(
+                f"{self.core}/api/events",
+                json={
+                    "service": self.service,
+                    "type": type,
+                    "severity": severity,
+                    "message": message,
+                    "meta": meta or {},
+                },
+                headers=self._headers(),
+                timeout=5.0,
+            )
+            return resp.status_code == 200
+        except Exception:
+            return False

watchman_sdk-0.1.0/watchman_sdk/error_capture.py

@@ -0,0 +1,17 @@
+import sys
+import traceback
+
+def setup_error_capture(client) -> None:
+    original_hook = sys.excepthook
+
+    def custom_excepthook(exc_type, exc_value, exc_tb):
+        tb_str = "".join(traceback.format_tb(exc_tb))
+        client.send_event(
+            type="error",
+            severity="critical",
+            message=f"{exc_type.__name__}: {exc_value}",
+            meta={"traceback": tb_str},
+        )
+        original_hook(exc_type, exc_value, exc_tb)
+
+    sys.excepthook = custom_excepthook

watchman_sdk-0.1.0/watchman_sdk/heartbeat.py

@@ -0,0 +1,16 @@
+import threading
+
+class HeartbeatThread(threading.Thread):
+    def __init__(self, client, interval_seconds: int = 60):
+        super().__init__(daemon=True)
+        self.client = client
+        self.interval = interval_seconds
+        self._stop_event = threading.Event()
+
+    def run(self) -> None:
+        while not self._stop_event.is_set():
+            self.client.send_heartbeat()
+            self._stop_event.wait(self.interval)
+
+    def stop(self) -> None:
+        self._stop_event.set()

watchman_sdk-0.1.0/watchman_sdk.egg-info/PKG-INFO

@@ -0,0 +1,5 @@
+Metadata-Version: 2.4
+Name: watchman-sdk
+Version: 0.1.0
+Requires-Python: >=3.9
+Requires-Dist: httpx>=0.27.0

watchman_sdk-0.1.0/watchman_sdk.egg-info/SOURCES.txt

@@ -0,0 +1,11 @@
+pyproject.toml
+tests/test_client.py
+watchman_sdk/__init__.py
+watchman_sdk/client.py
+watchman_sdk/error_capture.py
+watchman_sdk/heartbeat.py
+watchman_sdk.egg-info/PKG-INFO
+watchman_sdk.egg-info/SOURCES.txt
+watchman_sdk.egg-info/dependency_links.txt
+watchman_sdk.egg-info/requires.txt
+watchman_sdk.egg-info/top_level.txt

watchman_sdk-0.1.0/watchman_sdk.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

watchman_sdk-0.1.0/watchman_sdk.egg-info/requires.txt

@@ -0,0 +1 @@
+httpx>=0.27.0

watchman_sdk-0.1.0/watchman_sdk.egg-info/top_level.txt

@@ -0,0 +1 @@
+watchman_sdk