def createStateSaver(name, setting):
''' create state saver '''
if 'sql' == name:
from ultrafinance.backTest.stateSaver.sqlSaver import SqlSaver
saver = SqlSaver()
else:
raise UfException(Errors.INVALID_SAVER_NAME,
"Saver name is invalid %s" % name)
saver.setup(setting)
return saver
def createStateSaver(name, setting, tableName = None):
''' create state saver '''
if 'habse' == name:
from ultrafinance.backTest.stateSaver.hbaseSaver import HbaseSaver
saver = HbaseSaver()
elif 'sql' == name:
from ultrafinance.backTest.stateSaver.sqlSaver import SqlSaver
saver = SqlSaver()
else:
raise UfException(Errors.INVALID_SAVER_NAME,
"Saver name is invalid %s" % name)
if not tableName:
saver.tableName = 'output'
else:
saver.tableName = tableName
saver.setup(setting)
return saver
