fix #111, chrome not installed bug

dexplo · Oct 9, 2024 · 2aec126 · 2aec126
1 parent 77cbc07
commit 2aec126
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/dataframe_image/converter/browser/chrome_converter.py b/dataframe_image/converter/browser/chrome_converter.py
@@ -67,8 +67,11 @@ def get_chrome_path(chrome_path=None):
             r"SOFTWARE\Microsoft\Windows\CurrentVersion\App Paths\brave.exe",
         ]
         for loc in locs:
-            handle = winreg.OpenKey(winreg.HKEY_LOCAL_MACHINE, loc)
-            num_values = winreg.QueryInfoKey(handle)[1]
+            try:
+                handle = winreg.OpenKey(winreg.HKEY_LOCAL_MACHINE, loc)
+                num_values = winreg.QueryInfoKey(handle)[1]
+            except FileNotFoundError:
+                num_values = 0
             if num_values > 0:
                 return winreg.EnumValue(handle, 0)[1]
         raise OSError("Cannot find chrome.exe on your windows machine")