Refectoring of the main classes

[collecty.git] / src / collecty / plugins / conntrack.py

diff --git a/src/collecty/plugins/conntrack.py b/src/collecty/plugins/conntrack.py
index 1e9bb79819a1934077466e15cbf9ddc471dd5136..22fcd60195b48ba03cf60179c715b152f9087f8e 100644 (file)
--- a/src/collecty/plugins/conntrack.py
+++ b/src/collecty/plugins/conntrack.py
@@ -136,8 +136,10 @@ class ConntrackLayer3ProtocolsGraphTemplate(base.GraphTemplate):
                "ipv4"  : "#cccc33",
        }
 
-       def get_object(self, *args):
-               return self.plugin.get_object("layer3-protocols")
+       def get_objects(self, *args):
+               return [
+                       self.plugin.get_object("layer3-protocols"),
+               ]
 
        @property
        def protocols(self):
@@ -248,8 +250,10 @@ class ConntrackLayer4ProtocolsGraphTemplate(ConntrackLayer3ProtocolsGraphTemplat
                "dccp"    : 7,
        }
 
-       def get_object(self, *args):
-               return self.plugin.get_object("layer4-protocols")
+       def get_objects(self, *args):
+               return [
+                       self.plugin.get_object("layer4-protocols"),
+               ]
 
        @property
        def graph_title(self):