在Ruby中编辑文件中的每一行

Question

我正在尝试找到一种编辑文件中每一行的简单方法,而我在理解如何使用File该类时遇到了一些麻烦.

我要编辑的文件有几百行,每行有逗号分隔值.我只对每行中的第一个值感兴趣,并且我想在第一个值之后删除所有值.我试着做以下事情:

File.open('filename.txt', 'r+') do |file|
  file.each_line { |line| line = line.split(",")[0] }
  file.write
  file.close
end

哪个不起作用,因为File.write方法要求将内容写为参数.

有人可以告诉我如何达到预期的效果吗？

Answer 1

一个更好的解决方案(也是最安全的)是使用TempFile创建一个临时文件,并在完成后将其移动到原始位置(使用FileUtils):

   require 'fileutils'
   require 'tempfile'

    t_file = Tempfile.new('filename_temp.txt')
    File.open("filename.txt", 'r') do |f|
      f.each_line{|line| t_file.puts line.split(",")[0].to_s }
    end
    t_file.close
    FileUtils.mv(t_file.path, "filename.txt")

Answer 2

另一种修改文件的方法是使用-i交换机

ruby -F"," -i.bak -ane 'puts $F[0]' file

Answer 3

使用代码进行文件处理与我们在文本编辑器中编辑文件时的操作大不相同.操作系统提供的文件操作在这方面非常有限(由于许多,部分历史原因 - 认为磁带).

简而言之,您应该创建另一个文件并向其写入数据(Mike提供了相应的代码),或者将整个文件加载到内存中(如果文件很大,这可能是个坏主意)并用处理过的数据覆盖它.

只是为了练习,这里是你如何实际编辑文件的原因.如你所见,不是最漂亮的景象:

File.open('foo', 'r+') do |file|
  write_pos = 0
  file.each do |line|
    word = line.chomp.split(',').first
    read_pos = file.pos
    file.pos = write_pos
    file.puts word
    write_pos = file.pos
    file.pos = read_pos
  end
  file.truncate write_pos
end