artiq/artiq/devices/lda/lda-client.py

#!/usr/bin/env python3
import argparse

from artiq.management.pc_rpc import Client


if __name__ == "__main__":

    parser = argparse.ArgumentParser()
    parser.add_argument('-s', '--server', default="::1",
                        help="The IP address or hostname of the controller")
    parser.add_argument('-p', '--port', default=8890, type=int,
                        help="The TCP port the controller listens to")
    parser.add_argument('-a', '--attenuation', type=float,
                        help="The attenuation value you want to set")
    args = parser.parse_args()

    remote = Client(args.server, args.port, "lda")

    try:
        if args.attenuation is None:
            print("Current attenuation: {}".format(remote.get_attenuation()))
        else:
            remote.set_attenuation(args.attenuation)
    finally:
        remote.close_rpc()