From 2b3c3f9d1f7187c40ccb2077168dd541bd36fbf2 Mon Sep 17 00:00:00 2001
From: cllcode <2440893398@qq.com>
Date: Mon, 8 Jul 2024 22:20:15 +0800
Subject: [PATCH] =?UTF-8?q?=E7=AE=80=E5=8C=96=E4=BA=86=20Windows=20?=
 =?UTF-8?q?=E5=B9=B3=E5=8F=B0=E7=89=B9=E5=AE=9A=E6=A8=A1=E5=9D=97=E7=9A=84?=
 =?UTF-8?q?=E5=AF=BC=E5=85=A5=E7=9A=84=E5=AF=BC=E5=85=A5=E6=96=B9=E5=BC=8F?=
 =?UTF-8?q?=EF=BC=8C=E4=BD=BF=E7=94=A8=E6=9D=A1=E4=BB=B6=E8=A1=A8=E8=BE=BE?=
 =?UTF-8?q?=E5=BC=8F=E5=87=8F=E5=B0=91=E4=BB=A3=E7=A0=81=E5=86=97=E4=BD=99?=
 =?UTF-8?q?=E3=80=82?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 pywxdump/api/api.py               |  5 +----
 pywxdump/wx_info/get_bias_addr.py |  5 +----
 pywxdump/wx_info/get_wx_info.py   | 17 +++++------------
 3 files changed, 7 insertions(+), 20 deletions(-)

diff --git a/pywxdump/api/api.py b/pywxdump/api/api.py
index 6a8fd70..70f4ccd 100644
--- a/pywxdump/api/api.py
+++ b/pywxdump/api/api.py
@@ -13,10 +13,7 @@ import re
 import time
 import shutil
 import sys
-if sys.platform == "win32":
-    import pythoncom
-else:
-    pythoncom = None
+pythoncom = __import__('pythoncom') if sys.platform == "win32" else None
 import pywxdump
 from pywxdump.file import AttachmentContext
 
diff --git a/pywxdump/wx_info/get_bias_addr.py b/pywxdump/wx_info/get_bias_addr.py
index e0c77b3..54099d8 100644
--- a/pywxdump/wx_info/get_bias_addr.py
+++ b/pywxdump/wx_info/get_bias_addr.py
@@ -15,10 +15,7 @@ import pymem
 
 from .utils import get_exe_version, get_exe_bit, verify_key
 
-if sys.platform == "win32":
-    ReadProcessMemory = ctypes.windll.kernel32.ReadProcessMemory
-else:
-    ReadProcessMemory = None
+ReadProcessMemory = ctypes.windll.kernel32.ReadProcessMemory if sys.platform == "win32" else None
 void_p = ctypes.c_void_p
 
 
diff --git a/pywxdump/wx_info/get_wx_info.py b/pywxdump/wx_info/get_wx_info.py
index 2c82780..26c3d78 100644
--- a/pywxdump/wx_info/get_wx_info.py
+++ b/pywxdump/wx_info/get_wx_info.py
@@ -10,19 +10,14 @@ import json
 import os
 import re
 import sys
-if sys.platform == "win32":
-    import winreg
-else:
-    winreg = None
+winreg = __import__('winreg') if sys.platform == "win32" else None
 import psutil
 import pymem
 from typing import List, Union
 from .utils import pattern_scan_all, verify_key, get_exe_version, get_exe_bit, info_error
 
-if sys.platform == "win32":
-    ReadProcessMemory = ctypes.windll.kernel32.ReadProcessMemory
-else:
-    ReadProcessMemory = None
+ReadProcessMemory = ctypes.windll.kernel32.ReadProcessMemory if sys.platform == "win32" else None
+
 
 void_p = ctypes.c_void_p
 
@@ -213,10 +208,8 @@ def get_details(process, version_list: dict = None, is_logging: bool = False):
           "account": "None", "mobile": "None", "name": "None", "mail": "None",
           "wxid": "None", "key": "None", "filePath": "None"}
     try:
-        if sys.platform == "win32":
-            Handle = ctypes.windll.kernel32.OpenProcess(0x1F0FFF, False, process.pid)
-        else:
-            Handle = None
+        Handle = ctypes.windll.kernel32.OpenProcess(0x1F0FFF, False, process.pid) if sys.platform == "win32" else None
+
         bias_list = version_list.get(rd['version'], None)
 
         addrLen = get_exe_bit(process.exe()) // 8