From c0b829dfb4831a37b6bf7f5c2efa39121e07b17d Mon Sep 17 00:00:00 2001 From: kf_wuhao <15392746632@qq.com> Date: Wed, 23 Dec 2020 13:35:42 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- task/summary_func.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/task/summary_func.py b/task/summary_func.py index 4b35204..d8d94b8 100644 --- a/task/summary_func.py +++ b/task/summary_func.py @@ -40,10 +40,11 @@ class SummaryFunc(Task): item['cdate'] = int(pd.Timestamp(item['_event_time'], unit='s', tz=self.timezone) \ .normalize().timestamp()) model = self.Model(**item) - data = model.dict() + data = model.dict(by_alias=True) bulk_data.append(UpdateOne({'_id': data['_id']}, {'$set': data}, upsert=True)) except Exception as e: logger.error(f'ftype {item["ftype"]} msg:{e}') # pass - self.remote_db[self.task_name].bulk_write(bulk_data, ordered=False) + if bulk_data: + self.remote_db[self.dest_coll].bulk_write(bulk_data, ordered=False) self.set_cursor(cursor_st=ts['cursor_st'], cursor_et=ts['cursor_et'])