python程序，用来批量添加图片水印。输入一个文件夹、水印位置(左下角、底部中间、右下角)、用户名(用户名是中文的)，批量给文件夹里所有的jpg和png图片在指定位置添加水印。

水印内容是：用户名 拍摄时间 拍摄地点。其中用户名是入参，拍摄时间和地点都从照片里读出来，如果读不出来拍摄时间的话就用文件最近修改时间代替，如果读不出来地点就留空。

'''

输入: 1.文件夹

2.水印位置

3.用户名

输出：1.用户名

2.拍摄时间

3.拍摄地点

'''

from PIL import Image, ImageDraw, ImageFont, ImageOps

import imghdr

import os

import exifread

import time as ostime

class AddWaterMark(object):

def __init__(self, file_name, location, user):

self.file_name = file_name

self.location = location

self.user = user

# 防止图片发生旋转

def exif_transpose(self, img):

if not img:

return img

exif_orientation_tag = 274

# Check for EXIF data (only present on some files)

if hasattr(img, "_getexif") and isinstance(img._getexif(), dict) and exif_orientation_tag in img._getexif():

exif_data = img._getexif()

orientation = exif_data[exif_orientation_tag]

if orientation == 1:

pass

elif orientation == 2:

img = img.transpose(Image.FLIP_LEFT_RIGHT)

elif orientation == 3:

img = img.rotate(180)

elif orientation == 4:

img = img.rotate(180).transpose(Image.FLIP_LEFT_RIGHT)

elif orientation == 5:

img = img.rotate(-90, expand=True).transpose(Image.FLIP_LEFT_RIGHT)

elif orientation == 6:

img = img.rotate(-90, expand=True)

elif orientation == 7:

img = img.rotate(90, expand=True).transpose(Image.FLIP_LEFT_RIGHT)

elif orientation == 8:

img = img.rotate(90, expand=True)

return img

# 获取文件夹下的所有jpg、png图片

def get_all_pic(self):

imgType_list = {'jpg','bmp','png','jpeg','rgb','tif'}

# imghdr 可以用来判断文件是否是图片

for item in os.listdir(self.file_name):

if os.path.isfile(item):

if imghdr.what(item) in imgType_list:

self.add_watermark_content(item)

# 水印位置

def get_watermark_location(self, img):

width, height = img.size # 图片大小

fnt_size = max(width//50, height//50) # 字体大小

if self.location == '左下角':

return (0, height-fnt_size*5)

elif self.location == '右下角':

return (width-fnt_size*12, height-fnt_size*5)

else: # 底部中间

return (width//2-fnt_size*6, height-fnt_size*5)

# 获取照片拍摄时间

def get_pic_time(self, img_name):

img = exifread.process_file(open(img_name,'rb'))

if 'Image DateTime' in img.keys():

time = img['Image DateTime'] # 拍摄时间

else:

file_path = os.path.join(self.file_name, img_name)

time = ostime.ctime(os.stat(file_path).st_mtime) #文件的修改时间

return time

# 获取照片拍摄位置

def get_pic_address(self, img_name):

img = exifread.process_file(open(img_name,'rb'))

if 'GPS GPSLatitude' in img.keys():

latitude = img['GPS GPSLatitude']

else:

latitude = ''

if 'GPS GPSLongitude' in img.keys():

longitude = img['GPS GPSLongitude']

else:

longitude = ''

position = f'{latitude}\n{longitude}\n'

return position

# 设置水印内容

def set_content(self, img_name):

user = self.user

time = self.get_pic_time(img_name)

position = self.get_pic_address(img_name)

content = f'{user}\n{time}\n{position}\n'

return content

# 添加水印的内容

def add_watermark_content(self, img_name):

img = Image.open(img_name)

# 防止图片发生旋转

# if hasattr(ImageOps, 'exif_transpose'):

# img = ImageOps.exif_transpose(img)

# else:

# img = self.exif_transpose(img)

# img = img.convert('RGB')

draw = ImageDraw.Draw(img)

# 字体和大小

fnt_size = max(img.size[0]//50, img.size[1]//50)

font = ImageFont.truetype('C:\Windows\Fonts\simkai.ttf', size=fnt_size)

# 水印位置

(x,y) = self.get_watermark_location(img)

# 水印内容

content = self.set_content(img_name)

# 位置，内容，字体，颜色，可以自行调节到合适

draw.text((x,y), content, font=font, fill=(255, 255, 255))

draw.text((x+2, y+2), content, font=font, fill=(0, 0, 0))

self.save_watermark_pic(img, img_name)

# 保存水印图片

def save_watermark_pic(self, img, img_name):

new_file = os.path.join(self.file_name, '水印图片')

if not os.path.exists(new_file):

os.mkdir(new_file)

img.save(os.path.join(new_file, img_name), quality=95)

if __name__ == "__main__":

file_name = 'D:\Desktop\picture' # 文件名称

location = '左下角' # 水印位置：左下角、底部中间、右下角

user = '超级无敌奥特曼' # 用户名

WM = AddWaterMark(file_name, location, user)

WM.get_all_pic()

print('hello world...')