Python使用标准输出和文件输入写入文件

问题描述：

我有以下代码，它通过进行正则表达式替换来修改文件test.tex的每一行。Python使用标准输出和文件输入写入文件

import re 
import fileinput 

regex=re.compile(r'^([^&]*)(&)([^&]*)(&)([^&]*)') 

for line in fileinput.input('test.tex',inplace=1): 
    print regex.sub(r'\3\2\1\4\5',line),

唯一的问题是，我只想替换应用到文件中的某些行，而且也没有办法定义一个模式来选择正确的线路。因此，我想要显示每行并在命令行中提示用户，询问是否在当前行进行替换。如果用户输入“y”，则进行替换。如果用户根本没有输入任何内容，则替换为而不是。

问题当然是，通过使用代码inplace=1我已经有效地将stdout重定向到打开的文件。所以没有办法显示输出（例如询问是否进行替换）到没有发送到文件的命令行。

任何想法？

使用stderr ..... –

'fileinput'不是这个工作的正确工具。只需使用标准的读取 - 修改 - 写入模式 –

@EliBendersky您能否指点我做一个提及的例子？对不起，我对Python很陌生。 – synaptik

答

文件输入模块实际上是处理多个输入文件。您可以使用常规的open（）函数。

这样的事情应该工作。

通过阅读文件，然后重新与求（指针），我们可以覆盖的文件，而不是追加到最后，所以编辑就地文件

import re 

regex = re.compile(r'^([^&]*)(&)([^&]*)(&)([^&]*)') 

with open('test.tex', 'r+') as f: 
    old = f.readlines() # Pull the file contents to a list 
    f.seek(0) # Jump to start, so we overwrite instead of appending 
    for line in old: 
     s = raw_input(line) 
     if s == 'y': 
      f.write(regex.sub(r'\3\2\1\4\5',line)) 
     else: 
      f.write(line)

http://docs.python.org/tutorial/inputoutput.html

当然，如果你有一个太大而无法加载到内存中的大文件，那么你可以一次读一行，然后写入一个临时文件。 –

非常感谢！ :) – synaptik

答

基础的关于大家提供的帮助，这里是我最终的结果：

#!/usr/bin/python 

import re 
import sys 
import os 

# regular expression 
regex = re.compile(r'^([^&]*)(&)([^&]*)(&)([^&]*)') 

# name of input and output files 
if len(sys.argv)==1: 
    print 'No file specified. Exiting.' 
    sys.exit() 
ifilename = sys.argv[1] 
ofilename = ifilename+'.MODIFIED' 

# read input file 
ifile = open(ifilename) 
lines = ifile.readlines() 

ofile = open(ofilename,'w') 

# prompt to make substitutions wherever a regex match occurs 
for line in lines: 
    match = regex.search(line)  
    if match is not None: 
     print '' 
     print '***CANDIDATE FOR SUBSTITUTION***' 
     print '--: '+line, 
     print '++: '+regex.sub(r'\3\2\1\4\5',line), 
     print '********************************' 
     input = raw_input('Make subsitution (enter y for yes)? ') 
     if input == 'y': 
      ofile.write(regex.sub(r'\3\2\1\4\5',line)) 
     else: 
      ofile.write(line) 
    else: 
     ofile.write(line) 

# replace original file with modified file 
os.remove(ifilename) 
os.rename(ofilename, ifilename)

非常感谢！

Python使用标准输出和文件输入写入文件

相关推荐