[ruby/optparse] Add exact: keyword argument

https://github.com/ruby/optparse/commit/07e83673a8
This commit is contained in:
Nobuyoshi Nakada 2024-02-21 13:42:11 +09:00
parent 4831bb5bab
commit 41c0fb6991
2 changed files with 58 additions and 29 deletions

View File

@ -1654,21 +1654,21 @@ XXX
# #
# Returns the rest of +argv+ left unparsed. # Returns the rest of +argv+ left unparsed.
# #
def order(*argv, into: nil, &nonopt) def order(*argv, **keywords, &nonopt)
argv = argv[0].dup if argv.size == 1 and Array === argv[0] argv = argv[0].dup if argv.size == 1 and Array === argv[0]
order!(argv, into: into, &nonopt) order!(argv, **keywords, &nonopt)
end end
# #
# Same as #order, but removes switches destructively. # Same as #order, but removes switches destructively.
# Non-option arguments remain in +argv+. # Non-option arguments remain in +argv+.
# #
def order!(argv = default_argv, into: nil, &nonopt) def order!(argv = default_argv, into: nil, **keywords, &nonopt)
setter = ->(name, val) {into[name.to_sym] = val} if into setter = ->(name, val) {into[name.to_sym] = val} if into
parse_in_order(argv, setter, &nonopt) parse_in_order(argv, setter, **keywords, &nonopt)
end end
def parse_in_order(argv = default_argv, setter = nil, &nonopt) # :nodoc: def parse_in_order(argv = default_argv, setter = nil, exact: require_exact, **, &nonopt) # :nodoc:
opt, arg, val, rest = nil opt, arg, val, rest = nil
nonopt ||= proc {|a| throw :terminate, a} nonopt ||= proc {|a| throw :terminate, a}
argv.unshift(arg) if arg = catch(:terminate) { argv.unshift(arg) if arg = catch(:terminate) {
@ -1679,7 +1679,7 @@ XXX
opt, rest = $1, $2 opt, rest = $1, $2
opt.tr!('_', '-') opt.tr!('_', '-')
begin begin
if require_exact if exact
sw, = search(:long, opt) sw, = search(:long, opt)
else else
sw, = complete(:long, opt, true) sw, = complete(:long, opt, true)
@ -1714,7 +1714,7 @@ XXX
val = arg.delete_prefix('-') val = arg.delete_prefix('-')
has_arg = true has_arg = true
rescue InvalidOption rescue InvalidOption
raise if require_exact raise if exact
# if no short options match, try completion with long # if no short options match, try completion with long
# options. # options.
sw, = complete(:long, opt) sw, = complete(:long, opt)
@ -1779,18 +1779,18 @@ XXX
# <code>[]=</code> method (so it can be Hash, or OpenStruct, or other # <code>[]=</code> method (so it can be Hash, or OpenStruct, or other
# similar object). # similar object).
# #
def permute(*argv, into: nil) def permute(*argv, **keywords)
argv = argv[0].dup if argv.size == 1 and Array === argv[0] argv = argv[0].dup if argv.size == 1 and Array === argv[0]
permute!(argv, into: into) permute!(argv, **keywords)
end end
# #
# Same as #permute, but removes switches destructively. # Same as #permute, but removes switches destructively.
# Non-option arguments remain in +argv+. # Non-option arguments remain in +argv+.
# #
def permute!(argv = default_argv, into: nil) def permute!(argv = default_argv, **keywords)
nonopts = [] nonopts = []
order!(argv, into: into, &nonopts.method(:<<)) order!(argv, **keywords, &nonopts.method(:<<))
argv[0, 0] = nonopts argv[0, 0] = nonopts
argv argv
end end
@ -1802,20 +1802,20 @@ XXX
# values are stored there via <code>[]=</code> method (so it can be Hash, # values are stored there via <code>[]=</code> method (so it can be Hash,
# or OpenStruct, or other similar object). # or OpenStruct, or other similar object).
# #
def parse(*argv, into: nil) def parse(*argv, **keywords)
argv = argv[0].dup if argv.size == 1 and Array === argv[0] argv = argv[0].dup if argv.size == 1 and Array === argv[0]
parse!(argv, into: into) parse!(argv, **keywords)
end end
# #
# Same as #parse, but removes switches destructively. # Same as #parse, but removes switches destructively.
# Non-option arguments remain in +argv+. # Non-option arguments remain in +argv+.
# #
def parse!(argv = default_argv, into: nil) def parse!(argv = default_argv, **keywords)
if ENV.include?('POSIXLY_CORRECT') if ENV.include?('POSIXLY_CORRECT')
order!(argv, into: into) order!(argv, **keywords)
else else
permute!(argv, into: into) permute!(argv, **keywords)
end end
end end
@ -1838,7 +1838,7 @@ XXX
# # params[:bar] = "x" # --bar x # # params[:bar] = "x" # --bar x
# # params[:zot] = "z" # --zot Z # # params[:zot] = "z" # --zot Z
# #
def getopts(*args, symbolize_names: false) def getopts(*args, symbolize_names: false, **keywords)
argv = Array === args.first ? args.shift : default_argv argv = Array === args.first ? args.shift : default_argv
single_options, *long_options = *args single_options, *long_options = *args
@ -1866,7 +1866,7 @@ XXX
end end
end end
parse_in_order(argv, result.method(:[]=)) parse_in_order(argv, result.method(:[]=), **keywords)
symbolize_names ? result.transform_keys(&:to_sym) : result symbolize_names ? result.transform_keys(&:to_sym) : result
end end
@ -1979,10 +1979,10 @@ XXX
# The optional +into+ keyword argument works exactly like that accepted in # The optional +into+ keyword argument works exactly like that accepted in
# method #parse. # method #parse.
# #
def load(filename = nil, into: nil) def load(filename = nil, **keywords)
unless filename unless filename
basename = File.basename($0, '.*') basename = File.basename($0, '.*')
return true if load(File.expand_path(basename, '~/.options'), into: into) rescue nil return true if load(File.expand_path(basename, '~/.options'), **keywords) rescue nil
basename << ".options" basename << ".options"
return [ return [
# XDG # XDG
@ -1994,11 +1994,11 @@ XXX
'~/config/settings', '~/config/settings',
].any? {|dir| ].any? {|dir|
next if !dir or dir.empty? next if !dir or dir.empty?
load(File.expand_path(basename, dir), into: into) rescue nil load(File.expand_path(basename, dir), **keywords) rescue nil
} }
end end
begin begin
parse(*File.readlines(filename, chomp: true), into: into) parse(*File.readlines(filename, chomp: true), **keywords)
true true
rescue Errno::ENOENT, Errno::ENOTDIR rescue Errno::ENOENT, Errno::ENOTDIR
false false
@ -2011,10 +2011,10 @@ XXX
# #
# +env+ defaults to the basename of the program. # +env+ defaults to the basename of the program.
# #
def environment(env = File.basename($0, '.*')) def environment(env = File.basename($0, '.*'), **keywords)
env = ENV[env] || ENV[env.upcase] or return env = ENV[env] || ENV[env.upcase] or return
require 'shellwords' require 'shellwords'
parse(*Shellwords.shellwords(env)) parse(*Shellwords.shellwords(env), **keywords)
end end
# #
@ -2331,19 +2331,19 @@ XXX
# Parses +self+ destructively in order and returns +self+ containing the # Parses +self+ destructively in order and returns +self+ containing the
# rest arguments left unparsed. # rest arguments left unparsed.
# #
def order!(&blk) options.order!(self, &blk) end def order!(**keywords, &blk) options.order!(self, **keywords, &blk) end
# #
# Parses +self+ destructively in permutation mode and returns +self+ # Parses +self+ destructively in permutation mode and returns +self+
# containing the rest arguments left unparsed. # containing the rest arguments left unparsed.
# #
def permute!() options.permute!(self) end def permute!(**keywords) options.permute!(self, **keywords) end
# #
# Parses +self+ destructively and returns +self+ containing the # Parses +self+ destructively and returns +self+ containing the
# rest arguments left unparsed. # rest arguments left unparsed.
# #
def parse!() options.parse!(self) end def parse!(**keywords) options.parse!(self, **keywords) end
# #
# Substitution of getopts is possible as follows. Also see # Substitution of getopts is possible as follows. Also see
@ -2356,8 +2356,8 @@ XXX
# rescue OptionParser::ParseError # rescue OptionParser::ParseError
# end # end
# #
def getopts(*args, symbolize_names: false) def getopts(*args, symbolize_names: false, **keywords)
options.getopts(self, *args, symbolize_names: symbolize_names) options.getopts(self, *args, symbolize_names: symbolize_names, **keywords)
end end
# #

View File

@ -116,6 +116,35 @@ class TestOptionParser < Test::Unit::TestCase
assert_equal(false, @foo) assert_equal(false, @foo)
end end
def test_exact_option
@opt.def_option('-F', '--zrs=IRS', 'zrs')
%w(--zrs --zr --z -zfoo -z -F -Ffoo).each do |arg|
result = {}
@opt.parse([arg, 'foo'], into: result)
assert_equal({zrs: 'foo'}, result)
end
[%w(--zrs foo), %w(--zrs=foo), %w(-F foo), %w(-Ffoo)].each do |args|
result = {}
@opt.parse(args, into: result, exact: true)
assert_equal({zrs: 'foo'}, result)
end
assert_raise(OptionParser::InvalidOption) {@opt.parse(%w(--zr foo), exact: true)}
assert_raise(OptionParser::InvalidOption) {@opt.parse(%w(--z foo), exact: true)}
assert_raise(OptionParser::InvalidOption) {@opt.parse(%w(-zrs foo), exact: true)}
assert_raise(OptionParser::InvalidOption) {@opt.parse(%w(-zr foo), exact: true)}
assert_raise(OptionParser::InvalidOption) {@opt.parse(%w(-z foo), exact: true)}
@opt.def_option('-f', '--[no-]foo', 'foo') {|arg| @foo = arg}
@opt.parse(%w[-f], exact: true)
assert_equal(true, @foo)
@opt.parse(%w[--foo], exact: true)
assert_equal(true, @foo)
@opt.parse(%w[--no-foo], exact: true)
assert_equal(false, @foo)
end
def test_raise_unknown def test_raise_unknown
@opt.def_option('--my-foo [ARG]') {|arg| @foo = arg} @opt.def_option('--my-foo [ARG]') {|arg| @foo = arg}
assert @opt.raise_unknown assert @opt.raise_unknown