Fixes #3: Proton in a non-default library path

simons-public · Oct 9, 2018 · ff766fe · ff766fe
1 parent dd679ec
commit ff766fe
Showing 1 changed file with 11 additions and 4 deletions.
diff --git a/protonfixes/util.py b/protonfixes/util.py
@@ -25,10 +25,17 @@ def which(appname):
 def protondir():
     """ Returns the path to proton
     """
-
-    return glob.glob(os.path.join(
-        os.environ['STEAM_COMPAT_CLIENT_INSTALL_PATH'],
-        'steamapps/common/Proton*'))[0]
+
+    try:
+        proton_dir = glob.glob(os.path.join(
+            os.environ['STEAM_COMPAT_CLIENT_INSTALL_PATH'],
+            'steamapps/common/Proton*'))[0]
+        return proton_dir
+    except IndexError:
+        base = '/'.join(env['STEAM_COMPAT_DATA_PATH'].split('/')[:-2])
+        proton_dir = glob.glob(os.path.join(
+            base, 'common/Proton*'))[0]
+        return proton_dir
 
 
 def protonprefix():