ExternalVendorCode/tahoe-lafs
1
0
Fork 0
mirror of https://github.com/tahoe-lafs/tahoe-lafs.git synced 2024-12-29 17:28:53 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

New name for module.

Browse Source
This commit is contained in:
Itamar Turner-Trauring 2021-05-05 11:31:19 -04:00
parent 1ed24a1616
commit fa4be104a6
1 changed files with 12 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
src/allmydata/windows/registry.py
View File

@ -1,5 +1,5 @@
import sys
import _winreg
import winreg
_AMD_KEY = r"Software\Allmydata"
_BDIR_KEY = 'Base Dir Path'
@ -22,19 +22,19 @@ def get_registry_setting(key, name, _topkey=None):
@param name: The name of the setting we are querying.
@type name: String
"""
topkeys = [_winreg.HKEY_CURRENT_USER, _winreg.HKEY_LOCAL_MACHINE]
topkeys = [winreg.HKEY_CURRENT_USER, winreg.HKEY_LOCAL_MACHINE]
if _topkey:
topkeys.insert(0, _topkey)
for topkey in topkeys:
try:
regkey = _winreg.OpenKey(topkey, key)
regkey = winreg.OpenKey(topkey, key)
sublen, vallen, timestamp = _winreg.QueryInfoKey(regkey)
sublen, vallen, timestamp = winreg.QueryInfoKey(regkey)
for validx in xrange(vallen):
keyname, value, keytype = _winreg.EnumValue(regkey, validx)
if keyname == name and keytype == _winreg.REG_SZ:
keyname, value, keytype = winreg.EnumValue(regkey, validx)
if keyname == name and keytype == winreg.REG_SZ:
return value
except WindowsError:
@ -42,27 +42,27 @@ def get_registry_setting(key, name, _topkey=None):
# We didn't find the key:
raise KeyError(key, name, "registry setting not found")
def set_registry_setting(key, name, data, reg_type=_winreg.REG_SZ,
_topkey=_winreg.HKEY_LOCAL_MACHINE, create_key_if_missing=True):
def set_registry_setting(key, name, data, reg_type=winreg.REG_SZ,
_topkey=winreg.HKEY_LOCAL_MACHINE, create_key_if_missing=True):
"""
Sets a registry setting.
defaults to string values (REG_SZ) - overridable with reg_type.
"""
try:
regkey = _winreg.OpenKey(_topkey, key, 0, _winreg.KEY_SET_VALUE)
regkey = winreg.OpenKey(_topkey, key, 0, winreg.KEY_SET_VALUE)
except WindowsError:
if create_key_if_missing:
regkey = _winreg.CreateKey(_topkey, key)
regkey = winreg.CreateKey(_topkey, key)
else:
raise KeyError(key, "registry key not found")
try:
_winreg.DeleteValue(regkey, name)
winreg.DeleteValue(regkey, name)
except:
pass
_winreg.SetValueEx(regkey, name, 0, reg_type, data)
winreg.SetValueEx(regkey, name, 0, reg_type, data)
def get_registry_value(keyname):
"""