thumbnail is only calculated if sha1 change, track and lap count is now correct

This commit is contained in:
raphael60650 2021-08-02 18:08:51 +02:00
parent 844297ff9d
commit 76d47cd916

View file

@ -25,7 +25,6 @@ EMOTE_2STAR = 843109881385058325
EMOTE_3STAR = 843109892330881107 EMOTE_3STAR = 843109892330881107
placeholder_image_url = "https://media.discordapp.net/attachments/871469647617216652/871487829023289404/Placeholder.png" placeholder_image_url = "https://media.discordapp.net/attachments/871469647617216652/871487829023289404/Placeholder.png"
download_icon_url = "https://media.discordapp.net/attachments/842865834090037310/871467284009480202/img_68704.png"
def get_track_minimap(track: Track): def get_track_minimap(track: Track):
@ -71,23 +70,17 @@ async def on_ready():
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: track_technical_data = szs.analyze(track.file_szs)
image = get_track_minimap(track)
image.save(image_binary, "PNG")
image_binary.seek(0)
message_minimap = await data_channel.send(file=discord.File(fp=image_binary, filename="minimap.png"))
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)
embed.add_field(name="Lap count", value="3") # TODO embed.add_field(name="Lap count", value=track_technical_data["lap_count"])
embed.add_field(name="Speed multiplier", value="x1.0") # TODO embed.add_field(name="Speed multiplier", value=track_technical_data["speed_factor"])
embed.set_image(url=placeholder_image_url) # TODO embed.set_image(url=placeholder_image_url) # TODO
@ -95,11 +88,23 @@ async def on_ready():
embed.add_field(name="sha1", value=track.sha1) embed.add_field(name="sha1", value=track.sha1)
if track.sha1 not in old_sha1: if track.sha1 not in old_sha1:
with io.BytesIO() as image_binary:
image = get_track_minimap(track)
image.save(image_binary, "PNG")
image_binary.seek(0)
message_minimap = await data_channel.send(
file=discord.File(fp=image_binary, filename="minimap.png"))
embed.set_thumbnail(url=message_minimap.attachments[0].url)
message = await track_channel.send(embed=embed) message = await track_channel.send(embed=embed)
await message.add_reaction(bot.get_emoji(EMOTE_1STAR)) 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_2STAR))
await message.add_reaction(bot.get_emoji(EMOTE_3STAR)) await message.add_reaction(bot.get_emoji(EMOTE_3STAR))
await message.add_reaction("") await message.add_reaction("")
else: else:
if hasattr(track, "sha1"): if hasattr(track, "sha1"):
await message_from_sha1[track.sha1].edit(embed=embed) await message_from_sha1[track.sha1].edit(embed=embed)