Add Cheese to /food command

EthanC · May 27, 2024 · b76c15e · b76c15e
1 parent 17c6322
commit b76c15e
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/extensions/food.py b/extensions/food.py
@@ -19,6 +19,7 @@
 plugin: GatewayPlugin = GatewayPlugin("food")
 foodTypes: list[str] = [
     "Burger",
+    "Cheese",
     "Chicken",
     "Dessert",
     "Hot Dog",
@@ -80,6 +81,20 @@ async def CommandFood(
                         logger.warning(
                             f"Recieved unknown source {source} for food type {type}"
                         )
+            case "Cheese":
+                source = random.randint(1, 3)
+
+                match source:
+                    case 1:
+                        result = await GetRandomImage("Cheese")
+                    case 2:
+                        result = await GetRandomImage("cheesemaking")
+                    case 3:
+                        result = await GetRandomImage("grilledcheese")
+                    case _:
+                        logger.warning(
+                            f"Recieved unknown source {source} for food type {type}"
+                        )
             case "Chicken":
                 source = random.randint(1, 3)