From ed6f188eb2a66c712d94203430aa2352d2ef9fba Mon Sep 17 00:00:00 2001
From: 2Shirt <2xShirt@gmail.com>
Date: Thu, 15 Apr 2021 20:04:33 -0600
Subject: [PATCH] Avoid pylint errors under Linux/macOS

---
 scripts/wk/os/win.py | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/scripts/wk/os/win.py b/scripts/wk/os/win.py
index 8f1cc52c..44e7efb4 100644
--- a/scripts/wk/os/win.py
+++ b/scripts/wk/os/win.py
@@ -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)