Ruby 中字符串的一部分递增

Question

我在 Ruby 脚本中有一个方法，它尝试在保存文件之前重命名文件。它看起来像这样：

def increment (path)
    if path[-3,2] == "_#"
        print "    Incremented file with that name already exists, renaming\n"
        count = path[-1].chr.to_i + 1
        return path.chop! << count.to_s
    else
        print "    A file with that name already exists, renaming\n"
        return path << "_#1"
    end
end

假设您有 3 个同名文件保存到一个目录中，我们会说该文件名为 example.mp3。这个想法是，第一个将保存为 example.mp3 （因为它不会被 if File.exists?("#{file_path}.mp3") 捕获> 脚本中的其他位置），第二个将保存为 example_#1.mp3 （因为它被上述方法的 else 部分捕获），第三个将保存为example_#2.mp3（因为它被上述方法的 if 部分捕获）。

我遇到的问题是双重的。

1) if path[-3,2] == "_#" 不适用于整数超过一位的文件 (example_#11.mp3例如），因为字符位置将是错误的（您需要它是 path[-4,2] 但这样就不能处理 3 位数字等）。

2）我永远不会遇到问题1），因为该方法不能可靠地捕获文件名。目前，它将第一个重命名为 example_#1.mp3，但第二个将重命名为相同的内容（导致它覆盖以前保存的文件）。

这对于 Stack Overflow 来说可能太模糊了，但我找不到任何可以解决递增字符串特定部分问题的内容。

提前致谢！

编辑/更新：

韦恩的下面的方法似乎可以单独工作，但当作为整个脚本的一部分包含时则不行 - 它可以将文件增加一次（从 example.mp3 到example_#1.mp3），但无法处理获取 example_#1.mp3 并将其递增到 example_#2.mp3。为了提供更多上下文 - 当前，当脚本找到要保存的文件时，会将名称传递给 Wayne 的方法，如下所示：

file_name = increment(image_name)
File.open("images/#{file_name}.jpeg", 'w') do |output|
    open(image_url) do |input|
        output << input.read
    end
end    

我对 Wayne 的脚本进行了一些编辑，所以现在看起来像这样：

def increment (name)
    name = name.gsub(/\s{2,}|(http:\/\/)|(www.)/i, '')
    if File.exists?("images/#{name}.jpeg")
        _, filename, count, extension = *name.match(/(\A.*?)(?:_#(\d+))?(\.[^.]*)?\Z/)
        count = (count || '0').to_i + 1
        "#{name}_##{count}#{extension}"
    else
        return name
    end
end

我哪里出错了？再次提前致谢。

Answer 1

正则表达式将完成：

#!/usr/bin/ruby1.8

def increment(path)
  _, filename, count, extension = *path.match(/(\A.*?)(?:_#(\d+))?(\.[^.]*)?\Z/)
  count = (count || '0').to_i + 1
  "#{filename}_##{count}#{extension}"
end

p increment('example')        # => "example_#1"
p increment('example.')       # => "example_#1."
p increment('example.mp3')    # => "example_#1.mp3"
p increment('example_#1.mp3') # => "example_#2.mp3"
p increment('example_#2.mp3') # => "example_#3.mp3"

这对于您正在编写的代码来说可能并不重要，但是如果您可能有多个线程或进程在同一文件上使用此算法，则在检查之前是否存在时会出现竞争条件保存：两个写入者都可以找到未使用的相同文件名并写入它。如果这对您很重要，请以某种模式打开该文件（如果存在），该模式会失败，从而挽救异常。当异常发生时，选择不同的名称。大致：

loop do
  begin
    File.open(filename, File::CREAT | File::EXCL | File::WRONLY) do |file|
      file.puts "Your content goes here"
    end
    break
  rescue Errno::EEXIST
    filename = increment(filename)
    redo
  end
end

Answer 2

这是不接受具有现有计数的文件名的变体：

def non_colliding_filename( filename )
  if File.exists?(filename)
    base,ext = /\A(.+?)(\.[^.]+)?\Z/.match( filename ).to_a[1..-1]
    i = 1
    i += 1 while File.exists?( filename="#{base}_##{i}#{ext}" )
  end
  filename
end

证明：

%w[ foo bar.mp3 jim.bob.mp3 ].each do |desired|
  3.times{
    file = non_colliding_filename( desired )
    p file
    File.open( file, 'w' ){ |f| f << "tmp" }
  }
end
#=> "foo"
#=> "foo_#1"
#=> "foo_#2"
#=> "bar.mp3"
#=> "bar_#1.mp3"
#=> "bar_#2.mp3"
#=> "jim.bob.mp3"
#=> "jim.bob_#1.mp3"
#=> "jim.bob_#2.mp3"