Power/cli.py at master · p404/Power · GitHub

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
79
80
81
# -*- coding: UTF-8 -*-
from core import Power
from config import PowerConfig
from cement.core.foundation import CementApp
from cement.ext.ext_argparse import ArgparseController, expose

class BaseController(ArgparseController):
    class Meta:
        label = 'base'
        description = "Power, Tool for managing power state of multiples machines using Wakeonlan/SSH"
        ignore_unknown_arguments = True

    def default(self):
        self.app.args.print_help()

class PowerController(ArgparseController):
    class Meta:
        label = 'power'
        stacked_on = 'base'
        stacked_type = 'embedded'

    @expose(arguments=[
                (['--hosts', '-hosts'], {'help': 'Turn on machines by hostname', 'action':'store', 'nargs': '*'}),
                (['--tag', '-t'], {'help': 'Turn on machines by a tag', 'action': 'store'})
            ],
            help="Turn on machines"
    )
    def on(self):
        servers_obj = PowerConfig()
        if self.app.pargs.hosts or self.app.pargs.tag:
            if self.app.pargs.hosts:
                self.app.log.info('Turning on machines')
                for host in self.app.pargs.hosts:
                    server = servers_obj.filter_by_host(host)
                    if bool(server) == False:
                        self.app.log.fatal("Servers {} not found".format(host))
                    else:
                        Power.turn_on(server)
            elif self.app.pargs.tag:
                servers = servers_obj.filter_by_tag(self.app.pargs.tag)
                if bool(servers) == False:
                    self.app.log.fatal("Servers with a tag {} not found".format(self.app.pargs.tag))
                else:
                    self.app.log.info('Turning on machines')
                    Power.turn_on(servers)
        else:
            self.app.log.info('Turning on all machines')
            Power.turn_on(servers_obj.SERVERS)

    @expose(arguments=[
                (['--hosts', '-hosts'], {'help': 'Turn off machines by hostname', 'action':'store', 'nargs': '*'}),
                (['--tag', '-t'], {'help': 'Turn off machines by tag', 'action': 'store'})
            ],
            help="Turn off machines"
    )
    def off(self):
        servers_obj = PowerConfig()
        if self.app.pargs.hosts or self.app.pargs.tag:
            if self.app.pargs.hosts:
                self.app.log.info('Turning off machines')
                for host in self.app.pargs.hosts:
                    server = servers_obj.filter_by_host(host)
                    if bool(server) == False:
                        self.app.log.fatal("Servers {} not found".format(host))
                    else:
                        Power.turn_off(server)
            elif self.app.pargs.tag:
                servers = servers_obj.filter_by_tag(self.app.pargs.tag)
                if bool(servers) == False:
                    self.app.log.fatal("Servers with a tag {} not found".format(self.app.pargs.tag))
                else:
                    self.app.log.info('Turning off machines')
                    Power.turn_off(servers)
        else:
            self.app.log.info('Turning off all machines')
            Power.turn_off(servers_obj.SERVERS)

class PowerCLI(CementApp):
    class Meta:
        label = 'power'
        handlers = [BaseController, PowerController]