4.28

图像信息导入.py

@@ -0,0 +1,108 @@
+import pandas as pd
+import psycopg2
+import json
+import os
+
+# ================= 配置区 =================
+DB_CONFIG = {
+    'dbname': 'inventory_system',
+    'user': 'test',
+    'password': '1234',
+    'host': 'localhost',
+    'port': '5435'
+}
+
+EXCEL_FILE = "Odoo_Archive/Odoo产品_终极大满贯版.xlsx"
+
+
+# ================= 辅助函数 =================
+def process_paths_only(json_str):
+    """
+    将爬虫的绝对路径，转换为现有后端接口完美支持的纯文件名格式！
+    """
+    if not json_str or str(json_str).strip() in ['[]', 'nan', 'None']:
+        return '[]'
+
+    try:
+        paths = json.loads(json_str)
+        new_paths = []
+
+        for path in paths:
+            if path.startswith('http://') or path.startswith('https://'):
+                new_paths.append(path)
+            else:
+                filename = os.path.basename(path)
+
+                # 【终极修复】去掉中间的子文件夹，直接请求文件名！
+                web_path = f"/api/v1/common/files/{filename}"
+                new_paths.append(web_path)
+
+        return json.dumps(new_paths, ensure_ascii=False)
+
+    except Exception as e:
+        return '[]'
+
+
+# ================= 主程序 =================
+def process_excel_to_db():
+    if not os.path.exists(EXCEL_FILE):
+        print(f"❌ 找不到 Excel 文件: {EXCEL_FILE}")
+        return
+
+    try:
+        df = pd.read_excel(EXCEL_FILE, dtype=str)
+        df = df.where(pd.notnull(df), None)
+        print(f"✅ 成功读取 Excel，共 {len(df)} 行数据。")
+
+        conn = psycopg2.connect(**DB_CONFIG)
+        cur = conn.cursor()
+        success_count = 0
+
+        for index, row in df.iterrows():
+            internal_ref = row.get('内部参考')
+            barcode = row.get('条码')
+
+            spec_model = ""
+            if barcode and internal_ref:
+                spec_model = f"{barcode}/{internal_ref}"
+            elif barcode:
+                spec_model = f"{barcode}"
+            elif internal_ref:
+                spec_model = f"{internal_ref}"
+            else:
+                continue
+
+            raw_image_json = row.get('generalImage')
+            raw_manual_json = row.get('generalManual')
+
+            if (not raw_image_json or raw_image_json == '[]') and (not raw_manual_json or raw_manual_json == '[]'):
+                continue
+
+            product_image = process_paths_only(raw_image_json)
+            manual_link = process_paths_only(raw_manual_json)
+
+            update_query = """
+                           UPDATE material_base
+                           SET product_image = %s, \
+                               manual_link   = %s
+                           WHERE spec_model = %s
+                           """
+            cur.execute(update_query, (product_image, manual_link, spec_model))
+
+            if cur.rowcount > 0:
+                success_count += 1
+
+        conn.commit()
+        print(f"\n🎉 导入完成！成功更新了 {success_count} 条数据的正确路径。")
+        print("💡 赶快去刷新前端看看吧！这次图片一定能刷出来！")
+
+    except Exception as e:
+        print(f"❌ 发生致命错误: {e}")
+        if 'conn' in locals() and conn: conn.rollback()
+    finally:
+        if 'cur' in locals() and cur: cur.close()
+        if 'conn' in locals() and conn: conn.close()
+
+
+if __name__ == "__main__":
+    process_excel_to_db()