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

You are viewing proxied material from sdf.org. The copyright of proxied material belongs to its original authors. Any comments or complaints in relation to proxied material should be directed to the original authors of the content concerned. Please see the disclaimer for more details.