Improve code quality of check_redis_cluster.py

src/check_redis_cluster.py

@@ -32,24 +32,26 @@
 from argparse import ArgumentParser
 from sys import exit
 
+
 def main():
     """Main entrypoint for script"""
 
     args = get_parser().parse_args()
 
-    master_addr, master_state, cluster_state_master, failed_master = _get_cluster_status(
-        args.master_port, args.password)
-    slave_addr, slave_state, cluster_state_slave, failed_slave = _get_cluster_status(
-        args.slave_port, args.password)
+    master_addr, master_state, cluster_state_master, failed_master = \
+        _get_cluster_status(args.master_port, args.password)
+    slave_addr, slave_state, cluster_state_slave, failed_slave = \
+        _get_cluster_status(args.slave_port, args.password)
     if master_state != 'unknown' and slave_state != 'unknown':
+        failed_hosts = failed_master + failed_slave
         if cluster_state_master != 'ok' and cluster_state_slave != 'ok':
             print('CRITICAL - cluster is broken')
             code = 2
-        elif len(failed_master + failed_slave) > 0:
+        elif failed_hosts:
             print('WARNING - cluster status is degraded')
-            for host in failed_master + failed_slave:
+            code = 1
+            for host in failed_hosts:
                 print('{} is in a failed state'.format(host))
-                code = 1
         else:
             print('OK - cluster status is OK')
             print('{} - {}'.format(master_addr, master_state))
@@ -109,12 +111,26 @@ def _get_cluster_status(port, password):
     The status of the local instances will be checked
     """
 
+    """The output from redis-cli cluser nodes has the following format:
+    6 lines where each line is a space seperated list with the following data:
+        7a8517281ff3039a99fc28a9cd5b7bb5cded8dcf (hash of the node)
+        10.44.1.13:7001@17001                    (ip:port@remote port)
+        slave,fail                               (master|slave)(,fail)?
+        9a168ff82b0a87a7b0af78159e2efa6dec483a76 (remote node hash)
+        1600863309527                            (performance data)
+        1600863307000                            (performance data)
+        2                                        (num clients)
+        connected
+
+    The string ",fail" is optional. It will only exists on failed hosts
+    """
     try:
         role = subprocess.check_output(
             'redis-cli -p {0} -a {1} cluster nodes'.format(
                 port, password), shell=True).decode().split()
         state_index = [i for i, s in enumerate(role) if 'myself' in s][0]
-        failed_hosts = [role[i-1] for i, s in enumerate(role) if 'fail' in s and str(port) in role[i-1]]
+        failed_hosts = [role[i-1] for i, s in enumerate(role)
+                        if 'fail' in s and str(port) in role[i-1]]
         role_state = role[state_index].replace('myself,', '')
         role_addr = role[state_index - 1]