From 856d2ae1234e4e16bc19d810591848c8d84c73a7 Mon Sep 17 00:00:00 2001
From: xaoyaoo <xaoyaoo@163.com>
Date: Wed, 5 Jun 2024 13:35:20 +0800
Subject: [PATCH] get info add decorator info_error

---
 pywxdump/wx_info/utils.py | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/pywxdump/wx_info/utils.py b/pywxdump/wx_info/utils.py
index 6920f7f..d5a1eee 100644
--- a/pywxdump/wx_info/utils.py
+++ b/pywxdump/wx_info/utils.py
@@ -9,6 +9,8 @@ import os
 import re
 import sys
 import hmac
+import traceback
+
 import pymem
 import hashlib
 from win32com.client import Dispatch
@@ -25,7 +27,9 @@ def info_error(func):
         try:
             return func(*args, **kwargs)
         except Exception as e:
-            print(f"info_error: {e}")
+            traceback_data = traceback.format_exc()
+            rdata = f"{traceback_data}"
+            print(f"info_error: \n{rdata}")
             return "None"
 
     return wrapper
@@ -51,7 +55,7 @@ def verify_key(key, wx_db_path):
         return False
     return True
 
-
+@info_error
 def get_exe_version(file_path):
     """
     获取 PE 文件的版本号