本文章是第一次更新,更新于 2022年3月28日
方案设想来自2019年的文章 使用 Python 对接 PicGo,实现自动添加水印并上传 (写代码的明哥)
此方案部分代码已过时,由于pyclipboard
库不支持windows
,遂做出修改。编写一个能在windows
上使用的剪贴板图片自动加水印工具,即按下快捷键自动给剪贴板(复制)的图片添加水印。 改动如下
Pynput
全局快捷键监听器win32clipboard
对windows
剪贴板进行操作plyer
进行windows
弹窗通知pyinstaller
进行exe包装PS:我踩了很多坑,这些坑都被记录下来,帮你们填完。
下面是我参考过的文档
如果文章字体过小,请调整浏览器页面缩放。Windows: Ctrl + 鼠标滚轮
本篇文章代码注释使用了 vscode 的 better-comments 拓展
按下快捷键后,你可以看到四层水印,分别在左上,左下,中间,右下。
我将程序分为了三个部分,通知处理模块Notify.py
,图片处理模块Image.py
,键盘监听模块Work.py
graph TB A(AutoWatermark.exe) --- Notify[Notify.py] A --- Image[Image.py] Notify --- Work(Work.py) Image --- Work
用于右下角的系统通知
Install the plyer
module with pip
.
pip install plyer
plyer
comes with a class called notification
, which helps us create a notification. Import it like this:
from plyer import notification
For Example
#import notification from plyer module 导入 player 里的 notification 方法
from plyer import notification
#import time 导入时间库
import time
#Use while loop to create notifications indefinetly 死循环
while(True):
#notification 进行系统提示
notification.notify(
title = "Reminder to take a break",
message = '''Drink water, take a walk''',
timeout = 60
)
#System pause the execution of this programm for 60 minutes 每60分钟执行一次
time.sleep(60*60)
Windows有效的可选参数:
title (str): 显示通知标题
message (str): 显示通知信息
app_icon (str): 显示通知图标,注意windows只能使用ico
timeout (int): 通知显示时间
对notification进行简单封装
from plyer import notification
import os
import random
import sys
path = os.path.dirname(__file__)
appname = "自动水印"
def resource_path(relative_path):
""" Get absolute path to resource, works for dev and for PyInstaller """
try:
# PyInstaller creates a temp folder and stores path in _MEIPASS
base_path = sys._MEIPASS
except Exception:
base_path = os.path.abspath(".")
return os.path.join(base_path, relative_path)
def notify(title, message, icon):
notification.notify(
title=title,
message=message,
app_name=appname,
app_icon=resource_path('{}.ico'.format(icon)),
timeout=2
)
代码当中有一个 resource_path
方法,这个是取当前文件夹路径,是为了打包成exe之后能够使用依赖资源,因为使用的不是exe当前的目录,而是temp缓存目录,pyinstaller
打包出来的可执行程序会将temp
目录保存到 sys._MEIPASS
,os.path.abspath(".")
通过绝对路径方法获取当前目录位置。
注意:执行exe时 os.path.dirname(__file__)
将会出错无法得到正确路径,只能用于本地测试。
PIL 是 Python 的图像处理标准库
https://www.liaoxuefeng.com/wiki/1016959663602400/1017785454949568 PIL介绍与使用
https://blog.csdn.net/weixin_43790276/article/details/108478270 PIL简单图片处理
Windows 剪贴板仅支持 BMP 位图格式,所以我们需要对Image进行格式转换。存入剪贴板的数据需要去除BMP格式的头部信息(位图信息)。
图片转化后,我们需要一组二进制字节流,用于存储图片的数据,再从二进制流里读取数据放入剪贴板中。
img_byte_arr = io.BytesIO() # 生成二进制字节流
img.save(img_byte_arr, format='BMP') # 将图片输入二进制字节流中
img_byte_arr = img_byte_arr.getvalue()[14:] # 取出二进制字节流并切片截取
存入剪贴板需要使用 win32clipboard
库,需要 pip install pywin32
https://stackoverflow.com/questions/1815165/draw-bold-italic-text-with-pil
添加文字框的方法中发现并没有字体变粗的属性,具体实现有两种
切换字体,像微软雅黑UI 有三种字体,分别是 粗、普通、细,文件名也不同
描边,对 drawTXT.text
添加参数,最好颜色一致,这种可以自定义字体粗的程度
drawTXT.text((x/2, y/2), "xiaoneng1024", font=font, fill=(220, 220, 220, 16), anchor='mm',stroke_width=2,stroke_fill = "black")
import io
import os
import random
import win32clipboard
import time
from PIL import Image, ImageFont, ImageDraw, ImageGrab
from notify import notify
path = os.path.dirname(file)
def main():
# ^ 运行主流程
img = getClipboard()
if img == None:
notify('AutoWatermark', '剪贴板不是图片数据', 'warn')
return
# print(img.size, img.mode)
try:
img = imgProcessing(img)
img_bytes = imgToBytes(img)
res = setClipboard(img_bytes)
except:
notify('AutoWatermark', '异常错误 ERROR', 'file')
return
if res == True:
notify('AutoWatermark', '图片已自动添加水印', 'file')
else:
notify('AutoWatermark', '剪贴板读取失败:拒绝访问', 'warn')
def getClipboard():
# ^ 获取剪贴板数据
try:
img = ImageGrab.grabclipboard()
except:
return None
return img
def imgProcessing(img):
# ^ 图像处理
# @ 将图像转换为RGBA,添加通道值方便后期与字体图层叠加
img = img.convert("RGBA")
x, y = img.size
fontsize = 18
# @ 根据截取大小调整字体大小
if x*y > 1280 * 720:
fontsize = 28
elif x*y > 1600 * 900:
fontsize = 36
elif x*y > 1920 * 1080:
fontsize = 48
# @ 选择字体,这里我选择微软雅黑
font = ImageFont.truetype("msyh.ttc", fontsize)
# @ 新建一个存文字框的图层
txt = Image.new('RGBA', img.size, (0, 0, 0, 0))
# @ 创建Draw对象,可以对图层绘图操作
drawTXT = ImageDraw.Draw(txt)
drawTXT.text((x-4, y-4), "cnblogs.com/linxiaoxu", font=font, fill=(220, 220, 220, 156), anchor='rs') # stroke_width=2,stroke_fill = "black"
drawTXT.text((x/2, y/2), "xiaoneng1024", font=font, fill=(220, 220, 220, 16), anchor='mm')
drawTXT.text((4, 4), time.strftime("%Y-%m-%d", time.gmtime()), font=font, fill=(220, 220, 220, 24), anchor='la') # stroke_width=2,stroke_fill = "black"
drawTXT.text((4, y), "小能1024", font=font, fill=(220, 220, 220, 24), anchor='lb')
# @ 将两个图层合并
img = Image.alpha_composite(img, txt)
return img
def imgToBytes(img):
# ^ 将图像转换为二进制流,并裁剪头部信息
img_byte_arr = io.BytesIO()
img.save(img_byte_arr, format='BMP')
img_byte_arr = img_byte_arr.getvalue()[14:]
# print(img_byte_arr, type(img_byte_arr))
return img_byte_arr
def setClipboard(img_byte_arr):
# ^ 设置系统剪贴板数据
try:
win32clipboard.OpenClipboard()
win32clipboard.EmptyClipboard()
win32clipboard.SetClipboardData(win32clipboard.CF_DIB, img_byte_arr)
win32clipboard.CloseClipboard()
return True
except:
return False
if name == 'main':
main()
我们需要有一个对全局键盘事件进行监听并响应事件的线程,这时 pynput
包作用就出来了,用于控制和监视鼠标、键盘的类
https://pynput.readthedocs.io/en/latest/index.html
from pynput import keyboard
from image import main
from notify import notify
def on_activate():
print('Global hotkey activated!')
main()
def for_canonical(f):
return lambda k: f(l.canonical(k))
notify('AutoWatermark', '程序已启动!', 'warn')
hotkey = keyboard.HotKey(
keyboard.HotKey.parse('<ctrl>+<shift>+,'),
on_activate)
with keyboard.Listener(
on_press=for_canonical(hotkey.press),
on_release=for_canonical(hotkey.release)) as l:
l.join()
on_activate
是触发快捷键的方法
for_canonical(f)
是修饰器,返回一个匿名函数,为了调用 listener
的方法 canonical
使用户输入的按键规范化
奇数行是直接调用,不使用修饰器。偶数行使用修饰器,可以看到原本的 Key.alt_l
被规范成 Key.alt
第1次 Key.alt_l <enum 'Key'>
第1次 Key.alt <enum 'Key'>
第2次 Key.ctrl_l <enum 'Key'>
第2次 Key.ctrl <enum 'Key'>
hotkey = keyboard.HotKey(
keyboard.HotKey.parse('<ctrl>+<shift>+,'),
on_activate)
这个是设置HotKey对象,parse
方法方便转换组合的快捷键,on_activate
是触发时调用的方法
with keyboard.Listener(
on_press=for_canonical(hotkey.press),
on_release=for_canonical(hotkey.release)) as l:
l.join()
这个是阻塞式的用法,开始一个新线程监听键盘事件
pip install pyinstaller
在安装 pyinstaller
的过程中控制台报错,ModuleNotFoundError: No module named ‘pip‘
解决方法
python -m ensurepip
python -m pip install --upgrade pip
我们将三个py合并为一个py
在当前终端执行如下命令
pyinstaller --windowed --icon cat.ico -i cat.ico --onefile --add-data "L:/IT/Python/Codes/Work/220327_AutoWatermark/file.ico;." --add-data "L:/IT/Python/Codes/Work/220327_AutoWatermark/warn.ico;." --hidden-import plyer.platforms.win.notification --hidden-import PIL final.py
参数介绍
在当前终端执行如下命令
pyinstaller --windowed --icon cat.ico -i cat.ico --onefile --add-data "L:/IT/Python/Codes/Work/220327_AutoWatermark/file.ico;." --add-data "L:/IT/Python/Codes/Work/220327_AutoWatermark/warn.ico;." --hidden-import plyer.platforms.win.notification --hidden-import PIL work.py -p notify.py -p image.py
参数介绍
Traceback (most recent call last):
...
ModuleNotFoundError: No module named 'plyer.platforms'
Traceback (most recent call last):
File "notification test.py", line 3, in <module>
File "plyer\facades\notification.py", line 79, in notify
File "plyer\facades\notification.py", line 88, in _notify
NotImplementedError: No usable implementation found!
[12520] Failed to execute script 'notification test' due to unhandled exception!
https://stackoverflow.com/questions/67489963/failed-to-execute-script-pyinstaller
plyer 模块不会自动被打包,需要手动添加命令 --hidden-import plyer.platforms.win.notification
PIL库也可能未被打包,先打包一次,在当前目录下找到被打包py的spec文件,打开修改添加库所在的位置。我是win10,库文件夹目录在C:\Users\Administrator\AppData\Local\Programs\Python\Python310\Lib\site-packages
datas=[('C:/Users/Administrator/AppData/Local/Programs/Python/Python310/Lib/site-packages/PIL','PIL'),],
https://pypi.org/project/auto-py-to-exe/ 打包工具
https://www.imooc.com/article/286538 打包指南
https://www.zhihu.com/question/281858271/answer/611320245 打包文件太大了
import os
import io
import win32clipboard
import time
import sys
from PIL import Image, ImageFont, ImageDraw, ImageGrab
from pynput import keyboard
from plyer import notification
# path = os.path.dirname(__file__)
appname = "自动水印"
def resource_path(relative_path):
""" Get absolute path to resource, works for dev and for PyInstaller """
try:
# PyInstaller creates a temp folder and stores path in _MEIPASS
base_path = sys._MEIPASS
except Exception:
base_path = os.path.abspath(".")
return os.path.join(base_path, relative_path)
def notify(title, message, icon):
notification.notify(
title=title,
message=message,
app_name=appname,
app_icon=resource_path('{}.ico'.format(icon)),
timeout=19
)
def main():
img = getClipboard()
if img == None:
notify('AutoWatermark', '剪贴板不是图片数据', 'warn')
return
# print(img.size, img.mode)
try:
img = imgProcessing(img)
img_bytes = imgToBytes(img)
res = setClipboard(img_bytes)
except:
notify('AutoWatermark', '异常错误 ERROR', 'file')
return
if res == True:
notify('AutoWatermark', '图片已自动添加水印', 'file')
else:
notify('AutoWatermark', '剪贴板读取失败:拒绝访问', 'warn')
def getClipboard():
try:
img = ImageGrab.grabclipboard()
except:
return None
return img
def imgProcessing(img):
img = img.convert("RGBA")
x, y = img.size
fontsize = 18
if x*y > 1280 * 720:
fontsize = 28
elif x*y > 1600 * 900:
fontsize = 36
elif x*y > 1920 * 1080:
fontsize = 48
font = ImageFont.truetype("msyh.ttc", fontsize)
txt = Image.new('RGBA', img.size, (0, 0, 0, 0))
drawTXT = ImageDraw.Draw(txt)
drawTXT.text((x-4, y-4), "cnblogs.com/linxiaoxu", font=font, fill=(220, 220, 220, 156), anchor='rs') # stroke_width=2,stroke_fill = "black"
drawTXT.text((x/2, y/2), "xiaoneng1024", font=font, fill=(220, 220, 220, 16), anchor='mm')
drawTXT.text((4, 4), time.strftime("%Y-%m-%d", time.gmtime()), font=font, fill=(220, 220, 220, 24), anchor='la') # stroke_width=2,stroke_fill = "black"
drawTXT.text((4, y), "小能1024", font=font, fill=(220, 220, 220, 24), anchor='lb')
img = Image.alpha_composite(img, txt)
return img
def imgToBytes(img):
img_byte_arr = io.BytesIO()
img.save(img_byte_arr, format='BMP')
img_byte_arr = img_byte_arr.getvalue()[14:]
# print(img_byte_arr, type(img_byte_arr))
return img_byte_arr
def setClipboard(img_byte_arr):
try:
win32clipboard.OpenClipboard()
win32clipboard.EmptyClipboard()
win32clipboard.SetClipboardData(win32clipboard.CF_DIB, img_byte_arr)
win32clipboard.CloseClipboard()
return True
except:
return False
def on_activate():
print('Global hotkey activated!')
main()
def for_canonical(f):
return lambda k: f(l.canonical(k))
notify('AutoWatermark', '程序已启动!', 'warn')
hotkey = keyboard.HotKey(
keyboard.HotKey.parse('<ctrl>+<shift>+,'),
on_activate)
with keyboard.Listener(
on_press=for_canonical(hotkey.press),
on_release=for_canonical(hotkey.release)) as l:
l.join()
手机扫一扫
移动阅读更方便
你可能感兴趣的文章