From 2f07a1347635e3426136917a24b4c03831080574 Mon Sep 17 00:00:00 2001
From: Summer Mousa <3171936+smousa@users.noreply.github.com>
Date: Sat, 16 Mar 2024 22:26:26 -0500
Subject: [PATCH] add ENDPOINT_URL environment option

---
 parquet_tools/commands/csv.py     | 3 ++-
 parquet_tools/commands/inspect.py | 2 ++
 parquet_tools/commands/show.py    | 3 ++-
 3 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/parquet_tools/commands/csv.py b/parquet_tools/commands/csv.py
index eb46af4..d2e7cf0 100644
--- a/parquet_tools/commands/csv.py
+++ b/parquet_tools/commands/csv.py
@@ -1,3 +1,4 @@
+import os
 import sys
 from argparse import ArgumentParser, Namespace
 from typing import List, Optional
@@ -34,7 +35,7 @@ def configure_parser(paser: ArgumentParser) -> ArgumentParser:
     paser.add_argument('--endpoint-url',
                        type=str,
                        required=False,
-                       default=None,
+                       default=os.environ.get('ENDPOINT_URL'),
                        help='A custom S3 endpoint URL')
     paser.set_defaults(handler=_cli)
     return paser
diff --git a/parquet_tools/commands/inspect.py b/parquet_tools/commands/inspect.py
index 5b0da65..d776079 100644
--- a/parquet_tools/commands/inspect.py
+++ b/parquet_tools/commands/inspect.py
@@ -1,3 +1,4 @@
+import os
 import sys
 from argparse import ArgumentParser, Namespace
 from typing import List
@@ -33,6 +34,7 @@ def configure_parser(paser: ArgumentParser) -> ArgumentParser:
                        type=str,
                        required=False,
                        default=None,
+                       default=os.environ.get('ENDPOINT_URL'),
                        help='A custom S3 endpoint URL')
 
     paser.set_defaults(handler=_cli)
diff --git a/parquet_tools/commands/show.py b/parquet_tools/commands/show.py
index 9ab01a2..965ddb1 100644
--- a/parquet_tools/commands/show.py
+++ b/parquet_tools/commands/show.py
@@ -1,3 +1,4 @@
+import os
 import sys
 from argparse import ArgumentParser, Namespace
 from logging import getLogger
@@ -49,7 +50,7 @@ def configure_parser(paser: ArgumentParser) -> ArgumentParser:
     paser.add_argument('--endpoint-url',
                        type=str,
                        required=False,
-                       default=None,
+                       default=os.environ.get('ENDPOINT_URL'),
                        help='A custom S3 endpoint URL')
 
     paser.set_defaults(handler=_cli)