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'
class DatabaseTest(unittest.TestCase): def setUp(self): self.database = Database() def tearDown(self): self.database = None def test_get_stock_symbol_list(self): stock_symbol_list = self.database.get_stock_symbol_list() self.assertEqual(stock_symbol_list[0], ('1101', datetime.date(1962, 2, 9))) self.assertEqual(stock_symbol_list[1], ('1102', datetime.date(1962, 6, 8)))
class StockSymbolAnalyzer(): def __init__(self): self.database = Database() self.market_category = { '\xe4\xb8\x8a\xe5\xb8\x82' : 'TW', '\xe4\xb8\x8a\xe6\xab\x83' : 'TWO', } def get_stock_symbol_list(self): result = [] records = self.database.get('StockSymbolList', None) for record in records: entry = { 'stock_symbol' : record[0], 'listing_date' : record[1], 'market_category' : self.market_category[record[2]] } result.append(entry) return result
def __init__(self): self.database = Database() self.market_category = { '\xe4\xb8\x8a\xe5\xb8\x82' : 'TW', '\xe4\xb8\x8a\xe6\xab\x83' : 'TWO', }
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 setUp(self): self.database = Database()
def __init_value(self, param): output = {} database = Database() for account in param["account_list"]: output[account] = TimeSeries.create(database.get(account, param)) return output