Python中临时音频文件删除策略：利用内存文件对象解决文件占用问题

import os
import io
import time
import pygame

# 假设 speech.save() 是一个将音频保存到文件的函数
# 这里的 speech.lang 和 speech.save() 是示例，你需要替换为你的实际音频生成逻辑
class MockSpeech:
    def __init__(self):
        self.lang = "en"
    def save(self, filename):
        with open(filename, "wb") as f:
            f.write(b"RIFF\x00\x00\x00\x00WAVEfmt \x10\x00\x00\x00\x01\x00\x01\x00\x44\xac\x00\x00\x88\x58\x01\x00\x02\x00\x10\x00data\x00\x00\x00\x00") # 这是一个非常小的mock wav头，实际音频数据会更大
            f.write(b"\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00") # 示例音频数据

speech = MockSpeech()
pygame.mixer.init()

temp_file = f"temp.{speech.lang}.wav" # 假设是wav文件，更通用
speech.save(temp_file) # 先将音频保存到磁盘文件

try:
    # 读取磁盘文件内容到内存
    with open(temp_file, "rb") as f:
        buf = io.BytesIO(f.read())

    # 将内存文件对象指针重置到开头
    buf.seek(0) 

    # 使用内存文件对象加载音频
    # 注意：pygame.mixer.music.load()可以接受文件对象
    # 第二个参数是可选的名称提示，有时有助于pygame识别文件类型
    pygame.mixer.music.load(buf, namehint="wav") 

    pygame.mixer.music.play()
    while pygame.mixer.music.get_busy():
        pygame.time.Clock().tick(10)
    pygame.mixer.music.stop()

except pygame.error as e:
    print(f"[Playback Error]: {e}")
except Exception as e:
    print(f"[Unexpected Playback Error]: {e}")
finally:
    # 确保在播放结束后，pygame不再持有对内存对象的引用
    # 这一步不是严格必需的，因为buf是局部变量，但养成好习惯有助于资源管理
    pygame.mixer.music.unload() 

time.sleep(1) # 给系统一点时间释放资源，虽然对于内存对象通常不是问题

# 现在可以安全地删除磁盘上的临时文件了
if os.path.exists(temp_file):
    try:
        os.remove(temp_file)
        print(f"Successfully deleted temporary file: {temp_file}")
    except OSError as e:
        print(f"Error deleting file {temp_file}: {e}")

pygame.mixer.quit()

import os
import io
import time
import pygame

class MockSpeech:
    def __init__(self):
        self.lang = "en"
    # 修改 save 方法以接受文件对象
    def save(self, file_obj):
        file_obj.write(b"RIFF\x00\x00\x00\x00WAVEfmt \x10\x00\x00\x00\x01\x00\x01\x00\x44\xac\x00\x00\x88\x58\x01\x00\x02\x00\x10\x00data\x00\x00\x00\x00")
        file_obj.write(b"\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00")

speech = MockSpeech()
pygame.mixer.init()

# 直接创建内存文件对象
buf = io.BytesIO()
speech.save(buf) # 将音频数据直接保存到内存对象

# 将内存文件对象指针重置到开头
buf.seek(0) 

try:
    pygame.mixer.music.load(buf, namehint="wav")

    pygame.mixer.music.play()
    while pygame.mixer.music.get_busy():
        pygame.time.Clock().tick(10)
    pygame.mixer.music.stop()

except pygame.error as e:
    print(f"[Playback Error]: {e}")
except Exception as e:
    print(f"[Unexpected Playback Error]: {e}")
finally:
    pygame.mixer.music.unload()
    # 内存文件对象无需显式删除，它会在不再被引用时被Python垃圾回收
    # buf.close() # 可选，显式关闭，但通常不是必需的

print("Audio played from memory. No temporary disk file created or deleted.")

pygame.mixer.quit()