Skip to content

Commit e470622

Browse files
committed
chore: ruff format.
1 parent 62a12d1 commit e470622

2 files changed

Lines changed: 13 additions & 7 deletions

File tree

src/sumo/wrapper/_auth_provider.py

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -67,8 +67,10 @@ def get_authorization(self):
6767
return {"Authorization": "Bearer " + token}
6868

6969
def store_shared_access_key_for_case(self, case_uuid, token):
70-
with open(get_token_path(self._resource_id + "+" + case_uuid,
71-
".sharedkey"), "w") as f:
70+
with open(
71+
get_token_path(self._resource_id + "+" + case_uuid, ".sharedkey"),
72+
"w",
73+
) as f:
7274
f.write(token)
7375

7476
pass
@@ -399,7 +401,7 @@ def get_auth_provider(
399401
access_token=None,
400402
refresh_token=None,
401403
devicecode=False,
402-
case_uuid = None,
404+
case_uuid=None,
403405
):
404406
if refresh_token:
405407
return AuthProviderRefreshToken(
@@ -409,7 +411,9 @@ def get_auth_provider(
409411
if access_token:
410412
return AuthProviderAccessToken(access_token)
411413
# ELSE
412-
if case_uuid is not None and os.path.exists(get_token_path(resource_id + "+" + case_uuid, ".sharedkey")):
414+
if case_uuid is not None and os.path.exists(
415+
get_token_path(resource_id + "+" + case_uuid, ".sharedkey")
416+
):
413417
return AuthProviderSumoToken(resource_id + "+" + case_uuid)
414418
# ELSE
415419
if os.path.exists(get_token_path(resource_id, ".sharedkey")):

src/sumo/wrapper/sumo_client.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@ def __init__(
3333
verbosity: str = "CRITICAL",
3434
retry_strategy=RetryStrategy(),
3535
timeout=DEFAULT_TIMEOUT,
36-
case_uuid = None,
36+
case_uuid=None,
3737
):
3838
"""Initialize a new Sumo object
3939
@@ -86,7 +86,7 @@ def __init__(
8686
refresh_token=refresh_token,
8787
access_token=access_token,
8888
devicecode=devicecode,
89-
case_uuid = case_uuid,
89+
case_uuid=case_uuid,
9090
)
9191

9292
if env == "localhost":
@@ -413,7 +413,9 @@ def create_shared_access_key_for_case(self, case_uuid):
413413
Side effects:
414414
Creates a new file in ~/.sumo, named {app_id}+{case_uuid}
415415
"""
416-
token = self.get(f"/objects('{case_uuid}')/make-shared-access-key").text
416+
token = self.get(
417+
f"/objects('{case_uuid}')/make-shared-access-key"
418+
).text
417419
self.auth.store_shared_access_key_for_case(case_uuid, token)
418420

419421
@raise_for_status_async

0 commit comments

Comments
 (0)