Source code for labgrid.driver.power.tinycontrol_tcpdu

"""A driver to control the Tinycontrol tcPDU
Reference: https://docs.tinycontrol.pl/en/tcpdu/api/commands

Example configuration to use port #3 on a device with URL 'http://172.17.180.53/'

NetworkPowerPort:
  model: tinycontrol_tcpdu
  host: 'http://172.17.180.53'
  index: 3
"""

from urllib.parse import urljoin

import requests




def power_set(host, port, index, value):
    assert port is None

    index = int(index)
    value = 1 if value else 0
    r = requests.get(urljoin(host, f"/api/v1/save/?out{index}={value}"))
    r.raise_for_status()





def power_get(host, port, index):
    assert port is None

    index = int(index)
    r = requests.get(urljoin(host, "/api/v1/read/status/?outValues"))
    r.raise_for_status()
    json_decoded = r.json()
    return json_decoded[f"out{index}"] == 1