/mandos/trunk

To get this branch, use:
bzr branch http://bzr.recompile.se/loggerhead/mandos/trunk
24.1.116 by Björn Påhlsson
added a mandos list client program
1
#!/usr/bin/python
240 by Teddy Hogeborn
Merge "mandos-list" from belorn.
2
# -*- mode: python; coding: utf-8 -*-
24.1.116 by Björn Påhlsson
added a mandos list client program
3
4
import dbus
5
from optparse import OptionParser
240 by Teddy Hogeborn
Merge "mandos-list" from belorn.
6
import locale
7
8
locale.setlocale(locale.LC_ALL, u'')
24.1.116 by Björn Påhlsson
added a mandos list client program
9
10
tablewords = {
240 by Teddy Hogeborn
Merge "mandos-list" from belorn.
11
    'name': u'Name',
12
    'enabled': u'Enabled',
13
    'timeout': u'Timeout',
14
    'last_checked_ok': u'Last Successful Check',
15
    'created': u'Created',
16
    'interval': u'Interval',
17
    'host': u'Host',
18
    'fingerprint': u'Fingerprint',
19
    'checker_running': u'Check Is Running',
20
    'last_enabled': u'Last Enabled',
21
    'checker': u'Checker',
24.1.116 by Björn Påhlsson
added a mandos list client program
22
    }
23
busname = 'org.mandos-system.Mandos'
24.1.118 by Björn Påhlsson
Added enable/disable
24
server_path = '/Mandos'
25
server_interface = 'org.mandos_system.Mandos'
26
client_interface = 'org.mandos_system.Mandos.Client'
240 by Teddy Hogeborn
Merge "mandos-list" from belorn.
27
version = "1.0.2"
28
defaultkeywords = ('name', 'enabled', 'timeout', 'last_checked_ok',
29
                   'checker')
24.1.118 by Björn Påhlsson
Added enable/disable
30
bus = dbus.SystemBus()
31
mandos_dbus_objc = bus.get_object(busname, server_path)
32
mandos_serv = dbus.Interface(mandos_dbus_objc,
33
                             dbus_interface = server_interface)
34
mandos_clients = mandos_serv.GetAllClientsWithProperties()
35
36
def Getclientbyname(name):
37
    client_path = None
38
    for cli in mandos_clients.iteritems():
39
        if cli[1]['name'] == name:
40
            client_path = cli[0]
41
    if client_path is None:
42
        print "Client not found on server"
43
        return
44
45
    client_objc = bus.get_object(busname, client_path)
46
    return dbus.Interface(client_objc, dbus_interface = client_interface)
47
48
    
49
def Enable(options):
50
    client = Getclientbyname(options.enable)
51
    if client is not None:
52
        client.Enable()
53
54
def Disable(options):
55
    client = Getclientbyname(options.disable)
56
    if client is not None:
57
        client.Disable()
58
59
def List(foo):
60
    def valuetostring(x):
61
        if type(x) is dbus.Boolean:
62
            return u"Yes" if x else u"No"
63
        else:
64
            return unicode(x)
65
    format_string = u' '.join(u'%%-%ds' %
66
                              max(len(tablewords[key]),
67
                                  max(len(valuetostring(client[key]))
68
                                      for client in
69
                                      mandos_clients.itervalues()))
70
                              for key in keywords)
71
    print format_string % tuple(tablewords[key] for key in keywords) 
72
    for client in mandos_clients.itervalues():
73
        print format_string % tuple(valuetostring(client[key])
74
                                    for key in keywords)
75
76
action = List
24.1.116 by Björn Påhlsson
added a mandos list client program
77
parser = OptionParser(version = "%%prog %s" % version)
78
parser.add_option("-a", "--all", action="store_true", default=False,
240 by Teddy Hogeborn
Merge "mandos-list" from belorn.
79
                  help="Print all fields")
24.1.118 by Björn Påhlsson
Added enable/disable
80
parser.add_option("-e", "--enable", type="string",
81
                  help="Enable specified client")
82
parser.add_option("-d", "--disable", type="string",
83
                  help="disable specified client")
24.1.116 by Björn Påhlsson
added a mandos list client program
84
options = parser.parse_args()[0]
24.1.118 by Björn Påhlsson
Added enable/disable
85
if options.enable:
86
    action = Enable
87
elif options.disable:
88
    action = Disable
89
elif options.all:
240 by Teddy Hogeborn
Merge "mandos-list" from belorn.
90
    keywords = ('name', 'enabled', 'timeout', 'last_checked_ok',
91
                'created', 'interval', 'host', 'fingerprint',
92
                'checker_running', 'last_enabled', 'checker')
24.1.116 by Björn Påhlsson
added a mandos list client program
93
else:
94
    keywords = defaultkeywords
95
24.1.118 by Björn Påhlsson
Added enable/disable
96
action(options)
97
98