Add graph info functionality
[collecty.git] / src / collecty / bus.py
CommitLineData
f37913e8 1#!/usr/bin/python3
c968f6d9
MT
2###############################################################################
3# #
4# collecty - A system statistics collection daemon for IPFire #
5# Copyright (C) 2015 IPFire development team #
6# #
7# This program is free software: you can redistribute it and/or modify #
8# it under the terms of the GNU General Public License as published by #
9# the Free Software Foundation, either version 3 of the License, or #
10# (at your option) any later version. #
11# #
12# This program is distributed in the hope that it will be useful, #
13# but WITHOUT ANY WARRANTY; without even the implied warranty of #
14# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the #
15# GNU General Public License for more details. #
16# #
17# You should have received a copy of the GNU General Public License #
18# along with this program. If not, see <http://www.gnu.org/licenses/>. #
19# #
20###############################################################################
21
22import dbus
23import dbus.mainloop.glib
24import dbus.service
f37913e8
MT
25import gi.repository.GLib
26import gi.repository.GObject
c968f6d9
MT
27import threading
28
f37913e8
MT
29from .constants import *
30from .i18n import _
c968f6d9
MT
31
32import logging
33log = logging.getLogger("collecty.bus")
34log.propagate = 1
35
c968f6d9
MT
36class Bus(threading.Thread):
37 def __init__(self, collecty):
38 threading.Thread.__init__(self)
39 self.daemon = True
40
41 self.collecty = collecty
42
43 # Initialise the main loop
f37913e8
MT
44 gi.repository.GObject.threads_init()
45 dbus.mainloop.glib.threads_init()
46 dbus.mainloop.glib.DBusGMainLoop(set_as_default=True)
47
48 self.loop = gi.repository.GLib.MainLoop()
c968f6d9
MT
49
50 # Register the GraphGenerator interface
51 self.generator = GraphGenerator(self.collecty)
52
53 def run(self):
54 log.debug(_("Bus thread has started"))
55
56 # Run the main loop
f37913e8
MT
57 try:
58 self.loop.run()
59 except KeyboardInterrupt:
60 self.collecty.shutdown()
61
62 log.debug(_("Bus thread has ended"))
c968f6d9
MT
63
64 def shutdown(self):
65 log.debug(_("Stopping bus thread"))
66
67 # End the main loop
68 self.loop.quit()
69
70 # Return when this thread has finished
71 return self.join()
72
73
74class GraphGenerator(dbus.service.Object):
75 def __init__(self, collecty):
76 bus_name = dbus.service.BusName(BUS_DOMAIN, bus=dbus.SystemBus())
77 dbus.service.Object.__init__(self, bus_name, "/%s" % self.__class__.__name__)
78
79 self.collecty = collecty
80
a3864812 81 @dbus.service.method(BUS_DOMAIN, in_signature="sa{sv}", out_signature="a{sv}")
c968f6d9
MT
82 def GenerateGraph(self, template_name, kwargs):
83 """
84 Returns a graph generated from the given template and object.
85 """
86 graph = self.collecty.generate_graph(template_name, **kwargs)
87
a3864812
MT
88 # Convert the graph back to normal Python format
89 if graph:
90 graph["image"] = dbus.ByteArray(graph["image"] or [])
91
92 return graph
93
94 @dbus.service.method(BUS_DOMAIN, in_signature="", out_signature="a{sv}")
95 def GraphInfo(self, template_name, kwargs):
96 """
97 Returns a dictionary with information about the graph.
98 """
99 return self.collecty.graph_info(template_name, **kwargs)
c968f6d9
MT
100
101 @dbus.service.method(BUS_DOMAIN, in_signature="", out_signature="as")
102 def ListTemplates(self):
103 """
104 Returns a list of all available templates
105 """
106 return [t.name for t in self.collecty.templates]
73241420
MT
107
108 @dbus.service.method(BUS_DOMAIN, in_signature="", out_signature="s")
109 def Version(self):
110 return COLLECTY_VERSION