bug fix with reader

This commit is contained in:
Steve Nyemba 2020-09-22 13:30:04 -05:00
parent 5676c4991f
commit cbf641a986
1 changed files with 18 additions and 0 deletions

View File

@ -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):