Skip to content

feat(verify): 还从 \SOFTWARE\WOW6432Node\Microsoft\Windows\CurrentVersion\Uninstall\ 读取安装信息 #294

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Draft
DuckDuckStudio wants to merge 1 commit into
base: main
Choose a base branch
from
Draft

feat(verify): 还从 \SOFTWARE\WOW6432Node\Microsoft\Windows\CurrentVersion\Uninstall\ 读取安装信息 #294

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
47 changes: 20 additions & 27 deletions src/tools/verify.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,7 @@
import winreg
import shutil
import subprocess
from typing import Any
from colorama import Fore
from catfood.functions.print import 消息头
from function.github.token import read_token
Expand Down Expand Up @@ -210,20 +211,24 @@ def 测试安装与卸载(清单目录: str, 操作: str) -> int:

return 0

def 读取AAF字段():
# 机器范围 AAF: 计算机\HKEY_LOCAL_MACHINE\SOFTWARE\Microsoft\Windows\CurrentVersion\Uninstall\<ProductCode>
# 用户范围 AAF: 计算机\HKEY_CURRENT_USER\Software\Microsoft\Windows\CurrentVersion\Uninstall\<ProductCode>
# 此函数皆在实现获取机器/用户范围的 AAF，并返回获取到的结果
# 对应清单中的 AppsAndFeaturesEntries
def 读取AAF字段() -> list[dict[str, str | int]]:
"""
此函数从注册表的 `HKEY_LOCAL_MACHINE` (机器范围) 和 `HKEY_CURRENT_USER` (用户范围) 下的以下两处读取安装信息（对应清单中的 `AppsAndFeaturesEntries` 字段）。
- `\\SOFTWARE\\WOW6432Node\\Microsoft\\Windows\\CurrentVersion\\Uninstall\\<ProductCode>`
- `\\SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Uninstall\\<ProductCode>`

:return: 读取到的字段列表
:rtype: list[dict[str, str | int]]
"""

# 定义关心的字段列表
关心的字段 = [
关心的字段: list[str] = [
"DisplayName", "DisplayVersion", "Publisher", "UninstallString", "HelpLink",
"InstallLocation", "SystemComponent", "WindowsInstaller", "NoRemove", "NoModify",
"InstallSource", "EstimatedSize", "URLInfoAbout", "Comments"
]

def _读取注册表值(键: winreg.HKEYType, 值名: str):
def _读取注册表值(键: winreg.HKEYType, 值名: str) -> Any | str | None:
try:
值, 注册类型 = winreg.QueryValueEx(键, 值名)
if 注册类型 == winreg.REG_EXPAND_SZ:
Expand All @@ -239,7 +244,7 @@ def _读取注册表值(键: winreg.HKEYType, 值名: str):
except FileNotFoundError:
return None

def _读取注册表条目(hive: int, subkey: str, access: int=0):
def _读取注册表条目(hive: int, subkey: str, access: int=0) -> list[dict[str, str | int]]:
entries: list[dict[str, str | int]] = []
try:
with winreg.ConnectRegistry(None, hive) as reg:
Expand Down Expand Up @@ -270,25 +275,13 @@ def _读取注册表条目(hive: int, subkey: str, access: int=0):

所有条目: list[dict[str, str | int]] = []

# 读取机器范围 (64位和32位) - WOW
所有条目.extend(_读取注册表条目(
winreg.HKEY_LOCAL_MACHINE,
"SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Uninstall",
winreg.KEY_WOW64_64KEY
))

所有条目.extend(_读取注册表条目(
winreg.HKEY_LOCAL_MACHINE,
"SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Uninstall",
winreg.KEY_WOW64_32KEY
))

# 读取用户范围
所有条目.extend(_读取注册表条目(
winreg.HKEY_CURRENT_USER,
"Software\\Microsoft\\Windows\\CurrentVersion\\Uninstall"
))

for scope in (winreg.HKEY_LOCAL_MACHINE, winreg.HKEY_CURRENT_USER):
for regPath in (
"SOFTWARE\\WOW6432Node\\Microsoft\\Windows\\CurrentVersion\\Uninstall",
"SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Uninstall"
):
所有条目.extend(_读取注册表条目(scope, regPath, winreg.KEY_WOW64_64KEY if scope == winreg.HKEY_LOCAL_MACHINE else 0))

return 所有条目

def 转换AAF条目为YAML(AAF条目: dict[str, str | int]):
Expand Down