Как отправить обработанное фото через discord.py?

Question

[Роман Гончаров]

У меня есть функция задавания контраста фото.
Я попробовал что-то типа этого:

#Если что то это Pillow
@bot.command()
async def imag(ctx: context):
    # enh = ImageEnhance.Contrast(file)
    for attach in ctx.message.attachments:
        await ctx.send(file=ImageEnhance.Contrast(attach).enhance(100))

Но не получилось из-за ошибки:

discord.ext.commands.errors.CommandInvokeError: Command raised an exception: AttributeError: 'Attachment' object has no attribute 'convert'

Какого атрибута ему не хватает я так и не понял. Может кто сможет помочь? Тут я просто меняю контраст. Я бы не хотел каждый раз сохранять фото, поэтому лучше обойтись без скачивания.

Comments

[shurshur]

Скорее всего, ImageEnhance хочет класса Image, а получает класс Attachment. У класса Image действительно есть метод convert.

Решение - надо аттач скачать и превратить в Image. Что-то типа

r = requests.get(url_of_image)
image_data = r.read()
image_bytesio = io.BytesIO(image_data)
attach_image = Image.open(image_bytesio)

[Роман Гончаров]

shurshur, Метода read() нет в модуле. Есть метод .content. Как раз он и сработал:

@bot.command()
async def contrast(ctx: context, contrast:int):
    await ctx.message.delete()
    for attach in ctx.message.attachments:
        # print(attach.url)
        r = requests.get(attach.url)
        image_data = r.content
        image_bytesio = io.BytesIO(image_data)
        attach_image = Image.open(image_bytesio)
        enh = ImageEnhance.Contrast(attach_image)
        enh.enhance(contrast).save('1.png')
        await ctx.send(file=discord.File('1.png'))

@contrast.error
async def contrast_error(ctx:context, error:Exception):
    await ctx.send('Возможно вы ввели не число, либо отправили не картинку.')

Без сохранения не обошлось, но зато я получаю заведомый результат. Спасибо!

[shurshur]

Роман Гончаров, вроде я испытывал проблемы где-то с доступом к полю content - он мог быть пустым, пока не сделаешь read. Не помню обстоятельств. Возможно, от версии библиотеки зависит.

[Максим Невзоров]

Роман Гончаров, небольшой совет: вы можете сохранять файлы в память через BytesIO, но помните что их так же нужно закрывать, иначе есть вероятность создать утечку памяти.

А так же - не используйте requests в асинхронном коде. requests будет блокировать весь код до окончания запроса. В некоторых случаях это может занять слишком много времени, вплоть до разрыва соединения бота.

import aiohttp

@bot.command()
async def contrast(ctx: context, contrast:int):
    await ctx.message.delete()
    for attach in ctx.message.attachments:
        # print(attach.url)
        async with aiohttp.ClientSession() as session:
            async with session.get(attach.url) as resp:
                source_image_file = io.BytesIO(await resp.read())
        result_image_file = io.BytesIO()
        attach_image = Image.open(source_image_file)
        enh = ImageEnhance.Contrast(attach_image)
        enh.enhance(contrast).save(result_image_file)
        source_image_file.close()   # закрываем исходный файл
        attach_image.close()        # закрываем файл изображения, созданный Pillow
        await ctx.send(file=discord.File(result_image_file, filename=attach.filename))
        result_image_file.close()   # закрываем отправленный файл