Deseasonify: add colour attr to seasonal classes

We'll use this to colour the branding embed, as it currently looks very
plain. Seasons can either provide their own, or just use the default
green.

7 files changed, 19 insertions, 7 deletions

diff --git a/bot/seasons/__init__.py b/bot/seasons/__init__.py
index 28920476..13fec76b 100644
--- a/bot/seasons/__init__.py
+++ b/bot/seasons/__init__.py
@@ -4,7 +4,7 @@ from datetime import datetime
 from pathlib import Path
 from typing import List, Optional, Set, Type
 
-from bot.constants import Month
+from bot.constants import Colours, Month
 
 __all__ = ("SeasonBase", "get_seasons", "get_extensions", "get_current_season", "get_season")
 
@@ -58,6 +58,7 @@ class SeasonBase:
     season_name: str = "Evergreen"
     bot_name: str = "SeasonalBot"
 
+    colour: str = Colours.soft_green
     description: str = "The default season!"
 
     branding_path: str = "seasonal/evergreen"
diff --git a/bot/seasons/christmas/__init__.py b/bot/seasons/christmas/__init__.py
index b8fb19f0..33a8cfea 100644
--- a/bot/seasons/christmas/__init__.py
+++ b/bot/seasons/christmas/__init__.py
@@ -1,4 +1,4 @@
-from bot.constants import Month
+from bot.constants import Colours, Month
 from bot.seasons import SeasonBase
 
 
@@ -16,6 +16,8 @@ class Christmas(SeasonBase):
     season_name = "Festive season"
     bot_name = "Merrybot"
 
+    colour = Colours.soft_red
+
     branding_path = "seasonal/christmas"
 
     months = {Month.december}
diff --git a/bot/seasons/easter/__init__.py b/bot/seasons/easter/__init__.py
index 5056bd7e..eeac45a1 100644
--- a/bot/seasons/easter/__init__.py
+++ b/bot/seasons/easter/__init__.py
@@ -1,4 +1,4 @@
-from bot.constants import Month
+from bot.constants import Colours, Month
 from bot.seasons import SeasonBase
 
 
@@ -24,6 +24,7 @@ class Easter(SeasonBase):
     season_name = "Easter"
     bot_name = "BunnyBot"
 
+    colour = Colours.bright_green
     description = "Bunny here, bunny there, bunny everywhere!"
 
     branding_path = "seasonal/easter"
diff --git a/bot/seasons/halloween/__init__.py b/bot/seasons/halloween/__init__.py
index b20da9ac..23923eec 100644
--- a/bot/seasons/halloween/__init__.py
+++ b/bot/seasons/halloween/__init__.py
@@ -1,4 +1,4 @@
-from bot.constants import Month
+from bot.constants import Colours, Month
 from bot.seasons import SeasonBase
 
 
@@ -14,6 +14,8 @@ class Halloween(SeasonBase):
     season_name = "Halloween"
     bot_name = "NeonBot"
 
+    colour = Colours.orange
+
     branding_path = "seasonal/halloween"
 
     months = {Month.october}
diff --git a/bot/seasons/pride/__init__.py b/bot/seasons/pride/__init__.py
index 6509ac9b..cc79a72b 100644
--- a/bot/seasons/pride/__init__.py
+++ b/bot/seasons/pride/__init__.py
@@ -1,4 +1,4 @@
-from bot.constants import Month
+from bot.constants import Colours, Month
 from bot.seasons import SeasonBase
 
 
@@ -24,6 +24,8 @@ class Pride(SeasonBase):
     season_name = "Pride"
     bot_name = "ProudBot"
 
+    colour = Colours.pink
+
     branding_path = "seasonal/pride"
 
     months = {Month.june}
diff --git a/bot/seasons/valentines/__init__.py b/bot/seasons/valentines/__init__.py
index fb3d02af..67385c94 100644
--- a/bot/seasons/valentines/__init__.py
+++ b/bot/seasons/valentines/__init__.py
@@ -1,4 +1,4 @@
-from bot.constants import Month
+from bot.constants import Colours, Month
 from bot.seasons import SeasonBase
 
 
@@ -12,6 +12,8 @@ class Valentines(SeasonBase):
     season_name = "Valentines"
     bot_name = "TenderBot"
 
+    colour = Colours.pink
+
     branding_path = "seasonal/valentines"
 
     months = {Month.february}
diff --git a/bot/seasons/wildcard/__init__.py b/bot/seasons/wildcard/__init__.py
index 48491ce2..c628d62e 100644
--- a/bot/seasons/wildcard/__init__.py
+++ b/bot/seasons/wildcard/__init__.py
@@ -1,4 +1,4 @@
-from bot.constants import Month
+from bot.constants import Colours, Month
 from bot.seasons import SeasonBase
 
 
@@ -21,4 +21,6 @@ class Wildcard(SeasonBase):
     season_name = "Wildcard"
     bot_name = "RetroBot"
 
+    colour = Colours.purple
+
     months = {Month.august}