Manually define the color of an endpoint from the config file

Author: kloostert

dashboard/colors.py

@@ -1,4 +1,7 @@
 from colorhash import ColorHash
+from dashboard import config
+
+import ast
 
 
 def get_color(hash):
@@ -7,7 +10,9 @@ def get_color(hash):
     :param hash: the string that is translated into a color
     :return: a color (as string)
     """
-    c = ColorHash(hash)
-    rgb = c.rgb
+    if hash in config.colors:
+        rgb = ast.literal_eval(config.colors[hash])
+    else:
+        rgb = ColorHash(hash).rgb
     return 'rgb({0}, {1}, {2})'.format(rgb[0], rgb[1], rgb[2])
 

dashboard/config.py

@@ -1,5 +1,6 @@
 import configparser
 import os
+import ast
 
 
 class Config(object):
@@ -21,6 +22,7 @@ def __init__(self):
         self.guest_username = 'guest'
         self.guest_password = 'guest_password'
         self.outlier_detection_constant = 2.5
+        self.colors = {}
 
         # define a custom function to retrieve the session_id or username
         self.get_group_by = None
@@ -66,6 +68,10 @@ def from_file(self, config_file):
             if parser.has_option('dashboard', 'TEST_DIR'):
                 self.test_dir = parser.get('dashboard', 'TEST_DIR')
 
+            # For manually defining colors of specific endpoints
+            if parser.has_option('dashboard', 'COLORS'):
+                self.colors = ast.literal_eval(parser.get('dashboard', 'COLORS'))
+
             # When the option git is selected, it overrides the given version
             if parser.has_option('dashboard', 'GIT'):
                 git = parser.get('dashboard', 'GIT')