mirror of
https://github.com/lutris/lutris
synced 2024-10-04 14:59:37 +00:00
Prefix scanning implementation
This commit is contained in:
parent
03581a121b
commit
40cd119aea
|
@ -1,6 +1,78 @@
|
||||||
import os
|
import os
|
||||||
import shutil
|
import shutil
|
||||||
import sys
|
import sys
|
||||||
|
from copy import copy
|
||||||
|
|
||||||
|
from lutris.util import magic
|
||||||
|
|
||||||
|
PROGRAM_FILES_IGNORES = {
|
||||||
|
"Common Files": {
|
||||||
|
"Microsoft Shared": "*",
|
||||||
|
"System": "*",
|
||||||
|
"InstallShield": "*"
|
||||||
|
},
|
||||||
|
"Internet Explorer": "*",
|
||||||
|
"Windows Media Player": "*",
|
||||||
|
"Windows NT": "*",
|
||||||
|
"InstallShield Installation Information": "*",
|
||||||
|
"Microsoft XNA": "*",
|
||||||
|
"Microsoft.NET": "*",
|
||||||
|
}
|
||||||
|
|
||||||
|
IGNORED_DIRS = {
|
||||||
|
"ProgramData": {
|
||||||
|
"Microsoft": { "Windows": "*" },
|
||||||
|
"GOG.com": "*",
|
||||||
|
"Package Cache": "*",
|
||||||
|
},
|
||||||
|
"Program Files": PROGRAM_FILES_IGNORES,
|
||||||
|
"Program Files (x86)": PROGRAM_FILES_IGNORES,
|
||||||
|
"windows": "*",
|
||||||
|
"users": {
|
||||||
|
"Public": "*",
|
||||||
|
os.getenv("USER"): {
|
||||||
|
"Desktop": "*",
|
||||||
|
"Videos": "*",
|
||||||
|
"Temp": "*",
|
||||||
|
"Cookies": "*",
|
||||||
|
"AppData": {
|
||||||
|
"LocalLow": "*",
|
||||||
|
"Local": {
|
||||||
|
"Microsoft": "*"
|
||||||
|
},
|
||||||
|
"Roaming": { "Microsoft": "*", "wine_gecko": "*" },
|
||||||
|
},
|
||||||
|
"Local Settings": {
|
||||||
|
"Application Data": { "Microsoft": "*" },
|
||||||
|
"History": "*",
|
||||||
|
"Temporary Internet Files": "*"
|
||||||
|
},
|
||||||
|
"Application Data": {
|
||||||
|
"Microsoft": "*",
|
||||||
|
"wine_gecko": "*"
|
||||||
|
},
|
||||||
|
"Start Menu": "*",
|
||||||
|
"PrintHood": "*",
|
||||||
|
"Favorites": "*",
|
||||||
|
"Recent": "*",
|
||||||
|
"Downloads": "*",
|
||||||
|
"Templates": "*",
|
||||||
|
"NetHood": "*",
|
||||||
|
"My Pictures": "*",
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
IGNORED_EXES = [
|
||||||
|
"UNWISE.EXE",
|
||||||
|
"unins000.exe",
|
||||||
|
"Uninstall.exe",
|
||||||
|
"UnSetup.exe",
|
||||||
|
"UE3Redist.exe",
|
||||||
|
"dotNetFx40_Full_setup.exe",
|
||||||
|
"sysinfo.exe",
|
||||||
|
"register.exe",
|
||||||
|
]
|
||||||
|
|
||||||
KNOWN_DIRS = [
|
KNOWN_DIRS = [
|
||||||
"ProgramData/Microsoft/Windows",
|
"ProgramData/Microsoft/Windows",
|
||||||
|
@ -33,9 +105,74 @@ def remove_empty_dirs(dirname):
|
||||||
return empty_folders
|
return empty_folders
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
def cleanup_prefix(path):
|
||||||
dirname = sys.argv[1]
|
delete_known_dirs(path)
|
||||||
delete_known_dirs(dirname)
|
|
||||||
empty_folders = True
|
empty_folders = True
|
||||||
while empty_folders:
|
while empty_folders:
|
||||||
empty_folders = remove_empty_dirs(dirname)
|
empty_folders = remove_empty_dirs(path)
|
||||||
|
|
||||||
|
|
||||||
|
def is_ignored_path(path_parts):
|
||||||
|
ignored_dirs = copy(IGNORED_DIRS)
|
||||||
|
if len(path_parts) in (0, 1):
|
||||||
|
return True
|
||||||
|
for level, part in enumerate(path_parts):
|
||||||
|
if level == 0:
|
||||||
|
if part == "dosdevices":
|
||||||
|
return True
|
||||||
|
if part in ignored_dirs:
|
||||||
|
if ignored_dirs[part] == "*":
|
||||||
|
return True
|
||||||
|
ignored_dirs = ignored_dirs[part]
|
||||||
|
return False
|
||||||
|
|
||||||
|
def get_content_folders(path):
|
||||||
|
found_dirs = []
|
||||||
|
for root, _dirs, files in os.walk(path, topdown=True):
|
||||||
|
# print(root, files, dirs)
|
||||||
|
relpath = root[len(path):].strip("/")
|
||||||
|
path_parts = relpath.split("/")
|
||||||
|
if is_ignored_path(path_parts):
|
||||||
|
continue
|
||||||
|
if files:
|
||||||
|
found_dirs.append(root)
|
||||||
|
folders = []
|
||||||
|
for dir in found_dirs:
|
||||||
|
skip = False
|
||||||
|
for _dir in folders:
|
||||||
|
if dir.startswith(_dir):
|
||||||
|
skip = True
|
||||||
|
if skip:
|
||||||
|
continue
|
||||||
|
folders.append(dir)
|
||||||
|
return folders
|
||||||
|
|
||||||
|
|
||||||
|
def find_exes_in_path(folder):
|
||||||
|
exes = []
|
||||||
|
for filename in os.listdir(folder):
|
||||||
|
abspath = os.path.join(folder, filename)
|
||||||
|
if os.path.isdir(abspath):
|
||||||
|
exes += find_exes_in_path(abspath)
|
||||||
|
if os.path.isfile(abspath):
|
||||||
|
if filename in IGNORED_EXES:
|
||||||
|
continue
|
||||||
|
if filename.lower().endswith(".exe"):
|
||||||
|
exes.append(os.path.join(folder, filename))
|
||||||
|
# if "executable" in file_type:
|
||||||
|
# print(filename, file_type)
|
||||||
|
return exes
|
||||||
|
|
||||||
|
|
||||||
|
def scan_prefix(path):
|
||||||
|
folders = get_content_folders(path)
|
||||||
|
exes = []
|
||||||
|
for folder in folders:
|
||||||
|
if "drive_c/users" in folder:
|
||||||
|
continue
|
||||||
|
exes += find_exes_in_path(folder)
|
||||||
|
for exe in exes:
|
||||||
|
print("EXE", exe)
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
scan_prefix(sys.argv[1])
|
||||||
|
|
Loading…
Reference in a new issue