Merge pull request #8 from Garulf/fix-fallback-system-drive

(MINOR) Fallback to C drive if not found in environ
Garulf · Sep 5, 2021 · 5ccf0b5 · 5ccf0b5
2 parents 6f2821d + 05b7151
commit 5ccf0b5
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/plugin/steam_search.py b/plugin/steam_search.py
@@ -8,8 +8,10 @@
 import vdf
 from flox import Flox
 
+DEFAULT_DRIVE = "C:"
+SYSTEM_DRIVE = os.environ.get("SYSTEMDRIVE", DEFAULT_DRIVE)
 STEAM_FOLDER = os.path.join(
-    f"{os.environ['SYSTEMDRIVE']}\\", "Program Files (x86)", "Steam"
+    f"{SYSTEM_DRIVE}\\", "Program Files (x86)", "Steam"
 )
 LIBRARIES_CONFIG = os.path.join(STEAM_FOLDER, "config", "libraryfolders.vdf")
 LIBRARIES_STEAMAPPS = os.path.join(STEAM_FOLDER, "steamapps", "libraryfolders.vdf")