1 files changed, 16 insertions, 10 deletions
diff --git a/netlib/odict.py b/netlib/odict.py
index ea95a586..61448e6d 100644
--- a/netlib/odict.py
+++ b/netlib/odict.py
@@ -1,3 +1,4 @@
+from __future__ import (absolute_import, print_function, division)
 import re, copy
 
 
@@ -23,6 +24,9 @@ class ODict:
     def __eq__(self, other):
         return self.lst == other.lst
 
+    def __ne__(self, other):
+        return not self.__eq__(other)
+
     def __iter__(self):
         return self.lst.__iter__()
 
@@ -97,16 +101,6 @@ class ODict:
     def items(self):
         return self.lst[:]
 
-    def _get_state(self):
-        return [tuple(i) for i in self.lst]
-
-    def _load_state(self, state):
-        self.list = [list(i) for i in state]
-
-    @classmethod
-    def _from_state(klass, state):
-        return klass([list(i) for i in state])
-
     def copy(self):
         """
             Returns a copy of this object.
@@ -167,6 +161,18 @@ class ODict:
         self.lst = nlst
         return count
 
+    # Implement the StateObject protocol from mitmproxy
+    def get_state(self, short=False):
+        return [tuple(i) for i in self.lst]
+
+    def load_state(self, state):
+        self.list = [list(i) for i in state]
+
+    @classmethod
+    def from_state(klass, state):
+        return klass([list(i) for i in state])
+
+
 
 class ODictCaseless(ODict):
     """