Merge branch 'msopko81-patch-1'

Author: asadurski

dateparser/timezone_parser.py

@@ -28,6 +28,9 @@ def localize(self, dt, is_dst=False):
             raise ValueError('Not naive datetime (tzinfo is already set)')
         return dt.replace(tzinfo=self)
 
+    def __getinitargs__(self):
+        return self.__name, self.__offset
+
 
 def pop_tz_offset_from_string(date_string, as_offset=True):
     for name, info in _tz_offsets:

tests/test_pickle.py

@@ -0,0 +1,33 @@
+import pickle
+from dateparser import parse
+from dateparser.timezone_parser import StaticTzInfo
+from datetime import datetime
+from tests import BaseTestCase
+
+
+class TestPickle(BaseTestCase):
+    def test_pickling_basic_parsed_object(self):
+        now = parse('now')
+        self.assertIsInstance(now, datetime)
+        pickle_dumps = pickle.dumps(now)
+        self.assertIsInstance(pickle_dumps, bytes)
+
+    def test_pickling_parsed_object_with_tz(self):
+        now_in_utc = parse('now in UTC')
+        self.assertIsInstance(now_in_utc, datetime)
+        self.assertIsInstance(now_in_utc.tzinfo, StaticTzInfo)
+        pickle_dumps = pickle.dumps(now_in_utc)
+        self.assertIsInstance(pickle_dumps, bytes)
+
+    def test_unpickling_basic_parsed_object(self):
+        now = parse('now')
+        pickle_dumps = pickle.dumps(now)
+        pickle_loads = pickle.loads(pickle_dumps)
+        self.assertIsInstance(pickle_loads, datetime)
+
+    def test_unpickling_parsed_object_with_tz(self):
+        now_in_utc = parse('now in UTC')
+        pickle_dumps = pickle.dumps(now_in_utc)
+        pickle_loads = pickle.loads(pickle_dumps)
+        self.assertIsInstance(pickle_loads, datetime)
+        self.assertIsInstance(pickle_loads.tzinfo, StaticTzInfo)