[Fixed] DatahubHepler params

WISE-AIFS · May 3, 2021 · 749fcfd · 749fcfd
1 parent 80fe22f
commit 749fcfd
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/afs2datasource/DBManager.py b/afs2datasource/DBManager.py
@@ -101,6 +101,22 @@ def _get_credential_from_config(self, config):
           'containers': containers
         }
       }
+    elif db_type == const.DB_TYPE['DATAHUB']:
+      dataDir = {
+        'type': db_type,
+        'data': {
+          'timeRange': config.get('timeRange'),
+          'timeLast': config.get('timeLast'),
+          'datahub_config': config.get('datahub_config'),
+          'datahub_url': config.get('datahub_url'),
+          'credential': {
+            'uri': config.get('mongouri'),
+            'influxuri': config.get('influxuri'),
+          },
+          'username': config.get('username'),
+          'password': config.get('password'),
+        }
+      }
     else:
       username = config.get('username', None)
       password = config.get('password', None)