hope I got the merge wright...

[monitor.git] / commands / plcquery.py

diff --git a/commands/plcquery.py b/commands/plcquery.py
new file mode 100755 (executable)
index 0000000..8643b81
--- /dev/null
+++ b/commands/plcquery.py
@@ -0,0 +1,149 @@
+#!/usr/bin/python
+
+
+import sys
+from monitor.wrapper import plc
+api = plc.api
+
+from monitor import config
+
+def parse_fields(fields):
+    if fields:
+        s = fields.split(",")
+        if s == []:
+            return None
+        else:
+            return s
+    else:
+        return None
+
+def parse_filter(filter):
+    if filter:
+        s = filter.split("=")
+        if len(s) == 2:
+            return {s[0] : s[1]}
+        else:
+            raise Exception("filter format should be name=value")
+    else:
+        return None
+
+def conv(s):
+    # strip non-ascii characters to prvent errors
+    r = "".join([x for x in s if ord(x) < 128])
+    return r
+
+def print_fields(obj, fields, format):
+    if format:
+        for f in obj:
+            if type(obj[f]) in (str, unicode): 
+                obj[f] = conv(obj[f])
+            
+        print format % obj
+    else:
+        for f in fields:
+            if f in obj:
+                obj[f] = conv(obj[f])
+                print obj[f],
+        print ""
+
+def list_fields(l):
+    if len(l) > 0:
+        o = l[0]
+        for k in o.keys():
+            print k
+        sys.exit(1)
+    else:
+        print "no objects returned to list fields"
+        sys.exit(1)
+
+def main():
+
+    from monitor import parser as parsermodule
+    parser = parsermodule.getParser()
+
+    parser.set_defaults(get=True,
+                        type='node',
+                        filter=None,
+                        fields=None,
+                        format=None,
+                        listfields=False,
+                        withsitename=False,
+                        byloginbase=None,
+                        byrole=None,
+                        )
+
+    parser.add_option("", "--get", dest="get", action="store_true",
+                        help="just get values for object")
+
+    parser.add_option("", "--type", dest="type", metavar="[node|person|site]", 
+                        help="object type to query")
+
+    parser.add_option("", "--fields", dest="fields", metavar="key,list,...", 
+                        help="a list of keys to display for each object.")
+
+    parser.add_option("", "--filter", dest="filter", metavar="name=value", 
+                        help="Filter passed to Get* calls")
+
+    parser.add_option("", "--format", dest="format",
+                        help="Format string to use to print")
+
+    parser.add_option("", "--byloginbase", dest="byloginbase",
+                        help="")
+    parser.add_option("", "--byrole", dest="byrole",
+                        help="")
+    parser.add_option("", "--withsitename", dest="withsitename",
+                        action="store_true", help="")
+
+    parser.add_option("", "--nodelist", dest="nodelist", metavar="nodelist.txt", 
+                        help="A list of nodes to bring out of debug mode.")
+    parser.add_option("", "--listfields", dest="listfields", action="store_true",
+                        help="A list of nodes to bring out of debug mode.")
+
+    #parser = parsermodule.getParser(['defaults'], parser)
+    config = parsermodule.parse_args(parser)
+    
+    if config.nodelist:
+        nodelist = utilfile.getListFromFile(config.nodelist)
+
+    if config.get:
+        f = parse_filter(config.filter)
+        fields = parse_fields(config.fields)
+
+    if config.type == 'node': 
+        if config.fields is None: 
+            fields='node_id,hostname,last_contact',
+            fields = parse_fields(config.fields)
+
+        n = api.GetNodes(f, fields)
+        if config.listfields: list_fields(n)
+        for i in n:
+            print_fields(i, fields, config.format)
+
+    if config.type == 'person': 
+            
+        if config.byloginbase:
+            s = api.GetSites({'login_base' : config.byloginbase}, ['person_ids'])
+            f = s[0]['person_ids']
+        if config.byrole:
+            p = api.GetPersons(None, ['person_id', 'roles'])
+            p = filter(lambda x: config.byrole in x['roles'], p)
+            f = [ x['person_id'] for x in  p ]
+
+        if config.withsitename:
+            n = api.GetPersons(f, fields)
+            if config.listfields: list_fields(n)
+            for i in n:
+                sitelist = api.GetSites(i['site_ids'], ['person_ids', 'name'])
+                if len(sitelist) > 0:
+                    s = sitelist[0]
+                    if i['person_id'] in s['person_ids']:
+                        i['name'] = conv(s['name'])
+                        print_fields(i, fields, config.format)
+        else:
+            n = api.GetPersons(f, fields)
+            if config.listfields: list_fields(n)
+            for i in n:
+                print_fields(i, fields, config.format)
+
+if __name__ == "__main__":
+    main()