'''Module for accessing MWR. This module provides an advanced, easy-to-use API for accessing and mutating the Microsoft Windows Registry.''' __version__ = 1.2 ################################################################################ import _winreg import os import random import sys import time class HKEY: 'Hive Constants' CLASSES_ROOT = -2147483648 CURRENT_USER = -2147483647 LOCAL_MACHINE = -2147483646 USERS = -2147483645 CURRENT_CONFIG = -2147483643 class KEY: 'SAM Constants' QUERY_VALUE = 1 SET_VALUE = 2 CREATE_SUB_KEY = 4 ENUMERATE_SUB_KEYS = 8 NOTIFY = 16 CREATE_LINK = 32 WRITE = 131078 EXECUTE = 131097 READ = 131097 ALL_ACCESS = 983103 class REG: 'Type Constants' NONE = 0 SZ = 1 EXPAND_SZ = 2 BINARY = 3 DWORD = 4 DWORD_BIG_ENDIAN = 5 LINK = 6 MULTI_SZ = 7 RESOURCE_LIST = 8 FULL_RESOURCE_DESCRIPTOR = 9 RESOURCE_REQUIREMENTS_LIST = 10 QWORD = 11 ################################################################################ class Value(object): 'Value(value) -> Value' def __init__(self, value): 'Initialize the Value object.' self.__value = value def __repr__(self): 'Return the object\'s representation.' return '%s(%r)' % (self.__class__.__name__, self.__value) def __get_value(self): 'Private class method.' return self.__value value = property(__get_value, doc='Value of this object.') class REG_NONE(Value): pass class REG_SZ(Value): pass class REG_EXPAND_SZ(Value): pass class REG_BINARY(Value): pass class REG_DWORD(Value): pass class REG_DWORD_BIG_ENDIAN(Value): pass class REG_LINK(Value): pass class REG_MULTI_SZ(Value): pass class REG_RESOURCE_LIST(Value): pass class REG_FULL_RESOURCE_DESCRIPTOR(Value): pass class REG_RESOURCE_REQUIREMENTS_LIST(Value): pass class REG_QWORD(Value): pass ################################################################################ class Registry(object): 'Registry([computer_name]) -> Registry' def __init__(self, computer_name=None): 'Initialize the Registry object.' self.__computer_name = computer_name def __repr__(self): 'Return the object\'s representation.' if self.__computer_name is None: return '%s()' % self.__class__.__name__ return '%s(%r)' % (self.__class__.__name__, self.__computer_name) def __iter__(self): 'Iterate over hives defined in HKEY.' for key in HKEY.__dict__: if not key.startswith('_'): yield self.__hive(HKEY.__dict__[key]) def __hive(self, key): 'Private class method.' if self.__computer_name is None: return Key(key=key) return Key(key=key, computer_name=self.__computer_name) def __HKEY_CLASSES_ROOT(self): 'Private class method.' return self.__hive(HKEY.CLASSES_ROOT) def __HKEY_CURRENT_USER(self): 'Private class method.' return self.__hive(HKEY.CURRENT_USER) def __HKEY_LOCAL_MACHINE(self): 'Private class method.' return self.__hive(HKEY.LOCAL_MACHINE) def __HKEY_USERS(self): 'Private class method.' return self.__hive(HKEY.USERS) def __HKEY_CURRENT_CONFIG(self): 'Private class method.' return self.__hive(HKEY.CURRENT_CONFIG) HKEY_CLASSES_ROOT = property(__HKEY_CLASSES_ROOT, doc='The CLASSES_ROOT hive.') HKEY_CURRENT_USER = property(__HKEY_CURRENT_USER, doc='The CURRENT_USER hive.') HKEY_LOCAL_MACHINE = property(__HKEY_LOCAL_MACHINE, doc='The LOCAL_MACHINE hive.') HKEY_USERS = property(__HKEY_USERS, doc='The USERS hive.') HKEY_CURRENT_CONFIG = property(__HKEY_CURRENT_CONFIG, doc='The CURRENT_CONFIG hive.') class Key(object): '''Key(**kwargs) -> Key Key(key=int) -> Key Key(key=PyHKEY) -> Key Key(key=Key) -> Key Key(key=int, sub_key=str) -> Key Key(key=PyHKEY, sub_key=str) -> Key Key(key=Key, sub_key=str) -> Key Key(key=int, sam=int) -> Key Key(key=PyHKEY, sam=int) -> Key Key(key=Key, sam=int) -> Key Key(key=int, sub_key=str, sam=int) -> Key Key(key=PyHKEY, sub_key=str, sam=int) -> Key Key(key=Key, sub_key=str, sam=int) -> Key Key(key=int, computer_name=str) -> Key Key(key=int, sub_key=str, computer_name=str) -> Key Key(key=int, sam=int, computer_name=str) -> Key Key(key=int, sub_key=str, sam=int, computer_name=str) -> Key''' def __init__(self, **kwargs): 'Initialize the Key object.' assert kwargs, 'No Keyword Arguments Were Found' self.__repr, key, sub_key, sam, computer_name = '%s(%s)' % (self.__class__.__name__, ', '.join(['%s=%r' % (key, kwargs[key]) for key in kwargs])), kwargs.pop('key', None), kwargs.pop('sub_key', None), kwargs.pop('sam', None), kwargs.pop('computer_name', None) assert not kwargs, 'Invalid Keyword Arguments Were Found' if isinstance(key, (int, _winreg.HKEYType)) and sub_key is None and sam is None and computer_name is None: self.__self = _winreg.OpenKey(key, '') elif isinstance(key, Key) and sub_key is None and sam is None and computer_name is None: self.__self = _winreg.OpenKey(key.__self, '') elif isinstance(key, (int, _winreg.HKEYType)) and isinstance(sub_key, str) and sam is None and computer_name is None: self.__self = _winreg.OpenKey(key, sub_key) elif isinstance(key, Key) and isinstance(sub_key, str) and sam is None and computer_name is None: self.__self = _winreg.OpenKey(key.__self, sub_key) elif isinstance(key, (int, _winreg.HKEYType)) and sub_key is None and isinstance(sam, int) and computer_name is None: self.__self = _winreg.OpenKey(key, '', 0, sam) elif isinstance(key, Key) and sub_key is None and isinstance(sam, int) and computer_name is None: self.__self = _winreg.OpenKey(key.__self, '', 0, sam) elif isinstance(key, (int, _winreg.HKEYType)) and isinstance(sub_key, str) and isinstance(sam, int) and computer_name is None: self.__self = _winreg.OpenKey(key, sub_key, 0, sam) elif isinstance(key, Key) and isinstance(sub_key, str) and isinstance(sam, int) and computer_name is None: self.__self = _winreg.OpenKey(key.__self, sub_key, 0, sam) elif isinstance(key, int) and sub_key is None and sam is None and isinstance(computer_name, str): self.__self = _winreg.ConnectRegistry(computer_name, key) elif isinstance(key, int) and isinstance(sub_key, str) and sam is None and isinstance(computer_name, str): self.__self = _winreg.OpenKey(_winreg.ConnectRegistry(computer_name, key), sub_key) elif isinstance(key, int) and sub_key is None and isinstance(sam, int) and isinstance(computer_name, str): self.__self = _winreg.OpenKey(_winreg.ConnectRegistry(computer_name, key), '', 0, sam) elif isinstance(key, int) and isinstance(sub_key, str) and isinstance(sam, int) and isinstance(computer_name, str): self.__self = _winreg.OpenKey(_winreg.ConnectRegistry(computer_name, key), sub_key, 0, sam) else: raise TypeError, 'Key Could Not Be Initialized' def __repr__(self): 'Return the object\'s representation.' return self.__repr def save(self, file_name): 'Save this key to file.' _winreg.SaveKey(self.__self, file_name) def load(self, sub_key, file_name): 'Load subkey from file.' _winreg.LoadKey(self.__self, sub_key, file_name) def __get_keys(self): 'Private class method.' return Keys(self.__self) def __set_keys(self, keys): 'Private class method.' if isinstance(keys, str): _winreg.CreateKey(self.__self, keys) elif isinstance(keys, (list, tuple)): for key in keys: self.keys = key else: raise TypeError, 'Key Could Not Be Created' def __del_keys(self): 'Private class method.' try: while True: _winreg.DeleteKey(self.__self, _winreg.EnumKey(self.__self, 0)) except EnvironmentError: pass def __get_values(self): 'Private class method.' return Values(self.__self) def __set_values(self, values): 'Private class method.' if isinstance(values, str): _winreg.SetValueEx(self.__self, values, 0, REG.SZ, _winreg.QueryValue(self.__self, '')) elif isinstance(values, (list, tuple)): for value in values: self.values = value else: raise TypeError, 'Value Could Not Be Created' def __del_values(self): 'Private class method.' try: while True: _winreg.DeleteValue(self.__self, _winreg.EnumValue(self.__self, 0)[0]) except EnvironmentError: pass def __get_value(self): 'Private class method.' return _winreg.QueryValue(self.__self, '') def __set_value(self, value): 'Private class method.' _winreg.SetValue(self.__self, '', REG.SZ, value) def __del_value(self): 'Private class method.' _winreg.DeleteValue(self.__self, '') def __get_info(self): 'Private class method.' return Info(*_winreg.QueryInfoKey(self.__self)) keys = property(__get_keys, __set_keys, __del_keys, 'Keys of this key.') values = property(__get_values, __set_values, __del_values, 'Values of this key.') value = property(__get_value, __set_value, __del_value, 'Value of this key.') info = property(__get_info, doc='Information about this key.') class Keys(object): 'Keys(key) -> Keys' def __init__(self, key): 'Initialize the Keys object.' self.__self = key def __repr__(self): 'Return the object\'s representation.' return '%s(%r)' % (self.__class__.__name__, self.__self) def __len__(self): 'Return the number of keys.' return _winreg.QueryInfoKey(self.__self)[0] def __getitem__(self, key): 'Return the specified key.' return Key(key=self.__self, sub_key=key) def __setitem__(self, key, value): 'Assign the item to a key.' file_name = os.path.join(os.path.dirname(sys.argv[0]), ''.join(random.sample('0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz', 8))) while os.path.exists(file_name): file_name = os.path.join(os.path.dirname(sys.argv[0]), ''.join(random.sample('0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz', 8))) try: value.save(file_name) _winreg.LoadKey(self.__self, key, file_name) finally: os.remove(file_name) def __delitem__(self, key): 'Delete the specified key.' _winreg.DeleteKey(self.__self, key) def __iter__(self): 'Iterate over the key names.' keys, index = [], 0 try: while True: keys.append(_winreg.EnumKey(self.__self, index)) index += 1 except EnvironmentError: for key in keys: yield key def __contains__(self, item): 'Check for a key\'s existence.' item, index = item.lower(), 0 try: while True: if _winreg.EnumKey(self.__self, index).lower() == item: return True index += 1 except EnvironmentError: return False class Values(object): 'Values(key) -> Values' TYPES = REG_NONE, REG_SZ, REG_EXPAND_SZ, REG_BINARY, REG_DWORD, REG_DWORD_BIG_ENDIAN, REG_LINK, REG_MULTI_SZ, REG_RESOURCE_LIST, REG_FULL_RESOURCE_DESCRIPTOR, REG_RESOURCE_REQUIREMENTS_LIST, REG_QWORD def __init__(self, key): 'Initialize the Values object.' self.__self = key def __repr__(self): 'Return the object\'s representation.' return '%s(%r)' % (self.__class__.__name__, self.__self) def __len__(self): 'Return the number of values.' return _winreg.QueryInfoKey(self.__self)[1] def __getitem__(self, key): 'Return the specified value.' item_value, item_type = _winreg.QueryValueEx(self.__self, key) return self.TYPES[item_type](item_value) def __setitem__(self, key, value): 'Assign the item to a value.' if isinstance(value, self.TYPES): _winreg.SetValueEx(self.__self, key, 0, list(self.TYPES).index(value.__class__), value.value) else: _winreg.SetValueEx(self.__self, key, 0, _winreg.QueryValueEx(self.__self, key)[1], value) def __delitem__(self, key): 'Delete the specified value.' _winreg.DeleteValue(self.__self, key) def __iter__(self): 'Iterate over the value names.' values, index = [], 0 try: while True: values.append(_winreg.EnumValue(self.__self, index)[0]) index += 1 except EnvironmentError: for value in values: yield value def __contains__(self, item): 'Check for a value\'s existence.' item, index = item.lower(), 0 try: while True: if _winreg.EnumValue(self.__self, index)[0].lower() == item: return True index += 1 except EnvironmentError: return False class Info(object): 'Info(keys, values, modified) -> Info' def __init__(self, keys, values, modified): 'Initialize the Info object.' self.__keys, self.__values, self.__modified = keys, values, modified def __repr__(self): 'Return the object\'s representation.' return '%s(%r, %r, %r)' % (self.__class__.__name__, self.__keys, self.__values, self.__modified) def __get_keys(self): 'Private class method.' return self.__keys def __get_values(self): 'Private class method.' return self.__values def __get_modified(self): 'Private class method.' return self.__modified def __get_difference(self): 'Private class method.' return time.time() + 11644473600.0 - self.__modified / 10000000.0 keys = property(__get_keys, doc='Number of keys.') values = property(__get_values, doc='Number of values.') modified = property(__get_modified, doc='Time last modified.') difference = property(__get_difference, doc='Seconds since modified.') ################################################################################ if __name__ == '__main__': print 'Content-Type: text/plain' print print file(sys.argv[0]).read()