/
niswitch_connect_channels.py
46 lines (30 loc) · 1.64 KB
/
niswitch_connect_channels.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
#!/usr/bin/python
import argparse
import niswitch
import sys
def example(resource_name, channel1, channel2, topology, simulate):
# if we are simulating resource name must be blank
resource_name = '' if simulate else resource_name
with niswitch.Session(resource_name=resource_name, topology=topology, simulate=simulate) as session:
session.connect(channel1=channel1, channel2=channel2)
print('Channel ', channel1, ' and ', channel2, ' are now connected.')
session.disconnect(channel1=channel1, channel2=channel2)
print('Channel ', channel1, ' and ', channel2, ' are now disconnected.')
def _main(argsv):
parser = argparse.ArgumentParser(description='Performs a connection with NI-SWITCH Channels.', formatter_class=argparse.ArgumentDefaultsHelpFormatter)
parser.add_argument('-n', '--resource-name', default='PXI1Slot2', help='Resource name of an NI switch.')
parser.add_argument('-ch1', '--channel1', default='c0', help='Channel One.')
parser.add_argument('-ch2', '--channel2', default='r0', help='Channel Two.')
parser.add_argument('-t', '--topology', default='Configured Topology', help='Topology.')
parser.add_argument('-s', '--simulate', default=False, action='store_true', help='Simulate device.')
args = parser.parse_args(argsv)
example(args.resource_name, args.channel1, args.channel2, args.topology, args.simulate)
def test_example():
example('', 'c0', 'r0', '2737/2-Wire 4x64 Matrix', True)
def test_main():
cmd_line = ['--topology', '2737/2-Wire 4x64 Matrix', '--simulate']
_main(cmd_line)
def main():
_main(sys.argv[1:])
if __name__ == '__main__':
main()