def testReadExcel(self): self.testWriteExcel() readDam = ExcelDAM() readDam.setDir(self.targetPath) readDam.setSymbol(self.symbol) print(readDam.readQuotes(0, 10000000000)) print(readDam.readTicks(1320676201, 1320676201))
def testWriteExcel(self): writeDam = ExcelDAM() writeDam.setDir(self.targetPath) writeDam.setSymbol(self.symbol) for f in [ writeDam.targetPath(ExcelDAM.QUOTE), writeDam.targetPath(ExcelDAM.TICK) ]: if os.path.exists(f): os.remove(f) quote1 = Quote('1320676200', '32.58', '32.58', '32.57', '32.57', '65212', None) quote2 = Quote('1320676201', '32.59', '32.59', '32.58', '32.58', '65213', None) tick1 = Tick('1320676200', '32.58', '32.58', '32.57', '32.57', '65212') tick2 = Tick('1320676201', '32.59', '32.59', '32.58', '32.58', '65213') writeDam.writeQuotes([quote1, quote2]) writeDam.writeTicks([tick1, tick2])
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 testReadExcel(self): self.testWriteExcel() readDam = ExcelDAM() readDam.setDir(self.targetPath) readDam.setSymbol(self.symbol)