Question

我有这个脚本，它成功地将水印应用于图像，调整水印大小并设置图像“质量”。我发现它实际上降低了质量并将其调整为比我想要的小。

这是我正在使用的电报机器人，我将图像发送给机器人，机器人正在添加水印，然后调整大小。我根本不想要调整大小的部分。我要原尺寸+水印。

问题：如何更改此设置，使其不会调整添加水印的原始图像的大小？

电流

我将图像发送到我的机器人以激活脚本
脚本添加水印并调整图像大小
脚本将调整后的图像连同水印一起发送给我

期望的流量

我将图像发送到我的机器人以激活脚本
脚本添加了水印但不会调整原始图像的大小
脚本将包含水印的图像发送回给我

代码 1

# -*- coding: utf-8 -*-
import telebot
import config
import random
import os
import time
from tools.tools import getting_ready

bot = telebot.TeleBot(config.BOT_TOKEN)


@bot.message_handler(commands=['start', 'help'])
def send_welcome(message):
    bot.send_message(message.chat.id, 'Hi, Send or Forward me a PHOTO!')


@bot.message_handler(func=lambda message: True)
def echo_all(message):
    bot.reply_to(message, 'Send /start or /help')


@bot.message_handler(content_types=["photo"])
def send_watermark(message):
    chat_id = message.chat.id
    sent_message = bot.reply_to(message, 'Downloading...')
    message_id = sent_message.message_id
    file = bot.get_file(message.photo[0].file_id)
    downloaded_file = bot.download_file(file.file_path)
    bot.edit_message_text(f'Downloaded!\n\nNow, Genarating Watermarks...\n\ncurrent watermark: {config.WATERMARK}', chat_id=chat_id, message_id=message_id)
    __path = 'images/' + str(random.randint(100000, 999999)) + '.tmp'
    with open(__path, 'wb') as f:
        f.write(downloaded_file)
    fname = getting_ready(__path)
    os.remove(__path)
    time.sleep(2)
    bot.edit_message_text('Now, Uploading...', chat_id=chat_id, message_id=message_id)
    for i in ('black','white',):
        __file = 'images/out/{}/{}'.format(i, fname)
        __photo = open(__file, 'rb')
        bot.send_photo(message.chat.id, __photo, reply_to_message_id=message.message_id)
        os.remove(__file)
    bot.delete_message(chat_id=chat_id, message_id=message_id)


print('watermark bot started successfully!')
bot.polling()

代码 2

from PIL import Image, ImageFont, ImageDraw, ImageEnhance
import hashlib
import piexif
import config


def getting_ready(path):
    fname = md5(path) + '.jpg'
    image = Image.open(path).convert('RGB')
    try:
        exif_dict = piexif.load(image.info['exif'])
        orientation = exif_dict['0th'][274]
    except KeyError:
        orientation = None
    rotate_values = {3: 180, 6: 270, 8: 90}
    if orientation in rotate_values:
        image = image.rotate(rotate_values[orientation], expand=True)
    for color in ['black', 'white']:
        watermark(image, fname, color)
    return fname


def watermark(img, new_fname, color):
    text = config.WATERMARK
    wm = Image.new('RGBA', img.size, (0, 0, 0, 0))
    fontsize = img.size[1] // 100 * config.FONT_SIZE
    font = ImageFont.truetype(f'tools/fonts/{config.FONT_NAME}', fontsize)
    indent = fontsize // 8
    w, h = font.getsize(text)
    text_position = (img.size[0] - w - indent, img.size[1] - h - indent)
    draw = ImageDraw.Draw(wm, 'RGBA')
    draw.text(text_position, text, font=font, fill=color)
    alpha = wm.split()[3]
    alpha = ImageEnhance.Brightness(alpha).enhance(config.TRANSPARENCY)
    wm.putalpha(alpha)
    out_path = 'images/out/{}/{}'.format(color, new_fname)
    Image.composite(wm, img, wm).save(out_path, 'JPEG', optimize=False, quality=config.QUALITY)


def md5(path):
    with open(path, 'rb') as f:
        md5hash = hashlib.md5()
        for chunk in iter(lambda: f.read(4096), b''):
            md5hash.update(chunk)
    return md5hash.hexdigest()

Answer 1

这可能不是解决此问题的最佳方法，但它对我来说很有效，可以将其恢复到正常大小

double = 2.0
    Image.composite(wm, img, wm).resize( [int(double * s) for s in img.size] ).save(out_path, 'JPEG', optimize=True, quality=config.QUALITY)

Python 水印 - 防止图像尺寸和质量降低

1 个答案: