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 fix: adding types to read function

This commit is contained in:
Steve Nyemba 2021-06-15 12:30:48 -05:00
parent 91434ec32d
commit b5eb5e0991
2 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
setup.py
View File

@ -8,7 +8,7 @@ def read(fname):
return open(os.path.join(os.path.dirname(__file__), fname)).read()
args = {
"name":"data-transport",
"version":"1.3.8.3",
"version":"1.3.8.4",
"author":"The Phi Technology LLC","author_email":"info@the-phi.com",
"license":"MIT",
"packages":["transport"]}

5
transport/sql.py
View File

@ -228,10 +228,11 @@ class BQReader(BigQuery,Reader) :
SQL = SQL.replace(':dataset',self.dataset).replace(':DATASET',self.dataset)
_info = {'credentials':self.credentials,'dialect':'standard'}
if 'dtypes' in _args or self.dtypes :
self.dtypes = _args ['dtypes'] if 'dtypes' in self.dtypes else None
if not self.dtypes :
self.dtypes = _args ['dtypes'] if 'dtypes' in _args else None
if self.dtypes :
_info['dtypes'] = self.dtypes
return pd.read_gbq(SQL,*_info) if SQL else None
return pd.read_gbq(SQL,**_info) if SQL else None
# return pd.read_gbq(SQL,credentials=self.credentials,dialect='standard') if SQL else None
class BQWriter(BigQuery,Writer):
lock = Lock()