@@ -376,8 +376,8 @@ def set_connection(self):
376
376
377
377
conn_info = {
378
378
"driver" : conn_type ,
379
- "project" : credentials .get ("project" ),
380
- "dataset" : credentials .get ("dataset" ),
379
+ "project" : credentials .get ("project" ) or credentials . get ( "database" ) ,
380
+ "dataset" : credentials .get ("dataset" ) or credentials . get ( "schema" ) ,
381
381
}
382
382
383
383
self .threads = credentials .get ("threads" )
@@ -402,13 +402,13 @@ def set_connection(self):
402
402
"user" : credentials .get ("user" ),
403
403
"password" : credentials .get ("password" ) or credentials .get ("pass" ),
404
404
"port" : credentials .get ("port" ),
405
- "dbname" : credentials .get ("dbname" ),
405
+ "dbname" : credentials .get ("dbname" ) or credentials . get ( "database" ) ,
406
406
}
407
407
self .threads = credentials .get ("threads" )
408
408
elif conn_type == "databricks" :
409
409
conn_info = {
410
410
"driver" : conn_type ,
411
- "catalog" : credentials .get ("catalog" ),
411
+ "catalog" : credentials .get ("catalog" ) or credentials . get ( "database" ) ,
412
412
"server_hostname" : credentials .get ("host" ),
413
413
"http_path" : credentials .get ("http_path" ),
414
414
"schema" : credentials .get ("schema" ),
@@ -420,7 +420,7 @@ def set_connection(self):
420
420
"driver" : "postgresql" ,
421
421
"host" : credentials .get ("host" ),
422
422
"user" : credentials .get ("user" ),
423
- "password" : credentials .get ("password" ),
423
+ "password" : credentials .get ("password" ) or credentials . get ( "pass" ) ,
424
424
"port" : credentials .get ("port" ),
425
425
"dbname" : credentials .get ("dbname" ) or credentials .get ("database" ),
426
426
}
0 commit comments