python，PIL给图片加水印

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 119 120 121 122 123

#! /usr/bin/env python2.7 # -*- coding: utf-8 -*-    import Image import os       class WaterMark(object):     '''     水印工具类，提供基本的给图片打水印的功能     使用：         marker = WaterMark(path) # path为水印图片的路径         marker.mark(pic) # pic为待打水印的图片         marker.mark_dir(pic_dir) # pic_dir为待打水印的目录     '''        POSITION_TOP_LEFT = 0     POSITION_TOP_RIGHT = 1     POSITION_BOTTOM_LEFT = 2     POSITION_BOTTOM_RIGHT = 3        # 在原图片中打水印的区域     box = (0, 0, 0, 0)        backup_suffix = '.bak'        def _gen_mosaic(self, arg):         '''打开水印图片'''            try:             mosaic = Image.open(arg).convert('RGBA')         except IOError:             print 'cannot convert file.'         return mosaic        def __init__(self, *args, **kwargs):            if args:             arg = args[0]             self._mosaic = self._gen_mosaic(arg)        def _locate(self, image,  position=POSITION_BOTTOM_RIGHT):         '''定位原图片打水印的位置'''            w, h = self._mosaic.size         i_w, i_h = image.size         if position == self.POSITION_TOP_LEFT:             self.box = (0, 0, w, h)         elif position == self.POSITION_TOP_RIGHT:             self.box = (i_w - w, 0, i_w, h)         elif position == self.POSITION_BOTTOM_LEFT:             self.box = (0, i_h - h, w, i_h)         elif position == self.POSITION_BOTTOM_RIGHT:             self.box = (i_w - w, i_h - h, i_w, i_h)         else:             self.box = (0, 0, 0, 0)        def mark(self, path, position=POSITION_BOTTOM_RIGHT):         '''给单个图片打水印'''            try:             img = Image.open(path)         except IOError:             return None            if img.size[0] < self._mosaic.size[0]:             print 'width', img.size[0], self._mosaic.size[0]             return None         if img.size[1] < self._mosaic.size[1]:             print 'height', img.size[1], self._mosaic.size[1]             return None         img_area = img.size[0] * img.size[1]         mosaic_area = self._mosaic.size[0] * self._mosaic.size[1]         ratio = 4         if img_area < mosaic_area * ratio:             return None            self._locate(img, position)         layer = Image.new('RGBA', img.size, (0, 0, 0, 0))         layer.paste(self._mosaic, self.box)            return Image.composite(layer, img, layer)        def mark_dir(self, path, backup=False, position=POSITION_BOTTOM_RIGHT):         '''批量打水印'''            for root, dirs, files in os.walk(path):                for f in files:                 path = os.path.join(root, f)                 if path.endswith(self.backup_suffix):                     continue                 print path                    f_marked = self.mark(path)                 if not f_marked:                     continue                    if backup:                     os.rename(path, path + self.backup_suffix)                    f_marked.save(path, f_marked.format)       def test():     '''     用来测试打水印功能     命令行使用：         python watermark.py 水印图片路径 待打水印图片的目录     '''        import sys     if len(sys.argv) < 3:         return False     mosaic_path = sys.argv[1]     directory = sys.argv[2]        marker = WaterMark(mosaic_path)     marker.mark_dir(directory, backup=True)       if __name__ == '__main__':     test()