修复无法正常显示图片并报错的问题

xaoyaoo · Jan 2, 2024 · 5d5ab76 · 5d5ab76
1 parent 401e42c
commit 5d5ab76
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/pywxdump/ui/view_chat.py b/pywxdump/ui/view_chat.py
@@ -183,9 +183,12 @@ def load_chat_records(selected_talker, start_index, page_size, user_list, MSG_AL
                 img_path = os.path.join(*img_path)
                 if FileStorage_path:
                     img_path = os.path.join(FileStorage_path, img_path)
-                    fomt, md5, out_bytes = read_img_dat(img_path)
-                    out_bytes = base64.b64encode(out_bytes).decode("utf-8")
-                    content["src"] = f"data:{fomt};base64,{out_bytes}"
+                    if os.path.exists(img_path):
+                        fomt, md5, out_bytes = read_img_dat(img_path)
+                        out_bytes = base64.b64encode(out_bytes).decode("utf-8")
+                        content["src"] = f"data:{fomt};base64,{out_bytes}"
+                    else:
+                        content["src"] = ""
                 else:
                     content["src"] = ""
             else: