def mytrade_digest(req): csrf_token = get_token(req) #holding mtu = MyTradeUtil() holding_stocks = MyTradeDigestStock.objects.filter(user__username=req.user,holding__gt=0) for s in holding_stocks: #s.stock.code = "%06d" % s.stock.code bret,close,profit = mtu.calc_single_stock_floatprofit(s,s.stock.code) if bret == False: close = '-' profit = '-' s.close = close s.market_price = s.close*s.holding s.float_profit = profit #history trade history_stocks = MyTradeDigestStock.objects.filter(user__username=req.user,holding=0) #for s in history_stocks: # s.stock.code = "%06d" % s.stock.code #error_code err_stocks = MyTradeDigestErrorStock.objects.filter(user__username=req.user) if len(err_stocks)>0: #print len(err_stocks) for s in err_stocks: s.code = "%06d" % s.code return render_to_response('mytrade/mytrade_digest.html',{'holding_stocks':holding_stocks,'history_stocks':history_stocks,'error_stocks':err_stocks,'csrf_token':csrf_token},context_instance=RequestContext(req)) else: return render_to_response('mytrade/mytrade_digest.html',{'holding_stocks':holding_stocks,'history_stocks':history_stocks,'csrf_token':csrf_token},context_instance=RequestContext(req))
def upload_complete(self, request, filename, *args, **kwargs): path = settings.MEDIA_URL + self.UPLOAD_DIR + "/" + filename self._dest.close() print "in mytrade upload backend upload successful :" + path # 文件上传成功之后,将交易记录导入到数据库中 mtu = MyTradeUtil() username = "******" csvfile = path # csvfile = 'e:/table0.csv' # stocks = [] # funds = [] bret, stocks, funds = mtu.import_csv(csvfile, username) if bret: print "import csv succeed" print stocks print funds if os.path.isfile(path): os.remove(path) for stock in stocks: print stock mtu.calc_stock_earning(username, stock) status, stockname = mtu.calc_stock_digest(username, stock) if status == False: print "calc_stock_digest failed %s" % stockname err_stocks = MyTradeDigestErrorStock.objects.filter(code=stock, user__username=username) print err_stocks if len(err_stocks) == 0: err_stock = MyTradeDigestErrorStock() else: err_stock = err_stocks[0] err_stock.code = stock err_stock.name = stockname err_stock.user = User.objects.get(username=username) err_stock.save() for fund in funds: if mtu.calc_fund_digest(username, fund) == False: print "calc_fund_digest failed" else: print "import csv failed" return {"path": path}