home *** CD-ROM | disk | FTP | other *** search
/ H4CK3R 14 / hacker14.iso / programacao / pythonwin / python.exe / PYTHON23.CHM / lib / required_2.txt < prev    next >
Encoding:
Text File  |  2003-10-02  |  1.3 KB  |  42 lines
  1. import optparse
  2.  
  3. class Option (optparse.Option):
  4.     ATTRS = optparse.Option.ATTRS + ['required']
  5.  
  6.     def _check_required (self):
  7.         if self.required and not self.takes_value():
  8.             raise OptionError(
  9.                 "required flag set for option that doesn't take a value",
  10.                  self)
  11.  
  12.     # Make sure _check_required() is called from the constructor!
  13.     CHECK_METHODS = optparse.Option.CHECK_METHODS + [_check_required]
  14.  
  15.     def process (self, opt, value, values, parser):
  16.         optparse.Option.process(self, opt, value, values, parser)
  17.         parser.option_seen[self] = 1
  18.  
  19.  
  20. class OptionParser (optparse.OptionParser):
  21.  
  22.     def _init_parsing_state (self):
  23.         optparse.OptionParser._init_parsing_state(self)
  24.         self.option_seen = {}
  25.  
  26.     def check_values (self, values, args):
  27.         for option in self.option_list:
  28.             if (isinstance(option, Option) and
  29.                 option.required and
  30.                 not self.option_seen.has_key(option)):
  31.                 self.error("%s not supplied" % option)
  32.         return (values, args)
  33.  
  34.  
  35. parser = OptionParser(option_list=[
  36.     Option("-v", action="count", dest="verbose"),
  37.     Option("-f", "--file", required=1)])
  38. (options, args) = parser.parse_args()
  39.  
  40. print "verbose:", options.verbose
  41. print "file:", options.file
  42.