From 89d38027b6bacf3f02d73b50efef77ba9d8844b7 Mon Sep 17 00:00:00 2001
From: gcavalcante8808 <gabriel.cavalcante88@gmail.com>
Date: Mon, 11 Jan 2016 12:46:26 -0200
Subject: [PATCH] Now is possible to query both root values and nested values.

---
 moodle_inspect.py | 77 +++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 77 insertions(+)
 create mode 100755 moodle_inspect.py

diff --git a/moodle_inspect.py b/moodle_inspect.py
new file mode 100755
index 0000000..5dfd21c
--- /dev/null
+++ b/moodle_inspect.py
@@ -0,0 +1,77 @@
+#!/usr/bin/env python
+
+# This Module allow zabbix and other solutions to inspect
+# moodle api.
+import requests
+import argparse
+import pdb
+
+def get_nested_elements(info, elements):
+    # Function to traverse dictionaries and print when value is 
+    # not a dict (instead it's a str)
+#    pdb.set_trace()
+    if isinstance(elements, str):
+        keys = elements.split('.')
+
+    if isinstance(info, dict) and len(keys) > 1:
+        keys.pop(0)
+        for key in keys:
+            value = info[key]
+            if isinstance(value, dict):
+                keys.pop(0)
+                if keys:
+                    get_nested_elements(value, keys)
+            elif value:
+                print(value)
+            else:
+                return('Not Encountered Value')
+    else:
+        print(info)
+
+def inspect_ws(**kwargs):
+    # Get Params and try to pick a result from the server.
+    server = kwargs.pop('server')
+    token = kwargs.pop('token')
+    function = kwargs.pop('function')
+    data = kwargs.pop('data')
+    ssl = kwargs.pop('ssl')
+
+    if ssl:
+       prefix = 'https://'
+    else:
+       prefix = 'http://'
+    
+    wsdata = { 'moodlewsrestformat': 'json',
+               'wsfunction': function.split('.')[0],
+               'wstoken': token
+    }
+
+    if data:
+        wsdata += data
+
+    result = requests.post(url=prefix+server+'/webservice/rest/server.php',
+                           data=wsdata).json()
+
+    if isinstance(result, dict) and result.has_key('errorcode'):
+       get_nested_elements(result, function)
+    elif isinstance(result, str) or isinstance(result, int):
+       return(result)
+    elif isinstance(result,list):
+       for value in result:
+           get_nested_elements(value, function)
+
+
+if __name__ == '__main__':
+    parser = argparse.ArgumentParser()
+    parser.add_argument('--server', type=str, required=True)
+    parser.add_argument('--token', type=str, required=True)
+    parser.add_argument('--function', type=str, required=True)
+    parser.add_argument('--data', type=dict, required=False)
+    parser.add_argument('--ssl', type=bool, required=False, default=False)
+
+    args = parser.parse_args()
+
+    inspect_ws(server=args.server, token=args.token,
+               function=args.function, data=args.data, 
+               ssl=args.ssl)
+