simclient.py

"""
Client to drive the JSON api implemented in driver.py
"""
import requests
import simapi

class Client:
    def __init__(self, url: str) -> None:
        self.url = url

    def set_link_state(self, node1: str, node2: str, up: bool) -> None:
        try:
            print(f"send link state {node1}, {node2}, state up {up}")
            data = simapi.Link(node1_name=node1, node2_name=node2, up="true" if up else "false")
            url = f"{self.url}/link"
            r = requests.put(url, 
                    json=data.model_dump())
            print(r.text)
        except requests.exceptions.ConnectionError as e:
            print(e)
            pass

    def set_uplinks(self, ground_node: str, links: list[tuple[str, int]]) -> None:
        try:
            print(f"send up links: {ground_node}")
            data = simapi.UpLinks(ground_node=ground_node, uplinks=[])
            for link in links:
                data.uplinks.append(simapi.UpLink(sat_node=link[0], distance=link[1]))
            url = f"{self.url}/uplinks"
            print(data.model_dump())
            r = requests.put(url, json=data.model_dump())
            print(r.text)
        except requests.exceptions.ConnectionError as e:
            print(e)