diff options
| -rw-r--r-- | unixreg/functions.py | 53 | ||||
| -rw-r--r-- | unixreg/utils.py | 13 |
2 files changed, 57 insertions, 9 deletions
diff --git a/unixreg/functions.py b/unixreg/functions.py index 65cd641..200a720 100644 --- a/unixreg/functions.py +++ b/unixreg/functions.py @@ -3,6 +3,7 @@ from typing import Union from .key import RegKey from .constants import STANDARD_RIGHTS_REQUIRED, KEY_WOW64_64KEY, KEY_WRITE, KEY_READ +from .utils import strict_types KEY_TYPE = Union[str, RegKey] SUBKEY_TYPE = KEY_TYPE | Union[None] @@ -48,15 +49,18 @@ def ConnectRegistry(computer: SUBKEY_TYPE, key: str): # any program that fails to catch this is to blame raise OSError("Not Implemented") -def OpenKeyEx(key: KEY_TYPE, sub_key: SUBKEY_TYPE, reserved=0, access=KEY_READ): +@strict_types +def OpenKeyEx(key: RegKey, sub_key: SUBKEY_TYPE, reserved=0, access=KEY_READ): return CreateKeyEx(key, sub_key, reserved, access) OpenKey = OpenKeyEx -def CreateKey(key: KEY_TYPE, sub_key: SUBKEY_TYPE): +@strict_types +def CreateKey(key: RegKey, sub_key: SUBKEY_TYPE): return CreateKeyEx(key, sub_key) -def CreateKeyEx(key: KEY_TYPE, sub_key: SUBKEY_TYPE, reserved=0, access=KEY_WRITE): +@strict_types +def CreateKeyEx(key: RegKey, sub_key: SUBKEY_TYPE, reserved=0, access=KEY_WRITE): key = __init_values(key, sub_key, access) __create_key(key) @@ -113,13 +117,15 @@ def QueryValueEx(key: KEY_TYPE, sub_key: SUBKEY_TYPE) -> str: QueryValue = QueryValueEx -def LoadKey(key: KEY_TYPE, sub_key: SUBKEY_TYPE, file_name: str): - # Not Implemented but keeping this functionality stubbed should not cause a problem - return +@strict_types +def LoadKey(key: RegKey, sub_key: SUBKEY_TYPE, file_name: str): + # this requires a win32 permission compatibility layer + raise OSError("Not Implemented") -def SaveKey(key: KEY_TYPE, file_name: str) -> None: - # Not Implemented but keeping this functionality stubbed should not cause a problem - return +@strict_types +def SaveKey(key: RegKey, file_name: str) -> None: + # this requires a win32 permission compatibility layer + raise OSError("Not Implemented") def SetValue(key: KEY_TYPE, sub_key: str, type: int, value: str): return SetValueEx(key, sub_key, 0, type, value) @@ -140,3 +146,32 @@ def EnableReflectionKey(key: KEY_TYPE): def QueryReflectionKey(key: KEY_TYPE): raise NotImplementedError("Not Implemented") + +# Non winreg functions +def LoadRegFile(file_name: str): str + with open(file_name, "r") as reg: + nextline = reg.readline() + + key = None + + while nextline: + line = nextline.strip() + nextline = reg.readline() + + if len(line) == 1: continue + split = line.split("=") + + keyline = strip_brackets(line) + if keyline: + key = keyline + elif key and len(split) == 2: + name, value = split + name = strip_quotes(name) + value = strip_quotes(value) + + os.makedirs(key, exist_ok=True) + + with open(os.path.join(_CONFIG_DIR, key.key, name), "w") as regvalue: + regvalue.write(value) + + print(f"[{key}] {name}={value}")
\ No newline at end of file diff --git a/unixreg/utils.py b/unixreg/utils.py new file mode 100644 index 0000000..a5dcf45 --- /dev/null +++ b/unixreg/utils.py @@ -0,0 +1,13 @@ + +def strict_types(function): + def _decorator(*args, **kwargs): + hints = function.__annotations__ + all_args = kwargs.copy() + all_args.update(dict(zip(function.__code__.co_varnames, args))) + for argument, argument_type in [(i, type(j)) for i, j in all_args.items()]: + if argument in hints: + if argument_type != hints[argument]: + raise TypeError('{} is not {}'.format(argument, hints[argument].__name__)) + result = function(*args, **kwargs) + return result + return _decorator |