futurehouse-client 0.3.18.dev102__tar.gz → 0.3.18.dev110__tar.gz

{futurehouse_client-0.3.18.dev102 → futurehouse_client-0.3.18.dev110}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: futurehouse-client
-Version: 0.3.18.dev102
+Version: 0.3.18.dev110
 Summary: A client for interacting with endpoints of the FutureHouse service.
 Author-email: FutureHouse technical staff <hello@futurehouse.org>
 Classifier: Operating System :: OS Independent

futurehouse_client-0.3.18.dev110/futurehouse_client/utils/__init__.py

@@ -0,0 +1,7 @@
+from .context import JOB_ID_CONTEXT_KEY, USER_JWT_CONTEXT_KEY, RequestContext
+
+__all__ = [
+    "JOB_ID_CONTEXT_KEY",
+    "USER_JWT_CONTEXT_KEY",
+    "RequestContext",
+]

futurehouse_client-0.3.18.dev110/futurehouse_client/utils/context.py

@@ -0,0 +1,45 @@
+from typing import Any, ClassVar
+
+USER_JWT_CONTEXT_KEY = "user_jwt"
+JOB_ID_CONTEXT_KEY = "job_id"
+
+
+class RequestContext:
+    """A context manager for storing information from the initial request."""
+
+    _context: ClassVar[dict[str, Any]] = {}
+
+    @classmethod
+    def set(cls, key: str, value: Any) -> None:
+        """Set a context variable.
+
+        Args:
+            key: The context variable name
+            value: The value to store
+        """
+        cls._context[key] = value
+
+    @classmethod
+    def get(cls, key: str) -> Any:
+        """Get a context variable.
+
+        Args:
+            key: The context variable name
+            default: Default value if key doesn't exist
+
+        Returns:
+            The stored value or default if not found
+        """
+        return cls._context.get(key, None)
+
+    @classmethod
+    def clear(cls, key: str | None = None) -> None:
+        """Clear a specific context variable or all variables.
+
+        Args:
+            key: Specific key to clear, or None to clear all
+        """
+        if key is None:
+            cls._context.clear()
+        elif key in cls._context:
+            del cls._context[key]

{futurehouse_client-0.3.18.dev102 → futurehouse_client-0.3.18.dev110}/futurehouse_client.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: futurehouse-client
-Version: 0.3.18.dev102
+Version: 0.3.18.dev110
 Summary: A client for interacting with endpoints of the FutureHouse service.
 Author-email: FutureHouse technical staff <hello@futurehouse.org>
 Classifier: Operating System :: OS Independent

{futurehouse_client-0.3.18.dev102 → futurehouse_client-0.3.18.dev110}/futurehouse_client.egg-info/SOURCES.txt

@@ -18,6 +18,7 @@ futurehouse_client/models/app.py
 futurehouse_client/models/client.py
 futurehouse_client/models/rest.py
 futurehouse_client/utils/__init__.py
+futurehouse_client/utils/context.py
 futurehouse_client/utils/general.py
 futurehouse_client/utils/module_utils.py
 futurehouse_client/utils/monitoring.py