1. 首頁 > 生活百科 > >

Python實現給圖片加水印功能

生活百科

前言最近忙得連軸轉，很久沒更新博客了，代碼倒是沒啥寫，積累了好些東西，接下來一有時間就來更新吧~
本文記錄使用Python實現給圖片添加水印的功能實現過程
先看效果把公眾號的封面作為素材
原圖是這樣的
【Python實現給圖片加水印功能】

Python實現給圖片加水印功能

文章插圖
加個水印的效果
Python實現給圖片加水印功能

文章插圖
實現代碼主要實現是參考GitHub上一個項目的：https://github.com/2Dou/watermarker
用到了Pillow庫，使用前請先安裝，一般要在Django中保存圖片也需要安裝這個庫的 。所以這個依賴可以說是比較友好的 。
這個項目是命令行工具，而我是在Django項目中用，所以我做了魔改
核心代碼如下（這段代碼是在我封裝的一個類里面）
origin_image = Image.open(self.image_path)origin_image = ImageOps.exif_transpose(origin_image)# 計算字體的寬度、高度width = len(self.text) * self.sizeheight = round(self.size * self.font_height_crop)# 創建水印圖片watermark_image = Image.new(mode='RGBA', size=(width, height))# 生成文字draw_table = ImageDraw.Draw(im=watermark_image)draw_table.text(xy=(0, 0),text=self.text,fill=self.color,font=ImageFont.truetype(self.font_file, size=self.size))del draw_table# 裁剪空白watermark_image = Watermarker.crop_image_edge(watermark_image)# 設置透明度Watermarker.set_image_opacity(watermark_image, self.opacity)# 計算斜邊長度c = int(math.sqrt(origin_image.size[0] * origin_image.size[0] + origin_image.size[1] * origin_image.size[1]))# 以斜邊長度為寬高創建大圖（旋轉后大圖才足以覆蓋原圖）用于覆蓋在原圖之上watermark_mask = Image.new(mode='RGBA', size=(c, c))# 在大圖上生成水印文字y, idx = 0, 0while y < c:# 制造x坐標錯位x = -int((watermark_image.size[0] + self.space) * 0.5 * idx)idx = (idx + 1) % 2while x < c:# 在該位置粘貼mark水印圖片watermark_mask.paste(watermark_image, (x, y))x = x + watermark_image.size[0] + self.spacey = y + watermark_image.size[1] + self.space# 將大圖旋轉一定角度watermark_mask = watermark_mask.rotate(self.angle)# 在原圖上添加大圖水印if origin_image.mode != 'RGBA':origin_image = origin_image.convert('RGBA')origin_image.paste(watermark_mask,# 大圖(int((origin_image.size[0] - c) / 2), int((origin_image.size[1] - c) / 2)),# 坐標mask=watermark_mask.split()[3])del watermark_mask我把這個加水印的功能封裝成了一個類
class Watermarker(object):"""圖片水印工具"""django_support = Falsedef __init__(self, image_path: str, text: str,angle=30,color='#8B8B1B',font_file='青鳥華光簡琥珀.ttf',font_height_crop=1.2,opacity=0.15,quality=80,size=50,space=75,):...@staticmethoddef set_image_opacity(image: Image, opacity: float):...@staticmethoddef crop_image_edge(image: Image):...@propertydef image(self):...def save(self, file_path: str, image_format: str = 'png'):...def show(self):...使用方法簡單使用w = Watermarker('codelab.png', '程序設計實驗室', size=200)# 顯示加了水印的圖片w.show()# 保存w.save('save.png')在Django中使用以我封裝的 DjangoStarter 框架為例，直接把Image對象寫入Http響應里
from io import BytesIOfrom django.shortcuts import get_object_or_404, renderfrom django.http import HttpResponsefrom django_starter.contrib.watermark import Watermarkerdef add_watermark(request, pk):text = request.GET.get('text', timezone.now().strftime('%Y-%m-%d %H:%M:%S'))photo = get_object_or_404(Photo, pk=pk)image = Watermarker(photo.photo.path, text).image# 將圖片保存到內存中with BytesIO() as f:image.save(f, 'png')# 返回圖片數據流return HttpResponse(f.getvalue(), content_type='image/png')路由配置
from django.urls import pathfrom . import viewsapp_name = 'photo'urlpatterns = [path('photo/<int:pk>/add_watermark/', views.add_watermark, name='add_watermark'),]這樣配置之后，訪問鏈接：http://[host]:[port]/photo/1/add_watermark/
就可以看到加了水印的圖片了
效果如下
Python實現給圖片加水印功能

文章插圖
搞定~
命令行使用本項目參考的那個項目
完整代碼GitHub gist：https://gist.github.com/Deali-Axy/e22ea79bfbe785f9017b2e3cd7fdb3eb
PS：Github gist是個代碼片段工具，最近剛剛用起來，感覺還蠻不錯的
有時候用來記錄和分享一些代碼片段很方便 ， 不需要專門創建一個倉庫

    推薦閱讀