100111-批量为图片添加水印-参数可自定义-支持子孙文件夹下图片添加水印【源代码】

原标题：100111-批量为图片添加水印-参数可自定义-支持子孙文件夹下图片添加水印【源代码】

导读：

水印添加工具代码解析功能概述：该代码实现了一个命令行工具，用于为指定文件夹及其子文件夹中的图片添加水印。支持多种图片格式，包括.jpg、.jpeg、.png、.bmp、....

水印添加工具代码解析

功能概述：该代码实现了一个命令行工具，用于为指定文件夹及其子文件夹中的图片添加水印。支持多种图片格式，包括.jpg、.jpeg、.png、.bmp、.tiff、.webp。用户可以自定义水印内容、文字大小、透明度、间隔、旋转角度、文本颜色以及输出图片质量等参数。程序会在输入文件夹下创建一个指定名称（默认为output）的子文件夹，将添加水印后的图片保存到该子文件夹中。
操作步骤
- 运行程序：启动代码后，程序首先会打印作者相关信息以及操作提示。
- 输入参数：
  - 水印内容：程序提示“请输入水印文字内容: ”，用户输入想要添加的水印文字。
  - 图片文件夹路径：接着提示“请输入图片所在文件夹路径支持子孙文件夹内图片: ”，用户输入包含图片的文件夹路径，程序会处理该文件夹及其所有子文件夹中的图片。
  - 水印文字大小：提示“请输入水印文字大小建议12可自定义: ”，用户输入水印文字的大小，例如12。
  - 水印透明度：提示“请输入水印透明度（0到1之间的小数）建议0.1可自定义: ”，用户输入0到1之间的小数来设置水印透明度，如0.1。
  - 水印间隔：提示“请输入两个水印文字之间的间隔建议16可自定义: ”，用户输入水印文字之间的间隔值，例如16。
  - 水印旋转角度：提示“请输入水印旋转角度建议35可自定义: ”，用户输入水印的旋转角度，如35度。
  - 输出图片质量：提示“请输入输出图片质量建议100可自定义: ”，用户输入输出图片的质量值，一般输入100。
  - 水印文本颜色：提示“请输入水印文本颜色（灰色#C0C0C0 白色#FFFFFF 金黄色#FFD700 建议白色）: ”，用户输入表示颜色的十六进制值，如#FFFFFF。
- 处理图片：程序根据用户输入的参数生成水印，并为指定文件夹及其子文件夹中的所有图片添加水印。添加水印后，图片会被保存到输入文件夹下的output文件夹（可自定义名称）中。处理过程中，控制台会显示每张图片添加水印成功及输出路径的信息。
- 决定是否继续：程序最后询问“是否要继续运行程序？(y/n): ”，若用户输入y（不区分大小写），可重复上述步骤处理其他文件夹的图片；若输入其他字符，程序终止运行。
代码结构分析
- 函数定义：
  - 图片类型判断：is_image_file函数用于检查文件是否为支持的图片类型，通过比较文件扩展名来判断。
  - 添加水印并保存：add_mark函数负责打开图片，调用传入的水印生成函数添加水印，然后根据图片格式和输出质量保存图片到指定的输出文件夹，并在控制台打印保存信息。
  - 设置透明度：set_opacity函数通过调整图片的alpha通道来设置水印的透明度。
  - 裁剪图片：crop_image函数用于裁剪图片边缘的空白部分，通过比较图片与空白背景的差异获取边界框来裁剪。
  - 生成水印：gen_mark函数创建水印图片，设置文字、裁剪空白、调整透明度，并返回一个用于在原图上添加水印的函数。该函数会根据斜边长度创建大图，在大图上按一定规律排列水印，旋转大图后粘贴到原图上。
  - 处理文件夹图片：process_images_in_folder函数使用os.walk遍历指定文件夹及其子文件夹，对其中的图片文件调用add_mark函数添加水印。
  - 主函数：main函数使用argparse处理命令行参数。若未通过命令行传入参数，则通过用户输入获取水印内容、图片文件夹路径等各项参数。调用gen_mark生成水印函数，再调用process_images_in_folder处理图片，并在每次处理后询问用户是否继续运行。
对子文件夹及穿透情况：代码使用os.walk函数遍历args.file指定的文件夹，这表明它不仅对主文件夹有效，还能穿透主文件夹下的所有子文件夹，对其中符合图片类型要求的文件进行水印添加操作。

实例验证：假设存在以下文件夹及文件结构：

主文件夹
│
├── 文件夹 A
│ ├── img1.jpg
│ ├── img2.png
│ └── img3.webp
├── 文件夹 B
│ ├── img4.bmp
│ └── img5.tiff

运行程序后，依次输入水印内容（如“版权所有”）、主文件夹路径、水印文字大小（如12）、水印透明度（如0.1）、水印间隔（如16）、水印旋转角度（如35）、输出图片质量（如100）、水印文本颜色（如#FFFFFF）。
程序会遍历主文件夹及其子文件夹，为img1.jpg、img2.png、img3.webp、img4.bmp、img5.tiff添加水印，并将添加水印后的图片保存到主文件夹下的output文件夹中。处理过程中，控制台会显示每张图片添加水印成功及输出路径的信息。
处理完成后，若用户选择继续运行程序，可再次输入不同的参数进行新一轮的图片水印添加操作。