watermarker/watermark_linux.py at master · bhaskar4n/watermarker · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
import os, sys
from PIL import Image
from PIL import ImageDraw
from PIL import ImageFont
import time
print "1. Text only"
print "2. Logo only"
print "3. Both text and logo"
choice = int(input("choose (1 or 2 or 3) : "))
path = "images/"
logo_path = "logo/"
paths = "water_marked_images"
if not os.path.exists(paths):
	os.makedirs('water_marked_images')

dirs = os.listdir(path)

if choice == 2:
	try:
		if not os.path.exists(logo_path):
			os.makedirs('logo')
		trans = int(input("Transparent level(default=100): "))
		dirs1 = os.listdir(logo_path)
		file = "logo".join(dirs1)
		lgo_pth = logo_path+file
		logo = Image.open(lgo_pth)
		lgo = logo.size
	except:
		print "please add logo image";

if choice == 1:
	txt = raw_input("enter water_marker text: ")
	txtsize = int(input("enter the text size(default=72): "))
	trans = int(input("Transparent level(default=100): "))

if choice ==3:
	txt = raw_input("enter water_marker text: ")
	txtsize = int(input("enter the text size(default=72): "))
	trans = int(input("Transparent level(default=100): "))
	dirs1 = os.listdir(logo_path)
	file = "logo".join(dirs1)
	lgo_pth = logo_path+file
	logo = Image.open(lgo_pth)
	lgo = logo.size


for file in dirs:
	image_path=path+file
	#a = f.append(file)
	#print a
	print image_path
	im = Image.open(image_path)
	print im.size
	if choice == 1:

		transparent = Image.new("RGBA", im.size)
		draw = ImageDraw.ImageDraw(transparent,"RGBA")
		font = ImageFont.truetype("/usr/share/fonts/truetype/droid/DroidSansMono.ttf",txtsize)


		a = font.getsize(txt)
		print "getsize",a

		draw.line((0, 0) + im.size, fill=(160,161,161))
		draw.line((0, im.size[1], im.size[0], 0), fill=(161,161,161))
		if choice == 1:
			#font = ImageFont.truetype("Arial.ttf",72)
			draw.text(((im.size[0]/2 - a[0]/2, im.size[1]/2)),txt,font=font,fill=(255,255,255))
			mask = transparent.convert("L").point(lambda x: min(x, trans))
			transparent.putalpha(mask)
			im.paste(transparent,None,transparent)
			print "water_marked"
			im.save("water_marked_images/WM_"+file+".jpg")


	elif choice == 2:

		masklogo = logo.convert("L").point(lambda x: min(x,trans))
		print "logo_water_marked"
		x = im.size[0]/2-lgo[0]/2
		y = im.size[1]/2-lgo[1]/2
		im.paste(logo,(x,y),masklogo)
		im.save("water_marked_images/WM_"+file+".jpg")

	if choice == 3:

		masklogo = logo.convert("L").point(lambda x: min(x,trans))
		print "logo_water_marked"
		x = im.size[0]/2-lgo[0]/2
		y = im.size[1]/2-lgo[1]/2
		im.paste(logo,(x,y),masklogo)
		#im.save("water_marked_images/WM_"+file+".jpg")
		transparent = Image.new("RGBA", im.size)
		draw = ImageDraw.ImageDraw(transparent,"RGBA")
		font = ImageFont.truetype("/usr/share/fonts/truetype/droid/DroidSansMono.ttf",txtsize)
		a = font.getsize(txt)
		print "getsize",a

		draw.line((0, 0) + im.size, fill=(160,161,161))
		draw.line((0, im.size[1], im.size[0], 0), fill=(161,161,161))
		#if choice == 1:
			#font = ImageFont.truetype("Arial.ttf",72)
		draw.text(((im.size[0]/2 - a[0]/2, im.size[1]/2+lgo[1]/2)),txt,font=font,fill=(255,255,255))
		mask = transparent.convert("L").point(lambda x: min(x, trans))
		transparent.putalpha(mask)
		im.paste(transparent,None,transparent)
		print "water_marked"
		im.save("water_marked_images/WM_"+file+".jpg")


print "saving....."
t =  time.sleep(2)
print "finished"