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