wishpal_ironfan
/
scripts
5
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix post log

This commit is contained in:
lwl0608 2024-07-01 15:18:00 +08:00
parent 0652d8f125
commit d5b237ebc7
1 changed files with 31 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
cronjob/vas/user_income_calc.py
View File

@ -1,4 +1,5 @@
from lib.all import *
from pymongo import UpdateOne
class S:
@ -32,12 +33,32 @@ group by t1.mid,t1.ymd
docs = self.mysql_db_vas.query(sql)
print("week, len(docs): {}".format(len(docs)))
mid_map = dict()
mid_docs_map = dict()
for doc in docs:
mid = safe_get_int(doc, "mid")
mid_map[mid] = 1
mids = list(mid_map.keys())
print("week, len(mids): {}".format(len(mids)))
if mid not in mid_docs_map.keys():
mid_docs_map[mid] = list()
mid_docs_map[mid].append(doc)
print("week, len(mid_docs_map): {}".format(len(mid_docs_map)))
reqs = list()
for mid, docs in mid_docs_map.items():
q = {"_id": mid}
lis = list()
for d in docs:
lis.append({
"date": safe_get_str(d, "ymd"),
"income": safe_get_int(d, "income")
})
up = {
"$set": {
"ut": int(time.time()),
"week_dashboard": lis
}
}
reqs.append(UpdateOne(q, up, upsert=True))
ret = self.col_user_income.bulk_write(reqs)
print("week, bulk_ret, match: {}, modify: {}".format(ret.matched_count, ret.modified_count, ret.inserted_count, ret.upserted_count))
# 获取收益来源
def get_income_from_dashboard(self):
@ -65,16 +86,17 @@ group by t1.mid,t1.type_id
docs = self.mysql_db_vas.query(sql)
print("from, len(docs): {}".format(len(docs)))
mid_map = dict()
mid_docs_map = dict()
for doc in docs:
mid = safe_get_int(doc, "mid")
mid_map[mid] = 1
mids = list(mid_map.keys())
print("from, len(mids): {}".format(len(mids)))
if mid not in mid_docs_map.keys():
mid_docs_map[mid] = list()
mid_docs_map[mid].append(doc)
print("from, len(mid_docs_map): {}".format(len(mid_docs_map)))
def proc(self):
self.get_week_dashboard()
self.get_income_from_dashboard()
# self.get_income_from_dashboard()
s = S()