如何使用 OptionParser 解析 rake 参数

Question

引用that answer 我试图使用OptionParser 来解析rake 参数。我从那里简化了示例，我必须添加两个 ARGV.shift 才能使其工作。

require 'optparse'

namespace :user do |args|

  # Fix I hate to have here
  puts "ARGV: #{ARGV}"
  ARGV.shift
  ARGV.shift
  puts "ARGV: #{ARGV}"

  desc 'Creates user account with given credentials: rake user:create'
  # environment is required to have access to Rails models
  task :create => :environment do
    options = {}
    OptionParser.new(args) do |opts|      
      opts.banner = "Usage: rake user:create [options]"
      opts.on("-u", "--user {username}","Username") { |user| options[:user] = user }
    end.parse!

    puts "user: #{options[:user]}"

    exit 0
  end
end

这是输出：

$ rake user:create -- -u foo
ARGV: ["user:create", "--", "-u", "foo"]
ARGV: ["-u", "foo"]
user: foo

我认为ARGV.shift 不应该这样做。我想知道为什么没有它就无法工作以及如何以适当的方式修复它。

Answer 1

您必须在 -u 和 foo 之间添加一个“=”：

$ rake user:create -- -u=foo

代替：

$ rake user:create -- -u foo

Answer 2

我知道这并不能严格回答您的问题，但您是否考虑过使用任务参数？

这样您就不必再摆弄OptionParser 和ARGV：

namespace :user do |args|
  desc 'Creates user account with given credentials: rake user:create'
  task :create, [:username] => :environment do |t, args|
    # when called with rake user:create[foo],
    # args is now {username: 'foo'} and you can access it with args[:username]
  end
end

有关详细信息，请参阅this answer here on SO。

Answer 3

您可以使用OptionParser#order! 方法，它返回没有错误参数的 ARGV：

options = {}

o = OptionParser.new

o.banner = "Usage: rake user:create [options]"
o.on("-u NAME", "--user NAME") { |username|
  options[:user] = username
}
args = o.order!(ARGV) {}
o.parse!(args)
puts "user: #{options[:user]}"

你可以这样传递参数：$ rake foo:bar -- '--user=john'

Answer 4

试试这个：

require 'optparse'

namespace :programs do |args|
  desc "Download whatever"
  task :download => [:environment] do

    # USAGE: rake programs:download -- rm
    #-- Setting options $ rake programs:download -- --rm
    options = {}
    option_parser = OptionParser.new
    option_parser.banner = "Usage: rake programs:download -- rm"
    option_parser.on("-r", "--rm","Remove s3 files downloaded") do |value|
      options[:remove] = value
    end
    args = option_parser.order!(ARGV) {}
    option_parser.parse!(args)
    #-- end

    # your code here
  end
end