crowetic/mev-inspect-py
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Move ENV names to variables. Make region and keys optional

Browse Source
This commit is contained in:
Luke Van Seters 2022-02-11 11:19:59 -05:00
parent 4dbe6ed2d7
commit 88b5e0ce2a
1 changed files with 12 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
mev_inspect/s3_export.py
View File

@ -8,6 +8,11 @@ import boto3
from mev_inspect.text_io import BytesIteratorIO from mev_inspect.text_io import BytesIteratorIO
AWS_ENDPOINT_URL_ENV = "AWS_ENDPOINT_URL" AWS_ENDPOINT_URL_ENV = "AWS_ENDPOINT_URL"
EXPORT_BUCKET_NAME_ENV = "EXPORT_BUCKET_NAME"
EXPORT_BUCKET_REGION_ENV = "EXPORT_BUCKET_REGION"
EXPORT_AWS_ACCESS_KEY_ID_ENV = "EXPORT_AWS_ACCESS_KEY_ID"
EXPORT_AWS_SECRET_ACCESS_KEY_ENV = "EXPORT_AWS_SECRET_ACCESS_KEY"
MEV_SUMMARY_EXPORT_QUERY = """ MEV_SUMMARY_EXPORT_QUERY = """
SELECT to_json(mev_summary) SELECT to_json(mev_summary)
FROM mev_summary FROM mev_summary
@ -64,16 +69,16 @@ def get_endpoint_url() -> Optional[str]:
def get_export_bucket_name() -> str: def get_export_bucket_name() -> str:
return os.environ["EXPORT_BUCKET_NAME"] return os.environ[EXPORT_BUCKET_NAME_ENV]
def get_export_bucket_region() -> str: def get_export_bucket_region() -> Optional[str]:
return os.environ["EXPORT_BUCKET_REGION"] return os.environ.get(EXPORT_BUCKET_REGION_ENV)
def get_export_aws_access_key_id() -> str: def get_export_aws_access_key_id() -> Optional[str]:
return os.environ["EXPORT_AWS_ACCESS_KEY_ID"] return os.environ.get(EXPORT_AWS_ACCESS_KEY_ID_ENV)
def get_export_aws_secret_access_key() -> str: def get_export_aws_secret_access_key() -> Optional[str]:
return os.environ["EXPORT_AWS_SECRET_ACCESS_KEY"] return os.environ.get(EXPORT_AWS_SECRET_ACCESS_KEY_ENV)