From 29ae3d26a7314b8630750d0a08c6f6622d90d2bc Mon Sep 17 00:00:00 2001
From: Ast3risk-ops <spacedude609@pm.me>
Date: Sat, 28 Dec 2024 19:25:17 -0500
Subject: [PATCH] count attachments and minor formatting fixes

---
 main.py       | 7 +++++--
 main_emoji.py | 9 +++++----
 2 files changed, 10 insertions(+), 6 deletions(-)

diff --git a/main.py b/main.py
index 457a634..0d6448f 100644
--- a/main.py
+++ b/main.py
@@ -23,7 +23,7 @@ async def on_ready():
 
 class TagSet(discord.ui.Modal):
     def __init__(self, *args, **kwargs) -> None:
-        super().__init__(title="Message Tag(s)")
+        super().__init__(timeout=None, *args, **kwargs)
         # Modal for tag input
         self.add_item(discord.ui.InputText(label="", required=False))
     async def callback(self, interaction: discord.Interaction):
@@ -116,7 +116,7 @@ async def bookmark_tag(
     ctx,
     message: discord.Message
 ):
-    modal = TagSet()
+    modal = TagSet(title="Message Tag(s)")
     await ctx.send_modal(modal)
     await modal.wait() # Wait for the modal to be submitted before archiving
     reactionlist = [] # Empty list of reactions to use later
@@ -147,6 +147,9 @@ async def bookmark_tag(
     if message.embeds:
         numembeds = len(message.embeds) # Number of embeds
         embed.add_field(name="<:mdicardtext:1311825458480021596> Embeds", value=f"{numembeds}", inline=True)
+    if message.attachments:
+        numfiles = len(message.attachments)
+        embed.add_field(name="<:mdifiledownload:1322695880637284514> Attachments", value=f"{numfiles}", inline=True)
     embed.add_field(name="<:mdicalendar:1311544097240121384> Send Date", value=f"{discord.utils.format_dt(message.created_at, 'F')}", inline=True)
     if modal.children[0].value:
         embed.add_field(name="<:mditag:1311505882047189012> Tags", value=f"{modal.children[0].value}", inline=True)
diff --git a/main_emoji.py b/main_emoji.py
index 6e62703..e3969a4 100644
--- a/main_emoji.py
+++ b/main_emoji.py
@@ -24,7 +24,7 @@ async def on_ready():
 
 class TagSet(discord.ui.Modal):
     def __init__(self, *args, **kwargs) -> None:
-        super().__init__(title="Message Tag(s)")
+        super().__init__(timeout=None, *args, **kwargs)
         # Modal for tag input
         self.add_item(discord.ui.InputText(label="", required=False))
     async def callback(self, interaction: discord.Interaction):
@@ -48,7 +48,6 @@ async def callback(self, interaction: discord.Interaction):
 
         # Edit the original message with the updated embed
         await interaction.response.edit_message(embed=embed)
-        return
 class DeleteBookmark(discord.ui.View):
     def __init__(self):
         super().__init__(timeout=None)
@@ -72,7 +71,6 @@ async def customizer(self, button, interaction):
 class About(discord.ui.View):
     def __init__(self):
         super().__init__(timeout=None)
-
         SiteButton = discord.ui.Button(label='Website', style=discord.ButtonStyle.url, url='https://archiver.asterisk.lol')
         self.add_item(SiteButton)
         CodeButton = discord.ui.Button(label='Source Code', style=discord.ButtonStyle.url, url='https://github.com/Ast3risk-ops/archiver')
@@ -118,7 +116,7 @@ async def bookmark_tag(
     ctx,
     message: discord.Message
 ):
-    modal = TagSet()
+    modal = TagSet(title="Message Tag(s)")
     await ctx.send_modal(modal)
     await modal.wait() # Wait for the modal to be submitted before archiving
     reactionlist = [] # Empty list of reactions to use later
@@ -149,6 +147,9 @@ async def bookmark_tag(
     if message.embeds:
         numembeds = len(message.embeds) # Number of embeds
         embed.add_field(name="🔲 Embeds", value=f"{numembeds}", inline=True)
+    if message.attachments:
+        numfiles = len(message.attachments)
+        embed.add_field(name="📸 Attachments", value=f"{numfiles}", inline=True)
 
     embed.add_field(name="📅 Send Date", value=f"{discord.utils.format_dt(message.created_at, 'F')}", inline=True)
     if modal.children[0].value: