sumo-wrapper-python 1.0.16__tar.gz → 1.0.18__tar.gz

{sumo_wrapper_python-1.0.16 → sumo_wrapper_python-1.0.18}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sumo-wrapper-python
-Version: 1.0.16
+Version: 1.0.18
 Summary: Python wrapper for the Sumo API
 Author: Equinor
 License:                                  Apache License

{sumo_wrapper_python-1.0.16 → sumo_wrapper_python-1.0.18}/src/sumo/wrapper/_auth_provider.py

@@ -66,6 +66,13 @@ class AuthProvider:
 
         return {"Authorization": "Bearer " + token}
 
+    def store_shared_access_key_for_case(self, case_uuid, token):
+        with open(
+            get_token_path(self._resource_id, ".sharedkey", case_uuid),
+            "w",
+        ) as f:
+            f.write(token)
+
     pass
 
 
@@ -86,6 +93,7 @@ class AuthProviderAccessToken(AuthProvider):
         self._access_token = access_token
         payload = jwt.decode(access_token, options={"verify_signature": False})
         self._expires = payload["exp"]
+        self._resource_id = payload["aud"]
         return
 
     def get_token(self):
@@ -110,10 +118,17 @@ class AuthProviderRefreshToken(AuthProvider):
     pass
 
 
-def get_token_path(resource_id, suffix):
-    return os.path.join(
-        os.path.expanduser("~"), ".sumo", str(resource_id) + suffix
-    )
+def get_token_path(resource_id, suffix, case_uuid=None):
+    if case_uuid is not None:
+        return os.path.join(
+            os.path.expanduser("~"),
+            ".sumo",
+            str(resource_id) + "+" + str(case_uuid) + suffix,
+        )
+    else:
+        return os.path.join(
+            os.path.expanduser("~"), ".sumo", str(resource_id) + suffix
+        )
 
 
 @tn.retry(
@@ -169,8 +184,8 @@ def get_token_cache(resource_id, suffix):
     retry_error_callback=_return_last_value,
     before_sleep=_log_retry_info,
 )
-def protect_token_cache(resource_id, suffix):
-    token_path = get_token_path(resource_id, suffix)
+def protect_token_cache(resource_id, suffix, case_uuid=None):
+    token_path = get_token_path(resource_id, suffix, case_uuid)
 
     if sys.platform.startswith("linux") or sys.platform == "darwin":
         filemode = stat.filemode(os.stat(token_path).st_mode)
@@ -362,9 +377,9 @@ class AuthProviderSumoToken(AuthProvider):
         retry_error_callback=_return_last_value,
         before_sleep=_log_retry_info,
     )
-    def __init__(self, resource_id):
-        protect_token_cache(resource_id, ".sharedkey")
-        token_path = get_token_path(resource_id, ".sharedkey")
+    def __init__(self, resource_id, case_uuid=None):
+        protect_token_cache(resource_id, ".sharedkey", case_uuid)
+        token_path = get_token_path(resource_id, ".sharedkey", case_uuid)
         with open(token_path, "r") as f:
             self._token = f.readline().strip()
         return
@@ -394,6 +409,7 @@ def get_auth_provider(
     access_token=None,
     refresh_token=None,
     devicecode=False,
+    case_uuid=None,
 ):
     if refresh_token:
         return AuthProviderRefreshToken(
@@ -403,8 +419,8 @@ def get_auth_provider(
     if access_token:
         return AuthProviderAccessToken(access_token)
     # ELSE
-    if os.path.exists(get_token_path(resource_id, ".sharedkey")):
-        return AuthProviderSumoToken(resource_id)
+    if os.path.exists(get_token_path(resource_id, ".sharedkey", case_uuid)):
+        return AuthProviderSumoToken(resource_id, case_uuid)
     # ELSE
     if os.path.exists(get_token_path(resource_id, ".token")):
         auth_silent = AuthProviderSilent(client_id, authority, resource_id)

{sumo_wrapper_python-1.0.16 → sumo_wrapper_python-1.0.18}/src/sumo/wrapper/_version.py

@@ -12,5 +12,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '1.0.16'
-__version_tuple__ = version_tuple = (1, 0, 16)
+__version__ = version = '1.0.18'
+__version_tuple__ = version_tuple = (1, 0, 18)

{sumo_wrapper_python-1.0.16 → sumo_wrapper_python-1.0.18}/src/sumo/wrapper/sumo_client.py

@@ -33,6 +33,7 @@ class SumoClient:
         verbosity: str = "CRITICAL",
         retry_strategy=RetryStrategy(),
         timeout=DEFAULT_TIMEOUT,
+        case_uuid=None,
     ):
         """Initialize a new Sumo object
 
@@ -49,6 +50,9 @@ class SumoClient:
         if env not in APP_REGISTRATION:
             raise ValueError(f"Invalid environment: {env}")
 
+        self.env = env
+        self._verbosity = verbosity
+
         self._retry_strategy = retry_strategy
         self._client = httpx.Client()
         self._async_client = httpx.AsyncClient()
@@ -85,6 +89,7 @@ class SumoClient:
             refresh_token=refresh_token,
             access_token=access_token,
             devicecode=devicecode,
+            case_uuid=case_uuid,
         )
 
         if env == "localhost":
@@ -398,6 +403,35 @@ class SumoClient:
             pass
         return logger
 
+    def create_shared_access_key_for_case(self, case_uuid):
+        """Creates and stores a shared access key that can be used to access
+        the case identified by *case_uuid*, in the current Sumo environment.
+
+        This shared access key can then be used by instantiating
+        SumoClient with the parameter case_uuid set accordingly.
+
+        Args:
+            case_uuid: the uuid for a case.
+
+        Side effects:
+            Creates a new file in ~/.sumo, named {app_id}+{case_uuid}
+        """
+        token = self.get(
+            f"/objects('{case_uuid}')/make-shared-access-key"
+        ).text
+        self.auth.store_shared_access_key_for_case(case_uuid, token)
+
+    def client_for_case(self, case_uuid):
+        """Instantiate and return new SumoClient for accessing the
+        case identified by *case_uuid*."""
+        return SumoClient(
+            env=self.env,
+            verbosity=self._verbosity,
+            retry_strategy=self._retry_strategy,
+            timeout=self._timeout,
+            case_uuid=case_uuid,
+        )
+
     @raise_for_status_async
     async def get_async(self, path: str, params: dict = None):
         """Performs an async GET-request to the Sumo API.

{sumo_wrapper_python-1.0.16 → sumo_wrapper_python-1.0.18}/src/sumo_wrapper_python.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sumo-wrapper-python
-Version: 1.0.16
+Version: 1.0.18
 Summary: Python wrapper for the Sumo API
 Author: Equinor
 License:                                  Apache License