environmental table dev

helpers/dynamodb_helper.py

@@ -19,7 +19,9 @@ AWS_ACCESS_KEY = os.getenv("AWS_ACCESS_KEY", None)
 AWS_SECRET_ACCESS_KEY = os.getenv("AWS_SECRET_ACCESS_KEY", None)
 DYNAMODB_ENDPOINT = os.getenv("DYNAMODB_ENDPOINT", None)
 DDB_TABLE = os.getenv("DDB_TABLE", "chatbot-conversations")
-ENVIRONMENT_IMPACT_TABLE = "environmental-impact"
+DDB_ENVIRONMENT_IMPACT_TABLE = os.getenv(
+    "DDB_ENVIRONMENT_IMPACT_TABLE", "environmental-impact"
+)
 USE_LOCAL_DDB = os.getenv("USE_LOCAL_DDB", "false").lower() == "true"
 
 logger = logging.getLogger("uvicorn")
@@ -113,7 +115,7 @@ def create_environmental_table_if_not_exists(dynamodb):
     global environment_table
     try:
         environment_table = dynamodb.create_table(
-            TableName=ENVIRONMENT_IMPACT_TABLE,
+            TableName=DDB_ENVIRONMENT_IMPACT_TABLE,
             # Schema for Single Table Design
             KeySchema=[
                 {
@@ -134,7 +136,7 @@ def create_environmental_table_if_not_exists(dynamodb):
         )
 
         # Wait for the table to be created before moving on
-        logger.info(f"Creating table {ENVIRONMENT_IMPACT_TABLE}...")
+        logger.info(f"Creating table {DDB_ENVIRONMENT_IMPACT_TABLE}...")
         environment_table.wait_until_exists()
         logger.info("Table is now ACTIVE.")
         return environment_table
@@ -142,9 +144,9 @@ def create_environmental_table_if_not_exists(dynamodb):
     except ClientError as e:
         if e.response["Error"]["Code"] == "ResourceInUseException":
             logger.info(
-                f"Table {ENVIRONMENT_IMPACT_TABLE} already exists. Skipping creation."
+                f"Table {DDB_ENVIRONMENT_IMPACT_TABLE} already exists. Skipping creation."
             )
-            return dynamodb.Table(ENVIRONMENT_IMPACT_TABLE)
+            return dynamodb.Table(DDB_ENVIRONMENT_IMPACT_TABLE)
         else:
             raise e