mirror of
https://github.com/hastagAB/Awesome-Python-Scripts.git
synced 2024-11-30 15:31:07 +00:00
49 lines
1.4 KiB
Python
49 lines
1.4 KiB
Python
|
import os
|
||
|
|
import sys
|
||
|
|
from os.path import join
|
||
|
|
from PIL import Image, ImageEnhance
|
||
|
|
|
||
|
|
def FolderSelectAndRun():
|
||
|
|
batch(
|
||
|
|
"<input folder>",
|
||
|
|
"<output folder>",
|
||
|
|
"<watermark image>"
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
basewidth = 2048
|
||
|
|
|
||
|
|
def batch(infolder, outfolder, watermark):
|
||
|
|
mark = Image.open(watermark)
|
||
|
|
count = 0
|
||
|
|
for root, dirs, files in os.walk(infolder):
|
||
|
|
for name in files:
|
||
|
|
try:
|
||
|
|
count += 1
|
||
|
|
im = Image.open(join(root, name))
|
||
|
|
|
||
|
|
# New image in the making
|
||
|
|
layer = Image.new('RGBA', im.size, (0, 0, 0, 0))
|
||
|
|
position = (im.size[0] - (mark.size[0] + 50),
|
||
|
|
im.size[1] - (mark.size[1] + 50))
|
||
|
|
layer.paste(mark, position)
|
||
|
|
new_image = Image.composite(layer, im, layer)
|
||
|
|
|
||
|
|
# Resize in perspective
|
||
|
|
wpercent = (basewidth / float(im.size[0]))
|
||
|
|
hsize = int((float(new_image.size[1]) * float(wpercent)))
|
||
|
|
smaller_new_image = new_image.resize(
|
||
|
|
(basewidth, hsize), Image.ANTIALIAS)
|
||
|
|
|
||
|
|
# Save new smaller image
|
||
|
|
smaller_new_image.save(
|
||
|
|
join(outfolder, ('with-watermark_' + name)), 'jpeg')
|
||
|
|
|
||
|
|
except Exception as error:
|
||
|
|
# Debug line while making changes
|
||
|
|
print('Caught this error: ' + repr(error))
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
FolderSelectAndRun()
|