wuhao/data_cleaning
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

update

Browse Source
This commit is contained in:
kf_wuhao 2021-01-14 10:09:22 +08:00
parent 048f9958dc
commit b814161c5f
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
task/summary3.py
View File

@ -207,9 +207,14 @@ class Summary3(Task):
for doc in cursor:
doc_date = doc['cdate']
age_day = (cdate - doc_date) // 86400 + 1
data = {f'{cat}_all_money_{age_day}': doc[f'{cat}_all_money_{age_day - 1}']}
self.local_db[f'summary_{cat}'].update_one({'_id': doc['_id']}, {'$set': data})
self.remote_db[f'summary_{cat}'].update_one({'_id': doc['_id']}, {'$set': data})
try:
age_idx = RETAIN_DAYS.index(age_day)
except ValueError:
continue
else:
data = {f'{cat}_all_money_{age_day}': doc[f'{cat}_all_money_{RETAIN_DAYS[age_idx - 1]}']}
self.local_db[f'summary_{cat}'].update_one({'_id': doc['_id']}, {'$set': data})
self.remote_db[f'summary_{cat}'].update_one({'_id': doc['_id']}, {'$set': data})
update('role')
update('account')