64 lines
2.3 KiB
Python
64 lines
2.3 KiB
Python
import io
|
|
|
|
import nerimity
|
|
|
|
import bot
|
|
import utils as u
|
|
|
|
# from catbox_async_uploader.catbox_async_uploader.enums import LitterboxDuration
|
|
# from pyzxz import ZeroXZero
|
|
|
|
def setup(bot: bot.Bot):
|
|
@bot.command(name="chart", aliases=["c", "chartalbum", "albumchart", "collage"])
|
|
@bot.slash_command(name="chartalbum", description="Generate an album collage.")
|
|
async def chart_album(ctx: nerimity.Context, size: str = "5x5", timeframe: str = "7day", username: str = None):
|
|
if 'x' not in size:
|
|
await ctx.send(u.error_msg("Please provide a valid size.\nie `/chart album 5x5`"))
|
|
return
|
|
|
|
if not username:
|
|
try: username = await bot.get_lastfm(ctx.author.id)
|
|
except Exception as e:
|
|
print(e)
|
|
await ctx.send(u.error_msg(f"Unknown database error:\n{e}"))
|
|
|
|
if not username:
|
|
await ctx.send(u.error_msg("Please provide a Last.fm username (or set yours with `/setfm`)."))
|
|
print("returned")
|
|
return
|
|
|
|
try:
|
|
temp_msg = await ctx.send(f"Generating album chart for **{username}**...")
|
|
|
|
image = bot.collage_generator.generate(
|
|
entity = "album",
|
|
username = username,
|
|
rows = int(size.split(sep="x")[0]),
|
|
cols = int(size.split(sep="x")[1]),
|
|
period = timeframe
|
|
)
|
|
|
|
img_bytes = io.BytesIO()
|
|
image.save(fp=img_bytes, format="png")
|
|
|
|
# link = await bot.catbox_uploader.upload_to_litterbox(
|
|
# file_path_or_bytes = img_bytes,
|
|
# file_name = "collage.png",
|
|
# duration = LitterboxDuration.H12
|
|
# )
|
|
|
|
# attachment = await u.construct_attachment_from_bytes(
|
|
# filename = "chart",
|
|
# file_type = "png",
|
|
# bytes_arr = img_bytes,
|
|
# ).upload()
|
|
|
|
link = await bot.zxz.upload_from_bytes(img_bytes.getvalue(), "chart.png")
|
|
|
|
await ctx.send(u.good_msg(f"{size} Chart for {username} successfully generated:\n{link}"))
|
|
temp_msg.delete()
|
|
|
|
except Exception as e:
|
|
if temp_msg: temp_msg.delete()
|
|
print(e)
|
|
await ctx.send(u.error_msg(f"Unknown error:\n`{e}`"))
|