忽略 python 中 libpng 的 iCCP 警告

问题描述#

在使用 Python 处理 PNG 图片时，经常会遇到以下警告:

libpng warning: iCCP: known incorrect sRGB profile

根据 libpng 官方的说法, 图片本来就是坏的, 但是 libpng 仍然会尝试去处理它, 导致这个警告.

但即使项目中没有用任何 png 图片素材, 仅仅用 Tkinter 创建一个窗口都会触发一大堆这个警告刷屏, 强迫症患者震怒.

libpng_warnings

而且这个错误是直接调用 c 函数输出的, 十分恶心.

尝试过程#

网上看了一堆垃圾教程, 没一个有用的. 总结一下这些垃圾教程教的方法:

使用 warning 库屏蔽输出. 不可能有用, warning 库只能做到 python 层面屏蔽, 输出都不经过 python.
重定向 stderr. python 内置的 sys.stderr 本质还是经过 python 包装过的, 无法重定向 c 的输出.

解决方法#

还是重定向 stderr 输出管道, 并不优雅, 但至少没有一堆警告刷屏了.

与直接用sts.stderr = open(os.devnull, 'w') 不同, 这里需要更底层的操作.

这里实现一个上下文管理器, 在进入时重定向 stderr, 离开时恢复原状.

1
class LibpngFilter:
2
    def __enter__(self):
3
        self.pipe_in, pipe_out = os.pipe()
4
        self.orig_fd = os.dup(2)
5
        os.dup2(pipe_out, 2)
6
        os.close(pipe_out)
7
        self.stop = False
8

9
        def reader():
10
            f = os.fdopen(self.pipe_in)
11
            for line in f:
12
                if self.stop:
13
                    break
14
                if line.startswith("libpng warning:"): # 屏蔽逻辑, 可以根据需要修改
15
                    break
16
                sys.__stderr__.write(line)
17
                sys.__stderr__.flush()
18
            f.close()
19

20
        self.thread = threading.Thread(target=reader, daemon=True)
21
        self.thread.start()
22
        return self
23

24
    def __exit__(self, type, val, tb):
25
        self.stop = True
26
        os.dup2(self.orig_fd, 2)
27
        os.close(self.orig_fd)
28
        self.thread.join()

使用方法#

1
with LibpngFilter():
2
    ...
3
    # 处理 png 图片的代码