基于Python开发一个图像水印批量添加工具

class ImageWatermarker:
    def __init__(self, config):
        self.font_cache = {}
        self.load_config(config)
        
    def load_config(self, config):
        """加载配置文件"""
        self.watermark_text = config.get('text', '')
        self.position = config.get('position', 'bottom-right')
        self.opacity = config.get('opacity', 0.7)
        self.font_path = config.get('font_path', 'arial.ttf')
        self.min_font_size = config.get('min_font_size', 12)
        
    def process_folder(self, input_dir, output_dir):
        """批量处理目录"""
        pass
        
    def _add_watermark(self, image, text):
        """核心水印添加逻辑"""
        pass
        
    def _calculate_position(self, img_size, text_size):
        """智能位置计算"""
        pass

base_size = min(image_width, image_height) * ratio
adjusted_size = max(min_size, min(base_size, max_size))
if image_width < image_height:  # 竖版图片
    adjusted_size *= 1.2

def calculate_position(img_width, img_height, text_width, text_height, position):
    margin = min(img_width, img_height) * 0.02  # 动态边距
    position_map = {
        'top-left': (margin, margin),
        'top-center': ((img_width - text_width)/2, margin),
        'top-right': (img_width - text_width - margin, margin),
        # 其他位置计算...
    }
    return position_map[position]

from PIL import ImageFilter

def render_text(draw, position, text, font, opacity):
    # 先渲染大尺寸再缩小实现抗锯齿
    large_size = (int(font.size * 1.5),) * 2
    large_layer = Image.new('RGBA', large_size)
    large_draw = ImageDraw.Draw(large_layer)
    large_draw.text((0,0), text, font=font, fill=(255,255,255,255))
    
    # 高质量缩小
    small_layer = large_layer.resize(
        (font.size, font.size), 
        Image.Resampling.LANCZOS)
    small_layer.putalpha(int(255*opacity))
    
    # 合成到目标位置
    base_image.paste(small_layer, position, small_layer)

import subprocess

def preserve_metadata(original_path, processed_path):
    try:
        # 使用exiftool转移元数据
        subprocess.run([
            'exiftool', 
            '-TagsFromFile', original_path,
            '-overwrite_original',
            processed_path
        ], check=True)
    except Exception as e:
        print(f"元数据转移失败: {str(e)}")

from multiprocessing import Pool, cpu_count

def batch_process(file_list, config):
    with Pool(processes=min(8, cpu_count())) as pool:
        results = []
        for file in file_list:
            res = pool.apply_async(
                process_single, 
                (file, config))
            results.append(res)
        
        for res in results:
            try:
                res.get(timeout=300)
            except Exception as e:
                print(f"处理超时: {str(e)}")

def smart_watermark(image):
    # 使用OpenCV分析图像特征区域
    import cv2
    gray = cv2.cvtColor(np.array(image), cv2.COLOR_RGB2GRAY)
    
    # 边缘检测
    edges = cv2.Canny(gray, 100, 200)
    
    # 寻找非重要区域
    contours, _ = cv2.findContours(edges, 
        cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
    
    # 计算最佳水印位置
    return optimal_position

import unittest
from io import BytesIO

class WatermarkTestCase(unittest.TestCase):
    def setUp(self):
        self.test_image = Image.new('RGB', (800,600), (255,255,255))
        
    def test_watermark_position(self):
        output = add_watermark(self.test_image, "TEST")
        # 使用图像识别验证水印位置
        
    def test_opacity_control(self):
        # 测试不同透明度效果
        pass
        
    def test_performance(self):
        # 性能基准测试
        start = time.time()
        for _ in range(100):
            add_watermark(self.test_image, "TEST")
        duration = time.time() - start
        self.assertLess(duration, 5.0)

def apply_color_profile(image, profile_path):
    try:
        with open(profile_path, 'rb') as f:
            profile = ImageCms.ImageCmsProfile(BytesIO(f.read()))
        return ImageCms.profileToProfile(
            image, profile, 'sRGB')
    except Exception as e:
        print(f"色彩管理失败: {str(e)}")
        return image

FROM python:3.9-slim

RUN apt-get update && \
    apt-get install -y libimage-exiftool-perl && \
    rm -rf /var/lib/apt/lists/*

WORKDIR /app
COPY requirements.txt .
RUN pip install -r requirements.txt

COPY . .
ENTRYPOINT ["python", "watermarker.py"]

import logging
from logging.handlers import RotatingFileHandler

def setup_logging():
    logger = logging.getLogger()
    logger.setLevel(logging.INFO)
    
    # 文件日志（自动轮转）
    file_handler = RotatingFileHandler(
        'watermark.log', maxBytes=10*1024*1024, backupCount=5)
    file_handler.setFormatter(logging.Formatter(
        '%(asctime)s - %(levelname)s - %(message)s'))
    
    # 控制台日志
    console_handler = logging.StreamHandler()
    
    logger.addHandler(file_handler)
    logger.addHandler(console_handler)