Python給照片添加水印時間和地點 Mac OS-有解無憂

這里寫目錄標題

- 效果
- 需要用到的模塊
- 處理步驟以及代碼講解
- - 1、獲得照片的拍攝時間
  - 2、獲取拍照地點
  - 3、圖片轉正
  - 4、添加水印
- 完整代碼

我們平時想列印圖片的時候，希望能給照片加上時間、地點，以便以后看到照片的時候不會忘記，
手工一張張添加太麻煩，而且容易出錯，以下是使用代碼獲取原始圖片的拍攝時間和地點，并添加水印的方法，在Mac OS下對jpeg、jpg檔案處理已經測驗通過了，

效果

首先我們看下效果，第一張是原圖，第二張是添加水印時間和地點后的圖：在這里插入圖片描述

需要用到的模塊

exifread
PIL
requests
json

處理步驟以及代碼講解

1、獲得照片的拍攝時間

這里我們需要用到exifread模塊以獲取照片的拍照時間，¹

def getPhotoTime(filename):
    '''得到照片的拍照時間（如果獲取不到拍照時間，則使用檔案的創建時間）
    '''
    try:
        if os.path.isfile(filename):
            fd = open(filename, 'rb')
        else:
            raise "[%s] is not a file!\n" % filename
    except:
        raise "unopen file[%s]\n" % filename
        
    #默認用影像檔案的創建日期作為拍攝日期（如果有照片的拍攝日期，則修改為拍攝日期
    state = os.stat(filename)
    dateStr = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime(state[-2]))

    data = exifread.process_file( fd )
    if data: #取得照片的拍攝日期，改為拍攝日期
        try:
            t = data['EXIF DateTimeOriginal'] #轉換成 yyyy-mm-dd_hh:mm:ss的格式
            dateStr = str(t).replace(":","-")[:10] + str(t)[10:]
        except:
            pass

    return dateStr

2、獲取拍照地點

這里我們依然要用到exifread模塊以獲取照片的拍照的經緯度，在使用baidu的api逆地理編碼得到對應的具體位置，²
注意在使用baidu的api之前需要去百度地圖開放平臺注冊賬號，申請secret_key，可以參考別人的申請ak³，

def format_lat_lng(data):
    '將exif得到的經緯度轉化成數值， 這個有點笨重'
    list_tmp=str(data).replace('[', '').replace(']', '').split(',')
    l= [ele.strip() for ele in list_tmp]
    data_sec = int(l[-1].split('/')[0]) /(int(l[-1].split('/')[1])*3600)# 秒的值
    data_minute = int(l[1])/60
    data_degree = int(l[0])
    result=data_degree + data_minute + data_sec
    return result

def getLocationBy_lat_lng(lat, lng):
    """
    使用Geocoding API把經緯度坐標轉換為結構化地址，需要注冊baidu map api的key
    """
    secret_key = 'XXXXX你得ak密鑰'  # 請修改這里
    
    # 使用說明http://lbsyun.baidu.com/index.php?title=webapi/guide/webservice-geocoding-abroad
    baidu_map_api = 'http://api.map.baidu.com/reverse_geocoding/v3/?ak={0}&output=json&coordtype=wgs84ll&location={1},{2}'.format(
        secret_key, lat, lng)
    response = requests.get(baidu_map_api)
    content = response.text
    baidu_map_address = json.loads(content)
    formatted_address = baidu_map_address["result"]["formatted_address"]
    return formatted_address

def getLocation(filename):
    '得到照片的拍照位置（如果獲取不到，則為空字串）'
    try:
        if os.path.isfile(filename):
            fd = open(filename, 'rb')
        else:
            raise "[%s] is not a file!\n" % filename
    except:
        raise "unopen file[%s]\n" % filename
        
    # 影像檔案的拍攝地址默認為空
    locationStr= ''
    
    data = exifread.process_file( fd )
    if data: #取得照片的拍攝位置
        try:
            lat= format_lat_lng(data['GPS GPSLatitude'])  # [34, 12, 9286743/200000] -> xx.xxxxx
            lng= format_lat_lng(data['GPS GPSLongitude']) # [108, 57, 56019287/1000000] -> xx.xxxx
            locationStr= getLocationBy_lat_lng(lat, lng)
        except:
            pass
        
    return locationStr

3、圖片轉正

對于手機、相機等設備拍攝的照片，由于手持方向的不同，拍出來的照片可能是旋轉0°、90°、180°和270°，即使在電腦上利用軟體將其轉正，他們的exif資訊中還是會保留方位資訊，
在用PIL讀取這些影像時，讀取的是原始資料，也就是說，即使電腦螢屏上顯示是正常的照片，用PIL讀進來后，也可能是旋轉的影像，并且圖片的size也可能與螢屏上的不一樣，
對于這種情況，可以利用PIL讀取exif中的orientation資訊，然后根據這個資訊將圖片轉正后，再進行后續操作，具體如下，⁴

def orientate(img):
    '''利用PIL讀取exif中的orientation資訊，然后根據這個資訊將圖片轉正后，再進行后續操作
    '''
    try:
        orientation= None
        for i in ExifTags.TAGS.keys() : 
            if ExifTags.TAGS[i]=='Orientation' : # 肯定會找到orientation，所以不需要對None做處理
                orientation= i
                break 
        exif=dict(img._getexif().items())
        if   exif[orientation] == 3 : 
            img=img.rotate(180, expand = True)
        elif exif[orientation] == 6 : 
            img=img.rotate(270, expand = True)
        elif exif[orientation] == 8 : 
            img=img.rotate(90, expand = True)
    except:
        pass
    return img

4、添加水印

最后是添加水印，這一步是將上面的步驟整合起來，做處理，
這一步根據不同的圖片需要調整字體大小（size），字體位置（d_width,d_height）和字體填充顏色（fillcolor），⁵

def add_watermark(filename, text):
    '為照片檔案添加水印，filename是照片檔案名，text是水印時間和位置'
    # 創建輸出檔案夾
    outdir= 'watermark/'
    mymkdir(outdir) 

    # 創建繪畫物件
    image= Image.open(filename)
    image= orientate(image) # 將圖片轉正
    draw= ImageDraw.Draw(image)
    width, height= image.size # 寬度，高度
    size= int(0.04*width)  # 字體大小(可以調整0.04)
    myfont= ImageFont.truetype('/Library/Fonts/Arial Unicode.ttf', size=size) # 80, 4032*3024
    #fillcolor= '#000000' # RGB黑色
    fillcolor= '#fbfafe'

    # 引數一：位置（x軸，y軸）；引數二：填寫內容；引數三：字體；引數四：顏色    
    d_width, d_height=0.5*width, 0.92*height # 字體的相對位置（0.5, 0.92可以根據圖片調整）
    draw.text((d_width, d_height), text, font=myfont, fill=fillcolor) # (-1200, -320)

    new_filename= get_new_filename(filename)
    image.save(outdir + new_filename)

完整代碼

以下是完整代碼，并附上github的鏈接（鏈接中有原始的示例圖片），

''' 
批量給圖片增加時間水印的Python腳本程式【目前測驗為Mac OS X下的jpeg檔案】，
遍歷指定目錄（含子目錄）的照片檔案，根據拍照時間給照片在右下角添加時間和地點的水印，


！該程式需要安裝exifread，PIL，requests等模塊，否則無法使用，
例如，Linux/Mac OS X下命令列安裝該模塊：sudo pip install exifread

！該程式需要加baidu api的secret_key（在getLocationBy_lat_lng函式中）；
！該程式可以手動調整字體大小（size），字體位置（d_width,d_height）和字體填充顏色（fillcolor）（在add_watermark函式中），

獲取拍照時間 From 'https://www.racksam.com/2014/05/14/python-script-to-change-pictures-filenames/'
拍照地點 參考 http://lbsyun.baidu.com/index.php?title=webapi/guide/webservice-geocoding-abroad
watermark 參考 https://blog.csdn.net/weixin_30561177/article/details/97609897 
orientation 參考 https://blog.csdn.net/mizhenpeng/article/details/82794112
'''
#!/usr/bin/env python
# -*- coding: utf-8 -*-  
import os
import stat
import time
import math
import exifread
from PIL import Image, ImageDraw, ImageFont, ExifTags
import requests
import json

'=================分割線==================='
# 創建檔案夾
def mymkdir(path):
    if not os.path.exists(path):
        os.mkdir(path)

# 判斷是否是支持的檔案型別
SUFFIX_FILTER = ['.jpg','.png','.mpg','.mp4','.thm','.bmp','.jpeg','.avi','.mov'] # 僅測驗了jpeg，jpg
def isTargetedFileType(filename):
    '根據檔案擴展名，判斷是否是需要處理的檔案型別；僅支持x.x格式'
    filename_nopath = os.path.basename(filename)
    f,e = os.path.splitext(filename_nopath)
    if e.lower() in SUFFIX_FILTER:
        return True
    else:
        return False

# 新檔案的名字
def get_new_filename(filename):
    f, e= filename.split('.')
    return "%s_watermark.%s" % (f, e)

'=================分割線==================='

'=================begin{得到照片的拍攝時間}==================='
def getPhotoTime(filename):
    '''得到照片的拍照時間（如果獲取不到拍照時間，則使用檔案的創建時間）
    '''
    try:
        if os.path.isfile(filename):
            fd = open(filename, 'rb')
        else:
            raise "[%s] is not a file!\n" % filename
    except:
        raise "unopen file[%s]\n" % filename
        
    #默認用影像檔案的創建日期作為拍攝日期（如果有照片的拍攝日期，則修改為拍攝日期
    state = os.stat(filename)
    dateStr = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime(state[-2]))

    data = exifread.process_file( fd )
    if data: #取得照片的拍攝日期，改為拍攝日期
        try:
            t = data['EXIF DateTimeOriginal'] #轉換成 yyyy-mm-dd_hh:mm:ss的格式
            dateStr = str(t).replace(":","-")[:10] + str(t)[10:]
        except:
            pass

    return dateStr
'=================end{得到照片的拍攝時間}==================='


'=================begin{得到照片的拍攝地點}==================='
def format_lat_lng(data):
    '將exif得到的經緯度轉化成數值， 這個有點笨重'
    list_tmp=str(data).replace('[', '').replace(']', '').split(',')
    l= [ele.strip() for ele in list_tmp]
    data_sec = int(l[-1].split('/')[0]) /(int(l[-1].split('/')[1])*3600)# 秒的值
    data_minute = int(l[1])/60
    data_degree = int(l[0])
    result=data_degree + data_minute + data_sec
    return result

def getLocationBy_lat_lng(lat, lng):
    """
    使用Geocoding API把經緯度坐標轉換為結構化地址，需要注冊baidu map api的key
    """
    secret_key = 'XXXXX你得ak密鑰'  # 請修改這里
    
    # 使用說明http://lbsyun.baidu.com/index.php?title=webapi/guide/webservice-geocoding-abroad
    baidu_map_api = 'http://api.map.baidu.com/reverse_geocoding/v3/?ak={0}&output=json&coordtype=wgs84ll&location={1},{2}'.format(
        secret_key, lat, lng)
    response = requests.get(baidu_map_api)
    content = response.text
    baidu_map_address = json.loads(content)
    formatted_address = baidu_map_address["result"]["formatted_address"]
    return formatted_address

def getLocation(filename):
    '得到照片的拍照位置（如果獲取不到，則為空字串）'
    try:
        if os.path.isfile(filename):
            fd = open(filename, 'rb')
        else:
            raise "[%s] is not a file!\n" % filename
    except:
        raise "unopen file[%s]\n" % filename
        
    # 影像檔案的拍攝地址默認為空
    locationStr= ''
    
    data = exifread.process_file( fd )
    if data: #取得照片的拍攝位置
        try:
            lat= format_lat_lng(data['GPS GPSLatitude'])  # [34, 12, 9286743/200000] -> xx.xxxxx
            lng= format_lat_lng(data['GPS GPSLongitude']) # [108, 57, 56019287/1000000] -> xx.xxxx
            locationStr= getLocationBy_lat_lng(lat, lng)
        except:
            pass
        
    return locationStr

'=================end{得到照片的拍攝地點}==================='

def orientate(img):
    '''對于手機、相機等設備拍攝的照片，由于手持方向的不同，拍出來的照片可能是旋轉0°、90°、180°和270°，即使在電腦上利用軟體將其轉正，他們的exif資訊中還是會保留方位資訊，
        在用PIL讀取這些影像時，讀取的是原始資料，也就是說，即使電腦螢屏上顯示是正常的照片，用PIL讀進來后，也可能是旋轉的影像，并且圖片的size也可能與螢屏上的不一樣，
        對于這種情況，可以利用PIL讀取exif中的orientation資訊，然后根據這個資訊將圖片轉正后，再進行后續操作，具體如下，
    '''
    try:
        orientation= None
        for i in ExifTags.TAGS.keys() : 
            if ExifTags.TAGS[i]=='Orientation' : # 肯定會找到orientation，所以不需要對None做處理
                orientation= i
                break 
        exif=dict(img._getexif().items())
        if   exif[orientation] == 3 : 
            img=img.rotate(180, expand = True)
        elif exif[orientation] == 6 : 
            img=img.rotate(270, expand = True)
        elif exif[orientation] == 8 : 
            img=img.rotate(90, expand = True)
    except:
        pass
    return img


def add_watermark(filename, text):
    '為照片檔案添加水印，filename是照片檔案名，text是水印時間和位置'
    # 創建輸出檔案夾
    outdir= 'watermark/'
    mymkdir(outdir) 

    # 創建繪畫物件
    image= Image.open(filename)
    image= orientate(image) # 將圖片轉正
    draw= ImageDraw.Draw(image)
    width, height= image.size # 寬度，高度
    size= int(0.04*width)  # 字體大小(可以調整0.04)
    myfont= ImageFont.truetype('/Library/Fonts/Arial Unicode.ttf', size=size) # 80, 4032*3024
    #fillcolor= '#000000' # RGB黑色
    fillcolor= '#fbfafe'

    # 引數一：位置（x軸，y軸）；引數二：填寫內容；引數三：字體；引數四：顏色    
    d_width, d_height=0.5*width, 0.92*height # 字體的相對位置（0.5, 0.92可以根據圖片調整）
    draw.text((d_width, d_height), text, font=myfont, fill=fillcolor) # (-1200, -320)

    new_filename= get_new_filename(filename)
    image.save(outdir + new_filename)


def scandir(startdir):
    '遍歷指定目錄，對滿足條件的檔案進行改名或洗掉處理'
    os.chdir(startdir) # 改變當前作業目錄
    for obj in os.listdir(os.curdir) :
        if os.path.isfile(obj):
            if isTargetedFileType(obj): # 對滿足過濾條件的檔案，加時間和地點水印
                photoTime = getPhotoTime(obj) # 獲得照片的拍攝時間,當作水印的內容
                location= getLocation(obj) # 獲得照片的拍攝位置,當作水印的內容

                print("%s    %s" % (obj, photoTime+' '+location))
                add_watermark(obj, location +'\n'+ photoTime) #加時間和地點水印


if __name__ == "__main__":
    path = "./fig" # 照片位置
    scandir(path)

https://www.racksam.com/2014/05/14/python-script-to-change-pictures-filenames/ ??
http://lbsyun.baidu.com/index.php?title=webapi/guide/webservice-geocoding-abroad ??
https://www.jianshu.com/p/6ad61317d988 ??
https://blog.csdn.net/mizhenpeng/article/details/82794112 ??
https://blog.csdn.net/weixin_30561177/article/details/97609897 ??

轉載請註明出處，本文鏈接：https://www.uj5u.com/qita/123909.html

標籤：其他

上一篇：從0實作python批量爬取p站插畫

下一篇：爬蟲開始路線

Python給照片添加水印時間和地點 Mac OS