(MINOR) Fallback to C drive if not found in environ

Garulf · Sep 5, 2021 · 05b7151 · 05b7151
1 parent 6f2821d
commit 05b7151
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")