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

[core] Formatting and type hint fixes

Browse source
This commit is contained in:
derrod 2021-09-01 15:29:25 +02:00
parent a80244ed37
commit f78674b3ed
1 changed files with 9 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
legendary/core.py
View file

@ -221,7 +221,7 @@ class LegendaryCore:
def get_game_and_dlc_list(self, update_assets=True, def get_game_and_dlc_list(self, update_assets=True,
platform_override=None, platform_override=None,
skip_ue=True) -> (List[Game], Dict[str, Game]): skip_ue=True) -> (List[Game], Dict[str, List[Game]]):
_ret = [] _ret = []
_dlc = defaultdict(list) _dlc = defaultdict(list)
@ -461,7 +461,7 @@ class LegendaryCore:
'{appdata}': os.path.expandvars('%APPDATA%'), '{appdata}': os.path.expandvars('%APPDATA%'),
'{userdir}': os.path.expandvars('%userprofile%/documents'), '{userdir}': os.path.expandvars('%userprofile%/documents'),
# '{userprofile}': os.path.expandvars('%userprofile%'), # possibly wrong # '{userprofile}': os.path.expandvars('%userprofile%'), # possibly wrong
'{usersavedgames}': os.path.expandvars('%userprofile%/Saved Games') '{usersavedgames}': os.path.expandvars('%userprofile%/Saved Games')
}) })
else: else:
# attempt to get WINE prefix from config # attempt to get WINE prefix from config
@ -619,7 +619,7 @@ class LegendaryCore:
self.log.debug(f'Writing "{fpath}"...') self.log.debug(f'Writing "{fpath}"...')
with open(fpath, 'wb') as fh: with open(fpath, 'wb') as fh:
for cp in fm.chunk_parts: for cp in fm.chunk_parts:
fh.write(chunks[cp.guid_num][cp.offset:cp.offset+cp.size]) fh.write(chunks[cp.guid_num][cp.offset:cp.offset + cp.size])
# set modified time to savegame creation timestamp # set modified time to savegame creation timestamp
m_date = datetime.strptime(f_parts[4], '%Y.%m.%d-%H.%M.%S.manifest') m_date = datetime.strptime(f_parts[4], '%Y.%m.%d-%H.%M.%S.manifest')
@ -807,10 +807,10 @@ class LegendaryCore:
else: else:
if not game_folder: if not game_folder:
if game.is_dlc: if game.is_dlc:
game_folder = base_game.metadata.get('customAttributes', {}).\ game_folder = base_game.metadata.get('customAttributes', {}). \
get('FolderName', {}).get('value', base_game.app_name) get('FolderName', {}).get('value', base_game.app_name)
else: else:
game_folder = game.metadata.get('customAttributes', {}).\ game_folder = game.metadata.get('customAttributes', {}). \
get('FolderName', {}).get('value', game.app_name) get('FolderName', {}).get('value', game.app_name)
if not base_path: if not base_path:
@ -1058,8 +1058,8 @@ class LegendaryCore:
# If there's no in-progress installation assume the game doesn't need to be verified # If there's no in-progress installation assume the game doesn't need to be verified
if mf and not os.path.exists(os.path.join(app_path, '.egstore', 'bps')): if mf and not os.path.exists(os.path.join(app_path, '.egstore', 'bps')):
needs_verify = False needs_verify = False
if os.path.exists(os.path.join(app_path, '.egstore', 'Pending')): if os.path.exists(os.path.join(app_path, '.egstore', 'Pending')):
if os.listdir(os.path.join(app_path, '.egstore', 'Pending')): if os.listdir(os.path.join(app_path, '.egstore', 'Pending')):
needs_verify = True needs_verify = True
if not needs_verify: if not needs_verify:
@ -1183,13 +1183,13 @@ class LegendaryCore:
os.makedirs(egstore_folder) os.makedirs(egstore_folder)
# copy manifest and create mancpn file in .egstore folder # copy manifest and create mancpn file in .egstore folder
with open(os.path.join(egstore_folder, f'{egl_game.installation_guid}.manifest',), 'wb') as mf: with open(os.path.join(egstore_folder, f'{egl_game.installation_guid}.manifest', ), 'wb') as mf:
mf.write(manifest_data) mf.write(manifest_data)
mancpn = dict(FormatVersion=0, AppName=app_name, mancpn = dict(FormatVersion=0, AppName=app_name,
CatalogItemId=lgd_game.asset_info.catalog_item_id, CatalogItemId=lgd_game.asset_info.catalog_item_id,
CatalogNamespace=lgd_game.asset_info.namespace) CatalogNamespace=lgd_game.asset_info.namespace)
with open(os.path.join(egstore_folder, f'{egl_game.installation_guid}.mancpn',), 'w') as mcpnf: with open(os.path.join(egstore_folder, f'{egl_game.installation_guid}.mancpn', ), 'w') as mcpnf:
json.dump(mancpn, mcpnf, indent=4, sort_keys=True) json.dump(mancpn, mcpnf, indent=4, sort_keys=True)
# And finally, write the file for EGL # And finally, write the file for EGL