def testCheckIfCannotBuy(self): dailyQuote = SourceDataDao.load_new_daily_quote('') #双休日 checkResult = SourceDataDao.check_if_cannot_buy(dailyQuote, '2004-01-31', '000049.SZ') print('checkResult:'+str(checkResult)) #print(signalData) self.assertTrue(checkResult) #双休日 checkResult = SourceDataDao.check_if_cannot_buy(dailyQuote, '2004-02-01', '000049.SZ') print('checkResult:' + str(checkResult)) # print(signalData) self.assertTrue(checkResult) #一字涨停 checkResult = SourceDataDao.check_if_cannot_buy(dailyQuote, '2004-02-02', '000049.SZ') print('checkResult:' + str(checkResult)) # print(signalData) self.assertTrue(checkResult) #正常 checkResult = SourceDataDao.check_if_cannot_buy(dailyQuote, '2004-02-05', '000049.SZ') print('checkResult:' + str(checkResult)) # print(signalData) self.assertFalse(checkResult)
def getActualBuyList(planBuyList,dailyQuote,tradingDate,cannotBuyList): if len(planBuyList) == 0: return planBuyList actualBuyList = [] for innerCode in planBuyList: isCannotBuy = SourceDataDao.check_if_cannot_buy(dailyQuote, tradingDate, innerCode) #buyFlg = dailyQuoteRow[StockConst.buyFlg] #可买 if not isCannotBuy: actualBuyList.append(innerCode) else: cannotBuyList.append(innerCode) return actualBuyList