1 files changed, 2 insertions, 32 deletions
diff --git a/pysite/views/main/bot/cleanlog.py b/pysite/views/main/bot/cleanlog.py
index abd1ee2e..1c90dc0b 100644
--- a/pysite/views/main/bot/cleanlog.py
+++ b/pysite/views/main/bot/cleanlog.py
@@ -1,8 +1,7 @@
 import logging
-import random
 
 from pysite.base_route import RouteView
-from pysite.constants import ALL_STAFF_ROLES
+from pysite.constants import ALL_STAFF_ROLES, ROLE_COLORS
 from pysite.decorators import require_roles
 from pysite.mixins import DBMixin, OAuthMixin
 
@@ -16,35 +15,6 @@ class CleanLogView(RouteView, DBMixin, OAuthMixin):
     table_name = "clean_logs"
     template = "main/bot/clean_logs.html"
 
-    # Colors
-    author_colors = {}
-    all_colors = [
-        "#afcfff",
-        "#93ff91",
-        "#ffe559",
-        "#ff9036",
-        "#ff6c5e",
-        "#ff65be",
-        "#9298ff",
-    ]
-    color_pool = all_colors
-
-    def _assign_color(self, author):
-        """
-        Assign a color to a specific author.
-        """
-
-        if not self.color_pool:
-            self.color_pool = self.all_colors
-
-        if author not in self.author_colors:
-            random_index = random.randint(0, len(self.color_pool))
-            color = self.color_pool.pop(random_index)
-            self.author_colors[author] = color
-            return color
-        else:
-            return self.author_colors[author]
-
     @require_roles(ALL_STAFF_ROLES)
     def get(self, log_id):
         """
@@ -60,6 +30,6 @@ class CleanLogView(RouteView, DBMixin, OAuthMixin):
         messages = data["log_data"]
 
         for message in messages:
-            message['color'] = self._assign_color(message['author'])
+            message['color'] = ROLE_COLORS.get(message['role'], ROLE_COLORS['developers'])
 
         return self.render(self.template, messages=messages)