python批量给图加水印

以上是用python批量给图片添加水印

# -*- coding: utf8 -*-

import os,Image, ImageEnhance

def set_opacity(im, opacity):
"""设置透明度"""

assert opacity >=0 and opacity < 1
if im.mode != "RGBA":
im = im.convert('RGBA')
else:
im = im.copy()
alpha = im.split()[3]
alpha = ImageEnhance.Brightness(alpha).enhance(opacity)
im.putalpha(alpha)
return im

def watermark(im, mark, position, opacity=1):
"""添加水印"""

try:
if opacity < 1:
mark = set_opacity(mark, opacity)
if im.mode != 'RGBA':
im = im.convert('RGBA')
#if im.size[0] < mark.size[0] or im.size[1] < mark.size[1]:
#print "The mark image size is larger size than original image file."
#return False

#设置水印位置
if position == 'left_top':
x = 0
y = 0
elif position == 'left_bottom':
x = 0
y = im.size[1] - mark.size[1]
elif position == 'right_top':
x = im.size[0] - mark.size[0]
y = 0
elif position == 'right_bottom':
x = im.size[0] - mark.size[0]+5
y = im.size[1] - mark.size[1]-16
else:
x = (im.size[0] - mark.size[0]) / 2
y = (im.size[1] - mark.size[1]) / 2

layer = Image.new('RGBA', im.size,)
layer.paste(mark,(x,y))
return Image.composite(layer, im, layer)
except Exception as e:
print ">>>>>>>>>>> WaterMark EXCEPTION: " + str(e)
return False

def water(filename,mark):
try:
im = Image.open(filename) #原图
image = watermark(im, mark, 'right_bottom', 1)
image.save(filename)
#image.show()
except Exception as e:
print ">>>>>>>>>>> WaterMark save: " + str(e)
return False

def dirlist(path,mark):
global num
filelist = os.listdir(path)
for filename in filelist:
filepath = os.path.join(path, filename)
if os.path.isdir(filepath):
dirlist(filepath,mark)
else:
water(filepath,mark)
num += 1
print filepath, num

num = 0
if __name__ == '__main__':
water_path = "E:/water/logo.jpg"
dir_path = "H:/hx/bj/"
mark = Image.open(water_path) #水印
dirlist(dir_path,mark)