def write_trade_data(self): stockbase_data=['ticker','secShortName','industryID1', 'industryID2','industryID3'] #在tradedata中的字段 stocktrade_data=['date','close'] db=DatabaseInterface() tickers=db.get_db_tickers() for t in tickers: db.getone_db(self.db_name,'stockbase',{'ticker':t}, stockbase_data)
def get_step(self,key): db=DatabaseInterface() col_name='currentinfo' clause={} field=[key] res=db.getone_db(self.db_name,col_name,clause,field) return res[key]
def get_db_stocktradeadj(self,step,beginDate,endDate=''): #如果没有输入截止日期,就取得今天的日期 if not endDate: endDate=self.today_as_str() #获取tickers列表 db=DatabaseInterface() tickers=db.get_db_tickers() #获取上次中断步数 tickers=tickers[step+1:] print 'started from ticker: '+tickers[0] col_name='stockbase' for t in tickers: ticker_trade={} ticker_trade['ticker']=t ticker_trade['tradedata']=self.stkif._getTradeDataAdj(t,beginDate=beginDate,endDate=endDate) ind=db.getone_db(self.db_name,col_name,{"ticker":t},['industryID3']) ticker_trade['industryID3']=ind['industryID3'] print 'get ticker:'+t+'.....' yield ticker_trade