diff python/optparse_undefined.py @ 193:3efe894f9d79

add a simple example for optparse_undefined
author Jeff Hammel <jhammel@mozilla.com>
date Fri, 20 Jan 2012 14:15:16 -0800
parents
children
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/python/optparse_undefined.py	Fri Jan 20 14:15:16 2012 -0800
@@ -0,0 +1,49 @@
+#!/usr/bin/env python
+
+"""
+illustrates usage of the undefined pattern from e.g.
+http://k0s.org/mozilla/hg/bzconsole/file/tip/bzconsole/command.py
+
+This is useful for scenarios similar to:
+- you have base configuration in a file
+- you have an OptionParser to read options in the command line
+- the CLI options should overwrite the file configuration iff
+  the user specifies an option
+"""
+
+from optparse import OptionParser
+
+class Undefined(object):
+    def __init__(self, default):
+        self.default=default
+
+class UndefinedOptionParser(OptionParser):
+
+    def add_option(self, *args, **kwargs):
+        kwargs['default'] = Undefined(kwargs.get('default'))
+        OptionParser.add_option(self, *args, **kwargs)
+
+    def parse_args(self, *args, **kwargs):
+        options, args = OptionParser.parse_args(self, *args, **kwargs)
+        return options, args
+
+if __name__ == '__main__':
+
+    myparser = UndefinedOptionParser()
+    myparser.add_option("--foo", dest='foo', default='hello')
+    myparser.add_option("--bar", dest='bar', default='goodbye')
+    myparser.add_option("--baz", dest='baz', default='helloagain')
+    myparser.add_option("--fleem", dest='fleem', default='aufwiedersehen')
+
+    myconfiguration = {'foo': 'hi',
+                       'bar': 'ciao'}
+    options, args = myparser.parse_args(['--foo', 'hello', '--baz', 'hola'])
+    for key, value in options.__dict__.items():
+        # XXX ideally you would move this to parse_args
+        if isinstance(value, Undefined):
+            options.__dict__[key] = myconfiguration.get(key, value.default)
+
+    assert options.foo == 'hello'
+    assert options.bar == 'ciao'
+    assert options.baz == 'hola'
+    assert options.fleem == 'aufwiedersehen'