bug fixes with sqlite and provider
This commit is contained in:
parent
7d29a69a23
commit
9da2894b07
|
@ -209,16 +209,19 @@ class SQLiteWriter(SQLite,DiskWriter) :
|
||||||
"""
|
"""
|
||||||
"""
|
"""
|
||||||
|
|
||||||
if not self.fields :
|
#if not self.fields :
|
||||||
if type(info) == pd.DataFrame :
|
# #if type(info) == pd.DataFrame :
|
||||||
_columns = list(info.columns)
|
# # _columns = list(info.columns)
|
||||||
self.init(list(info.keys()))
|
# #self.init(list(info.keys()))
|
||||||
|
|
||||||
if type(info) == dict :
|
if type(info) == dict :
|
||||||
info = [info]
|
info = [info]
|
||||||
elif type(info) == pd.DataFrame :
|
elif type(info) == pd.DataFrame :
|
||||||
info = info.fillna('')
|
info = info.fillna('')
|
||||||
info = info.to_dict(orient='records')
|
info = info.to_dict(orient='records')
|
||||||
|
if not self.fields :
|
||||||
|
_rec = info[0]
|
||||||
|
self.init(list(_rec.keys()))
|
||||||
|
|
||||||
SQLiteWriter.LOCK.acquire()
|
SQLiteWriter.LOCK.acquire()
|
||||||
try:
|
try:
|
||||||
|
|
|
@ -72,6 +72,7 @@ WRITE = {'sql':sql.SQLWriter,'nosql':{MONGODB:mongo.MongoWriter,COUCHDB:couch.Co
|
||||||
PROVIDERS = {
|
PROVIDERS = {
|
||||||
FILE:{'read':disk.DiskReader,'write':disk.DiskWriter},
|
FILE:{'read':disk.DiskReader,'write':disk.DiskWriter},
|
||||||
SQLITE:{'read':disk.SQLiteReader,'write':disk.SQLiteWriter,'driver':sqlite3},
|
SQLITE:{'read':disk.SQLiteReader,'write':disk.SQLiteWriter,'driver':sqlite3},
|
||||||
|
'sqlite3':{'read':disk.SQLiteReader,'write':disk.SQLiteWriter,'driver':sqlite3},
|
||||||
|
|
||||||
POSTGRESQL:{'read':sql.SQLReader,'write':sql.SQLWriter,'driver':pg,'default':{'host':'localhost','port':5432}},
|
POSTGRESQL:{'read':sql.SQLReader,'write':sql.SQLWriter,'driver':pg,'default':{'host':'localhost','port':5432}},
|
||||||
NETEZZA:{'read':sql.SQLReader,'write':sql.SQLWriter,'driver':nz,'default':{'port':5480}},
|
NETEZZA:{'read':sql.SQLReader,'write':sql.SQLWriter,'driver':nz,'default':{'port':5480}},
|
||||||
|
|
Loading…
Reference in New Issue