Python 版 freopen

qiuyouzhi

Python freopen

众所周知，Python 的 sys 库里面有两个东西，分别是 sys.stdin 和 sys.stdout。

看名字就知道它俩是干啥的8，一个输入，一个输出。

他俩本质是变量，所以是可以改变值的。比如，把 print 的输出定向到文件中，有两种方法：

第一种：

1. print("Hello world!", file = open("我是一个文件.txt", "w", encoding = "utf-8"))

复制代码

第二种：

1. import sys
   
2. 
   
3. sys.stdout = open("我是一个文件.txt", "w", encoding = "utf-8")
   
4. print("Hello world!")

复制代码

明显可以发现，后者在使用时不需要设置 file 参数，所以更加方便。

从文件里读取内容并传给 input 函数也是一个道理：

1. import sys
   
2. 
   
3. sys.stdin = open("我是一个文件.txt", "r", encoding = "utf-8")
   
4. s = input()
   
5. print(s)

复制代码

假设文件真的存在且有内容，运行一下，可以发现，打印出了文件第一行的内容。

所以不难发现，input 函数的本质其实是：

1. import sys
   
2. 
   
3. def input(prompt):
   
4.     print(prompt, end = "")
   
5.     return sys.stdin.readline().strip() # strip 是为了删除行尾附带的换行符

复制代码

综上所述，可以写出 freopen 函数原型：

1. import sys
   
2. 
   
3. # 由于使用 write 写入内容后，写入的内容会在缓冲区里而不是在文件里，所以使用 flush 刷新缓冲区
   
4. def print(*values, end = '\n', sep = ' ', file = sys.stdout, flush = True):
   
5.     file.write(sep.join(values) + end)
   
6.     if flush: file.flush()
   
7. 
   
8. def freopen(file, mode):
   
9.     # 由于 freopen 通常只用这两个文件模式，所以我就只写这俩了
   
10.     if mode == 'r':
    
11.         sys.stdin = open(file, mode, encoding = "utf-8")
    
12.     elif mode == 'w':
    
13.         sys.stdout = open(file, mode, encoding = "utf-8")

复制代码

如果你想搭建一个程序评测系统的话，这个很好用！

Rosemajor

拿去试试

矿洞菠菜

看看

sinaop