Skip to content

Commit 314f353

Browse files
emsearcyemsearcy
committed
Format files and update dependencies
Signed-off-by: Eric Searcy <[email protected]>
1 parent 61babc4 commit 314f353

File tree

2 files changed

+209
-134
lines changed

2 files changed

+209
-134
lines changed

‎src/lfx_v2_mockdata/__init__.py‎

Lines changed: 48 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -81,7 +81,9 @@ class UploadMockDataArgs(BaseModel):
8181
)
8282
jinja_env: contextvars.ContextVar[Environment] = contextvars.ContextVar("jinja_env")
8383
args: contextvars.ContextVar[UploadMockDataArgs] = contextvars.ContextVar("args")
84-
retries_remaining: contextvars.ContextVar[int] = contextvars.ContextVar("retries_remaining")
84+
retries_remaining: contextvars.ContextVar[int] = contextvars.ContextVar(
85+
"retries_remaining"
86+
)
8587

8688
# NATS connection variables.
8789
nats_client: None | NatsClient = None
@@ -178,7 +180,9 @@ def evaluate(self):
178180
# Attempt to evaluate expression against data context.
179181
value = jmespath.search(self.expression, data_context)
180182
if value is None:
181-
raise AttributeError(f"JMESPath expression '{self.expression}' not found in data")
183+
raise AttributeError(
184+
f"JMESPath expression '{self.expression}' not found in data"
185+
)
182186
return value
183187

184188

@@ -242,7 +246,9 @@ def replace_placeholder(match):
242246
# Attempt to evaluate expression against data context.
243247
value = jmespath.search(expression, data_context)
244248
if value is None:
245-
raise AttributeError(f"JMESPath expression '{expression}' not found in data")
249+
raise AttributeError(
250+
f"JMESPath expression '{expression}' not found in data"
251+
)
246252
return str(value)
247253

248254
# Find and replace all ${...} patterns with their evaluated values.
@@ -343,7 +349,9 @@ def generate_jwt(self) -> str:
343349
)
344350
# Ensure we have an RSA private key for PS256 algorithm.
345351
if not isinstance(loaded_key, rsa.RSAPrivateKey):
346-
raise ValueError("JWT signing requires an RSA private key for PS256 algorithm")
352+
raise ValueError(
353+
"JWT signing requires an RSA private key for PS256 algorithm"
354+
)
347355
private_key = loaded_key
348356
except Exception as e:
349357
raise ValueError(f"Failed to load RSA private key: {e}") from e
@@ -377,7 +385,9 @@ def _get_key_id(self, cli_args) -> str:
377385
return _jwks_kid_cache
378386

379387
# Fetch from Heimdall JWKS endpoint.
380-
jwks_url = "http://lfx-platform-heimdall.lfx.svc.cluster.local:4457/.well-known/jwks"
388+
jwks_url = (
389+
"http://lfx-platform-heimdall.lfx.svc.cluster.local:4457/.well-known/jwks"
390+
)
381391
try:
382392
response = requests.get(jwks_url, timeout=10)
383393
response.raise_for_status()
@@ -532,7 +542,9 @@ def yaml_render(template_dir, yaml_file):
532542
env.globals["fake"] = fake
533543
env.globals["timedelta"] = datetime.timedelta
534544
env.globals["now_z"] = (
535-
lambda: datetime.datetime.now(datetime.UTC).isoformat("T").replace("+00:00", "Z")
545+
lambda: datetime.datetime.now(datetime.UTC)
546+
.isoformat("T")
547+
.replace("+00:00", "Z")
536548
)
537549
env.globals["slug_from_project_name"] = slug_from_project_name
538550
# Store the environment in the context for use by the !include
@@ -743,7 +755,9 @@ def run_http_request_playbook(name: str, playbook: dict) -> None:
743755
except AttributeError as e:
744756
if cli_args.dry_run:
745757
if cli_args.force:
746-
logger.error("Error processing playbook", error=str(e), playbook=name)
758+
logger.error(
759+
"Error processing playbook", error=str(e), playbook=name
760+
)
747761
step_payload["_response"] = {}
748762
continue
749763
else:
@@ -752,7 +766,9 @@ def run_http_request_playbook(name: str, playbook: dict) -> None:
752766
if retries_remaining.get() > 0:
753767
continue
754768
if cli_args.force:
755-
logger.error("Error processing playbook", error=str(e), playbook=name)
769+
logger.error(
770+
"Error processing playbook", error=str(e), playbook=name
771+
)
756772
continue
757773
raise
758774
if request_data is None and "raw" in step_payload:
@@ -792,7 +808,9 @@ def run_http_request_playbook(name: str, playbook: dict) -> None:
792808
step_payload["_response"] = r_dict
793809
except json.decoder.JSONDecodeError as e:
794810
if cli_args.force:
795-
logger.error("Failed to parse response as JSON", error=str(e), playbook=name)
811+
logger.error(
812+
"Failed to parse response as JSON", error=str(e), playbook=name
813+
)
796814
# Add a placeholder response to prevent re-running.
797815
step_payload["_response"] = {}
798816
continue
@@ -848,7 +866,9 @@ async def run_nats_publish_playbook(name: str, playbook: dict) -> None:
848866
except AttributeError as e:
849867
if cli_args.dry_run:
850868
if cli_args.force:
851-
logger.error("Error processing playbook", error=str(e), playbook=name)
869+
logger.error(
870+
"Error processing playbook", error=str(e), playbook=name
871+
)
852872
step_payload["_response"] = {}
853873
continue
854874
else:
@@ -857,7 +877,9 @@ async def run_nats_publish_playbook(name: str, playbook: dict) -> None:
857877
if retries_remaining.get() > 0:
858878
continue
859879
if cli_args.force:
860-
logger.error("Error processing playbook", error=str(e), playbook=name)
880+
logger.error(
881+
"Error processing playbook", error=str(e), playbook=name
882+
)
861883
continue
862884
raise
863885
elif "raw" in step_payload:
@@ -956,7 +978,9 @@ async def run_nats_kv_put_playbook(name: str, playbook: dict) -> None:
956978
except AttributeError as e:
957979
if cli_args.dry_run:
958980
if cli_args.force:
959-
logger.error("Error processing playbook", error=str(e), playbook=name)
981+
logger.error(
982+
"Error processing playbook", error=str(e), playbook=name
983+
)
960984
step_payload["_response"] = {}
961985
continue
962986
else:
@@ -965,7 +989,9 @@ async def run_nats_kv_put_playbook(name: str, playbook: dict) -> None:
965989
if retries_remaining.get() > 0:
966990
continue
967991
if cli_args.force:
968-
logger.error("Error processing playbook", error=str(e), playbook=name)
992+
logger.error(
993+
"Error processing playbook", error=str(e), playbook=name
994+
)
969995
continue
970996
raise
971997
elif "raw" in step_payload:
@@ -1050,7 +1076,9 @@ async def run_nats_request_playbook(name: str, playbook: dict) -> None:
10501076
except AttributeError as e:
10511077
if cli_args.dry_run:
10521078
if cli_args.force:
1053-
logger.error("Error processing playbook", error=str(e), playbook=name)
1079+
logger.error(
1080+
"Error processing playbook", error=str(e), playbook=name
1081+
)
10541082
step_payload["_response"] = {}
10551083
continue
10561084
else:
@@ -1059,7 +1087,9 @@ async def run_nats_request_playbook(name: str, playbook: dict) -> None:
10591087
if retries_remaining.get() > 0:
10601088
continue
10611089
if cli_args.force:
1062-
logger.error("Error processing playbook", error=str(e), playbook=name)
1090+
logger.error(
1091+
"Error processing playbook", error=str(e), playbook=name
1092+
)
10631093
continue
10641094
raise
10651095
elif "raw" in step_payload:
@@ -1085,7 +1115,9 @@ async def run_nats_request_playbook(name: str, playbook: dict) -> None:
10851115
)
10861116

10871117
try:
1088-
response = await nats_client.request(params.subject, data, timeout=params.timeout)
1118+
response = await nats_client.request(
1119+
params.subject, data, timeout=params.timeout
1120+
)
10891121
# Parse the response data and store it.
10901122
try:
10911123
response_data = json.loads(response.data.decode())

0 commit comments

Comments
 (0)