简化了 Windows 平台特定模块的导入的导入方式,使用条件表达式减少代码冗余。

This commit is contained in:
cllcode 2024-07-08 22:20:15 +08:00
parent 4086cc53c8
commit 2b3c3f9d1f
3 changed files with 7 additions and 20 deletions

View File

@ -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

View File

@ -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

View File

@ -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