Use SUPERSET_DBUSER and SUPERSET_DBPASS to access superset-db

Author: savvan0h

docker-compose.yml

@@ -104,6 +104,8 @@ services:
     #  - ADMIN_LASTNAME=user
     #  - ADMIN_EMAIL=admin@fab.org
     #  - ADMIN_PASSWORD=admin
+      - SUPERSET_DBUSER=superset
+      - SUPERSET_DBPASS=superset
   superset-db:
     build: superset-db
     container_name: superset-db

superset/create-data-sources.py

@@ -25,12 +25,17 @@ def main():
         headers=headers
     )
     headers['X-CSRFToken'] = r.json()['result']
+
+    db_user = os.getenv('SUPERSET_DBUSER', 'postgres')
+    db_pass = os.getenv('SUPERSET_DBPASS', '')
+    base_uri = f'postgresql://{db_user}:{db_pass}@superset-db:5432'
+
     r = s.post(
         'http://localhost:8088/api/v1/database',
         headers=headers,
         json={
            'database_name': 'Learning Locker',
-           'sqlalchemy_uri': 'postgresql://postgres@superset-db:5432/learninglocker'
+           'sqlalchemy_uri': f'{base_uri}/learninglocker'
         }
     )
     print(r.json())
@@ -40,7 +45,7 @@ def main():
         headers=headers,
         json={
            'database_name': 'OpenLRW',
-           'sqlalchemy_uri': 'postgresql://postgres@superset-db:5432/openlrw'
+           'sqlalchemy_uri': f'{base_uri}/openlrw'
         }
     )
     print(r.json())
@@ -50,7 +55,7 @@ def main():
         headers=headers,
         json={
            'database_name': 'Jupyter',
-           'sqlalchemy_uri': 'postgresql://postgres@superset-db:5432/jupyter'
+           'sqlalchemy_uri': f'{base_uri}/jupyter'
         }
     )
     print(r.json())

superset/superset_config.py

@@ -20,7 +20,13 @@
 # superset metadata (slices, connections, tables, dashboards, ...).
 # Note that the connection information to connect to the datasources
 # you want to explore are managed directly in the web UI
-SQLALCHEMY_DATABASE_URI = 'postgresql+psycopg2://postgres@superset-db:5432/superset'
+SUPERSET_DBUSER = os.getenv('SUPERSET_DBUSER', 'postgres')
+SUPERSET_DBPASS = os.getenv('SUPERSET_DBPASS', '')
+SQLALCHEMY_DATABASE_URI = (
+    'postgresql+psycopg2://'
+    f'{SUPERSET_DBUSER}:{SUPERSET_DBPASS}'
+    '@superset-db:5432/superset'
+)
 
 # Flask-WTF flag for CSRF
 WTF_CSRF_ENABLED = True