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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: futurehouse-client
-Version: 0.3.18.dev109
+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.dev109 → 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.dev109
+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.dev109/futurehouse_client/utils/__init__.py

@@ -1,3 +0,0 @@
-from .context import UserContext
-
-__all__ = ["UserContext"]

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

@@ -1,16 +0,0 @@
-class UserContext:
-    """A context manager for storing user information from the initial request."""
-
-    _user_jwt = None
-
-    @classmethod
-    def set_user_jwt(cls, jwt: str) -> None:
-        cls._user_jwt = jwt
-
-    @classmethod
-    def get_user_jwt(cls) -> str | None:
-        return cls._user_jwt
-
-    @classmethod
-    def clear_user_jwt(cls) -> None:
-        cls._user_jwt = None