前言
本文提供给图片添加文字或者logo图片水印的python工具,打造专属图片。
环境依赖
ffmpeg环境安装,可以参考我的另一篇文章:windows ffmpeg安装部署_阿良的博客-CSDN博客
ffmpy安装:
pip install ffmpy -i https://pypi.douban.com/simple
代码
上代码。
#!/user/bin/env python # coding=utf-8 """ @project : csdn @author : 剑客阿良_ALiang @file : image_add_watermark_tool.py @ide : PyCharm @time : 2021-11-20 14:18:13 """ import os import uuid from ffmpy import FFmpeg # 图片加文字水印 def image_add_text( image_path: str, output_dir: str, text: str, font_name: str, font_size=100, position=(0, 0), font_color='blue', box=1, box_color='white'): ext = _check_format(image_path) result = os.path.join(output_dir, '{}.{}'.format(uuid.uuid4(), ext)) ff = FFmpeg( inputs={ image_path: None}, outputs={ result: '-vf drawtext=\"fontsize={}:fontfile={}:text=\'{}\':x={}:y={}:fontcolor={}:box={}:boxcolor={}\"'.format( font_size, font_name, text, position[0], position[1], font_color, box, box_color)}) print(ff.cmd) ff.run() return result # 图片添加logo def image_add_logo( image_path: str, output_dir: str, logo_path: str, position=(0, 0)): ext = _check_format(image_path) result = os.path.join(output_dir, '{}.{}'.format(uuid.uuid4(), ext)) filter_cmd = '-vf \"movie=\'{}\' [wm];[in] [wm]overlay={}:{} [out]\"' ff = FFmpeg( inputs={ image_path: None}, outputs={ result: filter_cmd.format(logo_path, position[0], position[1])}) print(ff.cmd) ff.run() return result def _check_format(image_path: str): ext = os.path.basename(image_path).strip().split('.')[-1] if ext not in ['png', 'jpg']: raise Exception('format error') return ext
代码说明
1、image_add_text方法给图片添加文字水印方法,主要参数为:图片路径、输出目录、
水印文字、字体名称、字体大小(默认100)、文字左上角坐标(默认0:0)、文字颜色(默认蓝色)、是否需要背景(默认需要为1,不需要为0)、背景色(默认白色)。
2、image_add_logo方法给图片添加logo,主要参数为:图片路径、输出目录、logo图片地址、文字左上角坐标(默认0:0)。
3、注意logo地址如果有类似C:/这种windows盘的路径情况,需要对":"进行转义。后面验证的时候,可以看看我给的参数。
4、文件名为了避免重复,采用了uuid作为文件名。
5、图片后缀格式校验只有两种,按需添加即可。