From ae1c2d24add1804a911153d8efc7db2e9f8e8ea3 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Johana=20Sup=C3=ADkov=C3=A1?= <johana.supikova@seznam.cz>
Date: Mon, 5 Jun 2023 09:52:37 +0200
Subject: [PATCH] feat: script to check RPC availability
---
perun/proxy/utils/nagios/check_rpc_status.py | 75 ++++++++++++++++++++
setup.py | 1 +
2 files changed, 76 insertions(+)
create mode 100644 perun/proxy/utils/nagios/check_rpc_status.py
diff --git a/perun/proxy/utils/nagios/check_rpc_status.py b/perun/proxy/utils/nagios/check_rpc_status.py
new file mode 100644
index 0000000..e13a976
--- /dev/null
+++ b/perun/proxy/utils/nagios/check_rpc_status.py
@@ -0,0 +1,75 @@
+#!/usr/bin/env python3
+
+import argparse
+import re
+import sys
+import time
+import requests
+
+
+"""
+check RPC API is available
+"""
+
+
+def get_args():
+ """
+ Supports the command-line arguments listed below.
+ """
+ parser = argparse.ArgumentParser(description="Check RPC status")
+ parser.add_argument(
+ "-u",
+ "--username",
+ required=True,
+ help="username for IdP",
+ )
+ parser.add_argument(
+ "-p",
+ "--password",
+ required=True,
+ help="password for IdP",
+ )
+ parser.add_argument(
+ "-d",
+ "--domain",
+ help="RPC domain with authentication method (e.g. 'perun.cesnet.cz/ba')",
+ required=True,
+ )
+ parser.add_argument(
+ "-i",
+ "--id",
+ type=int,
+ help="valid userId - This id will be used in getUserById call",
+ required=True,
+ )
+
+ return parser.parse_args()
+
+
+def call_api(auth, url, user_id):
+ start_time = time.time()
+ try:
+ response = requests.get(url, timeout=10, auth=auth)
+ rpc_result = response.text
+ except requests.Timeout:
+ rpc_result = "Request timeout"
+ end_time = time.time()
+ total_time = end_time - start_time
+ if re.search(r'"id":' + str(user_id), rpc_result):
+ print(f"0 check_rpc_status - total_time={total_time:.4f} OK")
+ return 0
+ else:
+ rpc_result = rpc_result.replace("\n", " ")
+ print(f"2 check_rpc_status - total_time={total_time:.4f} {rpc_result}")
+ return 2
+
+
+def main():
+ args = get_args()
+ auth = (args.username, args.password)
+ url = f"https://{args.domain}/rpc/json/usersManager/getUserById?id={args.id}"
+ return call_api(auth, url, args.id)
+
+
+if __name__ == "__main__":
+ sys.exit(main())
diff --git a/setup.py b/setup.py
index 6894cb4..0e4f0da 100644
--- a/setup.py
+++ b/setup.py
@@ -13,5 +13,6 @@ setuptools.setup(
"asyncssh~=2.13",
"docker~=6.0",
"beautifulsoup4~=4.12",
+ "requests~=2.31",
],
)
--
GitLab