Commit 47eb992a authored by Lluis Gifre Renom's avatar Lluis Gifre Renom
Browse files

Context component: 

- updated device_set and link_set to support placeholder creation
parent ae294326

src/context/service/database/Device.py

+18 −16
Original line number Diff line number Diff line
@@ -136,6 +136,7 @@ def device_set(db_engine : Engine, request : Device) -> Tuple[Dict, bool]: 
        created_at,updated_at = session.execute(stmt).fetchone()

        updated = updated_at > created_at



        if len(endpoints_data) > 0:

            stmt = insert(EndPointModel).values(endpoints_data)

            stmt = stmt.on_conflict_do_update(

                index_elements=[EndPointModel.endpoint_uuid],
@@ -150,6 +151,7 @@ def device_set(db_engine : Engine, request : Device) -> Tuple[Dict, bool]: 
            endpoint_updates = session.execute(stmt).fetchall()

            updated = updated or any([(updated_at > created_at) for created_at,updated_at in endpoint_updates])



        if len(related_topologies) > 0:

            session.execute(insert(TopologyDeviceModel).values(related_topologies).on_conflict_do_nothing(

                index_elements=[TopologyDeviceModel.topology_uuid, TopologyDeviceModel.device_uuid]

            ))

src/context/service/database/Link.py

+11 −9
Original line number Diff line number Diff line
@@ -100,6 +100,7 @@ def link_set(db_engine : Engine, request : Link) -> Tuple[Dict, bool]: 
        created_at,updated_at = session.execute(stmt).fetchone()

        updated = updated_at > created_at



        if len(link_endpoints_data) > 0:

            # TODO: manage add/remove of endpoints; manage changes in relations with topology

            stmt = insert(LinkEndPointModel).values(link_endpoints_data)

            stmt = stmt.on_conflict_do_nothing(
@@ -107,6 +108,7 @@ def link_set(db_engine : Engine, request : Link) -> Tuple[Dict, bool]: 
            )

            session.execute(stmt)



        if len(related_topologies) > 0:

            session.execute(insert(TopologyLinkModel).values(related_topologies).on_conflict_do_nothing(

                index_elements=[TopologyLinkModel.topology_uuid, TopologyLinkModel.link_uuid]

            ))