Proper sourcemod search on Linux

README.md

@@ -1,7 +1,10 @@
 # TF2CDownloader
+Self-developed improvements for TF2 Classic downloader
 Simple script for downloading and extracting TF2Classic quickly, simply, and efficiently.
 
+
 Won't function right on Windows unless it's compiled by PyInstaller, as it's hardcoded to use a variable only supplied to it *by* PyInstaller in order to find the Binaries folder.
+https://github.com/tf2classic/TF2CDownloader
 
 In the Binaries folder of the repository, Aria2 and its relevant dependencies are from MSYS2: https://packages.msys2.org/package/mingw-w64-x86_64-aria2
 

setup.py

@@ -27,16 +27,14 @@ def sourcemods_path():
 			return None
 	else:
 		try:
-			linux_dirs = {
-				'~/.steam/steam/steamapps/sourcemods/',
-				# these locations are mentioned in old posts on forums, so it's better to check them too
-				'~/.local/share/Steam/SteamApps/sourcemods',
-				'~/Steam/SteamApps/sourcemods'
-			}
-			for dir in linux_dirs:
-				if os_path.isdir(dir):
-					return dir
-			return None
+			sourcepath = None
+			with open(os_path.expanduser('~/.steam/registry.vdf'), 'r') as file:
+				for _, line in enumerate(file):
+					if 'SourceModInstallPath' in line:
+						sourcepath = line[line.index('/home'):-1].replace(r'\\', '/')
+						break
+				file.close()
+			return sourcepath
 		except Exception:
 			return None
 
@@ -117,4 +115,4 @@ def setup_binaries():
 		elif which('7zz') is not None:
 			vars.SEVENZ_BINARY = '7zz'
 		else:
-			vars.SEVENZ_BINARY = '7z'
+			vars.SEVENZ_BINARY = '7z'