diff options
Diffstat (limited to 'libmproxy')
| -rw-r--r-- | libmproxy/proxy.py | 23 | ||||
| -rw-r--r-- | libmproxy/utils.py | 7 |
2 files changed, 15 insertions, 15 deletions
diff --git a/libmproxy/proxy.py b/libmproxy/proxy.py index 44bc10e2..c08040e6 100644 --- a/libmproxy/proxy.py +++ b/libmproxy/proxy.py @@ -26,13 +26,6 @@ class Config: self.pemfile = pemfile -def try_del(dict, key): - try: - del dict[key] - except KeyError: - pass - - def parse_url(url): """ Returns a (scheme, host, port, path) tuple, or None on error. @@ -151,10 +144,10 @@ class Request(controller.Msg): modifications to make sure interception works properly. """ headers = self.headers.copy() - try_del(headers, 'accept-encoding') - try_del(headers, 'proxy-connection') - try_del(headers, 'keep-alive') - try_del(headers, 'connection') + utils.try_del(headers, 'accept-encoding') + utils.try_del(headers, 'proxy-connection') + utils.try_del(headers, 'keep-alive') + utils.try_del(headers, 'connection') headers["connection"] = ["close"] data = (self.method, self.path, str(headers), self.content) return self.FMT%data @@ -211,10 +204,10 @@ class Response(controller.Msg): modifications to make sure interception works properly. """ headers = self.headers.copy() - try_del(headers, 'accept-encoding') - try_del(headers, 'proxy-connection') - try_del(headers, 'connection') - try_del(headers, 'keep-alive') + utils.try_del(headers, 'accept-encoding') + utils.try_del(headers, 'proxy-connection') + utils.try_del(headers, 'connection') + utils.try_del(headers, 'keep-alive') headers["connection"] = ["close"] proto = "%s %s %s"%(self.proto, self.code, self.msg) data = (proto, str(headers), self.content) diff --git a/libmproxy/utils.py b/libmproxy/utils.py index e29910c5..22f98d12 100644 --- a/libmproxy/utils.py +++ b/libmproxy/utils.py @@ -138,6 +138,13 @@ def _caseless(s): return s.lower() +def try_del(dict, key): + try: + del dict[key] + except KeyError: + pass + + class MultiDict: """ Simple wrapper around a dictionary to make holding multiple objects per |