hiplab/data-transport
hiplab
/
data-transport
mirror of https://github.com/lnyemba/data-transport
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

bug fixes

This commit is contained in:
Steve Nyemba 2022-05-16 11:54:28 -05:00
parent f64b945245
commit db1496dd38
2 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
transport/__init__.py
View File

@ -151,7 +151,7 @@ def instance(**_args):
try: try:
host = '' host = ''
if provider not in ['bigquery','mongodb','couchdb','sqlite','console','etl','file','rabbitmq'] : if provider not in ['bigquery','mongodb','mongo','couchdb','sqlite','console','etl','file','rabbitmq'] :
# #
# In these cases we are assuming RDBMS and thus would exclude NoSQL and BigQuery # In these cases we are assuming RDBMS and thus would exclude NoSQL and BigQuery
username = args['username'] if 'username' in args else '' username = args['username'] if 'username' in args else ''
@ -169,7 +169,7 @@ def instance(**_args):
account = '' account = ''
host = '' host = ''
database = args['path'] if 'path' in args else args['database'] database = args['path'] if 'path' in args else args['database']
if provider not in ['mongodb','couchdb','bigquery','console','etl','file','rabbitmq'] : if provider not in ['mongodb','mongo','couchdb','bigquery','console','etl','file','rabbitmq'] :
uri = ''.join([provider,"://",account,host,'/',database]) uri = ''.join([provider,"://",account,host,'/',database])
e = sqlalchemy.create_engine (uri,future=True) e = sqlalchemy.create_engine (uri,future=True)

5
transport/mongo.py
View File

@ -38,7 +38,7 @@ class Mongo :
host = ":".join([host,port]) #-- Formatting host information here host = ":".join([host,port]) #-- Formatting host information here
self.uid = args['doc'] if 'doc' in args else None #-- document identifier self.uid = args['doc'] if 'doc' in args else None #-- document identifier
self.dbname = args['dbname'] if 'dbname' in args else args['db'] self.dbname = args['dbname'] if 'dbname' in args else args['db']
authMechanism= 'SCRAM-SHA-256' if 'mechanism' not in args else args['mechanism']
self._lock = False if 'lock' not in args else args['lock'] self._lock = False if 'lock' not in args else args['lock']
username = password = None username = password = None
@ -49,6 +49,9 @@ class Mongo :
_info = json.loads((open(args['auth_file'])).read()) _info = json.loads((open(args['auth_file'])).read())
username = _info['username'] username = _info['username']
password = _info['password'] password = _info['password']
if 'mechanism' in _info:
authMechanism = _info['mechanism']
authSource=(args['authSource'] if 'authSource' in args else self.dbname) authSource=(args['authSource'] if 'authSource' in args else self.dbname)
if username and password : if username and password :