Doc/lib/caseless.py

   1 from optparse import Option, OptionParser, _match_abbrev
   2
   3 # This case-insensitive option parser relies on having a
   4 # case-insensitive dictionary type available.  Here's one
   5 # for Python 2.2.  Note that a *real* case-insensitive
   6 # dictionary type would also have to implement __new__(),
   7 # update(), and setdefault() -- but that's not the point
   8 # of this exercise.
   9
  10 class caseless_dict (dict):
  11     def __setitem__ (self, key, value):
  12         dict.__setitem__(self, key.lower(), value)
  13
  14     def __getitem__ (self, key):
  15         return dict.__getitem__(self, key.lower())
  16
  17     def get (self, key, default=None):
  18         return dict.get(self, key.lower())
  19
  20     def has_key (self, key):
  21         return dict.has_key(self, key.lower())
  22
  23
  24 class CaselessOptionParser (OptionParser):
  25
  26     def _create_option_list (self):
  27         self.option_list = []
  28         self._short_opt = caseless_dict()
  29         self._long_opt = caseless_dict()
  30         self._long_opts = []
  31         self.defaults = {}
  32
  33     def _match_long_opt (self, opt):
  34         return _match_abbrev(opt.lower(), self._long_opt.keys())
  35
  36
  37 if __name__ == "__main__":
  38     from optik.errors import OptionConflictError
  39
  40     # test 1: no options to start with
  41     parser = CaselessOptionParser()
  42     try:
  43         parser.add_option("-H", dest="blah")
  44     except OptionConflictError:
  45         print "ok: got OptionConflictError for -H"
  46     else:
  47         print "not ok: no conflict between -h and -H"
  48
  49     parser.add_option("-f", "--file", dest="file")
  50     #print repr(parser.get_option("-f"))
  51     #print repr(parser.get_option("-F"))
  52     #print repr(parser.get_option("--file"))
  53     #print repr(parser.get_option("--fIlE"))
  54     (options, args) = parser.parse_args(["--FiLe", "foo"])
  55     assert options.file == "foo", options.file
  56     print "ok: case insensitive long options work"
  57
  58     (options, args) = parser.parse_args(["-F", "bar"])
  59     assert options.file == "bar", options.file
  60     print "ok: case insensitive short options work"