blob: 854a029f349ec8e601e8ebbc48e238083aa43311 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
|
# -*- coding: utf-8 -*-
from testexception import TestException
import _winreg
_registry = dict()
_registry["HKEY_CLASSES_ROOT"] = _winreg.HKEY_CLASSES_ROOT
_registry["HKEY_CURRENT_USER"] = _winreg.HKEY_CURRENT_USER
_registry["HKEY_LOCAL_MACHINE"] = _winreg.HKEY_LOCAL_MACHINE
_registry["HKEY_USERS"] = _winreg.HKEY_USERS
_registry["HKEY_CURRENT_CONFIG"] = _winreg.HKEY_CURRENT_CONFIG
def splitKey( key ):
key, seperator, subKey = key.partition( '\\' )
return _registry[key], subKey
def checkKey( key, value, expectedData ):
baseKey, subKey = splitKey( key )
keyHandle = _winreg.OpenKey( baseKey, subKey )
data, _ = _winreg.QueryValueEx( keyHandle, value )
if data != expectedData:
raise TestException( '{0}: unexpected registry data. Actual: {1} Expected: {2}'.format( key, data, expectedData ) )
|