我想要一个python脚本，它可以在不立即将其直接保存到磁盘的情况下截取屏幕截图。基本上是否有一个带有函数的模块，该函数返回原始字节，然后我可以自己手动将其写入文件？
import some_screenshot_module
raw_data = some_screenshot_module.return_raw_screenshot_bytes()
f = open('screenshot.png','wb')
f.write(raw_data)
f.close()
我已经检查了 mss、pyscreenshot 和 PIL，但我找不到我需要的东西。我找到了一个看起来像我要找的函数，叫做 frombytes。但是，在从 frombytes 函数检索字节并将其保存到文件中后，我无法将其视为 .BMP、.PNG、.JPG。是否有一个函数可以返回我可以自己保存到文件中的原始字节，或者可能是一个具有类似功能的模块？