Merge pull request #104 from der-michik/master

Allow unpickling of LazyStrings

Author: TkTech

flask_babel/speaklater.py

@@ -9,6 +9,8 @@ def __init__(self, func, *args, **kwargs):
         self._kwargs = kwargs
 
     def __getattr__(self, attr):
+        if attr == "__setstate__":
+            raise AttributeError(attr)
         string = text_type(self)
         if hasattr(string, attr):
             return getattr(string, attr)

tests/tests.py

@@ -5,6 +5,8 @@
 import os
 sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..'))
 
+import pickle
+
 import unittest
 from decimal import Decimal
 import flask
@@ -60,6 +62,13 @@ def test_lazy_old_style_formatting(self):
         lazy_string = lazy_gettext(u'test')
         assert u'Hello %s' % lazy_string == u'Hello test'
 
+    def test_lazy_pickling(self):
+        lazy_string = lazy_gettext(u'Foo')
+        pickled = pickle.dumps(lazy_string)
+        unpickled = pickle.loads(pickled)
+
+        assert unpickled == lazy_string
+
 
 class DateFormattingTestCase(unittest.TestCase):