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)
