def readinput(directory, stockdir): assert (directory != None) filename = os.path.join(directory, "predicted.csv") if not os.path.exists(filename): filename = os.path.join(directory, "predicted.cv") if not os.path.exists(filename): print "can not find file name : %s" % (filename) postfile = open(os.path.join(directory, "predicted_post.csv"), "w") for line in open(filename, "r"): terms = line.rstrip().split(",") if not len(terms) == 4: print "terms is not 4 splited:%s" % line sys.exit(1) sym = terms[0] strdate1 = terms[1].strip() assert (len(strdate1) == 10) span = int(terms[2]) predicted = terms[3] stock_prices1 = base.get_stock_data_one_day(sym, strdate1, stockdir) stock_prices2 = base.get_stock_data_span_day(sym, strdate1, span, stockdir) assert ("close" in stock_prices1) assert ("close" in stock_prices2) print >> postfile, "%s,%s,%d,%s,%s,%s" % ( sym, strdate1, span, stock_prices2["date"], predicted, stock_prices2["close"] * 1.0 / stock_prices1["close"] * 10000) postfile.close()
def readinput(directory, stockdir): assert(directory != None) filename = os.path.join(directory, "predicted.csv") if not os.path.exists(filename): filename = os.path.join(directory, "predicted.cv") if not os.path.exists(filename): print "can not find file name : %s" % (filename) postfile = open(os.path.join(directory, "predicted_post.csv"), "w") for line in open(filename, "r"): terms = line.rstrip().split(",") if not len(terms) == 4: print "terms is not 4 splited:%s" % line sys.exit(1) sym = terms[0] strdate1 = terms[1].strip(); assert(len(strdate1) == 10) span = int(terms[2]) predicted = terms[3] stock_prices1 = base.get_stock_data_one_day(sym, strdate1, stockdir) stock_prices2 = base.get_stock_data_span_day(sym, strdate1, span, stockdir) assert("close" in stock_prices1) assert("close" in stock_prices2) print >> postfile, "%s,%s,%d,%s,%s,%s" % (sym, strdate1, span, stock_prices2["date"], predicted, stock_prices2["close"] * 1.0 /stock_prices1["close"] * 10000) postfile.close()
def test_get_stock_data_one_day(self): stock_prices = base.get_stock_data_one_day('YHOO', '2013-11-01') print stock_prices