Add lastupdate()

[collecty.git] / src / collecty / client.py

diff --git a/src/collecty/client.py b/src/collecty/client.py
index 7989d1197dee30afa1fb754cc40a5d20791d1365..be898bb47cf97bcf0054078cdb0fc4010deea687 100644 (file)
--- a/src/collecty/client.py
+++ b/src/collecty/client.py
@@ -20,6 +20,7 @@
 ###############################################################################
 
 import argparse
+import datetime
 import dbus
 import os
 import platform
@@ -37,6 +38,23 @@ class CollectyClient(object):
 
                self.proxy = self.bus.get_object(BUS_DOMAIN, "/GraphGenerator")
 
+       def last_update(self, template_name, **kwargs):
+               last_update = self.proxy.LastUpdate(template_name, kwargs)
+
+               if last_update:
+                       last_update["timestamp"] = datetime.datetime.strptime(last_update["timestamp"], "%Y-%m-%dT%H:%M:%S")
+
+               return last_update
+
+       def last_update_cli(self, ns):
+               last_update = self.last_update(ns.template, object_id=ns.object)
+
+               print(_("Last update: %s") % last_update.get("timestamp"))
+
+               dataset = last_update.get("dataset")
+               for k, v in dataset.items():
+                       print("%16s = %s" % (k, v))
+
        def list_templates(self):
                templates = self.proxy.ListTemplates()
 
@@ -136,6 +154,15 @@ class CollectyClient(object):
                parser_generate_graph.add_argument("--width", type=int, default=0,
                        help=_("Width of the generated image"))
 
+               # last-update
+               parser_last_update = subparsers.add_parser("last-update",
+                       help=_("Fetch the last dataset in the database"))
+               parser_last_update.add_argument("--template",
+                       help=_("The graph template identifier"), required=True)
+               parser_last_update.add_argument("--object",
+                       help=_("Object identifier"), default="default")
+               parser_last_update.set_defaults(func=self.last_update_cli)
+
                # list-templates
                parser_list_templates = subparsers.add_parser("list-templates",
                        help=_("Lists all graph templates"))