Example #1
0
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()
Example #2
0
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()
Example #3
0
 def test_get_stock_data_one_day(self):
     stock_prices = base.get_stock_data_one_day('YHOO', '2013-11-01')
     print stock_prices
Example #4
0
 def test_get_stock_data_one_day(self):
     stock_prices = base.get_stock_data_one_day('YHOO', '2013-11-01')
     print stock_prices