sped up the function generic_visit that now skips unnecessary fields
added mentioning of skipped class names since they could not be found
This commit is contained in:
parent
a6356c2821
commit
82218d01f4
@ -25,11 +25,12 @@ def start_strategy_update(args: Dict[str, Any]) -> None:
|
||||
config, enum_failed=True, recursive=config.get('recursive_strategy_search', False))
|
||||
|
||||
filtered_strategy_objs = []
|
||||
for args_strategy in args['strategy_list']:
|
||||
for strategy_obj in strategy_objs:
|
||||
for args_strategy in args['strategy_list']:
|
||||
if strategy_obj['name'] == args_strategy and strategy_obj not in filtered_strategy_objs:
|
||||
filtered_strategy_objs.append(strategy_obj)
|
||||
break
|
||||
print(f"strategy {strategy_obj['name']} could not be loaded or found and is skipped.")
|
||||
|
||||
for filtered_strategy_obj in filtered_strategy_objs:
|
||||
# Initialize backtesting object
|
||||
|
@ -52,6 +52,7 @@ class StrategyUpdater:
|
||||
"""
|
||||
|
||||
source_file = strategy_obj['location']
|
||||
print(f"started conversion of {source_file}")
|
||||
strategies_backup_folder = Path.joinpath(config['user_data_dir'], "strategies_orig_updater")
|
||||
target_file = Path.joinpath(strategies_backup_folder, strategy_obj['location_rel'])
|
||||
|
||||
@ -106,6 +107,8 @@ class NameUpdater(ast.NodeTransformer):
|
||||
# traverse the AST recursively by calling the visitor method for each child node
|
||||
if hasattr(node, "_fields"):
|
||||
for field_name, field_value in ast.iter_fields(node):
|
||||
if not isinstance(field_value, ast.AST):
|
||||
continue # to avoid unnecessary loops
|
||||
self.visit(field_value)
|
||||
self.generic_visit(field_value)
|
||||
self.check_fields(field_value)
|
||||
@ -204,6 +207,7 @@ class NameUpdater(ast.NodeTransformer):
|
||||
node.slice.value = StrategyUpdater.rename_dict[node.slice.value]
|
||||
if hasattr(node.slice, "elts"):
|
||||
self.visit_slice_elts(node.slice.elts)
|
||||
if hasattr(node.slice, "value"):
|
||||
if hasattr(node.slice.value, "elts"):
|
||||
self.visit_slice_elts(node.slice.value.elts)
|
||||
return node
|
||||
|
Loading…
Reference in New Issue
Block a user