44 lines
1.4 KiB
Python
44 lines
1.4 KiB
Python
from redbot.core import commands, Config, checks
|
|
import discord
|
|
import requests
|
|
import imgkit
|
|
from PIL import Image, ImageOps
|
|
import os
|
|
|
|
UID = 0x181eff7aacab91fa4051d1d995d44050
|
|
|
|
|
|
def image(name, html):
|
|
open_image = imgkit.from_string(html, False)
|
|
open_image = Image.open(open_image)
|
|
image_box = ImageOps.invert(open_image.convert("RGB")).getbbox()
|
|
open_image.crop(image_box).save(f"{name}_cropped.png")
|
|
|
|
|
|
class Sarenrae(commands.Cog):
|
|
def __init__(self, bot):
|
|
self.bot = bot
|
|
self.conf = Config.get_conf(self, identifier=UID, force_registration=True)
|
|
default_global = {
|
|
"pf2_token": "",
|
|
}
|
|
self.conf.register_global(**default_global)
|
|
|
|
@commands.command()
|
|
async def ancestry(self, ctx, *args):
|
|
name = args[0]
|
|
headers = {"Authorization": self.conf.pf2_token()}
|
|
req = requests.get(f"https://api.pathfinder2.fr/v1/pf2/ancestry?name={name}", headers=headers).json()
|
|
html = req["results"][0]["data"]["description"]["value"]
|
|
filename = f"ancestry-{name}"
|
|
image(filename, html)
|
|
await ctx.send(file=discord.File(open(f"ancestry/{name}.png", "rb")))
|
|
os.remove(f"{filename}.png")
|
|
os.remove(f"{filename}_cropped.png")
|
|
|
|
@commands.command()
|
|
@checks.is_owner()
|
|
async def sarenrae_token(self, ctx, *args):
|
|
self.conf.pf2_token.set(args[0])
|
|
return await ctx.send(f"Set token to {self.conf.pf2_token()}")
|