from PIL import Image


# 处理两个图片的拼接
def join_two_image(img_1, img_2, flag='horizontal'):  # 默认是水平参数
    # 1、首先使用open创建Image对象,open()需要图片的路径作为参数
    # 2、然后获取size,size[0]代表宽,size[1]代表长,分别代表坐标轴的x,y
    # 3、使用Image.new创建一个新的对象
    # 4、设置地点,两个图片分别在大图的什么位置粘贴
    # 5、粘贴进大图,使用save()保存图像
    img1 = Image.open(img_1)
    img2 = Image.open(img_2)
    size1, size2 = img1.size, img2.size
    if flag == 'horizontal':
        joint = Image.new("RGB", (size1[0] + size2[0], size1[1]))
        loc1, loc2 = (0, 0), (size1[0], 0)
        joint.paste(img1, loc1)
        joint.paste(img2, loc2)
        joint.save('joint.jpg')
    # 同理垂直拼接,只要设置位置参数就行了


if __name__ == '__main__':
    png = 'img.png'
    join_two_image(png, png)