Selaa lähdekoodia

Fix TypeError raised in logging (validate_models)

Alexander 10 vuotta sitten
vanhempi
commit
d22e17ddc9
1 muutettua tiedostoa jossa 6 lisäystä ja 2 poistoa
  1. 6 2
      celery/fixups/django.py

+ 6 - 2
celery/fixups/django.py

@@ -1,10 +1,14 @@
 from __future__ import absolute_import
 from __future__ import absolute_import
 
 
-import io
 import os
 import os
 import sys
 import sys
 import warnings
 import warnings
 
 
+if sys.version_info[0] < 3 and not hasattr(sys, 'pypy_version_info'):
+    from StringIO import StringIO
+else:
+    from io import StringIO
+
 from kombu.utils import cached_property, symbol_by_name
 from kombu.utils import cached_property, symbol_by_name
 
 
 from datetime import datetime
 from datetime import datetime
@@ -152,7 +156,7 @@ class DjangoWorkerFixup(object):
             pass
             pass
         else:
         else:
             django_setup()
             django_setup()
-        s = io.StringIO()
+        s = StringIO()
         try:
         try:
             from django.core.management.validation import get_validation_errors
             from django.core.management.validation import get_validation_errors
         except ImportError:
         except ImportError: