Skip to content

Commit 224fe29

Browse files
IevIeievgeniia ieromenko
andauthored
Refactored GuardDuty code (#173)
* refactoring guardduty * linting fix --------- Co-authored-by: ievgeniia ieromenko <ieviero@amazon.com>
1 parent f141288 commit 224fe29

File tree

2 files changed

+220
-161
lines changed

2 files changed

+220
-161
lines changed

aws_sra_examples/solutions/guardduty/guardduty_org/lambda/src/app.py

Lines changed: 11 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -213,16 +213,20 @@ def process_create_update_event(params: dict, regions: list) -> None:
213213
enable_eks_addon_management = (params.get("ENABLE_EKS_ADDON_MANAGEMENT", "false")).lower() in "true"
214214
enable_lambda_network_logs = (params.get("ENABLE_LAMBDA_NETWORK_LOGS", "false")).lower() in "true"
215215

216+
gd_features = {
217+
"auto_enable_s3_logs": auto_enable_s3_logs,
218+
"enable_eks_audit_logs": enable_eks_audit_logs,
219+
"auto_enable_malware_protection": auto_enable_malware_protection,
220+
"enable_rds_login_events": enable_rds_login_events,
221+
"enable_eks_runtime_monitoring": enable_eks_runtime_monitoring,
222+
"enable_eks_addon_management": enable_eks_addon_management,
223+
"enable_lambda_network_logs": enable_lambda_network_logs,
224+
}
225+
216226
guardduty.configure_guardduty(
217227
session,
218228
params["DELEGATED_ADMIN_ACCOUNT_ID"],
219-
auto_enable_s3_logs,
220-
enable_eks_audit_logs,
221-
auto_enable_malware_protection,
222-
enable_rds_login_events,
223-
enable_eks_runtime_monitoring,
224-
enable_eks_addon_management,
225-
enable_lambda_network_logs,
229+
gd_features,
226230
regions,
227231
params.get("FINDING_PUBLISHING_FREQUENCY", "FIFTEEN_MINUTES"),
228232
params["KMS_KEY_ARN"],

0 commit comments

Comments
 (0)