All the db credentials are loaded by env variables.

madgik · Sep 19, 2023 · 869ed56 · 869ed56
1 parent 67db177
commit 869ed56
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 5 deletions.
diff --git a/mipdb/commands.py b/mipdb/commands.py
@@ -39,8 +39,8 @@ def __init__(self, *args, **kwargs):
         option_to_env_var = {
             "--ip": os.getenv("DB_IP"),
             "--port": os.getenv("DB_PORT"),
-            "--username": os.getenv("DB_USERNAME"),
-            "--password": os.getenv("DB_PASSWORD"),
+            "--username": os.getenv("MONETDB_ADMIN_USERNAME"),
+            "--password": os.getenv("MONETDB_LOCAL_PASSWORD"),
             "--db_name": os.getenv("DB_NAME"),
         }
         option = args[0][0]

diff --git a/mipdb/tables.py b/mipdb/tables.py
@@ -1,3 +1,4 @@
+import os
 from abc import ABC, abstractmethod
 import json
 from enum import Enum
@@ -18,9 +19,21 @@
 
 
 class User(Enum):
-    executor = "executor"
-    admin = "admin"
-    guest = "guest"
+    executor = (
+        "executor"
+        if not os.environ.get("MONETDB_LOCAL_USERNAME")
+        else os.environ.get("MY_ENV_VARIABLE")
+    )
+    admin = (
+        "admin"
+        if not os.environ.get("MONETDB_ADMIN_USERNAME")
+        else os.environ.get("MY_ENV_VARIABLE")
+    )
+    guest = (
+        "guest"
+        if not os.environ.get("MONETDB_PUBLIC_USERNAME")
+        else os.environ.get("MY_ENV_VARIABLE")
+    )
 
 
 @compiles(sql.types.JSON, "monetdb")