浏览目录，更改定界符并更改文件扩展名

Question

就像标题所示，我试图遍历.tsv文件的目录，我想将定界符从制表符更改为“ |” 并将扩展名更改为.csv。 当我从命令行运行此文件并传递文件路径，旧扩展名和新扩展名（例如“ convert.py C：/ Users / MyDir / files .tsv .csv”）时，我收到错误消息：“ WindowsError： [错误2]系统找不到指定的文件。

这是我当前的代码：

import csv
import sys
import os

if len(sys.argv) < 4:
     sys.exit("Usage: convert_to_psv.py ~/dir .tsv .csv")
else:    
     cur_dir = (sys.argv[1])
     old_ext = (sys.argv[2])
     new_ext = (sys.argv[3])
     print "here1"
     files = os.listdir(cur_dir)
     for filename in files:
         file_ext = os.path.splitext(filename)[1]
         if old_ext == file_ext:
             newfile = filename.replace(old_ext, new_ext)
             os.rename(filename, newfile)
             csv.field_size_limit(sys.maxsize)
             csv.writer(file(sys.argv[3], 'w+'), delimiter="|").writerows(csv.reader(open(sys.argv[2]), delimiter="\t"))

我在这里先向您的帮助表示感谢！

Answer 1

您不能同时读取和写入同一文件。 您要么必须将所有内容读取到内存中，要么使用单独的输出文件。

最简单的方法是不重命名文件，而是写入新文件名，然后删除旧文件：

import csv
import sys
import os

if len(sys.argv) < 4:
     sys.exit("Usage: convert_to_psv.py ~/dir .tsv .csv")
else:    
     cur_dir, old_ext, new_ext = sys.argv[1:]
     for filename in os.listdir(cur_dir):
         filename = os.path.join(cur_dir, filename)
         base, file_ext = os.path.splitext(filename)
         if file_ext == old_ext:
             newfile = base + new_ext
             csv.field_size_limit(sys.maxsize)
             with open(filename, 'rb') as ifh, open(newfile, 'wb') as ofh:
                 reader = csv.reader(ifh, delimiter='\t')
                 csv.writer(ofh, delimiter='|').writerows(reader)
             os.unlink(filename)

请注意，这不一定会创建具有相同权限的新文件。