#!/usr/bin/env python3

import argparse
import logging

from artiq.devices.pdq2.driver import Pdq2
from artiq.management.pc_rpc import simple_server_loop


def _get_args():
    parser = argparse.ArgumentParser(description="PDQ2 controller")
    parser.add_argument("--bind", default="::1",
                        help="hostname or IP address to bind to")
    parser.add_argument("-p", "--port", default=8889, type=int,
                        help="TCP port to listen to")
    parser.add_argument(
        "-s", "--serial", default=None,
        help="device (FT245R) serial string [first]")
    parser.add_argument(
        "-d", "--debug", default=False, action="store_true",
        help="debug communications")
    return parser.parse_args()


def main():
    args = _get_args()

    if args.debug:
        logging.basicConfig(level=logging.DEBUG)
    else:
        logging.basicConfig(level=logging.WARNING)

    dev = Pdq2(serial=args.serial)
    try:
        simple_server_loop(dev, "pdq2", args.bind, args.port,
                           id_parameters="serial="+str(args.serial))
    finally:
        dev.close()

if __name__ == "__main__":
    main()