Skip to content

Commit 7603599

Browse files
[IMP] upgrade_analysis: handle merge models
1 parent 5ff8742 commit 7603599

File tree

1 file changed

+14
-1
lines changed

1 file changed

+14
-1
lines changed

upgrade_analysis/compare.py

Lines changed: 14 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,8 +37,12 @@ def model_rename_map(model):
3737
return apriori.renamed_models.get(model, model)
3838

3939

40+
def model_merge_map(model):
41+
return apriori.merged_models.get(model, model)
42+
43+
4044
def model_map(model):
41-
return apriori.renamed_models.get(model, apriori.merged_models.get(model, model))
45+
return apriori.renamed_models.get(model, model_merge_map(model))
4246

4347

4448
def inv_model_map(model):
@@ -486,6 +490,15 @@ def compare_model_sets(old_records, new_records):
486490
f"obsolete model {model} "
487491
f"[module {module_map(column['module'])}]"
488492
)
493+
elif model_merge_map(model) in new_models:
494+
text = f"obsolete model {model}" f" (merged to {model_map(model)})"
495+
if column["model_type"]:
496+
text += f" [{column['model_type']}]"
497+
reprs[module_map(column["module"])].append(text)
498+
reprs["general"].append(
499+
f"obsolete model {model} (merged to {model_map(model)}) "
500+
f"[module {module_map(column['module'])}]"
501+
)
489502
else:
490503
moved_module = ""
491504
if module_map(column["module"]) != new_models[model_map(model)]:

0 commit comments

Comments
 (0)