improve configuration (with default values) - also for testing ejabberdrpc.py

1 files changed, 13 insertions, 3 deletions

diff --git a/ejabberdrpc.py b/ejabberdrpc.py
index 4cdaf88..0e20986 100644..100755
--- a/ejabberdrpc.py
+++ b/ejabberdrpc.py
@@ -319,8 +319,18 @@ class EjabberdMetrics():
 
 
 if __name__ == "__main__":
-    from json import dumps
-    metric = EjabberdMetrics("http://[::1]:5280/api", api="rest")
+    import os
+    import json
+    # load config
+    path = os.path.dirname(__file__)
+    with open("/".join([path, "config.json"]), "r", encoding="utf-8") as f:
+        config = json.load(f)
+
+    url = config['url'] if "url" in config else "http://[::1]:5280/api"
+    login = config['login'] if "login" in config else None
+    api = config['api'] if "api" in config else "rest"
+
+    metric = EjabberdMetrics(url, login=login, api=api)
    
     data = metric.get_all()
-    print(dumps(data, indent=True))
+    print(json.dumps(data, indent=True))