← Back to branch summary

~ubuntu-branches/ubuntu/karmic/tahoe-lafs/karmic

« back to all changes in this revision

Viewing changes to src/allmydata/windows/registry.py

  • Committer: Bazaar Package Importer
  • Author(s): Zooko O'Whielacronx (Hacker)
  • Date: 2009-09-24 00:00:05 UTC
  • Revision ID: james.westby@ubuntu.com-20090924000005-ixe2n4yngmk49ysz
Tags: upstream-1.5.0
Import upstream version 1.5.0

Show diffs side-by-side

added added

removed removed

Lines of Context:
src/allmydata/windows/registry.py
 
1
import sys
 
2
import _winreg
 
3
 
 
4
_AMD_KEY = r"Software\Allmydata"
 
5
_BDIR_KEY = 'Base Dir Path'
 
6
 
 
7
if sys.platform not in ('win32'):
 
8
    raise ImportError, "registry cannot be used on non-windows systems"
 
9
    class WindowsError(Exception): # stupid voodoo to appease pyflakes
 
10
        pass
 
11
 
 
12
def get_registry_setting(key, name, _topkey=None):
 
13
    """
 
14
    This function iterates through _topkey (if not None),
 
15
    HKEY_CURRENT_USER, and HKEY_LOCAL_MACHINE before giving up.
 
16
 
 
17
    @note: Only supports string values.
 
18
 
 
19
    @param key: The key we are searching.
 
20
    @type key: String
 
21
 
 
22
    @param name: The name of the setting we are querying.
 
23
    @type name: String
 
24
    """
 
25
    topkeys = [_winreg.HKEY_CURRENT_USER, _winreg.HKEY_LOCAL_MACHINE]
 
26
 
 
27
    if _topkey:
 
28
        topkeys.insert(0, _topkey)
 
29
 
 
30
    for topkey in topkeys:
 
31
        try:
 
32
            regkey = _winreg.OpenKey(topkey, key)
 
33
 
 
34
            sublen, vallen, timestamp = _winreg.QueryInfoKey(regkey)
 
35
            for validx in xrange(vallen):
 
36
                keyname, value, keytype = _winreg.EnumValue(regkey, validx)
 
37
                if keyname == name and keytype == _winreg.REG_SZ:
 
38
                    return value
 
39
 
 
40
        except WindowsError:
 
41
            continue
 
42
    # We didn't find the key:
 
43
    raise KeyError, (key, name, "registry setting not found")
 
44
 
 
45
def set_registry_setting(key, name, data, reg_type=_winreg.REG_SZ,
 
46
                         _topkey=_winreg.HKEY_LOCAL_MACHINE, create_key_if_missing=True):
 
47
    """
 
48
    Sets a registry setting.
 
49
 
 
50
    defaults to string values (REG_SZ) - overridable with reg_type.
 
51
    """
 
52
    try:
 
53
        regkey = _winreg.OpenKey(_topkey, key, 0, _winreg.KEY_SET_VALUE)
 
54
    except WindowsError:
 
55
        if create_key_if_missing:
 
56
            regkey = _winreg.CreateKey(_topkey, key)
 
57
        else:
 
58
            raise KeyError, (key, "registry key not found")
 
59
 
 
60
    try:
 
61
        _winreg.DeleteValue(regkey, name)
 
62
    except:
 
63
        pass
 
64
 
 
65
    _winreg.SetValueEx(regkey, name, 0, reg_type, data)
 
66
 
 
67
def get_registry_value(keyname):
 
68
    """
 
69
    retrieves a registry key value from within the Software/Allmydata Inc key
 
70
    """
 
71
    try:
 
72
        return get_registry_setting(_AMD_KEY, keyname)
 
73
    except KeyError:
 
74
        return None
 
75
 
 
76
def get_base_dir_path():
 
77
    return get_registry_value(_BDIR_KEY)