PY-11855 Run manage.py task improvements

[idea/community.git] / python / helpers / pycharm / django_obtainer.py

diff --git a/python/helpers/pycharm/django_obtainer.py b/python/helpers/pycharm/django_obtainer.py
index 15bf56cfe9c3455ce6d74657e6ab2fa9219ea63b..3f9849c4ad7fc8392c609ed5ddf5e536bfb46138 100644 (file)
--- a/python/helpers/pycharm/django_obtainer.py
+++ b/python/helpers/pycharm/django_obtainer.py
@@ -14,7 +14,11 @@ from _django_obtainer_core import XmlDumper
 __author__ = 'Ilya.Kazakevich'
 
 # TODO: Support Django 1.8 as well, it uses argparse, not optparse
-assert LooseVersion(django.get_version()) < LooseVersion('1.8a'), "Only Django <1.8 is supported now"
+version = LooseVersion(django.get_version())
+assert version < LooseVersion('1.8a'), "Only Django <1.8 is supported now"
+# Some django versions require setup
+if django.setup:
+    django.setup()
 dumper = XmlDumper()
 _django_obtainer_optparse.report_data(dumper)
 print(dumper.xml)
\ No newline at end of file