diff --git a/photo_review/photo_review.py b/photo_review/photo_review.py index 4e9a1dd..1fc9998 100644 --- a/photo_review/photo_review.py +++ b/photo_review/photo_review.py @@ -1,6 +1,6 @@ import json import logging -from time import sleep, time +from time import sleep from paddlenlp import Taskflow from sqlalchemy import update @@ -19,6 +19,7 @@ from photo_review.entity.zx_phhd import ZxPhhd from photo_review.entity.zx_phrec import ZxPhrec from photo_review.util.data_util import handle_date, handle_decimal, handle_department, handle_name from photo_review.util.ucloud import get_private_url +from photo_review.util.util import get_default_datetime # 关键信息提取 @@ -35,9 +36,10 @@ def information_extraction(schema, phrecs, task_path): if len(result_json) > 5000: result_json = result_json[:5000] # 提取完保存每张图片的结果 + now = get_default_datetime() session = MysqlSession() zx_ocr = ZxOcr(pk_phhd=phrec.pk_phhd, pk_phrec=phrec.pk_phrec, cfjaddress=phrec.cfjaddress, - content=result_json, create_time=time(), update_time=time()) + content=result_json, create_time=now, update_time=now) session.add(zx_ocr) session.commit() session.close() @@ -80,15 +82,16 @@ def save_or_update_ie(table, pk_phhd, data): obj = table(**data) session = MysqlSession() db_data = session.query(table).filter_by(pk_phhd=pk_phhd).one_or_none() + now = get_default_datetime() if db_data: # 更新 - db_data.update_time = time() + db_data.update_time = now for k, v in data.items(): setattr(db_data, k, v) else: # 新增 - obj.create_time = time() - obj.update_time = time() + obj.create_time = now + obj.update_time = now session.add(obj) session.commit() session.close() diff --git a/photo_review/util/util.py b/photo_review/util/util.py new file mode 100644 index 0000000..6d1c2b2 --- /dev/null +++ b/photo_review/util/util.py @@ -0,0 +1,6 @@ +from datetime import datetime + + +# 获取yyyy-MM-dd HH:mm:ss格式的当前时间 +def get_default_datetime(): + return datetime.now().strftime('%Y-%m-%d %H:%M:%S')