You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
35 lines
1.1 KiB
Python
35 lines
1.1 KiB
Python
import sqlalchemy
|
|
import pandas as pd
|
|
from transport.sql.common import Base, BaseReader, BaseWriter
|
|
from multiprocessing import RLock
|
|
def template():
|
|
return {'database':'path-to-database','table':'table'}
|
|
class SQLite3 :
|
|
lock = RLock()
|
|
def __init__(self,**_args):
|
|
super().__init__(**_args)
|
|
if 'path' in _args :
|
|
self._database = _args['path']
|
|
if 'database' in _args :
|
|
self._database = _args['database']
|
|
def _get_uri(self,**_args):
|
|
path = self._database
|
|
return f'sqlite:///{path}' # ensure this is the correct path for the sqlite file.
|
|
|
|
class Reader(SQLite3,BaseReader):
|
|
def __init__(self,**_args):
|
|
super().__init__(**_args)
|
|
# def read(self,**_args):
|
|
# sql = _args['sql']
|
|
# return pd.read_sql(sql,self._engine)
|
|
|
|
|
|
class Writer (SQLite3,BaseWriter):
|
|
def __init__(self,**_args):
|
|
super().__init__(**_args)
|
|
def write(self,_data,**_kwargs):
|
|
try:
|
|
SQLite3.lock.acquire()
|
|
super().write(_data,**_kwargs)
|
|
finally:
|
|
SQLite3.lock.release() |