可以使用文件、缓存或内存等方式来存储聊天数据,以下是一个使用文件存储聊天数据的示例:
import os
import json
class ChatData:
def __init__(self, filename):
self.filename = filename
def save_message(self, message):
if os.path.exists(self.filename):
with open(self.filename, 'r') as f:
data = json.load(f)
else:
data = []
data.append(message)
with open(self.filename, 'w') as f:
json.dump(data, f)
def get_messages(self):
if os.path.exists(self.filename):
with open(self.filename, 'r') as f:
data = json.load(f)
else:
data = []
return data
使用示例:
chat = ChatData('chat_data.json')
chat.save_message('hello')
chat.save_message('world')
messages = chat.get_messages()
print(messages) # ['hello', 'world']