bug fix with reader
This commit is contained in:
parent
5676c4991f
commit
cbf641a986
|
@ -105,6 +105,24 @@ class DiskWriter(Writer):
|
|||
pass
|
||||
finally:
|
||||
DiskWriter.THREAD_LOCK.release()
|
||||
class SQLiteReader (DiskReader):
|
||||
def __init__(self,**args)
|
||||
DiskReader.__init__(self,**args)
|
||||
self.conn = sqlite3.connect(self.path,isolation_level=None)
|
||||
self.conn.row_factory = sqlite3.Row
|
||||
self.table = args['table']
|
||||
def read(self,**args):
|
||||
if 'sql' in args :
|
||||
sql = args['sql']
|
||||
else if 'filter' in args :
|
||||
sql = "SELECT :fields FROM ",self.table, "WHERE (:filter)".replace(":filter",args['filter'])
|
||||
sql = sql.replace(":fields",args['fields']) if 'fields' in args else sql.replace(":fields","*")
|
||||
return = pd.read_sql(sql,self.conn)
|
||||
def close(self):
|
||||
try:
|
||||
self.conn.close()
|
||||
except Exception as e :
|
||||
pass
|
||||
|
||||
class SQLiteWriter(DiskWriter) :
|
||||
def __init__(self,**args):
|
||||
|
|
Loading…
Reference in New Issue