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
