def saveDB(self, data: pd.DataFrame, key): def callback(result): self.send_message(self.project_name, result, key + '_' + result[MONGODB_ID]) util.saveMongoDB(data, util.genEmptyFunc(), DB_NAME, COLLECTION_HEAD + key, callback)
def saveDB(self, data: pd.DataFrame, key): tableName = "{}{}".format(COLLECTION_HEAD, key) print("tableName: {}".format(tableName)) def callback(result): self.send_message(self.project_name, result, key + '_' + result[MONGODB_ID]) util.saveMongoDB(data, util.genEmptyFunc(), DB_NAME, tableName, callback)
def run(): DB_NAME = 'ald' COLLECTION_NAME = 'toplist_dict' data = GetList() tmp = [] for item in data: item['_id'] = item['key'] # series = pd.Series(item) tmp.append(item) # df = pd.DataFrame(tmp) util.saveMongoDB(tmp, util.genEmptyFunc(), DB_NAME, COLLECTION_NAME, None)
def calcOne(code, saveDB=True, saveFile=False, benchmark=False): print('calcOne %s' % (code)) s = stock.Stock(code) s.load(cwsj=True, yjyg=setting.yjyg_list) if benchmark: s.loadBenchmark(file=setting.PATH + 'out-adjust-' + code + '.xlsx') df = s.data # df = df.loc[:, [KN['date'], KN['zgb'], KN['jbmgsy']]] bdf = s.benchmark_data oneLoop = loop.AdjustLoop() oneLoop.addOP(marketValue.GenMarketValue(s)) oneLoop.addOP(marketValue.GenZGB(s)) oneLoop.addOP(marketValue.GenIndustry(s)) oneLoop.addOP(marketValue.GenLastPrice(s)) oneLoop.addOP(marketValue.GenCodeAndName(s)) oneLoop.addOP(forecastProfit.GenForecastProfit(s)) oneLoop.addOP(quarterProfit.GenQuarterProfit(s)) oneLoop.addOP(quarterProfitRatio.GenQuarterProfitRatio(s)) oneLoop.addOP(forecastQuarterProfit.GenForecastProfit(s)) oneLoop.addOP(lastYearProfit.GenLastYearProfit(s)) oneLoop.addOP(forecastMidGrowthRate.GenForecastMidGrowthRate(s)) oneLoop.addOP(forecastFinalGrowthRate.GenForecastFinalGrowthRate(s)) oneLoop.addOP(peMinMax.GenPEMinMax(s)) oneLoop.addOP(forecastPerShareProfit.GenForecastPerShareProfit(s)) oneLoop.addOP(forecastPE.GenForecastPE(s)) oneLoop.addOP(valueMinMax.GenValueMinMax(s)) oneLoop.addOP(distanceMinMax.GenDistanceMinMax(s)) dfOut = None if benchmark: dfOut = oneLoop.verify(df, bdf) else: dfOut = oneLoop.loop(df) c = oneLoop.columns # c.extend([KEY_NAME['date'], KEY_NAME['jbmgsy'], ADJUST_NAME['zgb'], const.YJYG_KEYWORD.KEY_NAME['forecastl']]) if saveDB: dfOut = oneLoop.genResult(dfOut, [ KEY_NAME['jbmgsy'], ADJUST_NAME['zgb'], const.YJYG_KEYWORD.KEY_NAME['increasel'] ]) # util.saveMongoDB(dfOut, util.genKeyDateFunc(const.MONGODB_ID), 'stock-out', 'cwsj-' + code) if saveFile: dfOut.to_excel(setting.PATH + 'out-' + code + '.xls') return dfOut
def saveDB(self, data: pd.DataFrame, handler): def callback(result): handler.send_message(handler.project_name, result, self._date + '_' + result['_id']) re = util.saveMongoDB(data, util.genEmptyFunc(), DB_NAME, COLLECTION_HEAD + self._date, callback) util.everydayChange(re, 'gpfh')
def saveDB(self, data: pd.DataFrame, key): util.saveMongoDB(data, util.genKeyDateFunc(MONGODB_ID), DB_NAME, COLLECTION_HEAD + key, None)
def saveDB(self, data: pd.DataFrame, date): re = util.saveMongoDB(data, util.genEmptyFunc(), self.DB_NAME, self.COLLECTION_HEAD + date, None) util.everydayChange(re, 'gpfh')
def saveDB(self, data): # def callback(result): # self.send_message(self.project_name, result, key + '_' + result[MONGODB_ID]) util.saveMongoDB(data, util.genEmptyFunc(), self.DB_NAME, self.COLLECTION_NAME, None)