ExternalVendorCode/tahoe-lafs
1
0
Fork 0
mirror of https://github.com/tahoe-lafs/tahoe-lafs.git synced 2025-01-04 12:14:11 +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 sys
import _winreg import winreg
_AMD_KEY = r"Software\Allmydata" _AMD_KEY = r"Software\Allmydata"
_BDIR_KEY = 'Base Dir Path' _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. @param name: The name of the setting we are querying.
@type name: String @type name: String
""" """
topkeys = [_winreg.HKEY_CURRENT_USER, _winreg.HKEY_LOCAL_MACHINE] topkeys = [winreg.HKEY_CURRENT_USER, winreg.HKEY_LOCAL_MACHINE]
if _topkey: if _topkey:
topkeys.insert(0, _topkey) topkeys.insert(0, _topkey)
for topkey in topkeys: for topkey in topkeys:
try: 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): for validx in xrange(vallen):
keyname, value, keytype = _winreg.EnumValue(regkey, validx) keyname, value, keytype = winreg.EnumValue(regkey, validx)
if keyname == name and keytype == _winreg.REG_SZ: if keyname == name and keytype == winreg.REG_SZ:
return value return value
except WindowsError: except WindowsError:
@ -42,27 +42,27 @@ def get_registry_setting(key, name, _topkey=None):
# We didn't find the key: # We didn't find the key:
raise KeyError(key, name, "registry setting not found") raise KeyError(key, name, "registry setting not found")
def set_registry_setting(key, name, data, reg_type=_winreg.REG_SZ, def set_registry_setting(key, name, data, reg_type=winreg.REG_SZ,
_topkey=_winreg.HKEY_LOCAL_MACHINE, create_key_if_missing=True): _topkey=winreg.HKEY_LOCAL_MACHINE, create_key_if_missing=True):
""" """
Sets a registry setting. Sets a registry setting.
defaults to string values (REG_SZ) - overridable with reg_type. defaults to string values (REG_SZ) - overridable with reg_type.
""" """
try: try:
regkey = _winreg.OpenKey(_topkey, key, 0, _winreg.KEY_SET_VALUE) regkey = winreg.OpenKey(_topkey, key, 0, winreg.KEY_SET_VALUE)
except WindowsError: except WindowsError:
if create_key_if_missing: if create_key_if_missing:
regkey = _winreg.CreateKey(_topkey, key) regkey = winreg.CreateKey(_topkey, key)
else: else:
raise KeyError(key, "registry key not found") raise KeyError(key, "registry key not found")
try: try:
_winreg.DeleteValue(regkey, name) winreg.DeleteValue(regkey, name)
except: except:
pass pass
_winreg.SetValueEx(regkey, name, 0, reg_type, data) winreg.SetValueEx(regkey, name, 0, reg_type, data)
def get_registry_value(keyname): def get_registry_value(keyname):
""" """