summaryrefslogtreecommitdiff
path: root/scripts/cxmux
blob: 938ad04165e6894606eb28568b09d37813c90ba4 (plain)
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
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
#!/usr/bin/env python


import os
import sys

from optparse import OptionParser

import cxmanage_api.fabric


def main():
    parser = OptionParser(
        "usage: %prog [options] COMMAND ecmeIP", conflict_handler="resolve"
    )
    parser.add_option(
        "-s", "--ssh",
        action="store_const", const=True, dest="ssh", default=False,
        help="Use the SPU IPs rather than ECME IPs"
    )
    parser.add_option(
        "-n", "--nosync",
        action="store_const", const=False, dest="sync", default=True,
        help="Do not syncronize input across terminals"
    )
    parser.add_option(
        "--virt-env",
        action="store", type="string", dest="virt_env",
        help="Calls workon <virtual_environment> before spawning a window"
    )
    parser.disable_interspersed_args()

    (options, args) = parser.parse_args()
    if len(args) == 0:
        parser.print_help()
        return -1
    elif len(args) < 2:
        parser.error("Need to specify COMMAND and ecmeIP")

    command = " ".join(args[:-1])

    if options.virt_env:
        command = 'workon %s; ' % options.virt_env + command

    ecmeip = args[-1]
    name = '%s@%s' % (args[0], ecmeip)
    fabric = cxmanage_api.fabric.Fabric(ecmeip)
    ips = [node.ip_address for node in fabric.nodes.values()]
    if options.ssh:
        ips = fabric.get_server_ip().values()

    for i, ip in enumerate(ips):
        if i == 0:
            os.system('tmux new-window -n "%s"' % name)
            os.system('tmux send-keys "%s %s"' % (command, ip))
            os.system('tmux send-keys Enter')
            continue

        os.system('tmux split-window -h')
        os.system('tmux send-keys "%s %s"' % (command, ip))
        os.system('tmux send-keys Enter')
        os.system('tmux select-layout -t "%s" tiled >/dev/null' % name)

    if options.sync:
        os.system(
            'tmux set-window-option -t "%s" synchronize-panes on >/dev/null' %
            name
        )


if __name__ == '__main__':
    sys.exit(main())