Grasscutters/mitmproxy
3
0
Fork 0
mirror of https://github.com/Grasscutters/mitmproxy.git synced 2024-11-23 00:01:36 +00:00
Code Issues Projects Releases Wiki Activity

optmanager: refactor for coverage and clarity

Browse Source
This commit is contained in:
Aldo Cortesi 2018-05-08 14:24:41 +12:00
parent 76a7883c8b
commit 717fbaa990
2 changed files with 6 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
mitmproxy/optmanager.py
View File

@ -298,7 +298,7 @@ class OptManager:
else:
optname, optval = parts[0], parts[1]
if optname in self._options:
vals[optname] = self.parse_setval(optname, optval)
vals[optname] = self.parse_setval(self._options[optname], optval)
else:
unknown[optname] = optval
if defer:
@ -315,19 +315,15 @@ class OptManager:
update = {}
for optname, optval in self._deferred.items():
if optname in self._options:
update[optname] = self.parse_setval(optname, optval)
update[optname] = self.parse_setval(self._options[optname], optval)
self.update(**update)
for k in update.keys():
del self._deferred[k]
def parse_setval(self, optname: str, optstr: typing.Optional[str]) -> typing.Any:
def parse_setval(self, o: _Option, optstr: typing.Optional[str]) -> typing.Any:
"""
Convert a string to a value appropriate for the option type.
"""
if optname not in self._options:
raise exceptions.OptionsError("No such option %s" % optname)
o = self._options[optname]
if o.typespec in (str, typing.Optional[str]):
return optstr
elif o.typespec in (int, typing.Optional[int]):
@ -337,7 +333,7 @@ class OptManager:
except ValueError:
raise exceptions.OptionsError("Not an integer: %s" % optstr)
elif o.typespec == int:
raise exceptions.OptionsError("Option is required: %s" % optname)
raise exceptions.OptionsError("Option is required: %s" % o.name)
else:
return None
elif o.typespec == bool:
@ -355,7 +351,7 @@ class OptManager:
if not optstr:
return []
else:
return getattr(self, optname) + [optstr]
return getattr(self, o.name) + [optstr]
raise NotImplementedError("Unsupported option type: %s", o.typespec)
def make_parser(self, parser, optname, metavar=None, short=None):

2
test/mitmproxy/test_optmanager.py
View File

@ -70,7 +70,7 @@ def test_defaults():
def test_required_int():
o = TO()
with pytest.raises(exceptions.OptionsError):
o.parse_setval("required_int", None)
o.parse_setval(o._options["required_int"], None)
def test_deepcopy():