From 79b8d42cc2629b53a7a696171221d7bd3a79b30c Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jarno=20Saarim=C3=A4ki?= Date: Fri, 27 Nov 2015 15:58:50 +0200 Subject: isodate.UTC remains the same object in a pickle --- src/isodate/tests/test_pickle.py | 6 ++++++ src/isodate/tzinfo.py | 14 ++++++++++++++ 2 files changed, 20 insertions(+) diff --git a/src/isodate/tests/test_pickle.py b/src/isodate/tests/test_pickle.py index b52f8cb..d24cf4a 100644 --- a/src/isodate/tests/test_pickle.py +++ b/src/isodate/tests/test_pickle.py @@ -36,6 +36,12 @@ class TestPickle(unittest.TestCase): self.assertEqual(len(failed), 0, "pickle protos failed: %s" % str(failed)) + def test_pickle_utc(self): + ''' + isodate.UTC objects remain the same after pickling. + ''' + self.assertTrue(isodate.UTC is pickle.loads(pickle.dumps(isodate.UTC))) + def test_suite(): ''' diff --git a/src/isodate/tzinfo.py b/src/isodate/tzinfo.py index b41f058..573b7b8 100644 --- a/src/isodate/tzinfo.py +++ b/src/isodate/tzinfo.py @@ -36,10 +36,24 @@ class Utc(tzinfo): ''' return ZERO + def __reduce__(self): + ''' + When unpickling a Utc object, return the default instance below, UTC. + ''' + return _Utc, () + + UTC = Utc() # the default instance for UTC. +def _Utc(): + ''' + Helper function for unpickling a Utc object. + ''' + return UTC + + class FixedOffset(tzinfo): ''' A class building tzinfo objects for fixed-offset time zones. -- cgit v1.2.1