Question

所以我有此功能可以返回用户头像的平均颜色：

import discord
from discord.ext import commands
import asyncio
from PIL import Image
import requests
from io import BytesIO

class Bot(commands.Bot):

    ...

    @staticmethod
    async def get_average_colour(image_url, default=0x696969):
        try:
            resp = requests.get(image_url)
            assert resp.ok
            img = Image.open(BytesIO(resp.content))
            img2 = img.resize((1, 1), Image.ANTIALIAS)
            colour = img2.getpixel((0, 0))
            res = "{:02x}{:02x}{:02x}".format(*colour)
            return int(res, 16)
        except:
            return default

    ...

这有效，但是问题在于它使用了requests，它被阻止了。所以我尝试改用aiohttp：

import discord
from discord.ext import commands
import asyncio
from PIL import Image
import aiohttp
from io import BytesIO

class Bot(commands.Bot):

    ...

    @staticmethod
    async def get_average_colour(image_url, default=0x696969):
        try:
            async with aiohttp.ClientSession() as session:
                async with session.get(image_url) as resp:
                    if resp.status != 200:
                        raise Exception
                    img = Image.open(BytesIO(await resp.read()))
            colour = img.resize((1, 1), Image.ANTIALIAS).getpixel((0, 0))
            return int("{:02x}{:02x}{:02x}".format(*colour), 16)
        except:
            return default

    ...

当我尝试查找random cat image link的平均颜色时，该函数可以正常工作，但是当我尝试使用用户的avatar_url调用此函数时，该函数始终返回默认值。有谁知道该功能出了什么问题？

Answer 1

似乎调用answer.upper()而不是async with session.get(str(image_url)) as resp:有用。

Answer 2

一个好方法是将image_url转换为字符串。这样，它将始终保留为字符串，而不是discord.Asset对象。

discord.py-无法获得用户头像的平均颜色

2 个答案: