merge

Kav-K · Feb 10, 2023 · cf538db · cf538db
2 parents ed43295 + 63b5f17
commit cf538db
Show file tree

Hide file tree

Showing 3 changed files with 31 additions and 7 deletions.
diff --git a/cogs/prompt_optimizer_cog.py b/cogs/prompt_optimizer_cog.py
@@ -182,7 +182,11 @@ class DrawButton(discord.ui.Button["OptimizeView"]):
     def __init__(
         self, converser_cog, image_service_cog, deletion_queue, custom_api_key
     ):
-        super().__init__(style=discord.ButtonStyle.green, label="Draw", custom_id="draw_button_optimizer")
+        super().__init__(
+            style=discord.ButtonStyle.green,
+            label="Draw",
+            custom_id="draw_button_optimizer",
+        )
         self.converser_cog = converser_cog
         self.image_service_cog = image_service_cog
         self.deletion_queue = deletion_queue
@@ -237,7 +241,11 @@ class RedoButton(discord.ui.Button["OptimizeView"]):
     def __init__(
         self, converser_cog, image_service_cog, deletion_queue, custom_api_key=None
     ):
-        super().__init__(style=discord.ButtonStyle.danger, label="Retry", custom_id="redo_button_optimizer")
+        super().__init__(
+            style=discord.ButtonStyle.danger,
+            label="Retry",
+            custom_id="redo_button_optimizer",
+        )
         self.converser_cog = converser_cog
         self.image_service_cog = image_service_cog
         self.deletion_queue = deletion_queue

diff --git a/services/image_service.py b/services/image_service.py
@@ -350,7 +350,11 @@ async def callback(self, interaction: discord.Interaction):
 
 class RedoButton(discord.ui.Button["SaveView"]):
     def __init__(self, cog, converser_cog, custom_api_key):
-        super().__init__(style=discord.ButtonStyle.danger, label="Retry", custom_id="redo_button_draw_main")
+        super().__init__(
+            style=discord.ButtonStyle.danger,
+            label="Retry",
+            custom_id="redo_button_draw_main",
+        )
         self.cog = cog
         self.converser_cog = converser_cog
         self.custom_api_key = custom_api_key

diff --git a/services/moderations_service.py b/services/moderations_service.py
@@ -323,7 +323,9 @@ def __init__(
 
 class ApproveMessageButton(discord.ui.Button["ModerationAdminView"]):
     def __init__(self, message, moderation_message, current_num):
-        super().__init__(style=discord.ButtonStyle.green, label="Approve", custom_id="approve_button")
+        super().__init__(
+            style=discord.ButtonStyle.green, label="Approve", custom_id="approve_button"
+        )
         self.message = message
         self.moderation_message = moderation_message
         self.current_num = current_num
@@ -336,7 +338,11 @@ async def callback(self, interaction: discord.Interaction):
 
 class DeleteMessageButton(discord.ui.Button["ModerationAdminView"]):
     def __init__(self, message, moderation_message, current_num):
-        super().__init__(style=discord.ButtonStyle.danger, label="Delete Message", custom_id="delete_button")
+        super().__init__(
+            style=discord.ButtonStyle.danger,
+            label="Delete Message",
+            custom_id="delete_button",
+        )
         self.message = message
         self.moderation_message = moderation_message
         self.current_num = current_num
@@ -361,7 +367,9 @@ async def callback(self, interaction: discord.Interaction):
 
 class KickUserButton(discord.ui.Button["ModerationAdminView"]):
     def __init__(self, message, moderation_message, current_num):
-        super().__init__(style=discord.ButtonStyle.danger, label="Kick User", custom_id="kick_button")
+        super().__init__(
+            style=discord.ButtonStyle.danger, label="Kick User", custom_id="kick_button"
+        )
         self.message = message
         self.moderation_message = moderation_message
         self.current_num = current_num
@@ -400,7 +408,11 @@ class TimeoutUserButton(discord.ui.Button["ModerationAdminView"]):
     def __init__(
         self, message, moderation_message, current_num, hours, nodelete, source_deleted
     ):
-        super().__init__(style=discord.ButtonStyle.danger, label=f"Timeout {hours}h", custom_id="timeout_button")
+        super().__init__(
+            style=discord.ButtonStyle.danger,
+            label=f"Timeout {hours}h",
+            custom_id="timeout_button",
+        )
         self.message = message
         self.moderation_message = moderation_message
         self.hours = hours