Make Range pickleable

lib/_range.py

@@ -171,6 +171,17 @@ class Range(object):
         else:
             return self.__gt__(other)
 
+    def __getstate__(self):
+        return dict(
+            (slot, getattr(self, slot))
+            for slot in self.__slots__
+            if hasattr(self, slot)
+        )
+
+    def __setstate__(self, state):
+        for slot, value in state.items():
+            setattr(self, slot, value)
+
 
 def register_range(pgrange, pyrange, conn_or_curs, globally=False):
     """Create and register an adapter and the typecasters to convert between

tests/test_types_extras.py

@@ -20,6 +20,7 @@ import sys
 from decimal import Decimal
 from datetime import date, datetime
 from functools import wraps
+from pickle import dumps, loads
 
 from testutils import unittest, skip_if_no_uuid, skip_before_postgres
 from testutils import ConnectingTestCase, decorate_all_tests
@@ -1397,6 +1398,12 @@ class RangeTestCase(unittest.TestCase):
         with py3_raises_typeerror():
             self.assert_(Range(1, 2) >= 1)
 
+    def test_pickling(self):
+        from psycopg2.extras import Range
+
+        r = Range(0, 4)
+        self.assertEqual(loads(dumps(r)), r)
+
 
 def skip_if_no_range(f):
     @wraps(f)