diff --git a/task/summary3.py b/task/summary3.py index 3ea182f..5e30711 100644 --- a/task/summary3.py +++ b/task/summary3.py @@ -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')