nex/college-bot-v1
1
0
Fork 0
mirror of https://github.com/nexy7574/LCC-bot.git synced 2024-09-19 18:16:34 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix file generator

Browse source
This commit is contained in:
eek7574 2022-12-08 11:33:58 +00:00
parent 9bcd159086
commit 6bcc2d4e14
1 changed files with 7 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
cogs/other.py
View file

@ -152,11 +152,15 @@ class OtherCog(commands.Cog):
@corrupt_file.command(name="generate") @corrupt_file.command(name="generate")
async def generate_corrupt_file(self, ctx: discord.ApplicationContext, file_name: str, size_in_megabytes: float): async def generate_corrupt_file(self, ctx: discord.ApplicationContext, file_name: str, size_in_megabytes: float):
"""Generates a "corrupted" file.""" """Generates a "corrupted" file."""
if size_in_megabytes > 8: limit_mb = round(ctx.guild.filesize_limit / 1024 / 1024)
return await ctx.respond("File size must be less than 8 MB.") if size_in_megabytes > limit_mb:
return await ctx.respond(f"File size must be less than {limit_mb} MB.")
await ctx.defer() await ctx.defer()
size = max(min(int(size_in_megabytes * 1024 * 1024), ctx.guild.filesize_limit), 1)
file = io.BytesIO() file = io.BytesIO()
file.write(os.urandom(int(size_in_megabytes * 1024 * 1024))) file.write(os.urandom(size))
file.seek(0) file.seek(0)
return await ctx.respond(file=discord.File(file, file_name)) return await ctx.respond(file=discord.File(file, file_name))