| OLD | NEW |
| (Empty) | |
| 1 # Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| 2 # Use of this source code is governed by a BSD-style license that can be |
| 3 # found in the LICENSE file. |
| 4 |
| 5 import getopt |
| 6 import sys |
| 7 |
| 8 from idl_log import ErrOut, InfoOut, WarnOut |
| 9 |
| 10 OptionMap = { } |
| 11 |
| 12 |
| 13 def GetOption(name): |
| 14 if name not in OptionMap: |
| 15 raise RuntimeError('Could not find option "%s".' % name) |
| 16 return OptionMap[name].Get() |
| 17 |
| 18 class Option(object): |
| 19 def __init__(self, name, desc, default = None, callfunc = None, |
| 20 testfunc = None, cookie = None): |
| 21 |
| 22 # Verify this option is not a duplicate |
| 23 if name in OptionMap: |
| 24 raise RuntimeError('Option "%s" already exists.' % name) |
| 25 self.name = name |
| 26 self.desc = desc |
| 27 self.default = default |
| 28 self.value = default |
| 29 self.callfunc = callfunc |
| 30 self.testfunc = testfunc |
| 31 self.cookie = cookie |
| 32 OptionMap[name] = self |
| 33 |
| 34 def Set(self, value): |
| 35 if self.testfunc: |
| 36 if not self.testfunc(self, value): return False |
| 37 # If this is a boolean option, set it to true |
| 38 if self.default is None: |
| 39 self.value = True |
| 40 else: |
| 41 self.value = value |
| 42 if self.callfunc: |
| 43 self.callfunc(self) |
| 44 return True |
| 45 |
| 46 def Get(self): |
| 47 return self.value |
| 48 |
| 49 |
| 50 def DumpOption(option): |
| 51 if len(option.name) > 1: |
| 52 out = ' --%-15.15s\t%s' % (option.name, option.desc) |
| 53 else: |
| 54 out = ' -%-15.15s\t%s' % (option.name, option.desc) |
| 55 if option.default: |
| 56 out = '%s\n\t\t\t(Default: %s)\n' % (out, option.default) |
| 57 InfoOut.Log(out) |
| 58 |
| 59 def DumpHelp(option=None): |
| 60 InfoOut.Log('Usage:') |
| 61 for opt in sorted(OptionMap.keys()): |
| 62 DumpOption(OptionMap[opt]) |
| 63 sys.exit(0) |
| 64 |
| 65 # |
| 66 # Default IDL options |
| 67 # |
| 68 # -h : Help, prints options |
| 69 # --verbose : use verbose output |
| 70 # --test : test this module |
| 71 # |
| 72 Option('h', 'Help', callfunc=DumpHelp) |
| 73 Option('help', 'Help', callfunc=DumpHelp) |
| 74 Option('verbose', 'Verbose') |
| 75 Option('test', 'Test the IDL scripts') |
| 76 |
| 77 def ParseOptions(args): |
| 78 short_opts= "" |
| 79 long_opts = [] |
| 80 |
| 81 # Build short and long option lists |
| 82 for name in sorted(OptionMap.keys()): |
| 83 option = OptionMap[name] |
| 84 if len(name) > 1: |
| 85 if option.default is None: |
| 86 long_opts.append('%s' % name) |
| 87 else: |
| 88 long_opts.append('%s=' % name) |
| 89 else: |
| 90 if option.default is None: |
| 91 short_opts += name |
| 92 else: |
| 93 short_opts += '%s:' % name |
| 94 |
| 95 try: |
| 96 opts, filenames = getopt.getopt(args, short_opts, long_opts) |
| 97 |
| 98 for opt, val in opts: |
| 99 if len(opt) == 2: opt = opt[1:] |
| 100 if opt[0:2] == '--': opt = opt[2:] |
| 101 OptionMap[opt].Set(val) |
| 102 |
| 103 except getopt.error, e: |
| 104 ErrOut.Log('Illegal option: %s\n' % str(e)) |
| 105 DumpHelp() |
| 106 sys.exit(-1) |
| 107 |
| 108 return filenames |
| OLD | NEW |