From fd56fb91b1b79d2b4247e5a6117e7bc5871306cc Mon Sep 17 00:00:00 2001
From: liuyebo <1515783401@qq.com>
Date: Wed, 29 May 2024 14:22:29 +0800
Subject: [PATCH] =?UTF-8?q?=E8=B0=83=E6=95=B4=E5=BD=93=E5=89=8D=E6=97=B6?=
 =?UTF-8?q?=E9=97=B4=E7=9A=84=E6=8F=92=E5=85=A5?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 photo_review/photo_review.py | 13 ++++++++-----
 photo_review/util/util.py    |  6 ++++++
 2 files changed, 14 insertions(+), 5 deletions(-)
 create mode 100644 photo_review/util/util.py

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')