def __init__(self, param):
self.logger = logging.getLogger(__name__)
# memento for state machine
self.memento = Memento(param['memento'])
self.memento.load()
# prepare database
self.spider = param['spider']
self.assembler = param['assembler']
self.feed_builder = param['feed_builder']
self.database = Database()
# all states are listed here
self.state_map = {
'InitialState': InitialState(self),
'LoadState': LoadState(self),
'SpiderState': SpiderState(self),
'AssemblerState': AssemblerState(self),
'DatabaseState': DatabaseState(self),
'FinalState': FinalState(self),
}
# set transition table
self.transition_table = param['transition_table']
# set current state
self.curr_state = 'InitialState'
def __init_value(self, param):
output = {}
database = Database()
for account in param['account_list']:
output[account] = TimeSeries.create(database.get(account, param))
return output
def __init__(self):
self.database = Database()
self.market_category = {
'\xe4\xb8\x8a\xe5\xb8\x82' : 'TW',
'\xe4\xb8\x8a\xe6\xab\x83' : 'TWO',
}
def setUp(self):
self.database = Database()
