欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

Python对接PicGo实现图片自动加水印并上传操作示例

程序员文章站 2022-03-20 13:49:08
目录1. 天下苦搬运党久矣2. 目前的图床管理工具3. 方案的设想4. 代码完整解析4.1 定义热键并监听键盘第一步:先定义好你的热键第二步:监听所有的键盘动作4.2 从内存中读取图像4.3 添加水印...

1. 天下苦搬运党久矣

对于我这样经常需要写点文章的技术自媒体来说,很经常早上我才在公众号首发了原创文章,中午就有人同步到了知乎、今日头条等外部平台,并且拿到了该篇文章在这些平台的首发。

Python对接PicGo实现图片自动加水印并上传操作示例

平台的首发很重要,以微信的公众号平台来说,一篇文章能否在公众号标原创,就是通过检测是否在公众号平台上首发,如果公众号的文章库里没有搜索到与你文章相似度较高的文章,那么你就可以标原创了。

为了解决了首发的问题,我付费使用了 openwrite 这个平台,每个月 20 块钱的费用,一键就可以分发各大平台,非常的省心~

首发固然重要,但有时候也没那么重要。

因为有专门的培训机构(这里就不点名,避免给他们反向营销)就是拿你的文章去给自己的帐号堆干货,吸引关注,他们才不在乎原创不原创,只要文章能发布就行了。

这种人在知乎这种没有原创检测机制的平台,可以活得很好。

之前有一无良培训机构在知乎上生产了10几个号,批量搬运我以及一些朋友的原创文章。那时候,我每天都可以举报好多。

Python对接PicGo实现图片自动加水印并上传操作示例

渐渐地,我累了,自己写了几百篇的文章,如果一篇一篇去检查,那我这一天基本啥事都做不了,违权成本实在太高了。

考虑到我的文章都有非常多的图片,为了让这些人在白嫖我文章的同时,也能给我带来点收益(当然人家是不可能付费的,但至少能给我的公众号打打广告也是非常奈斯的)。

于是我就想啊,是不是可以自己写个工具,给自己的每张配图上都加上自己的水印,看他们还盗不盗。。

2. 目前的图床管理工具

在开始讲如何利用 python 来实现我的需求之前 ,我有必要介绍下我的图床管理工具。

我在写文章的时候,主要用到三款工具:

typora :markdown 文案的编辑snipaste:非常好用的截图工具picgo:非常人性的图床管理工具

Python对接PicGo实现图片自动加水印并上传操作示例

其中今天的要参与的主角是 picgo

Python对接PicGo实现图片自动加水印并上传操作示例

它对当下主流的图床平台都提供了很好的支持

Python对接PicGo实现图片自动加水印并上传操作示例

当我使用了 snipaste 截图后,再按住快捷键(⌘ ⇧ p),就可以立即将你的图床上传至指定图床,并且将上传后的链接以 markdown 的图片格式复制到剪切板中,你可以直接粘贴使用。

3. 方案的设想

由于这一整工具,我已经使用了三年,各种操作都非常的熟悉,对他们我已经产生了极度的依赖,因此现在我想要实现自动加水印的功能,也一定是建立在这套工具的基础上完成的。

snipaste 和 picgo 本身都不支持自定义水印,也不提供第三方插件的开发入口。

snipaste 和 picgo 之所以能够在一起工作,是因为有了剪切板这个桥梁,因此想要实现自己的需求,只能从剪切板上寻找突破口。

多余的废话就不多说了,我直接说下我的方案:

在 snipaste 将图像放入剪切板后由我敲入自定义的热键去触发 python 脚本去从剪切板中读取图像然后使用 pil 去给该图像加水印,重新放入剪切板中接着利用 python 脚本去自动化触发 picgo 的快捷键picgo 被激活后,就能将带有水印的图片上传到图床

为了让你对这个方案,有一个直观的理解,我特地画了一张流程图,其中虚线就是我得自己实现的功能。

Python对接PicGo实现图片自动加水印并上传操作示例

4. 代码完整解析

4.1 定义热键并监听键盘

在 python 中有一个 pynput 库,利用它可以来监听系统的键盘。

在它的官方文档中,很快找到了一个可以自定义热键组合的方案。

from pynput import keyboard

def on_activate():
    print('global hotkey activated!')

def for_canonical(f):
    return lambda k: f(l.canonical(k))

hotkey = keyboard.hotkey(
    keyboard.hotkey.parse('<ctrl>+<alt>+h'),
    on_activate)
with keyboard.listener(
        on_press=for_canonical(hotkey.press),
        on_release=for_canonical(hotkey.release)) as l:
    l.join()

但是很遗憾的是,这个功能目前来看是有 bug 的,我在 mac 上亲测是没有效的,而在 github 的 issue 中也有人在 2020 年8月反应过问题, 没想到到现在还没有解决

Python对接PicGo实现图片自动加水印并上传操作示例

虽然它本身提供的组合键监听模式无法使用,但普通的监听模式还是可以使用的,只要有这个做为基础,那我自己造*也不难实现组合热键的功能。

第一步:先定义好你的热键

upload_pic_set = {
    keyboard.key.ctrl.value.vk,
    keyboard.key.cmd.value.vk,
    keyboard.key.alt.value.vk,
    keyboard.keycode(35).vk
}

第二步:监听所有的键盘动作

with keyboard.listener(
        on_press=on_press,
        on_release=on_release) as listener:
    listener.join()

只要有一个键处于 press 的状态,就往列表中存放这个键

key_list = []

def on_press(key):
    if isinstance(key, keyboard.keycode):
        key_list.append(key.vk)
    elif isinstance(key, keyboard.key):
        key_list.append(key.value.vk)

    if set(key_list) == upload_pic_set:
        image = get_image_from_clipboard()
        new_image = make_watermark(image)
        put_image_to_clip(new_image)
        upload_image_via_picgo()
        notify_to_mac("成功添加水印并上传到图床")

但是一旦有键释放了,就要清空这个列表

def on_release(key):
    key_list.clear()

每一次按下键都会检查,key_list 是否等于 定义好的快捷键,如果刚好是相等,就可以开始图片的处理逻辑了。

if set(key_list) == upload_pic_set:
  pass

4.2 从内存中读取图像

pil 有一个 imagegrab 模块,在这个模块中有一个 grabclipboard 函数,它实现了从剪切板中读取图像的功能,不过读取的 rgb 格式,由于我们后面加水印时,必须使用 rgba 格式才可以,因此再用 convert 转一下。

  from pil import  imagegrab
  
  img_rgb = imagegrab.grabclipboard()
  image = img_rgb.convert("rgba")

4.3 添加水印生成新图像

以下是添加水印的代码,其实可能要注意的一点就是如果你的文字里包含中文,那么选择字体时一定要是中文字体,否则会出现方块字

def make_watermark(image):
    txt = image.new('rgba', image.size, (0, 0, 0, 0))
    fnt = imagefont.truetype("/system/library/fonts/stheiti medium.ttc", 20)
    draw = imagedraw.draw(txt)
    draw.text(((txt.size[0]-300)//2, txt.size[1]-40), "微信公众号: python编程时光", font=fnt, fill=(240, 49, 48, 255))
    draw.text(((txt.size[0]-300)//2, txt.size[1]-70), "未经授权请勿转载", font=fnt, fill=(240, 49, 48,  255))
    out = image.alpha_composite(image, txt)
    return out

4.4 将新图像重新放入剪切板

内置的 io 模块支持在内存中读写 bytes,只要pil 的 image 对象在 save 的时候保存保存在 bytesio 对象中,然后通过 pasteboard 模块从 bytesio 对象中载入数据,就可以实现往剪切板中放入图像的功能。

def put_image_to_clip(image):
    img_byte_arr = io.bytesio() 
    pb = pasteboard.pasteboard()

    image.save(img_byte_arr, format='png')
    img_byte_arr = img_byte_arr.getvalue()
    pb.set_contents(img_byte_arr, pasteboard.png)

pasteboard 载入的图像只支持 png 格式,因此在保存时,一定要指定 png。

此外 pasteboard 还支持更多格式的数据,比如 pdf,音频数据,html、颜色数据等等

更多格式可查看:https://developer.apple.com/documentation/appkit/nspasteboardtypestring

4.5 模拟触发 picgo

正常情况下,我们是通过快捷键来触发 picgo 去从剪切板中上传图像的,因此想要在程序中激活 picgo,也只需要在 python 脚本中模拟键盘动作即可。

具体的代码如下:

from pykeyboard import pykeyboard

def upload_image_via_picgo():
    k = pykeyboard()
    k.press_keys(['command', 'shift', 'p'])

4.6 通知 mac 通知台

上面整个过程都是脚本在后台默默运行的,如果没有任何通知,作为用户,很难知道我们的图片是否处理好,是否上传成功,因此建议加一个通知的函数。

import os

def notify_to_mac(message):
    os.system("osascript -e 'display notification \"{}\"\'".format(message))

不过其实 picgo 上传完图片后,本身就会通知,所以这个通知并不是必要的,看个人需求啦~

5. 其他设置工作

5.1 设置程序权限

如果你在使用如上脚本时,发现有的键无法捕捉,那一定是系统没有给予权限,需要你手动开启。

Python对接PicGo实现图片自动加水印并上传操作示例

5.2 设置开机自启

在这里添加一个开机启动项,而这个 init.sh 是一个 shell 脚本。

Python对接PicGo实现图片自动加水印并上传操作示例

这个脚本的内容如下,注意最后那个 & 一定不能省略。

Python对接PicGo实现图片自动加水印并上传操作示例

6. 运行效果

代码全部解析完了,是不是很想看这个程序运行后,可以实现怎样的效果呢?

我录制了个 gif 动态图,你可以瞧一瞧,真的太方便了。

Python对接PicGo实现图片自动加水印并上传操作示例

7. 写在最后

对于有和我一样写博客习惯的朋友来说,我相信这篇文章的思路一定会有帮助,另外,即使你没有防搬运的需求,代码的实现依然值得学习,比如

如何监听键盘并定义程序的热键?

如何从剪切板中读取图像?

如何给图像添加水印?

如何将图像再放入到剪切板?

如何模拟键盘来激活程序?

在编码的时候,也遇到了不少的坑,有的第三方库并不适用于 mac,有的有 bug 至今也还没修复,有的甚至要阅读源码才能知道如何使用,经过多轮的调试和搜索,最终才完成这个脚本。

以上就是python对接picgo实现图片自动加水印并上传操作示例的详细内容,更多关于python对接picgo图片自动加水印并上传的资料请关注其它相关文章!