From 6acf1beec88cc286d9e4ad5c741d779a10a8abfc Mon Sep 17 00:00:00 2001
From: daniel_sagi <daniel.sagi@aquasec.com>
Date: Tue, 8 May 2018 10:07:51 +0300
Subject: [PATCH] changed function describe_service_type in services, to use a
 dictionary

---
 services.py | 20 +++++++-------------
 1 file changed, 7 insertions(+), 13 deletions(-)

diff --git a/services.py b/services.py
index 4e93e34..c68e57b 100644
--- a/services.py
+++ b/services.py
@@ -11,19 +11,13 @@ disable_warnings()
 
 
 def describe_service_type(service_type):
-    if service_type == KUBERNETES_DASHBOARD:
-        return "Kubernetes Dashboard"
-
-    if service_type == KUBERNETES_PROXY:
-        return "Kubernetes Proxy"
-
-    if service_type == KUBERNETES_KUBELET_HTTPS:
-        return "Kubernetes Kubelet"
-
-    if service_type == KUBERNETES_KUBELET_HTTP:
-        return "Kubernetes Kubelet (Read only)"
-
-    return "Unknown Service"
+    service_types = {
+        KUBERNETES_DASHBOARD: "Kubernetes Dashboard",
+        KUBERNETES_PROXY: "Kubernetes Proxy",
+        KUBERNETES_KUBELET_HTTPS: "Kubernetes Kubelet",
+        KUBERNETES_KUBELET_HTTP: "Kubernetes Kubelet (Read only)",
+    }
+    return service_types.get(service_type, "Unknown Service")
 
 
 def is_dashboard(host):