Python PIL图片合成,图片合并,添加文字集合

曾经终败给现在 2022-04-23 05:50 358阅读 0赞

#!/usr/bin/env python
    # encoding: utf-8
    """
    @author: morgan
    @time: 1/23/19 6:48 PM
    Python PIL图片合成,图片合并,添加文字集合
    """
    import ImageFont
    import cStringIO
    import urllib2
    from PIL import Image, ImageDraw
    
    
    def get_base_image(image_url):
        '''
        获取图片对象
        :param image_url:
        :return:
        '''
        file = urllib2.urlopen(image_url)
        tmpIm = cStringIO.StringIO(file.read())
        base_img = Image.open(tmpIm)
        return base_img
    
    
    def get_info_image(image_url, header_url, qr_url, items=None):
        '''
        获取图片
        :param image_url:
        :param header_url:
        :param qr_url:
        :param items:
        :return:
        '''
        base_image = get_base_image(image_url)
        width, height = base_image.size
        copy_base_image = base_image.copy()
        #
        back_height = 150
        # 白色背景
        target = Image.new('RGB', (width, height + back_height), '#ffffff')
        target.paste(copy_base_image, (0, 0))
        # # 头像 数据
        header_image = get_header_image(header_url)
        target.paste(header_image, (15, height + 32))
        qr_image = get_base_image(qr_url)
        copy_qr_image = qr_image.copy()
        target.paste(copy_qr_image, (width - 110, height + 32))
        draw_text(target, height, items)
        target.show()
        # draw.save('./out.png')
    
    
    def get_header_image(header_url):
        '''
        返回头像图片 填充透明 为白色
        :param header_url:
        :return:
        '''
        header_image = get_base_image(header_url)
        x, y = header_image.size
        p = Image.new('RGBA', header_image.size, (255, 255, 255))
        p.paste(header_image, (0, 0, x, y), header_image)
        return p.copy()
    
    
    def draw_text(target, height, items):
        '''
        图上写字
        :param target:
        :param height:
        :return:
        '''
        user_name = items.get('user_name') if items else u'韩寒'
        team_name = items.get('team_name') if items else '学堂'
        team_name = '邀您加入{0}'.format(team_name)
        description = '扫描二维码即可免费体验'
        draw = ImageDraw.Draw(target)
        font_path = 'huawenxihei.ttf'
        font = ImageFont.truetype(font_path, 28)
        font_14 = ImageFont.truetype(font_path, 18)
        draw.text((110, height + 30), user_name, fill=(0, 0, 0), font=font)
        draw.text((110, height + 70), unicode(team_name, 'utf-8'), fill=(0, 0, 0), font=font_14)
        draw.text((110, height + 90), unicode(description, 'utf-8'), fill=(0, 0, 0), font=font_14)
        ImageDraw.Draw(target)
        return target
    
    
    def main():
        image_url = ''
        header_url = ''
        qr_url = ''
        get_info_image(image_url, header_url, qr_url)
    
    
    if __name__ == '__main__':
        main()

新代码片段

class ImageClass:
        def __init__(self, image_url, qr_url, items=None):
            self.image_url = image_url
            self.qr_url = qr_url
            self.items = items
            self.team_id = self.items.get('team_id') if self.items else None
            self.user_id = self.items.get('user_id') if self.items else None
            self.get_user_team_info()
            self.header_url = '{0}?roundPic/radius/10000'.format(self.data.get('header_url'))
    
        def get_base_image(self, url):
            '''
            获取图片对象
            :param image_url:
            :return:
            '''
            file = urllib2.urlopen(url)
            tmpIm = cStringIO.StringIO(file.read())
            base_img = Image.open(tmpIm)
            re_base_img = base_img.convert('RGBA')  # 兼容多种图片
            return re_base_img
    
        def get_header(self):
            '''
            :return:
            '''
            header_image = self.get_header_image(self.header_url)
            header_image = header_image.resize((90, 90), Image.ANTIALIAS)
            x, y = header_image.size
            p = Image.new('RGBA', header_image.size, (255, 255, 255))
            p.paste(header_image, (0, 0, x, y), header_image)
            header_image = p.copy()
            return header_image
    
        def get_info_image(self, team_id, user_id, image_number, items):
            '''
            获取图片
            :param image_url:
            :param header_url:
            :param qr_url:
            :param items:
            :return:
            '''
            base_image = self.get_base_image(self.image_url)
            width, height = base_image.size
            copy_base_image = base_image.copy()
            #
            back_height = 150
            # 白色背景
            target = Image.new('RGBA', (width, height + back_height), '#ffffff')
            target.paste(copy_base_image, (0, 0))
            # # 头像 数据
            header_image = self.get_header()
            # 图像
            target.paste(header_image, (15, height + 32))
            qr_image = self.get_header_image(self.qr_url)
            copy_qr_image = qr_image.copy()
            target.paste(copy_qr_image, (width - 110, height + 32))
            self.draw_text(target, height)
            root_dir = os.path.abspath(os.path.join(os.getcwd(), ".."))
            dir_name = team_id + user_id + '{0}'.format(image_number)
            path_image = '{0}/static/upload/shop/{1}.png'.format(root_dir, dir_name)
            image_path = '{0}/static/upload/shop/'.format(root_dir)
            if not os.path.exists(image_path):
                os.makedirs(image_path)
            target.save(path_image)
            # target.show()
            items['_id'] = team_id + user_id + str(image_number)
            return upload_file(team_id, dir_name, image_number, items, path_image)
    
        def get_header_image(self, url):
            '''
            返回头像图片 填充透明 为白色
            :param url:
            :return:
            '''
            header_image = self.get_base_image(url)
            x, y = header_image.size
            p = Image.new('RGBA', header_image.size, (255, 255, 255))
            p.paste(header_image, (0, 0, x, y), header_image)
            return p.copy()
    
        def draw_text(self, target, height):
            '''
            图上写字
            :param target:
            :param height:
            :return:
            '''
            items = self.data
            user_name = items.get('user_name')[:30] if items else u'Morgan'
            team_name = items.get('team_name')[:30] if items else ''
            team_name = '邀您加入{0}'.format(team_name)
            description = '扫描二维码即可免费体验'
            draw = ImageDraw.Draw(target)
            root_dir = os.path.abspath(os.path.join(os.getcwd(), "."))
            font_path = '{0}/tmp/huawenxihei.ttf'.format(root_dir)
            font = ImageFont.truetype(font_path, 28)
            font_14 = ImageFont.truetype(font_path, 16)
            draw.text((110, height + 30), user_name, fill=(0, 0, 0), font=font)
            draw.text((110, height + 70), unicode(team_name, 'utf-8'), fill=(0, 0, 0), font=font_14)
            draw.text((110, height + 90), unicode(description, 'utf-8'), fill=(0, 0, 0), font=font_14)
            ImageDraw.Draw(target)
            return target