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

Merge pull request #1 from pacnpal/pixeebot/drip-2024-12-10-pixee-python/secure-random

Browse Source
This commit is contained in:
pacnpal
2024-12-10 19:16:45 -05:00
committed by GitHub
parent d1541cda77 5ec043de09
commit 52897a1f70
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: