使用python往png写入文本 将信息隐藏于图片

使用python往png写入文本 将信息隐藏于图片 而且要能重新提取出来

方法一：使用Stegano https://pypi.org/project/stegano/

代码如下：

"""
往png图片里写入文本，保存为新图片
新图片可解构出文本内容，且肉眼看起来与原图没区别
"""

from stegano import lsb  # pip install stegano

message = "万里悲秋常作客，百年多病独登台。"  # 要写入的文本

origin_file = "origin.png"  # 原图
new_name = "b.png"  # 要保存的新图片名称
encoding = "UTF-32LE"  # 文本包含中文的话，需指定为该编码
secret = lsb.hide(origin_file, message, encoding=encoding)  # 往图片写入文本
secret.save(new_name)  # 保持到本地

unpack_msg: str = lsb.reveal(new_name, encoding=encoding)  # 从本地图片里读出文本
print(f"{unpack_msg = }")
# unpack_msg = '万里悲秋常作客，百年多病独登台。'

方法二：使用pillow

代码直接从这里https://www.cnblogs.com/andy9468/p/10490160.html拷贝过来的，python3.11 Mac系统测试通过

from PIL import Image


def makeImageEven(image):
    """
    取得一个 PIL 图像并且更改所有值为偶数（使最低有效位为0）
    """
    # 得到一个这样的列表：[(r,g,b,t),(r,g,b,t)...]
    pixels = list(image.getdata())
    # 更改所有值为偶数（魔法般的移位）
    evenPixels = [
        (r >> 1 << 1, g >> 1 << 1, b >> 1 << 1, t >> 1 << 1) for [r, g, b, t] in pixels
    ]
    # 创建一个相同大小的图片副本
    evenImage = Image.new(image.mode, image.size)
    # 把上面的像素放入到图片副本
    evenImage.putdata(evenPixels)
    return evenImage


def constLenBin(int):
    """
    内置函数bin()的替代，返回固定长度的二进制字符串
    """
    # 去掉bin()返回的二进制字符串中的'0b'，并在左边补足'0'直到字符串长度为8
    binary = "0" * (8 - (len(bin(int)) - 2)) + bin(int).replace("0b", "")
    return binary


def encodeDataInImage(image, data):
    """
    将字符串编码到图片中
    """
    # 获得最低有效位为 0 的图片副本
    evenImage = makeImageEven(image)
    # 将需要被隐藏的字符串转换成二进制字符串
    binary = "".join(map(constLenBin, bytearray(data, "utf-8")))
    if len(binary) > len(image.getdata()) * 4:
        # 如果不可能编码全部数据，跑出异常
        raise Exception(
            "Error: Can't encode more than"
            + len(evenImage.getdata()) * 4
            + " bits in this image. "
        )
    # 将binary中的二进制字符串信息编码进像素里
    encodedPixels = [
        (
            r + int(binary[index * 4 + 0]),
            g + int(binary[index * 4 + 1]),
            b + int(binary[index * 4 + 2]),
            t + int(binary[index * 4 + 3]),
        )
        if index * 4 < len(binary)
        else (r, g, b, t)
        for index, (r, g, b, t) in enumerate(list(evenImage.getdata()))
    ]
    # 创建新图片以存放编码后的像素
    encodedImage = Image.new(evenImage.mode, evenImage.size)
    # 添加编码后的数据
    encodedImage.putdata(encodedPixels)
    return encodedImage


def binaryToString(binary):
    """
    从二进制字符串转为 UTF-8 字符串
    """
    index = 0
    string = []
    rec = lambda x, i: x[2:8] + (rec(x[8:], i - 1) if i > 1 else "") if x else ""
    fun = lambda x, i: x[i + 1 : 8] + rec(x[8:], i - 1)
    while index + 1 < len(binary):
        chartype = binary[index:].index(
            "0"
        )  # 存放字符所占字节数，一个字节的字符会存为0
        length = chartype * 8 if chartype else 8
        string.append(chr(int(fun(binary[index : index + length], chartype), 2)))
        index += length
    return "".join(string)


def decodeImage(image):
    """
    解码隐藏数据
    """
    pixels = list(image.getdata())  # 获得像素列表
    # 提取图片中所有最低有效位中的数据
    binary = "".join(
        [
            str(int(r >> 1 << 1 != r))
            + str(int(g >> 1 << 1 != g))
            + str(int(b >> 1 << 1 != b))
            + str(int(t >> 1 << 1 != t))
            for (r, g, b, t) in pixels
        ]
    )
    # 找到数据截止处的索引
    locationDoubleNull = binary.find("0000000000000000")
    endIndex = (
        locationDoubleNull + (8 - (locationDoubleNull % 8))
        if locationDoubleNull % 8 != 0
        else locationDoubleNull
    )
    data = binaryToString(binary[0:endIndex])
    return data


str1 = "哆啦A梦的世界，Doraemon World!哆啦A梦的世界，Doraemon World!哆啦A梦的世界，Doraemon World!哆啦A梦的世界，Doraemon World!哆啦A梦的世界，Doraemon World!哆啦A梦的世界，Doraemon World!哆啦A梦的世界，Doraemon World!哆啦A梦的世界，Doraemon World!哆啦A梦的世界，Doraemon World!哆啦A梦的世界，Doraemon World!哆啦A梦的世界，Doraemon World!哆啦A梦的世界，Doraemon World!哆啦A梦的世界，Doraemon World!哆啦A梦的世界，Doraemon World!12345678901234567890123455678"
# 可以隐藏的数据量是有限的目前可以隐藏350个字符
print(len(str1))
# png为不失真图片，jpg为失真图片。jpg不可隐藏信息。
encodeDataInImage(Image.open("coffee.png"), str1).save("encodeImage.png")
print(decodeImage(Image.open("encodeImage.png")))