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
73
74
75
76
77
78
|
import argparse,sys
from clusterman.config import *
from clusterman.commands import node, plugins, frontend
def main():
parser = argparse.ArgumentParser()
subparsers = parser.add_subparsers(dest="target", help='Target')
##### Node commands #####
target_node = subparsers.add_parser("node")
node_subparsers=target_node.add_subparsers(dest="command", help='Command')
# Scan
node_cmd_scan=node_subparsers.add_parser("scan")
node_cmd_scan.add_argument("-t", "--timeout" ,help="Timeout", type=float)
# Check
node_cmd_scan=node_subparsers.add_parser("check")
node_cmd_scan.add_argument("-t", "--timeout" ,help="Timeout", type=float)
# List
node_cmd_list=node_subparsers.add_parser("list")
node_cmd_list.add_argument("-g", "--group" ,help="Group to list")
# Exec
node_cmd_list=node_subparsers.add_parser("exec")
node_cmd_list.add_argument("cmd",help="Command to run",nargs=argparse.REMAINDER)
##### Frontend commands #####
target_frontend = subparsers.add_parser("frontend")
frontend_subparsers=target_frontend.add_subparsers(dest="command", help='Command')
# Info
node_cmd_scan=frontend_subparsers.add_parser("info")
##### Plugins commands #####
target_plugins = subparsers.add_parser("plugins")
target_plugins.add_argument("name", help="Plugin's name")
target_plugins.add_argument("parameters", help="Plugin's parameters",nargs=argparse.REMAINDER)
# Check if command specified:
if len(sys.argv)==1:
parser.print_help(sys.stderr)
sys.exit(1)
# Parse arguments:
args = parser.parse_args()
# Run the proper handler
if args.target == "node":
if args.command == "scan":
if args.timeout:
node.scan(node_cmd_scan.timeout)
else:
node.scan(CONF["timeout"])
elif args.command == "check":
if args.timeout:
node.check(node_cmd_scan.timeout)
else:
node.check(CONF["timeout"])
elif args.command == "list":
if args.group:
node.ls(args.group)
else:
node.ls()
elif args.command == "exec":
node.exec(args.cmd)
else:
target_node.print_help(sys.stderr)
sys.exit(1)
elif args.target == "frontend":
if args.command == "info":
frontend.info()
else:
target_frontend.print_help(sys.stderr)
sys.exit(1)
elif args.target == "plugins":
plugins.execute(args.name,args.parameters)
|
