标签：__ winreg 封装 python self KEY ._ REG

import winreg
from collections import namedtuple

__all__ = ['Reg', 'REG_ROOTS', 'query_reg']

_empty_generator = (_ele for _ele in ())
_backslash = '\\'

_REG_VALUE = namedtuple("REG_VALUE", ('name', 'value', 'data_type'))

REG_PERMISSION_DESC = [
    'KEY_ALL_ACCESS',
    'KEY_WRITE',
    'KEY_SET_VALUE',
    'KEY_EXECUTE',
    'KEY_WOW64_32KEY',
    'KEY_WOW64_64KEY',
    'KEY_CREATE_LINK',
    'KEY_CREATE_SUB_KEY',
    'KEY_ENUMERATE_SUB_KEYS',
    'KEY_READ',
    'KEY_QUERY_VALUE',
    'KEY_NOTIFY'
]

REG_PERMISSION = namedtuple("REG_PERMISSION", REG_PERMISSION_DESC)(*(
    winreg.KEY_ALL_ACCESS,
    winreg.KEY_WRITE,
    winreg.KEY_SET_VALUE,
    winreg.KEY_EXECUTE,
    winreg.KEY_WOW64_32KEY,
    winreg.KEY_WOW64_64KEY,
    winreg.KEY_CREATE_LINK,
    winreg.KEY_CREATE_SUB_KEY,
    winreg.KEY_ENUMERATE_SUB_KEYS,
    winreg.KEY_READ,
    winreg.KEY_QUERY_VALUE,
    winreg.KEY_NOTIFY
))

REG_ROOTS_PATH = (
    'HKEY_CLASSES_ROOT',
    'HKEY_CURRENT_USER',
    'HKEY_LOCAL_MACHINE',
    'HKEY_USERS',
    'HKEY_CURRENT_CONFIG'
)

REG_ROOTS = namedtuple('REG_ROOTS', REG_ROOTS_PATH)(*[
    winreg.HKEY_CLASSES_ROOT,
    winreg.HKEY_CURRENT_USER,
    winreg.HKEY_LOCAL_MACHINE,
    winreg.HKEY_USERS,
    winreg.HKEY_CURRENT_CONFIG
])


class Reg:
    def __init__(self, reg_key, subkey='', parent=None):
        self._parent = parent
        if not self._parent:
            if isinstance(reg_key, int) and REG_ROOTS.count(reg_key) > 0:
                self._parent = REG_ROOTS_PATH[REG_ROOTS.index(reg_key)]
        self._key = f"{self._parent}{_backslash if subkey else ''}{subkey.encode('utf-8', 'ignore').decode('utf-8', 'ignore')}"
        self.keyname = subkey
        self._self = None
        self._permission = None
        self._permission_desc = None
        for permission in REG_PERMISSION:
            try:
                self._self = winreg.OpenKey(reg_key, subkey, 0, permission)
                self._permission = permission
                self._permission_desc = REG_PERMISSION_DESC[REG_PERMISSION.index(permission)]
                break
            except Exception:
                continue

    def __str__(self):
        return f"Reg<\"{self._key}\", {self._permission_desc}>"

    __repr__ = __str__

    def __iter__(self):
        if self.has_keys:
            for _key in self.keys:
                yield _key
        if self.has_values:
            for _value in self.values:
                yield _value

    @property
    def keys(self):
        if not self._self:
            return _empty_generator
        _count = 0
        while True:
            try:
                yield Reg(self._self, winreg.EnumKey(self._self, _count), self._key)
            except WindowsError:
                break
            _count += 1

    @property
    def values(self):
        if not self._self:
            return _empty_generator
        _count = 0
        while True:
            try:
                yield _REG_VALUE(*winreg.EnumValue(self._self, _count))
            except WindowsError:
                break
            _count += 1

    @property
    def has_values(self):
        try:
            self.values.__next__()
        except StopIteration:
            return False
        return True

    @property
    def has_keys(self):
        try:
            self.keys.__next__()
        except StopIteration:
            return False
        return True

    def delete(self, warning=True):
        """删除注册表键值，包含子键值"""
        if self.has_keys:
            for subkey in self.keys:
                subkey.delete()
        winreg.DeleteKeyEx(self._self, sub_key='', access=self._permission)


def query_reg(reg: Reg, *query_texts: str):
    assert isinstance(reg, Reg)
    if query_texts is None or len(query_texts) == 0:
        return _empty_generator
    for _mainitem in reg:
        if isinstance(_mainitem, Reg):
            if query_texts.count(_mainitem.keyname) > 0:
                yield _mainitem
            if _mainitem.has_keys:
                for _subitem in query_reg(_mainitem, *query_texts):
                    yield _subitem
        if isinstance(_mainitem, _REG_VALUE):
            if query_texts.count(_mainitem.name) > 0:
                yield _mainitem


if __name__ == '__main__':
    def test_query():
        for _item in query_reg(Reg(REG_ROOTS.HKEY_LOCAL_MACHINE), 'Google'):
            print(_item)
            if isinstance(_item, Reg):
                print(list(_item.values))


    for item in Reg(REG_ROOTS.HKEY_CLASSES_ROOT):
        print(item)

    # test_query()

标签：__,winreg,封装,python,self,KEY,._,REG
来源： https://www.cnblogs.com/mintx/p/15774881.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。