42 lines
1.3 KiB
Python
42 lines
1.3 KiB
Python
|
import optparse
|
||
|
|
|
||
|
|
class Option (optparse.Option):
|
||
|
|
ATTRS = optparse.Option.ATTRS + ['required']
|
||
|
|
|
||
|
|
def _check_required (self):
|
||
|
|
if self.required and not self.takes_value():
|
||
|
|
raise OptionError(
|
||
|
|
"required flag set for option that doesn't take a value",
|
||
|
|
self)
|
||
|
|
|
||
|
|
# Make sure _check_required() is called from the constructor!
|
||
|
|
CHECK_METHODS = optparse.Option.CHECK_METHODS + [_check_required]
|
||
|
|
|
||
|
|
def process (self, opt, value, values, parser):
|
||
|
|
optparse.Option.process(self, opt, value, values, parser)
|
||
|
|
parser.option_seen[self] = 1
|
||
|
|
|
||
|
|
|
||
|
|
class OptionParser (optparse.OptionParser):
|
||
|
|
|
||
|
|
def _init_parsing_state (self):
|
||
|
|
optparse.OptionParser._init_parsing_state(self)
|
||
|
|
self.option_seen = {}
|
||
|
|
|
||
|
|
def check_values (self, values, args):
|
||
|
|
for option in self.option_list:
|
||
|
|
if (isinstance(option, Option) and
|
||
|
|
option.required and
|
||
|
|
not self.option_seen.has_key(option)):
|
||
|
|
self.error("%s not supplied" % option)
|
||
|
|
return (values, args)
|
||
|
|
|
||
|
|
|
||
|
|
parser = OptionParser(option_list=[
|
||
|
|
Option("-v", action="count", dest="verbose"),
|
||
|
|
Option("-f", "--file", required=1)])
|
||
|
|
(options, args) = parser.parse_args()
|
||
|
|
|
||
|
|
print "verbose:", options.verbose
|
||
|
|
print "file:", options.file
|