Avoid pylint errors under Linux/macOS

This commit is contained in:
2Shirt 2021-04-15 20:04:33 -06:00
parent 759cd12379
commit ed6f188eb2
Signed by: 2Shirt
GPG key ID: 152FAC923B0E132C

View file

@ -5,10 +5,15 @@ import logging
import os import os
import pathlib import pathlib
import platform import platform
import winreg
from contextlib import suppress from contextlib import suppress
try:
import winreg
except ImportError as err:
if platform.system() == 'Windows':
raise err
from wk.borrowed import acpi from wk.borrowed import acpi
from wk.exe import run_program from wk.exe import run_program
from wk.io import non_clobber_path from wk.io import non_clobber_path
@ -126,7 +131,7 @@ def reg_delete_key(hive, key, recurse=False):
# Delete subkeys first # Delete subkeys first
if recurse: if recurse:
with suppress(WindowsError), winreg.OpenKey(hive, key) as open_key: with suppress(OSError), winreg.OpenKey(hive, key) as open_key:
while True: while True:
subkey = fr'{key}\{winreg.EnumKey(open_key, 0)}' subkey = fr'{key}\{winreg.EnumKey(open_key, 0)}'
reg_delete_key(hive, subkey, recurse=recurse) reg_delete_key(hive, subkey, recurse=recurse)