from PIL import Image, ImageDraw, ImageFont
import os
import glob

# 透かしで表示したい文字
watermark = '© sunfish record' 
# 透かしの透明度
opacity = 80
# 透かし文字の色
color = (255, 255, 255)

# 透かしを追加したい画像を格納したフォルダ名
inputFolder = "input"
# 透かしを追加した画像の出力先のフォルダ名
outputFolder = "output"

# inputフォルダにある全てのJPGファイルを取り込んで編集
i = 1
for path in glob.glob(os.path.join(inputFolder, '*.JPG')):
    base = Image.open(path).convert('RGBA')

    # フォントサイズ指定
    fontsize = base.width * 0.037

    txt = Image.new('RGBA', base.size, (255, 255, 255, 0))
    draw = ImageDraw.Draw(txt)

    # あらかじめどっかからダウンロードしたフォント(ここではInk Free)を指定
    fnt = ImageFont.truetype(font='Ink-Free-Font/InkFree.ttf', size=round(fontsize))

    textw, texth = draw.textsize(watermark, font=fnt)

    # 透かし文字右下
    # 写真のサイズを取得してそれぞれの画像に応じた場所に配置
    draw.text((base.width * 0.708, base.height * 0.9178), watermark, font = fnt,  fill= color + (opacity,))

    out = Image.alpha_composite(base, txt)
    # この状態だとpngで出力される、
    # 今回はjpgで出力するように変換(pngで出そうとするとめっちゃ時間かかる)
    jpgOut = out.convert('RGB')

    # アウトプット先に指定したフォルダにtest~~っていうファイル名で出力される
    # jpgOut.save(outputFolder + '/'+ path + '.JPG', quality=100, optimize=True)
    jpgOut.save(outputFolder + '/WMoutput' + str(i) + '.JPG', quality=100, optimize=True)
    # pngで出力しようと思ったら↑を消して↓のコメントを外す
    # out.save(outputFolder + '/test' + str(i) + '.png', 'png', quality=100, optimize=True)
    print(str(i) + '回目の処理完了')
    i+=1