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)
