diff --git a/test/urbackup_api_test.py b/test/urbackup_api_test.py index 860040e..6c5d4d9 100644 --- a/test/urbackup_api_test.py +++ b/test/urbackup_api_test.py @@ -16,6 +16,10 @@ for line in computernames: clients = server.get_status() +usage = server.get_usage() + +if len(clients) != len(usage): + print("Failed to retreive usage or status information. Length of both lists is different.") # Uncomment to format time differently # locale.setlocale(locale.LC_TIME, "german") @@ -62,4 +66,4 @@ for key in settings: if server.set_global_setting("max_image_incr", "40"): print("Changed global setting successfully") else: - print("Failed to change global setting") \ No newline at end of file + print("Failed to change global setting") diff --git a/urbackup_api/__init__.py b/urbackup_api/__init__.py index d931fb9..de30f82 100644 --- a/urbackup_api/__init__.py +++ b/urbackup_api/__init__.py @@ -339,6 +339,20 @@ class urbackup_server: return status["status"] + def get_usage(self): + if not self.login(): + return None + + usage = self._get_json("usage") + + if not usage: + return None + + if not "usage" in usage: + return None + + return usage["usage"] + def get_extra_clients(self): if not self.login(): return None