From 8abb1bc79d5b5d4a9e8ab57f970e70b918e9f5d6 Mon Sep 17 00:00:00 2001
From: "pixeebot[bot]" <104101892+pixeebot[bot]@users.noreply.github.com>
Date: Tue, 10 Dec 2024 17:54:51 +0000
Subject: [PATCH] Secure Source of Randomness

---
 birthday/birthday.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/birthday/birthday.py b/birthday/birthday.py
index a254127..4102659 100644
--- a/birthday/birthday.py
+++ b/birthday/birthday.py
@@ -4,9 +4,9 @@ from redbot.core.bot import Red # type: ignore
 from redbot.core.config import Config # type: ignore
 from datetime import datetime, time, timedelta
 from zoneinfo import ZoneInfo, ZoneInfoNotFoundError
-import random
 import logging
 import asyncio
+import secrets
 
 # Set up logging
 logger = logging.getLogger("red.birthday")
@@ -52,7 +52,7 @@ async def birthday_context_menu(interaction: discord.Interaction, member: discor
             return await interaction.followup.send("Failed to assign the birthday role due to a Discord error.", ephemeral=True)
 
         # Generate birthday message with random cakes (or pie)
-        cakes = random.randint(0, 5)
+        cakes = secrets.SystemRandom().randint(0, 5)
         if cakes == 0:
             message = f"🎉 Happy Birthday, {member.mention}! Sorry, out of cake today! Here's pie instead: 🥧"
         else:
@@ -368,7 +368,7 @@ class Birthday(commands.Cog):
                 return await ctx.send("Failed to assign the birthday role due to a Discord error.", ephemeral=True)
 
             # Generate birthday message with random cakes (or pie)
-            cakes = random.randint(0, 5)
+            cakes = secrets.SystemRandom().randint(0, 5)
             if cakes == 0:
                 message = f"🎉 Happy Birthday, {member.mention}! Sorry, out of cake today! Here's pie instead: 🥧"
             else: