diff options
author | Aldo Cortesi <aldo@nullcube.com> | 2016-05-31 17:16:31 +1200 |
---|---|---|
committer | Aldo Cortesi <aldo@nullcube.com> | 2016-05-31 17:16:31 +1200 |
commit | 08fbe6f1118455bc44d05db30b83bdf81feda2a0 (patch) | |
tree | 38db5d8f3e1e52e09a9348d15db368ae3baa9b86 /netlib/http/message.py | |
parent | 2f526393d29b6a03e43d1f6240175b4dfb13dc7d (diff) | |
download | mitmproxy-08fbe6f1118455bc44d05db30b83bdf81feda2a0.tar.gz mitmproxy-08fbe6f1118455bc44d05db30b83bdf81feda2a0.tar.bz2 mitmproxy-08fbe6f1118455bc44d05db30b83bdf81feda2a0.zip |
Start cleaning up netlib.utils
- Remove http2 functions, move to http2.frame
- Remove Serializable, move to netlib.basetypes
Diffstat (limited to 'netlib/http/message.py')
-rw-r--r-- | netlib/http/message.py | 17 |
1 files changed, 8 insertions, 9 deletions
diff --git a/netlib/http/message.py b/netlib/http/message.py index 13d401a7..d9654f26 100644 --- a/netlib/http/message.py +++ b/netlib/http/message.py @@ -4,9 +4,8 @@ import warnings import six -from .headers import Headers -from .. import encoding, utils -from ..utils import always_bytes +from .. import encoding, utils, basetypes +from . import headers if six.PY2: # pragma: no cover def _native(x): @@ -20,10 +19,10 @@ else: return x.decode("utf-8", "surrogateescape") def _always_bytes(x): - return always_bytes(x, "utf-8", "surrogateescape") + return utils.always_bytes(x, "utf-8", "surrogateescape") -class MessageData(utils.Serializable): +class MessageData(basetypes.Serializable): def __eq__(self, other): if isinstance(other, MessageData): return self.__dict__ == other.__dict__ @@ -38,7 +37,7 @@ class MessageData(utils.Serializable): def set_state(self, state): for k, v in state.items(): if k == "headers": - v = Headers.from_state(v) + v = headers.Headers.from_state(v) setattr(self, k, v) def get_state(self): @@ -48,11 +47,11 @@ class MessageData(utils.Serializable): @classmethod def from_state(cls, state): - state["headers"] = Headers.from_state(state["headers"]) + state["headers"] = headers.Headers.from_state(state["headers"]) return cls(**state) -class Message(utils.Serializable): +class Message(basetypes.Serializable): def __eq__(self, other): if isinstance(other, Message): return self.data == other.data @@ -72,7 +71,7 @@ class Message(utils.Serializable): @classmethod def from_state(cls, state): - state["headers"] = Headers.from_state(state["headers"]) + state["headers"] = headers.Headers.from_state(state["headers"]) return cls(**state) @property |