diff options
author | AnInternetTroll <lucafulger@gmail.com> | 2020-07-07 16:32:50 +0000 |
---|---|---|
committer | AnInternetTroll <lucafulger@gmail.com> | 2020-07-07 16:32:50 +0000 |
commit | 5c499b146bf73b337040ceb1ae4bdb02dd0480ad (patch) | |
tree | a490a4c404c67442e55e9abdd230638d0bf61d29 /cogs | |
parent | 55cef1c7b6c6f4703eedae29965f5b1df5573a56 (diff) | |
download | steve-bot-5c499b146bf73b337040ceb1ae4bdb02dd0480ad.tar steve-bot-5c499b146bf73b337040ceb1ae4bdb02dd0480ad.tar.gz steve-bot-5c499b146bf73b337040ceb1ae4bdb02dd0480ad.tar.bz2 steve-bot-5c499b146bf73b337040ceb1ae4bdb02dd0480ad.tar.lz steve-bot-5c499b146bf73b337040ceb1ae4bdb02dd0480ad.tar.xz steve-bot-5c499b146bf73b337040ceb1ae4bdb02dd0480ad.tar.zst steve-bot-5c499b146bf73b337040ceb1ae4bdb02dd0480ad.zip |
/no longer blocks the bot
Diffstat (limited to 'cogs')
-rwxr-xr-x | cogs/utils.py | 80 |
1 files changed, 42 insertions, 38 deletions
diff --git a/cogs/utils.py b/cogs/utils.py index be0d09c..012a8c3 100755 --- a/cogs/utils.py +++ b/cogs/utils.py @@ -16,6 +16,7 @@ from selenium.webdriver.chrome.options import Options #import image as Image from PIL import Image from PIL import ImageFilter +import functools def set_viewport_size(driver, width, height): window_size = driver.execute_script(""" @@ -35,6 +36,45 @@ async def reportStuff(self, ctx, message): await channel.send(embed=embed) await ctx.author.send("Report has been submitted") +def save_leaderboard(): + DRIVER = '/usr/lib/chromium-browser/chromedriver' + chrome_options = Options() + chrome_options.add_argument("--disable-dev-shm-usage") + chrome_options.add_argument("--headless") + chrome_options.add_argument("--no-sandbox") + chrome_options.add_argument("--disable-gpu") + #chrome_options.binary_location = "" + driver = webdriver.Chrome(DRIVER, chrome_options=chrome_options) + set_viewport_size(driver, 1000, 1000) + driver.get('https://aninternettroll.github.io/mcbeVerifierLeaderboard/') + screenshot = driver.find_element_by_id('table').screenshot('leaderboard.png') + driver.quit() + #transparency time + img = Image.open('leaderboard.png') + img = img.convert("RGB") + pallette = Image.open("palette.png") + pallette = pallette.convert("P") + img = img.quantize(colors=256, method=3, kmeans=0, palette=pallette) + img = img.convert("RGBA") + datas = img.getdata() + + newData = [] + for item in datas: + if item[0] == 255 and item[1] == 255 and item[2] == 255: + newData.append((255, 255, 255, 0)) + else: + newData.append(item) + + img.putdata(newData) + """ + img = img.filter(ImageFilter.SHARPEN) + img = img.filter(ImageFilter.SHARPEN) + img = img.filter(ImageFilter.SHARPEN) + """ + #height, width = img.size + #img = img.resize((height*10,width*10), resample=Image.BOX) + img.save("leaderboard.png", "PNG") + class Utils(commands.Cog): def __init__(self, bot): @@ -177,44 +217,8 @@ class Utils(commands.Cog): async def leaderboard(self, ctx): """Leaderboard of the people that matter""" async with ctx.typing(): - DRIVER = '/usr/lib/chromium-browser/chromedriver' - chrome_options = Options() - chrome_options.add_argument("--disable-dev-shm-usage") - chrome_options.add_argument("--headless") - chrome_options.add_argument("--no-sandbox") - chrome_options.add_argument("--disable-gpu") - #chrome_options.binary_location = "" - driver = webdriver.Chrome(DRIVER, chrome_options=chrome_options) - set_viewport_size(driver, 1000, 1000) - driver.get('https://aninternettroll.github.io/mcbeVerifierLeaderboard/') - screenshot = driver.find_element_by_id('table').screenshot('leaderboard.png') - driver.quit() - #transparency time - img = Image.open('leaderboard.png') - img = img.convert("RGB") - pallette = Image.open("palette.png") - pallette = pallette.convert("P") - img = img.quantize(colors=256, method=3, kmeans=0, palette=pallette) - img = img.convert("RGBA") - datas = img.getdata() - - newData = [] - for item in datas: - if item[0] == 255 and item[1] == 255 and item[2] == 255: - newData.append((255, 255, 255, 0)) - else: - newData.append(item) - - img.putdata(newData) - """ - img = img.filter(ImageFilter.SHARPEN) - img = img.filter(ImageFilter.SHARPEN) - img = img.filter(ImageFilter.SHARPEN) - """ - #height, width = img.size - #img = img.resize((height*10,width*10), resample=Image.BOX) - img.save("leaderboard.png", "PNG") - + lbFunc = functools.partial(save_leaderboard) + await self.bot.loop.run_in_executor(None, lbFunc) await ctx.send(file=discord.File("leaderboard.png")) |