From 12d4bc687719c72d809808b8877b67a018601051 Mon Sep 17 00:00:00 2001 From: Sunil Sahu Date: Tue, 1 Apr 2025 20:13:08 +0530 Subject: [PATCH] Fix to respect user provided s3 source and destination path prefixes --- datadog_sync/utils/state.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/datadog_sync/utils/state.py b/datadog_sync/utils/state.py index baaa5350..e2ecfe8d 100644 --- a/datadog_sync/utils/state.py +++ b/datadog_sync/utils/state.py @@ -6,8 +6,6 @@ from datadog_sync.constants import ( Origin, - AWS_BUCKET_KEY_PREFIX_DESTINATION, - AWS_BUCKET_KEY_PREFIX_SOURCE, DESTINATION_PATH_DEFAULT, DESTINATION_PATH_PARAM, SOURCE_PATH_DEFAULT, @@ -21,16 +19,14 @@ class State: def __init__(self, type_: StorageType = StorageType.LOCAL_FILE, **kwargs: object) -> None: + source_resources_path = kwargs.get(SOURCE_PATH_PARAM, SOURCE_PATH_DEFAULT) + destination_resources_path = kwargs.get(DESTINATION_PATH_PARAM, DESTINATION_PATH_DEFAULT) if type_ == StorageType.LOCAL_FILE: - source_resources_path = kwargs.get(SOURCE_PATH_PARAM, SOURCE_PATH_DEFAULT) - destination_resources_path = kwargs.get(DESTINATION_PATH_PARAM, DESTINATION_PATH_DEFAULT) self._storage: BaseStorage = LocalFile( source_resources_path=source_resources_path, destination_resources_path=destination_resources_path, ) elif type_ == StorageType.AWS_S3_BUCKET: - source_resources_path = kwargs.get(AWS_BUCKET_KEY_PREFIX_SOURCE, SOURCE_PATH_DEFAULT) - destination_resources_path = kwargs.get(AWS_BUCKET_KEY_PREFIX_DESTINATION, DESTINATION_PATH_DEFAULT) config = kwargs.get("config", {}) if not config: raise ValueError("AWS configuration not found")