From b84c5434086c762a72cc50d3fd5c9fcfba72fd59 Mon Sep 17 00:00:00 2001 From: Luke Van Seters Date: Fri, 4 Feb 2022 16:32:42 -0500 Subject: [PATCH] Break env piece into a function --- mev_inspect/s3_export.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/mev_inspect/s3_export.py b/mev_inspect/s3_export.py index 272d6ad..1e979a1 100644 --- a/mev_inspect/s3_export.py +++ b/mev_inspect/s3_export.py @@ -22,7 +22,7 @@ logger = logging.getLogger(__name__) def export_block_range( inspect_db_session, after_block_number: int, before_block_number ) -> None: - export_bucket_name = os.environ["EXPORT_BUCKET_NAME"] + export_bucket_name = get_export_bucket_name() client = get_s3_client() mev_summary_json_results = inspect_db_session.execute( @@ -55,3 +55,7 @@ def get_s3_client(): def get_endpoint_url() -> Optional[str]: return os.environ.get(AWS_ENDPOINT_URL_ENV) + + +def get_export_bucket_name() -> str: + return os.environ["EXPORT_BUCKET_NAME"]