CodeMirror/legendary
1
0
Fork 0
mirror of https://github.com/derrod/legendary.git synced 2024-05-04 20:53:33 +12:00
Code Issues Projects Releases Wiki Activity

[core/utils] Respect include/exclude filter when deleting save data

Browse source 
Not doing this would result in some titles that save their save data in
the install directory having their game data deleted...

Fixes #497
This commit is contained in:
derrod 2022-11-06 17:14:02 +01:00
parent 032b7fc64f
commit c3eb6b4fe6
2 changed files with 33 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
legendary/core.py
View file

@ -999,9 +999,22 @@ class LegendaryCore:
if not os.path.exists(_save_dir):
os.makedirs(_save_dir)
if clean_dir:
if app_name and clean_dir:
game = self.lgd.get_game_meta(app_name)
custom_attr = game.metadata['customAttributes']
include_f = exclude_f = None
# Make sure to only delete files that match the include/exclude filters.
# This is particularly import for games that store save games in their install dir...
if (_include := custom_attr.get('CloudIncludeList', {}).get('value', None)) is not None:
include_f = _include.split(',')
if (_exclude := custom_attr.get('CloudExcludeList', {}).get('value', None)) is not None:
exclude_f = _exclude.split(',')
sgh = SaveGameHelper()
save_files = sgh.get_deletion_list(_save_dir, include_f, exclude_f)
self.log.info('Deleting old save files...')
delete_folder(_save_dir)
delete_filelist(_save_dir, save_files, silent=True)
self.log.info(f'Downloading "{fname.split("/", 2)[2]}"...')
# download manifest

18
legendary/utils/savegame_helper.py
View file

@ -167,3 +167,21 @@ class SaveGameHelper:
# return dict with created files for uploading/whatever
return self.files
def get_deletion_list(self, save_folder, include_filter=None, exclude_filter=None):
files = []
for _dir, _, _files in os.walk(save_folder):
for _file in _files:
_file_path = os.path.join(_dir, _file)
_file_path_rel = os.path.relpath(_file_path, save_folder).replace('\\', '/')
if include_filter and not _filename_matches(_file_path_rel, include_filter):
self.log.debug(f'Excluding "{_file_path_rel}" (does not match include filter)')
continue
elif exclude_filter and _filename_matches(_file_path_rel, exclude_filter):
self.log.debug(f'Excluding "{_file_path_rel}" (does match exclude filter)')
continue
files.append(_file_path_rel)
return files