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 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
def setUp(self):
self.database = Database()