66 lines
1.7 KiB
Python
66 lines
1.7 KiB
Python
|
#!/usr/bin/python3
|
||
|
|
|
||
|
|
import argparse
|
||
|
|
import logging
|
||
|
|
import sys
|
||
|
|
from xmlrpc.client import ServerProxy
|
||
|
|
|
||
|
|
DEFAULT_PORT = "8231"
|
||
|
|
|
||
|
|
logging.basicConfig(level=logging.INFO, format='%(levelname)s %(message)s')
|
||
|
|
LOG = logging.getLogger("att_ctrl_client")
|
||
|
|
|
||
|
|
def parse_args():
|
||
|
|
"""Command line argument parser"""
|
||
|
|
parser = argparse.ArgumentParser()
|
||
|
|
parser.add_argument("host", help="Hostname or ip address the server is \
|
||
|
|
serving on. (e.g. 'attctrl:8231 or 192.168.1.16'")
|
||
|
|
parser.add_argument('attenuator', help="Attenuator to be set [0..7]")
|
||
|
|
parser.add_argument('attenuation', help="Attenuation to be set [0.0 .. 31.5]")
|
||
|
|
|
||
|
|
return parser.parse_args()
|
||
|
|
|
||
|
|
def host_port(data):
|
||
|
|
'''Determine hostname/address and port.
|
||
|
|
|
||
|
|
Args:
|
||
|
|
data (str): Hostname/address with optional port (e.g. 'localhost:1234'
|
||
|
|
or '192.168.0.1').
|
||
|
|
|
||
|
|
Returns:
|
||
|
|
str: Hostname/address
|
||
|
|
str: Port
|
||
|
|
'''
|
||
|
|
tmp = data.split(":")
|
||
|
|
port = DEFAULT_PORT
|
||
|
|
host = tmp[0]
|
||
|
|
|
||
|
|
if len(tmp) > 1:
|
||
|
|
port = tmp[1]
|
||
|
|
return host, port
|
||
|
|
|
||
|
|
|
||
|
|
def main():
|
||
|
|
args = parse_args()
|
||
|
|
host, port = host_port(args.host)
|
||
|
|
client = ServerProxy('http://{}:{}'.format(host, port))
|
||
|
|
|
||
|
|
try:
|
||
|
|
dev = int(args.attenuator)
|
||
|
|
except ValueError:
|
||
|
|
LOG.error("attenuator needs to be convertible into int.")
|
||
|
|
return 1
|
||
|
|
try:
|
||
|
|
val = float(args.attenuation)
|
||
|
|
except ValueError:
|
||
|
|
LOG.error("attenuation needs to be convertible into float.")
|
||
|
|
return 1
|
||
|
|
|
||
|
|
ret = client.set_attenuation(dev, val)
|
||
|
|
|
||
|
|
LOG.info("xmlrpcclient: set_attenuation result: {}.".format(ret))
|
||
|
|
return ret
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
sys.exit(main())
|