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 update_stockbase_Equ(self,fields): col_name='stockbase' stkif=StockInterface(self.token) db=DatabaseInterface() tickers=db.get_db_tickers() for t in tickers: #获取字典 Equ_dics=stkif._getEqu(fields,t)[0] db.update_db(self.db_name,col_name,{"ticker":t},{"$set":Equ_dics}) print 'update stockbase data with ticker.no'+t+'....'
def get_newticker(self): col_name='stockbase' key='ticker' db=DatabaseInterface() tickers_new=db.get_db_tickers() print 'getting new tickers with length:'+str(len(tickers_new))+'....' clause={} field=[key] stockbase_data=db.get_db(self.db_name,col_name,clause,field) tickers_old=self.unpack_dic(key,stockbase_data) print 'getting old tickers with length:'+str(len(tickers_old))+'....' tickers_addition=list(set(tickers_new)-set(tickers_old)) return tickers_addition
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
def get_api_stockbase_all(self,step): #获取当前全部股票代码 db=DatabaseInterface() tickers=db.get_db_tickers() tickers=tickers[step+1:] return self.get_api_stockbase(tickers)