srenblad
/
artiq
forked from M-Labs/artiq
1
0
Fork 0
Code Pull Requests Activity

thorlabs_tcube: fix -P case handling

This commit is contained in:
Sebastien Bourdeauducq 2015-11-03 17:54:27 +08:00
parent 73fed53c09
commit 644a410c90
1 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
artiq/frontend/thorlabs_tcube_controller.py
View File

@ -11,8 +11,8 @@ from artiq.tools import verbosity_args, simple_network_args, init_logger
def get_argparser(): def get_argparser():
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument("-P", "--product", required=True, parser.add_argument("-P", "--product", required=True,
help="type of the Thorlabs T-Cube device to control", help="type of the Thorlabs T-Cube device to control: "
choices=["tdc001", "tpz001"]) "tdc001/tpz001")
parser.add_argument("-d", "--device", default=None, parser.add_argument("-d", "--device", default=None,
help="serial device. See documentation for how to " help="serial device. See documentation for how to "
"specify a USB Serial Number.") "specify a USB Serial Number.")
@ -39,11 +39,19 @@ def main():
dev = TdcSim() dev = TdcSim()
elif product == "tpz001": elif product == "tpz001":
dev = TpzSim() dev = TpzSim()
else:
print("Invalid product string (-P/--product), "
"choose from tdc001 or tpz001")
sys.exit(1)
else: else:
if product == "tdc001": if product == "tdc001":
dev = Tdc(args.device) dev = Tdc(args.device)
elif product == "tpz001": elif product == "tpz001":
dev = Tpz(args.device) dev = Tpz(args.device)
else:
print("Invalid product string (-P/--product), "
"choose from tdc001 or tpz001")
sys.exit(1)
try: try:
simple_server_loop({product: dev}, args.bind, args.port) simple_server_loop({product: dev}, args.bind, args.port)