installer fix Linux Documents folder

Die4Ever · Oct 10, 2023 · 291403f · 291403f
1 parent 3cb103b
commit 291403f
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 5 deletions.
diff --git a/installer/Install/__init__.py b/installer/Install/__init__.py
@@ -103,14 +103,19 @@ def GetSourcePath() -> Path:
     raise RuntimeError('failed to GetSourcePath()', p)
 
 
+def GetSteamPlayDocuments(system:Path) -> Path:
+    if 'Steam' in system.parts:
+        idx = system.parts.index('Steam')
+        p = system.parents[-idx -1] # parents array is backwards
+        info('GetSteamPlayDocuments() == ', p)
+        p = p /'steamapps'/'compatdata'/'6910'/'pfx'/'drive_c'/'users'/'steamuser'/'Documents'
+        info('GetSteamPlayDocuments() == ', p)
+        return p
+
 def GetDocumentsDir(system:Path) -> Path:
     if not IsWindows():
         if 'Steam' in system.parts:
-            idx = system.parts.index('Steam')
-            p = system.parents[idx]
-            info('GetDocumentsDir() == ', p)
-            p = p /'steamapps'/'compatdata'/'6910'/'pfx'/'drive_c'/'users'/'steamuser'/'Documents'
-            info('GetDocumentsDir() == ', p)
+            p = GetSteamPlayDocuments(system)
             Mkdir(p, True, True)
         else:
             p = Path.home()

diff --git a/installer/tests.py b/installer/tests.py
@@ -18,6 +18,10 @@
 @typechecked
 class DXRTestCase(unittest.TestCase):
     def test_documents(self):
+        system = Path('/home/deck/.local/share/Steam/steamapps/common/Deus Ex/System/')
+        d = Install.GetSteamPlayDocuments(system)
+        self.assertEqual(d, Path('/home/deck/.local/share/Steam/steamapps/compatdata/6910/pfx/drive_c/users/steamuser/Documents/'), 'GetSteamPlayDocuments')
+
         d = Install.GetDocumentsDir(Path.home())
         self.assertTrue(d.exists(), str(d) + ' exists')