supleed2/bot4koma
1
0
Fork 0
mirror of https://github.com/supleed2/bot4koma.git synced 2024-12-22 22:05:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update app.py to allow for overflow contents messages

Browse source 
Including keeping track of and deleting multiple contents messages
This commit is contained in:
Aadi Desai 2022-12-29 19:30:57 +00:00 committed by GitHub
parent 0711bdaf42
commit b93c401b77
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 30 additions and 20 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

40
app.py
View file

@ -10,38 +10,48 @@ listenChannelIDs = [
outputChannelID = 960995261570756690 # ICAS - 4koma-archive outputChannelID = 960995261570756690 # ICAS - 4koma-archive
fourKomaRegex = re.compile(r"^(?:[\s\S]*\[4KOMA\] (\d{4}\-(?:0[1-9]|1[012])\-(?:0[1-9]|[12][0-9]|3[01])))\s*([\s\S]*)$") fourKomaRegex = re.compile(r"^(?:[\s\S]*\[4KOMA\] (\d{4}\-(?:0[1-9]|1[012])\-(?:0[1-9]|[12][0-9]|3[01])))\s*([\s\S]*)$")
contentsRegex = re.compile(r"^[\S\s]*[cC]hapter[^\d]*([\w.]+)[\S\s]*$") contentsRegex = re.compile(r"^[\S\s]*[cC]hapter[^\d]*([\w.]+)[\S\s]*$")
splitSize = 10 contentsMsgIDs = None
contentsMsgID = None
with open("conf.json", "r") as conf: with open("conf.json", "r") as conf:
contentsMsgID = json.load(conf)["contentsMsgID"] contentsMsgIDs = json.load(conf)["contentsMsgIDs"]
async def generate_contents(): async def generate_contents():
global contentsMsgID global contentsMsgIDs
if contentsMsgID is not None: if contentsMsgIDs is not None:
for contentsMsgID in contentsMsgIDs:
try: try:
oldmsg = await outputChannel.fetch_message(contentsMsgID) oldmsg = await outputChannel.fetch_message(contentsMsgID)
except discord.errors.NotFound: except discord.errors.NotFound:
print(f"[ ERR ] {datetime.now().ctime()} Old Contents message not found, skipping") print(f"[ ERR ] {datetime.now().ctime()} Old Contents message ({contentsMsgID}) not found, skipping")
else: else:
await oldmsg.delete() await oldmsg.delete()
print(f"[INFO:] {datetime.now().ctime()} Deleted old Contents message") print(f"[INFO:] {datetime.now().ctime()} Deleted old Contents message ({contentsMsgID})")
bot_messages = [(contentsRegex.match(message.content).group(1), message.jump_url) bot_messages = [(contentsRegex.match(message.content).group(1), message.jump_url)
for message in await outputChannel.history(limit=1000).flatten() for message in await outputChannel.history(limit=1000).flatten()
if message.author == bot.user][::-1] if message.author == bot.user][::-1]
split_bot_messages = [bot_messages[i:i+splitSize] for i in range(0, len(bot_messages), splitSize)] split_bot_messages = [bot_messages[i:i+10] for i in range(0, len(bot_messages), 10)]
grouped_bot_messages = [split_bot_messages[i:i+6] for i in range(0, len(split_bot_messages), 6)]
embed = discord.Embed(title="4Koma Archive", embed = discord.Embed(title="4Koma Archive",
description="Hey guys! We have compiled all the past 4Koma here, come check them out and walk down memory lane <:2bhappy:370922383256715265>") description="Hey guys! We have compiled all the past 4Koma here, come check them out and walk down memory lane <:2bhappy:370922383256715265>")
for msgs in split_bot_messages: for msgs in grouped_bot_messages[0]:
embed.add_field( embed.add_field(
name=f"Chapters {msgs[0][0]} - {msgs[-1][0]}", name=f"Chapters {msgs[0][0]} - {msgs[-1][0]}",
value=", ".join([f"[{msg[0]}]({msg[1]})" for msg in msgs]), value=", ".join([f"[{msg[0]}]({msg[1]})" for msg in msgs]),
inline=False) inline=False)
newContentsMsg = await outputChannel.send(embed=embed) newContentsMsgs = [await outputChannel.send(embed=embed)]
contentsMsgID = newContentsMsg.id for part, extras in enumerate(grouped_bot_messages[1:]):
print(f"[INFO:] {datetime.now().ctime()} Regenerated Contents with {len(bot_messages)} entries, id: {contentsMsgID}") embed = discord.Embed(title=f"4Koma Archive (Part {part + 2})",
description="Another message for the remaining links *smh Discord with their 6000 character per message limit* <:aquaDismay:1007404394368745634>")
for msgs in extras:
embed.add_field(
name=f"Chapters {msgs[0][0]} - {msgs[-1][0]}",
value=", ".join([f"[{msg[0]}]({msg[1]})" for msg in msgs]),
inline=False)
newContentsMsgs.append(await outputChannel.send(embed=embed))
contentsMsgIDs = [msg.id for msg in newContentsMsgs]
print(f"[INFO:] {datetime.now().ctime()} Regenerated Contents with {len(bot_messages)} entries, IDs: {contentsMsgIDs}")
with open("conf.json", "w") as conf: with open("conf.json", "w") as conf:
json.dump({"contentsMsgID":contentsMsgID}, conf) json.dump({"contentsMsgIDs":contentsMsgIDs}, conf)
@bot.event @bot.event
async def on_ready(): async def on_ready():
@ -58,7 +68,7 @@ async def on_ready():
else: else:
print(f"[ ERR ] {datetime.now().ctime()} Failed to get output channel, ID: {outputChannelID}") print(f"[ ERR ] {datetime.now().ctime()} Failed to get output channel, ID: {outputChannelID}")
await bot.change_presence(status=discord.Status.do_not_disturb, activity=discord.Game(name="Failed to connect output channel")) await bot.change_presence(status=discord.Status.do_not_disturb, activity=discord.Game(name="Failed to connect output channel"))
print(f"[INFO:] {datetime.now().ctime()} Contents Message ID from conf.json: {contentsMsgID}") print(f"[INFO:] {datetime.now().ctime()} Contents Message IDs from conf.json: {contentsMsgIDs}")
committeeRole = bot.get_guild(249891637008793600).get_role(250056298710827008) committeeRole = bot.get_guild(249891637008793600).get_role(250056298710827008)
if committeeRole is not None: if committeeRole is not None:
print(f"[INFO:] {datetime.now().ctime()} Committee Role bound") print(f"[INFO:] {datetime.now().ctime()} Committee Role bound")
@ -72,7 +82,7 @@ async def on_message(message: discord.Message):
elif message.channel.id in listenChannelIDs: elif message.channel.id in listenChannelIDs:
if committeeRole in message.author.roles: if committeeRole in message.author.roles:
if message.content.startswith("!contents"): if message.content.startswith("!contents"):
print(f"[INFO:] {datetime.now().ctime()} Force regenerate contents message") print(f"[INFO:] {datetime.now().ctime()} Force regenerate contents messages")
await message.delete() await message.delete()
await generate_contents() await generate_contents()
return return