diff options
author | Aldo Cortesi <aldo@nullcube.com> | 2016-12-03 12:16:50 +1300 |
---|---|---|
committer | Aldo Cortesi <aldo@nullcube.com> | 2016-12-08 10:21:06 +1300 |
commit | 150372e29728459b93f0fb46028ced9c2afe1d15 (patch) | |
tree | 1b5576202801d4f23929085081f6209ccba3259d | |
parent | 2cb1f7038196d31ecdd4190c009dc63a1f6d0538 (diff) | |
download | mitmproxy-150372e29728459b93f0fb46028ced9c2afe1d15.tar.gz mitmproxy-150372e29728459b93f0fb46028ced9c2afe1d15.tar.bz2 mitmproxy-150372e29728459b93f0fb46028ced9c2afe1d15.zip |
options: defaults are a class attribute
-rw-r--r-- | mitmproxy/optmanager.py | 13 | ||||
-rw-r--r-- | test/mitmproxy/test_optmanager.py | 5 |
2 files changed, 9 insertions, 9 deletions
diff --git a/mitmproxy/optmanager.py b/mitmproxy/optmanager.py index fd777d74..61421ba7 100644 --- a/mitmproxy/optmanager.py +++ b/mitmproxy/optmanager.py @@ -54,15 +54,6 @@ class OptManager(metaclass=_DefaultsMeta): # ._initialized = True as the final operation. instance = super().__new__(cls) instance.__dict__["_opts"] = {} - - defaults = {} - for klass in reversed(inspect.getmro(cls)): - for p in inspect.signature(klass.__init__).parameters.values(): - if p.kind in (p.KEYWORD_ONLY, p.POSITIONAL_OR_KEYWORD): - if not p.default == p.empty: - defaults[p.name] = p.default - instance.__dict__["_defaults"] = defaults - return instance def __init__(self): @@ -143,6 +134,10 @@ class OptManager(metaclass=_DefaultsMeta): """ self.update(**self._defaults) + @classmethod + def default(klass, opt): + return copy.deepcopy(klass._defaults[opt]) + def update(self, **kwargs): updated = set(kwargs.keys()) for k, v in kwargs.items(): diff --git a/test/mitmproxy/test_optmanager.py b/test/mitmproxy/test_optmanager.py index 97124368..0c98daea 100644 --- a/test/mitmproxy/test_optmanager.py +++ b/test/mitmproxy/test_optmanager.py @@ -30,6 +30,11 @@ class TD2(TD): def test_defaults(): + assert TD2.default("one") == "done" + assert TD2.default("two") == "dtwo" + assert TD2.default("three") == "dthree" + assert TD2.default("four") == "dfour" + o = TD2() assert o._defaults == { "one": "done", |