Simplify start_strategy_update

This commit is contained in:
Matthias 2023-03-12 15:47:03 +01:00
parent cb086f79ff
commit d2a412d2c6
1 changed files with 13 additions and 17 deletions

View File

@ -29,31 +29,27 @@ def start_strategy_update(args: Dict[str, Any]) -> None:
config, enum_failed=False, recursive=config.get('recursive_strategy_search', False)) config, enum_failed=False, recursive=config.get('recursive_strategy_search', False))
filtered_strategy_objs = [] filtered_strategy_objs = []
if 'strategy_list' in args: if args['strategy_list']:
for args_strategy in args['strategy_list']: filtered_strategy_objs = [
for strategy_obj in strategy_objs: strategy_obj for strategy_obj in strategy_objs
if (strategy_obj['name'] == args_strategy if strategy_obj['name'] in args['strategy_list']
and strategy_obj not in filtered_strategy_objs): ]
filtered_strategy_objs.append(strategy_obj)
break
for filtered_strategy_obj in filtered_strategy_objs:
start_conversion(filtered_strategy_obj, config)
else: else:
# Use all available entries.
filtered_strategy_objs = strategy_objs
processed_locations = set() processed_locations = set()
for strategy_obj in strategy_objs: for strategy_obj in filtered_strategy_objs:
if strategy_obj['location'] not in processed_locations: if strategy_obj['location'] not in processed_locations:
processed_locations.add(strategy_obj['location']) processed_locations.add(strategy_obj['location'])
start_conversion(strategy_obj, config) start_conversion(strategy_obj, config)
def start_conversion(strategy_obj, config): def start_conversion(strategy_obj, config):
# try:
print(f"Conversion of {Path(strategy_obj['location']).name} started.") print(f"Conversion of {Path(strategy_obj['location']).name} started.")
instance_strategy_updater = StrategyUpdater() instance_strategy_updater = StrategyUpdater()
start = time.perf_counter() start = time.perf_counter()
instance_strategy_updater.start(config, strategy_obj) instance_strategy_updater.start(config, strategy_obj)
elapsed = time.perf_counter() - start elapsed = time.perf_counter() - start
print(f"Conversion of {Path(strategy_obj['location']).name} took {elapsed:.1f} seconds.") print(f"Conversion of {Path(strategy_obj['location']).name} took {elapsed:.1f} seconds.")
# except:
# pass