Commit d9816fd0 authored by Lluis Gifre Renom's avatar Lluis Gifre Renom
Browse files

Context component: 

- Add support for loading optical links in topology queries
parent 31551b44

src/context/service/database/Topology.py

+2 −0
Original line number Diff line number Diff line
@@ -51,6 +51,7 @@ def topology_list_objs(db_engine : Engine, request : ContextId) -> TopologyList: 
        obj_list : List[TopologyModel] = session.query(TopologyModel)\

            .options(selectinload(TopologyModel.topology_devices))\

            .options(selectinload(TopologyModel.topology_links))\

            .options(selectinload(TopologyModel.topology_optical_links))\

            .filter_by(context_uuid=context_uuid).all()

        return [obj.dump() for obj in obj_list]

    topologies = run_transaction(sessionmaker(bind=db_engine), callback)
@@ -62,6 +63,7 @@ def topology_get(db_engine : Engine, request : TopologyId) -> Topology: 
        obj : Optional[TopologyModel] = session.query(TopologyModel)\

            .options(selectinload(TopologyModel.topology_devices))\

            .options(selectinload(TopologyModel.topology_links))\

            .options(selectinload(TopologyModel.topology_optical_links))\

            .filter_by(topology_uuid=topology_uuid).one_or_none()

        return None if obj is None else obj.dump()

    obj = run_transaction(sessionmaker(bind=db_engine), callback)