On May 17, 2:06 am, Lie <[EMAIL PROTECTED]> wrote:
> On May 9, 7:36 pm, Unknown Hero <[EMAIL PROTECTED]> wrote:> Ah, never mind,
> got it to work. Here's the code now. I hope I won't
> > run into another problems later :D
>
> > <code snippet>
> > #Goes through all keys and subkeys in the selected hive (defined as
> > root) and replaces the value 'old' with the value 'new'
> > #
> > #IMPORTANT! You should always back up the registry before attempting
> > to modify it.
> > #The author of this script CANNOT BE HELD RESPONSIVE for any damage
> > caused by running this script.
>
> (snip)
>
> One thing though, the disclaimer should not said the author cannot be
> held responsive, or you would be Windows that is not responsive all
> the times. I think it should say "responsible".
>
> I'm quite confused though, the code could be made simpler by dozens
> using Python's high-level functionalities. Particularly the lengthy
> code that checked whether the string contained a substring and replace
> the substring with new could be done easily (and faster) using
> python's in and replace function. I'll post the code later when I
> finished checking places where the codes could be simplified, and I'll
> also polish the entry code and a few other things (and pythonify the
> code according to the PEP 8's guide lines).
The tidied code ++
'''
Goes through all keys and subkeys in the selected hive (defined
as root) and replaces the value 'old' with the value 'new'
IMPORTANT! You should always back up the registry before
attempting to modify it. The author of this script CANNOT BE
HELD RESPONSIBLE for any damage caused by running this script.
You can call the script from a command line and pass two or
three values: HIVE, OLD, and NEW.
OLD and NEW can be any string value
HIVE has to be one of the following:
HKEY_LOCAL_MACHINE / HKLM
HKEY_CURRENT_USER / HKCU
HKEY_CLASSES_ROOT / HKCR
HKEY_USERS / HKU
HKEY_CURRENT_CONFIG / HKCC
If NEW is not specified, values that matches OLD will be replaced
with empty string (i.e. deleted)
'''
import _winreg
import sys
HIVES = {
"HKEY_LOCAL_MACHINE" : _winreg.HKEY_LOCAL_MACHINE,
"HKEY_CURRENT_USER" : _winreg.HKEY_CURRENT_USER,
"HKEY_CLASSES_ROOT" : _winreg.HKEY_CLASSES_ROOT,
"HKEY_USERS" : _winreg.HKEY_USERS,
"HKEY_CURRENT_CONFIG" : _winreg.HKEY_CURRENT_CONFIG,
}
AccessError = False
class RegKey(object):
def __init__ (self, name, key):
self.name = name
self.key = key
def __str__ (self):
return self.name
def walk(top):
""" Walk through each key, subkey, and values
Walk the registry starting from the key top
in the form HIVE\\key\\subkey\\..\\subkey and generating
key, subkey_names, values at each level.
key is a lightly wrapped registry key, including the name
and the HKEY object.
subkey_names are simply names of the subkeys of that key
values are 3-tuples containing (name, data, data-type).
See the documentation for _winreg.EnumValue for more details.
"""
try:
if "\\" not in top:
top += "\\"
root, subkey = top.split ("\\", 1)
key = _winreg.OpenKey(HIVES[root],
subkey,
0,
_winreg.KEY_READ |
_winreg.KEY_SET_VALUE)
subkeys = []
i = 0
while True:
try:
subkeys.append(_winreg.EnumKey(key, i))
i += 1
except EnvironmentError:
break
values = []
i = 0
while True:
try:
values.append(_winreg.EnumValue(key, i))
i += 1
except EnvironmentError:
break
yield RegKey(top, key), subkeys, values
for subkey in subkeys:
for result in walk(top + "\\" + subkey):
yield result
except WindowsError:
global AccessError
AccessError = True
def main(start, startHandle, old, new):
basickeys = []
i = 0
while True:
try:
basickeys.append(_winreg.EnumKey(startHandle, i))
i += 1
except EnvironmentError:
break
for x in basickeys:
for key, subkey_names, values in walk(start + "\\" + x):
for (name, data, type) in values:
if type == _winreg.REG_SZ:
if old in data:
winreg.SetValueEx(key.key,
name,
0,
type,
data.replace(old, new))
print key.key, name, 0, type,
data.replace(old, new)
def help():
#Show help
print '''
USAGE: Registry.py HIVE OLD [NEW]
HIVE: The root key in registry you want to go through.
HKEY_CLASSES_ROOT / HKCR
HKEY_CURRENT_USER / HKCU
HKEY_LOCAL_MACHINE / HKLM
HKEY_USERS / HKU
HKEY_CURRENT_CONFIG / HKCC
OLD: The value to search for.
Wrap multiword strings with "".
NEW: The value which will replace OLD.
Wrap multiword strings with "".
If not supplied, it default to empty string
which means delete all occurence of OLD
in the registry.
'''
if __name__ == '__main__':
if len(sys.argv) < 3 or len(sys.argv) > 4:
print 'Invalid Number of Arguments'
help()
exit()
## Root Hive
try:
start = {
'HKCU' : 'HKEY_CURRENT_USER',
'HKLM' : 'HKEY_LOCAL_MACHINE',
'HKCR' : 'HKEY_CLASSES_ROOT',
'HKU' : 'HKEY_USERS',
'HKCC' : 'HKEY_CURRENT_CONFIG',
}[sys.argv[1].upper()]
except KeyError:
start = sys.argv[1].upper()
try:
startHandle = {
'HKEY_CURRENT_USER' : _winreg.HKEY_CURRENT_USER,
'HKEY_LOCAL_MACHINE' : _winreg.HKEY_LOCAL_MACHINE,
'HKEY_CLASSES_ROOT' : _winreg.HKEY_CLASSES_ROOT,
'HKEY_USERS' : _winreg.HKEY_USERS,
'HKEY_CURRENT_CONFIG' : _winreg.HKEY_CURRENT_CONFIG,
}[start]
except KeyError:
print >> sys.stderr, 'Invalid Hive'
help()
exit()
## The substring to be replaced
old = sys.argv[2]
## The replacement string
try:
new = sys.argv[3]
except:
new = ''
main(start, startHandle, old, new)
if AccessError:
print '''
Some keys cannot be changed because you don't have the appropriate
permission to modify those keys, please try again with a suitable
user account.
'''
--
http://mail.python.org/mailman/listinfo/python-list
