主页 / 问题 / 1351223
Accepted
biomotor
Asked:2022-04-18 18:12:46 +0000 UTC2022-04-18 18:12:46 +0000 UTC

帮助转换字符串

  • 772

我想使用 winreg 访问注册表。

来自互联网的例子：

Winreg(winreg.HKEY_CURRENT_USER, r"SOFTWARE\\RegTest\\test1")

我想使用这样的东西：

def addreg(r'HKEY_CURRENT_USER\\SOFTWARE\\RegTest\\test1')

问题 1：如何从字符串中去除 HKEY_CURRENT_USER？

问题2： python函数中是否有将字符串'HKEY_CURRENT_USER'转换为winreg.HKEY_CURRENT_USER引用的函数？

（我不想为每个注册表项写 if ... Winreg(winreg...) 5 次）

python

2 个回答

  1. Best Answer

    将匹配字符串的字典填充到键中，并通过它提取值。

    import winreg

keys_map = {
    "HKEY_CLASSES_ROOT"     : winreg.HKEY_CLASSES_ROOT    ,
    "HKEY_CURRENT_USER"     : winreg.HKEY_CURRENT_USER    ,
    "HKEY_LOCAL_MACHINE"    : winreg.HKEY_LOCAL_MACHINE   ,
    "HKEY_USERS"            : winreg.HKEY_USERS           ,
    "HKEY_PERFORMANCE_DATA" : winreg.HKEY_PERFORMANCE_DATA,
    "HKEY_CURRENT_CONFIG"   : winreg.HKEY_CURRENT_CONFIG
}
def str2key(fullkey):
    root, subkey = fullkey.split(r"\\", 1)
    if root in keys_map:
        return (keys_map[root], subkey)
    print(f"Unknown root key '{root}'")
    
print(str2key(r'HKEY_CURRENT_USER\\SOFTWARE\\RegTest\\test1'))
print(str2key(r'WTF\\SOFTWARE\\RegTest\\test2'))

    (18446744071562067969, 'SOFTWARE\\\\RegTest\\\\test1')
Unknown root key 'WTF'
None

    那么，您可以通过星号使用替换，或者您可以通过变量使用句柄。

    Winreg(*str2key(r'HKEY_CURRENT_USER\\SOFTWARE\\RegTest\\test1'))
# или
key, subkey = str2key(r'HKEY_CURRENT_USER\\SOFTWARE\\RegTest\\test1')
Winreg(key, subkey)
    • 4

  2. 对问题 1 的回答：（循环在字符串 str 中的 keys 列表中查找每个值，如果找到，则将值分配给 key，并使用 cut 键值重新定义 str 字符串。）

    str = "HKEY_CURRENT_USER\\SOFTWARE\\RegTest\\test1"
key = None
keys = {'HKEY_CLASSES_ROOT',
        'HKEY_CURRENT_USER',
        'HKEY_LOCAL_MACHINE',
        'HKEY_USERS',
        'HKEY_CURRENT_CONFIG'}

for i in keys:
    k = str.find(i)
    if k != -1:
        key = i
        str = str.replace(key, '')

if key == None:
    print('Error: key not found in str!')

print(key, str)
    • 0

相关问题