调整当前时间的插入

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

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