|
|
|
@ -56,11 +56,13 @@ class Base:
|
|
|
|
|
# _map = {'BIGINT':'INTEGER','TEXT':'STRING','DOUBLE_PRECISION':'FLOAT','NUMERIC':'FLOAT','DECIMAL':'FLOAT','REAL':'FLOAT'}
|
|
|
|
|
# _schema = [{"name":_attr.name,"type":_map.get(str(_attr.type),str(_attr.type))} for _attr in _handler.tables[_table].columns]
|
|
|
|
|
#
|
|
|
|
|
|
|
|
|
|
_inspector = inspect(self._engine)
|
|
|
|
|
_columns = _inspector.get_columns(_table)
|
|
|
|
|
_schema = [{'name':column['name'],'type':_map.get(str(column['type']),str(column['type'])) } for column in _columns]
|
|
|
|
|
return _schema
|
|
|
|
|
if _table :
|
|
|
|
|
_inspector = inspect(self._engine)
|
|
|
|
|
_columns = _inspector.get_columns(_table)
|
|
|
|
|
_schema = [{'name':column['name'],'type':_map.get(str(column['type']),str(column['type'])) } for column in _columns]
|
|
|
|
|
return _schema
|
|
|
|
|
else:
|
|
|
|
|
return []
|
|
|
|
|
def has(self,**_args):
|
|
|
|
|
return self.meta(**_args)
|
|
|
|
|
def apply(self,sql):
|
|
|
|
@ -74,7 +76,7 @@ class Base:
|
|
|
|
|
return pd.read_sql(sql,self._engine)
|
|
|
|
|
else:
|
|
|
|
|
_handler = self._engine.connect()
|
|
|
|
|
_handler.execute(text(sql))
|
|
|
|
|
_handler.execute(text(sql.strip()))
|
|
|
|
|
_handler.commit ()
|
|
|
|
|
_handler.close()
|
|
|
|
|
return None
|
|
|
|
@ -136,7 +138,7 @@ class BaseWriter (SQLBase):
|
|
|
|
|
super().__init__(**_args)
|
|
|
|
|
def write(self,_data,**_args):
|
|
|
|
|
if type(_data) == dict :
|
|
|
|
|
_df = pd.DataFrame(_data)
|
|
|
|
|
_df = pd.DataFrame([_data])
|
|
|
|
|
elif type(_data) == list :
|
|
|
|
|
_df = pd.DataFrame(_data)
|
|
|
|
|
else:
|
|
|
|
|