python - 如何创建内存文件对象

Question

我想制作一个内存文件以在 pygame 混音器中使用。我的意思是像 http://www.pygame.org/docs/ref/music.html#pygame.mixer.music.load其中说 load() 方法支持文件对象。

import requests
from pygame import mixer

r = requests.get("http://example.com/some_small_file.mp3")
in_memory_file = file(r.content) # something like this
mixer.music.init()
mixer.music.load(in_memory_file)
mixer.music.play()

Answer 1

您可能正在寻找 Python io 包中的 BytesIO 或 StringIO 类，两者都在 python 2 中提供和 python 3 .它们提供了一个类似文件的界面，您可以在代码中使用与与真实文件交互完全相同的方式。

StringIO 用于存储文本数据:

import io

f = io.StringIO("some initial text data")

BytesIO 必须用于二进制数据:

import io

f = io.BytesIO(b"\x00\x00\x00\x00\x00\x00\x00\x00\x01\x01\x01\x01\x01\x01")

要存储 MP3 文件数据，您可能需要 BytesIO 类。要将其从 GET 请求初始化到服务器，请执行以下操作:

import requests
from pygame import mixer
import io

r = requests.get("http://example.com/somesmallmp3file.mp3")
inmemoryfile = io.BytesIO(r.content)

mixer.music.init()
mixer.music.load(inmemoryfile)
mixer.music.play()

# This will free the memmory from any data
inmemoryfile.close()

附加说明:因为两个类都继承自 IOBase ，它们可以与 with 语句一起用作上下文管理器，因此您不再需要手动调用 close() 方法:

import requests
from pygame import mixer
import io

r = requests.get("http://example.com/somesmallmp3file.mp3")

with io.BytesIO(r.content) as inmemoryfile:
    mixer.music.init()
    mixer.music.load(inmemoryfile)
    mixer.music.play()