From 40d5a00f1b609b23a6cd77a3b7f1e4814ed7df82 Mon Sep 17 00:00:00 2001 From: ToxicKidz Date: Fri, 7 May 2021 11:01:24 -0400 Subject: chore: Get the next perfect square If the amount of squares is not a perfect square, get the next highest perfect square --- bot/exts/evergreen/avatar_modification/avatar_modify.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/bot/exts/evergreen/avatar_modification/avatar_modify.py b/bot/exts/evergreen/avatar_modification/avatar_modify.py index dffe43ce..6418eaee 100644 --- a/bot/exts/evergreen/avatar_modification/avatar_modify.py +++ b/bot/exts/evergreen/avatar_modification/avatar_modify.py @@ -325,8 +325,10 @@ class AvatarModify(commands.Cog): if 1 <= squares <= MAX_SQUARES: raise commands.BadArgument(f"Squares must be a positive number less than or equal to {MAX_SQUARES:,}.") - if not math.sqrt(squares).is_integer(): - raise commands.BadArgument("The number of squares must be a perfect square.") + sqrt = math.sqrt(squares) + + if not sqrt.is_integer(): + squares = math.ceil(sqrt) ** 2 # Get the next perfect square file_name = file_safe_name("mosaic_avatar", ctx.author.display_name) -- cgit v1.2.3