三体论坛

歌者文明清理员

玩玩三体论坛
首先，需要将论坛数据保存到二进制文件里，所以使用 pickle 库。
一般我们只使用 dump 和 load 函数。
dump 函数的一般格式如下

1. dump(data, file)
   
2. data：数据
   
3. file：文件（open()）

复制代码

load 是从文件中加载数据。参数

1. load(file)
   
2. file：文件（open()）

复制代码

如果你去翻 python 官网的说明，你会发现这并不是完整的参数形式，缩水了很多
所以代码如下

1. from pickle import load, dump

复制代码

自定义 writeInData 函数，它的调用形式为

1. writeInData(dataType, data, fileName)
   
2. dataType：数据类型
   
3. data：数据
   
4. fileName：文件名

复制代码

函数的定义如下

1. 
   
2. def writeInData(dataType, data, fileName):
   
3.     """
   
4.     在文件中写入数据。
   
5.     :param dataType: 数据类型
   
6.     :param data: 数据
   
7.     :param fileName: 文件名
   
8.     :return : None
   
9.     """
   
10.     # 打开文件
    
11.     # 如果文件不存在
    
12.     try:
    
13.         file = open(fileName, 'rb')
    
14.         # 源数据，为字典
    
15.         originData = load(file)
    
16.         file.close()
    
17.     except:
    
18.         originData = dict()  # 为了和集合区分
    
19.     file = open(fileName, 'wb')
    
20.     # 怕是空的
    
21.     if dataType not in originData:
    
22.         originData[dataType] = []
    
23.     # 字典的每个值都是列表
    
24.     originData[dataType].append(data)
    
25.     dump(originData, file)
    
26.     file.close()

复制代码

读取回复消息

1. def readReply(fileName, mode, index):
   
2.     """
   
3.     读取回复消息
   
4.     :param fileName: 文件名
   
5.     :param mode: 模式
   
6.     :param index: 索引
   
7.     :return : 读取的信息
   
8.     """
   
9.     file = open(fileName, 'rb')
   
10.     data = load(file)
    
11.     if mode not in data:
    
12.         # 我还真是故意的
    
13.         raise ValueError('读不出来啊')
    
14.     # 回复信息
    
15.     replyMessage = data[mode][index]
    
16.     return '%s 说：%s' % replyMessage
    

复制代码

哦对了这是 tools.py
然后

1. 
   
2. TEST = -1
   
3. FILENAME = 'data.dat'
   
4. if __name__ == '__main__':
   
5.     # (谁, 说了啥)
   
6.     writeInData(TEST, ('大史', '面壁者罗辑，我是您的破壁人'), FILENAME)
   
7.     print(readReply(FILENAME, TEST, 0))
   

复制代码

成功！
完整代码如下

1. 
   
2. 
   
3. from pickle import load, dump
   
4. 
   
5. def writeInData(dataType, data, fileName):
   
6.     """
   
7.     在文件中写入数据。
   
8.     :param dataType: 数据类型
   
9.     :param data: 数据
   
10.     :param fileName: 文件名
    
11.     :return : None
    
12.     """
    
13.     # 打开文件
    
14.     # 如果文件不存在
    
15.     try:
    
16.         file = open(fileName, 'rb')
    
17.         # 源数据，为字典
    
18.         originData = load(file)
    
19.         file.close()
    
20.     except:
    
21.         originData = dict()  # 为了和集合区分
    
22.     file = open(fileName, 'wb')
    
23.     # 怕是空的
    
24.     if dataType not in originData:
    
25.         originData[dataType] = []
    
26.     # 字典的每个值都是列表
    
27.     originData[dataType].append(data)
    
28.     dump(originData, file)
    
29.     file.close()
    
30. 
    
31. 
    
32. def readReply(fileName, mode, index):
    
33.     """
    
34.     读取回复消息
    
35.     :param fileName: 文件名
    
36.     :param mode: 模式
    
37.     :param index: 索引
    
38.     :return : 读取的信息
    
39.     """
    
40.     file = open(fileName, 'rb')
    
41.     data = load(file)
    
42.     if mode not in data:
    
43.         # 我还真是故意的
    
44.         raise ValueError('读不出来啊')
    
45.     # 回复信息
    
46.     replyMessage = data[mode][index]
    
47.     return '%s 说：%s' % replyMessage
    
48. 
    
49. 
    
50. TEST = -1
    
51. FILENAME = 'data.dat'
    
52. if __name__ == '__main__':
    
53.     # (谁, 说了啥)
    
54.     writeInData(TEST, ('大史', '面壁者罗辑，我是您的破壁人'), FILENAME)
    
55.     print(readReply(FILENAME, TEST, 0))
    

复制代码

[b]这只是第一期！！！[/b]