在命令行Ruby中使ARGV [0]可选

时间:2017-03-03 14:35:36

标签: ruby command-line args

我有以下代码:

#!/usr/bin/env ruby
require 'yaml'
require 'json'
require 'getoptlong'
DEFAULT_CONF_FILE = 'conf.yaml'

    opts = GetoptLong.new(
        [ '--asn', '-a', GetoptLong::OPTIONAL_ARGUMENT ],
        [ '--modify', '-m', GetoptLong::OPTIONAL_ARGUMENT ]
    )

    config_file = ARGV[0]
    if config_file.to_s.empty?
      config_file = DEFAULT_CONF_FILE
    end

    opts.each do |opt, arg|
      case opt
        when '--asn'
          write_asn_database(arg,config_file)
        when '--modify'
          generate_modify_conf_file(arg,config_file)
      end
    end

此代码根据给定的YAML配置文件生成一些json文件。而我想要做的是:

  • 如果在命令行中给出了conf文件,程序将使用该给定文件;
  • 如果没有,它将使用默认文件。

类似的东西:

示例1(带有conf文件):

$ ./my_script.rb new_conf_file.yaml -a

示例2(没有conf文件):

$ ./my_script.rb -a

第一个示例有效,第二个示例给出了以下错误:

 No such file or directory @ rb_sysopen - -a (Errno::ENOENT)

这是有道理的,因为该程序假定选项-a是一个文件。

我是Ruby的新手,我从未使用命令行来运行脚本。

2 个答案:

答案 0 :(得分:1)

好的,我会在这里发布。请不要这样做,因为它只是丑陋:

config_file = ARGV[0]
unless File.exist?(config_file)
  config_file = DEFAULT_CONF_FILE
end

答案 1 :(得分:0)

GetoptLong documentation的第一段开始:

  

空选项--(两个减号)用于结束选项处理。 如果选项包含可选参数,这一点尤其重要。

通过调用脚本来使用您的代码:

#                 ⇓⇓⇓⇓⇓⇓⇓⇓⇓⇓⇓⇓⇓⇓⇓⇓⇓⇓⇓⇓⇓  
./my_script.rb -a -- new_conf_file.yaml

另外,应该在opts阅读部分之前明确处理ARGV

不,使用GetoptLong,无法正确解析您提供的命令行。