from well_routing.classes import Node, GraphNode, Edge

from tagmapper import Model


asset = "SNA"
model_name = "8002 - Rate distribution"
model_name = Model.get_model_names("IOC")

for mod_type in model_name:
    print(mod_type)
    object_names = Model.get_object_names(
        model_owner=mod_type["model_owner"], model_name=mod_type["model_name"]
    )
    print(object_names)
print(model_name)


models_to_delete = [
    "8001 - THEORETICAL FLOW RATES",
    "8001 THEORETICAL FLOW RATES",
    "8002 RATE DISTRIBUTION",
    "DRY GAS SEALS (FROM TEMPLATE)",
]
