Python加水印

因为写在博客上的文章（部分）会给盗取，所以老早之前就写了一个加水印的程序

后来发现，因为防止文件名重复，我都是用一种随机的方式生成的文件名

代码语言：javascript复制

filename=sys.argv[i]
suffix=filename[-3:]
temp_name=d.datetime.now().strftime("%M-%S") "." str(suffix)

但是这样一来就会打乱文件顺序（因为我命名都是1 2 3 4这样），不好管理，所以就稍微修改了一下，发出来图一乐

先导入库

代码语言：javascript复制

import os
import sys
import time
import datetime as d
from PIL import Image, ImageDraw, ImageFont

因为一次性需要多个图片的问题，所以最开始写的时候想的是，文件拖过去就能执行了；获取文件名的方式就变成了

代码语言：javascript复制

...
for i in range(1,100):
    filename=sys.argv[i]

就一直循环，看有没有参数

接着设置水印内容

代码语言：javascript复制

...
text = "@blog.ernket.top"

下面就是加水印的操作了（水印固定右下角）

代码语言：javascript复制

...
image = Image.open(filename)
font = ImageFont.truetype('C:\Windows\Fonts\微软雅黑\msyh.ttc',18)
layer = image.convert('RGBA')
text_overlay = Image.new('RGBA', layer.size)
image_draw = ImageDraw.Draw(text_overlay)

text_size_x, test_size_y = image_draw.textsize(text,font=font)

text_xy = (layer.size[0] - text_size_x, layer.size[1] - test_size_y)
image_draw.text(text_xy,text,font=font,fill='#000000') #字体颜色
after = Image.alpha_composite(layer,text_overlay)
temp_name=d.datetime.now().strftime("%M-%S") "." str(suffix)
after.save(temp_name)

文件水印加完了，现在考虑的是文件名的问题，我的做法是删除源文件防止报 文件已存在的错误，然后在重命名 （也许可以直接覆盖吧，懒得写

代码语言：javascript复制

...
os.remove(filename)
os.rename(temp_name,filename)
print("[ ] " filename " 已完成水印工作，正在替换文件名....  " str(temp_name " --> " filename))
time.sleep(1)

到这就写完了，当然for循环的时候会报错，因为给的参数不同，加个 try就好了

代码语言：javascript复制

import os
import sys
import time
import datetime as d
from PIL import Image, ImageDraw, ImageFont

print("Auther:Elapse")
for i in range(1,100):
    try:
        filename=sys.argv[i]
        suffix=filename[-3:]
        image = Image.open(filename)
        text = "@blog.ernket.top"
        font = ImageFont.truetype('C:\Windows\Fonts\微软雅黑\msyh.ttc',18)
        layer = image.convert('RGBA')
        text_overlay = Image.new('RGBA', layer.size)
        image_draw = ImageDraw.Draw(text_overlay)

        text_size_x, test_size_y = image_draw.textsize(text,font=font)

        text_xy = (layer.size[0] - text_size_x, layer.size[1] - test_size_y)
        image_draw.text(text_xy,text,font=font,fill='#000000')
        after = Image.alpha_composite(layer,text_overlay)
        temp_name=d.datetime.now().strftime("%M-%S") "." str(suffix)
        after.save(temp_name)
        os.remove(filename)
        os.rename(temp_name,filename)
        print("[ ] " filename " 已完成水印工作，正在替换文件名....  " str(temp_name " --> " filename))
        time.sleep(1)
    except:
        break

for循环

0 人点赞