scalebox-sdk 0.1.24__py3-none-any.whl → 1.0.1__py3-none-any.whl

scalebox/sandbox/signature.py

@@ -1,40 +1,40 @@
-import base64
-import hashlib
-import time
-from typing import Literal, Optional, TypedDict
-
-Operation = Literal["read", "write"]
-
-
-class Signature(TypedDict):
-    signature: str
-    expiration: Optional[int]  # Unix timestamp or None
-
-
-def get_signature(
-    path: str,
-    operation: Operation,
-    user: str,
-    envd_access_token: Optional[str],
-    expiration_in_seconds: Optional[int] = None,
-) -> Signature:
-    """
-    Generate a v1 signature for sandbox file URLs.
-    """
-    if not envd_access_token:
-        raise ValueError("Access token is not set and signature cannot be generated!")
-
-    expiration = (
-        int(time.time()) + expiration_in_seconds if expiration_in_seconds else None
-    )
-
-    raw = (
-        f"{path}:{operation}:{user}:{envd_access_token}"
-        if expiration is None
-        else f"{path}:{operation}:{user}:{envd_access_token}:{expiration}"
-    )
-
-    digest = hashlib.sha256(raw.encode("utf-8")).digest()
-    encoded = base64.b64encode(digest).rstrip(b"=").decode("ascii")
-
-    return {"signature": f"v1_{encoded}", "expiration": expiration}
+import base64
+import hashlib
+import time
+from typing import Literal, Optional, TypedDict
+
+Operation = Literal["read", "write"]
+
+
+class Signature(TypedDict):
+    signature: str
+    expiration: Optional[int]  # Unix timestamp or None
+
+
+def get_signature(
+    path: str,
+    operation: Operation,
+    user: str,
+    envd_access_token: Optional[str],
+    expiration_in_seconds: Optional[int] = None,
+) -> Signature:
+    """
+    Generate a v1 signature for sandbox file URLs.
+    """
+    if not envd_access_token:
+        raise ValueError("Access token is not set and signature cannot be generated!")
+
+    expiration = (
+        int(time.time()) + expiration_in_seconds if expiration_in_seconds else None
+    )
+
+    raw = (
+        f"{path}:{operation}:{user}:{envd_access_token}"
+        if expiration is None
+        else f"{path}:{operation}:{user}:{envd_access_token}:{expiration}"
+    )
+    print(raw)
+    digest = hashlib.sha256(raw.encode("utf-8")).digest()
+    encoded = base64.b64encode(digest).rstrip(b"=").decode("ascii")
+
+    return {"signature": f"v1_{encoded}", "expiration": expiration}

scalebox/sandbox/utils.py

@@ -1,34 +1,34 @@
-import functools
-from typing import Any, Optional, Type, TypeVar, cast
-
-T = TypeVar("T")
-
-
-class class_method_variant(object):
-    def __init__(self, class_method_name):
-        self.class_method_name = class_method_name
-
-    method: Any
-
-    def __call__(self, method: T) -> T:
-        self.method = method
-        return cast(T, self)
-
-    def __get__(self, obj, objtype: Optional[Type[Any]] = None):
-        @functools.wraps(self.method)
-        def _wrapper(*args, **kwargs):
-            if obj is not None:
-                # Method was called as an instance method, e.g.
-                # instance.method(...)
-                return self.method(obj, *args, **kwargs)
-            elif len(args) > 0 and objtype is not None and isinstance(args[0], objtype):
-                # Method was called as a class method with the instance as the
-                # first argument, e.g. Class.method(instance, ...) which in
-                # Python is the same thing as calling an instance method
-                return self.method(args[0], *args[1:], **kwargs)
-            else:
-                # Method was called as a class method, e.g. Class.method(...)
-                class_method = getattr(objtype, self.class_method_name)
-                return class_method(*args, **kwargs)
-
-        return _wrapper
+import functools
+from typing import Any, Optional, Type, TypeVar, cast
+
+T = TypeVar("T")
+
+
+class class_method_variant(object):
+    def __init__(self, class_method_name):
+        self.class_method_name = class_method_name
+
+    method: Any
+
+    def __call__(self, method: T) -> T:
+        self.method = method
+        return cast(T, self)
+
+    def __get__(self, obj, objtype: Optional[Type[Any]] = None):
+        @functools.wraps(self.method)
+        def _wrapper(*args, **kwargs):
+            if obj is not None:
+                # Method was called as an instance method, e.g.
+                # instance.method(...)
+                return self.method(obj, *args, **kwargs)
+            elif len(args) > 0 and objtype is not None and isinstance(args[0], objtype):
+                # Method was called as a class method with the instance as the
+                # first argument, e.g. Class.method(instance, ...) which in
+                # Python is the same thing as calling an instance method
+                return self.method(args[0], *args[1:], **kwargs)
+            else:
+                # Method was called as a class method, e.g. Class.method(...)
+                class_method = getattr(objtype, self.class_method_name)
+                return class_method(*args, **kwargs)
+
+        return _wrapper