Source code for acq400_set_shot

#!/usr/bin/env python

"""
usage: acq400_abort.py [-h] uut [uut ...]

configure acq400_abort

positional arguments:
  uut         uut

optional arguments:
  -h, --help  show this help message and exit
"""

import acq400_hapi
import os
import argparse


def run_commands(args):
    uuts = [acq400_hapi.Acq400(u) for u in args.uuts]
    for uut in uuts:
        uut.s1.shot = args.shot

def run_main():
    parser = argparse.ArgumentParser(description='set shot number of all uuts')
    parser.add_argument('--shot', type=int, default=0)
    parser.add_argument('uuts', nargs='+', help='uut1 [uut2..]')
    run_commands(parser.parse_args())

# execution starts here
if __name__ == '__main__':
    run_main()