Skip to content

Commit e358556

Browse files
Avi-Robustaarikalon1
authored andcommitted
added krr params (#1775)
1 parent 1324bf5 commit e358556

1 file changed

Lines changed: 7 additions & 2 deletions

File tree

  • playbooks/robusta_playbooks

playbooks/robusta_playbooks/krr.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,6 @@
1010
from hikaru.model.rel_1_26 import Container, EnvVar, EnvVarSource, PodSpec, ResourceRequirements, SecretKeySelector
1111
from prometrix import AWSPrometheusConfig, CoralogixPrometheusConfig, PrometheusAuthorization, PrometheusConfig
1212
from pydantic import BaseModel, ValidationError, validator
13-
1413
from robusta.api import (
1514
IMAGE_REGISTRY,
1615
RUNNER_SERVICE_ACCOUNT,
@@ -32,6 +31,7 @@
3231
from robusta.core.reporting.consts import ScanState
3332
from robusta.integrations.openshift import IS_OPENSHIFT
3433
from robusta.integrations.prometheus.utils import generate_prometheus_config
34+
from robusta.utils.parsing import format_event_templated_string
3535

3636
IMAGE: str = os.getenv("KRR_IMAGE_OVERRIDE", f"{IMAGE_REGISTRY}/krr:v1.23.0")
3737
KRR_MEMORY_LIMIT: str = os.getenv("KRR_MEMORY_LIMIT", "2Gi")
@@ -279,8 +279,13 @@ def krr_scan(event: ExecutionBaseEvent, params: KRRParams):
279279
scan_id = str(uuid.uuid4())
280280
prom_config = generate_prometheus_config(params)
281281
additional_flags = get_krr_additional_flags(params)
282+
args_sanitized = params.args_sanitized
283+
284+
if params.args_sanitized and hasattr(event, "obj") and event.obj is not None:
285+
subject = event.get_subject()
286+
args_sanitized = format_event_templated_string(subject, params.args_sanitized)
282287

283-
python_command = f"python krr.py {params.strategy} {params.args_sanitized} {additional_flags} "
288+
python_command = f"python krr.py {params.strategy} {args_sanitized} {additional_flags} "
284289
verbose_str = "-v" if params.krr_verbose else ""
285290
python_command += f"--max-workers {params.max_workers} {verbose_str} -f json --width 2048"
286291

0 commit comments

Comments
 (0)