Atlas/Atlas-Install
1
0
Fork 0
mirror of https://github.com/Faraphel/Atlas-Install.git synced 2025-07-03 19:28:25 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

added a try to avoid exception stopping the process, track_channel.history had a limit of -1, making the loop instantly stop

Browse source
This commit is contained in:
raphael60650 2021-08-01 22:46:50 +02:00
parent ca01ded4e7
commit 3ba2fa92e7
1 changed files with 37 additions and 33 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

70
scripts/discord_add_track.py
View file

@ -50,7 +50,7 @@ async def on_ready():
old_sha1 = set() old_sha1 = set()
message: discord.Message message: discord.Message
async for message in track_channel.history(limit=-1): async for message in track_channel.history(limit=5000):
if message.author.id == bot.user.id: if message.author.id == bot.user.id:
for field in message.embeds[0].fields: for field in message.embeds[0].fields:
if "sha1" in field.name: if "sha1" in field.name:
@ -61,46 +61,50 @@ async def on_ready():
ct_config.load_ctconfig_file("./ct_config.json") ct_config.load_ctconfig_file("./ct_config.json")
for track in ct_config.all_tracks: for track in ct_config.all_tracks:
if track.name == "_": continue try:
if track.name == "_": continue
embed = discord.Embed(title=f"**{track.get_track_name()}**", embed = discord.Embed(title=f"**{track.get_track_name()}**",
description="", url=f"https://ct.wiimm.de/i/{track.sha1}") description="", url=f"https://ct.wiimm.de/i/{track.sha1}")
author_link = "" author_link = ""
if "," not in track.author: author_link = "http://wiki.tockdom.com/wiki/" + track.author if "," not in track.author: author_link = "http://wiki.tockdom.com/wiki/" + track.author
embed.set_author(name=track.author, url=author_link) embed.set_author(name=track.author, url=author_link)
with io.BytesIO() as image_binary: with io.BytesIO() as image_binary:
image = get_track_minimap(track) image = get_track_minimap(track)
image.save(image_binary, "PNG") image.save(image_binary, "PNG")
image_binary.seek(0) image_binary.seek(0)
message_minimap = await data_channel.send(file=discord.File(fp=image_binary, filename="minimap.png")) message_minimap = await data_channel.send(file=discord.File(fp=image_binary, filename="minimap.png"))
embed.set_thumbnail(url=message_minimap.attachments[0].url) embed.set_thumbnail(url=message_minimap.attachments[0].url)
if hasattr(track, "score"): if hasattr(track, "score"):
embed.add_field(name="Track Score", value=track.score) # TODO embed.add_field(name="Track Score", value=track.score) # TODO
if hasattr(track, "warning"): if hasattr(track, "warning"):
embed.add_field(name="Warning level", value=warning_level_message[track.warning]) embed.add_field(name="Warning level", value=warning_level_message[track.warning])
if hasattr(track, "since_version"): if hasattr(track, "since_version"):
embed.add_field(name="Here since version", value=track.since_version) # TODO embed.add_field(name="Here since version", value=track.since_version) # TODO
embed.add_field(name="Lap count", value="3") # TODO embed.add_field(name="Lap count", value="3") # TODO
embed.add_field(name="Speed multiplier", value="x1.0") # TODO embed.add_field(name="Speed multiplier", value="x1.0") # TODO
embed.set_image(url=placeholder_image_url) # TODO embed.set_image(url=placeholder_image_url) # TODO
if hasattr(track, "sha1"):
embed.add_field(name="sha1", value=track.sha1)
if track.sha1 not in old_sha1:
message = await track_channel.send(embed=embed)
await message.add_reaction(bot.get_emoji(EMOTE_1STAR))
await message.add_reaction(bot.get_emoji(EMOTE_2STAR))
await message.add_reaction(bot.get_emoji(EMOTE_3STAR))
await message.add_reaction("❌")
else:
if hasattr(track, "sha1"): if hasattr(track, "sha1"):
await message_from_sha1[track.sha1].edit(embed=embed) embed.add_field(name="sha1", value=track.sha1)
if track.sha1 not in old_sha1:
message = await track_channel.send(embed=embed)
await message.add_reaction(bot.get_emoji(EMOTE_1STAR))
await message.add_reaction(bot.get_emoji(EMOTE_2STAR))
await message.add_reaction(bot.get_emoji(EMOTE_3STAR))
await message.add_reaction("❌")
else:
if hasattr(track, "sha1"):
await message_from_sha1[track.sha1].edit(embed=embed)
except Exception as e:
print(f'error in track {track.name} : {str(e)}')
bot.run(os.environ['DISCORD_GR_TOKEN']) bot.run(os.environ['DISCORD_GR_TOKEN'])