def createDAM(damType, settings=None): ''' create DAM ''' if 'yahoo' == damType: from ultrafinance.dam.yahooDAM import YahooDAM dam = YahooDAM() elif 'google' == damType: from ultrafinance.dam.googleDAM import GoogleDAM dam = GoogleDAM() elif 'excel' == damType: from ultrafinance.dam.excelDAM import ExcelDAM dam = ExcelDAM() elif 'hbase' == damType: from ultrafinance.dam.hbaseDAM import HBaseDAM dam = HBaseDAM() elif 'sql' == damType: from ultrafinance.dam.sqlDAM import SqlDAM dam = SqlDAM() else: raise UfException(Errors.INVALID_DAM_TYPE, "DAM type is invalid %s" % damType) dam.setup(settings) return dam
def testReadQuotes(self): dam = YahooDAM() dam.setSymbol('^STI') data = dam.readQuotes('20110101', '20110110') print(data) self.assertNotEquals(len(data), 0)