pacnpal/Pac-cogs
1
0
Fork 0
mirror of https://github.com/pacnpal/Pac-cogs.git synced 2025-12-20 02:41:06 -05:00
Code Issues Packages Projects Releases Wiki Activity

Secure Source of Randomness

Browse Source
This commit is contained in:
pixeebot[bot]
2024-12-10 17:54:51 +00:00
committed by GitHub
parent 3097f994ea
commit 8abb1bc79d
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
birthday/birthday.py
View File

@@ -4,9 +4,9 @@ from redbot.core.bot import Red # type: ignore
from redbot.core.config import Config # type: ignore from redbot.core.config import Config # type: ignore
from datetime import datetime, time, timedelta from datetime import datetime, time, timedelta
from zoneinfo import ZoneInfo, ZoneInfoNotFoundError from zoneinfo import ZoneInfo, ZoneInfoNotFoundError
import random
import logging import logging
import asyncio import asyncio
import secrets
# Set up logging # Set up logging
logger = logging.getLogger("red.birthday") 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) 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) # Generate birthday message with random cakes (or pie)
cakes = random.randint(0, 5) cakes = secrets.SystemRandom().randint(0, 5)
if cakes == 0: if cakes == 0:
message = f"🎉 Happy Birthday, {member.mention}! Sorry, out of cake today! Here's pie instead: 🥧" message = f"🎉 Happy Birthday, {member.mention}! Sorry, out of cake today! Here's pie instead: 🥧"
else: 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) 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) # Generate birthday message with random cakes (or pie)
cakes = random.randint(0, 5) cakes = secrets.SystemRandom().randint(0, 5)
if cakes == 0: if cakes == 0:
message = f"🎉 Happy Birthday, {member.mention}! Sorry, out of cake today! Here's pie instead: 🥧" message = f"🎉 Happy Birthday, {member.mention}! Sorry, out of cake today! Here's pie instead: 🥧"
else: else: