Commit b7c134ef authored by Dom Sekotill's avatar Dom Sekotill
Browse files

Add support for zipped packages to the migration questioner

parent 607d436e
Loading
Loading
Loading
Loading
Loading
+2 −6
Original line number Diff line number Diff line
@@ -3,6 +3,7 @@ from __future__ import print_function, unicode_literals
import importlib
import os
import sys
from pkg_resources import resource_listdir

from django.apps import apps
from django.db.models.fields import NOT_PROVIDED
@@ -46,12 +47,7 @@ class MigrationQuestioner(object):
        except ImportError:
            return self.defaults.get("ask_initial", False)
        else:
            if hasattr(migrations_module, "__file__"):
                filenames = os.listdir(os.path.dirname(migrations_module.__file__))
            elif hasattr(migrations_module, "__path__"):
                if len(migrations_module.__path__) > 1:
                    return False
                filenames = os.listdir(list(migrations_module.__path__)[0])
            filenames = resource_listdir(migrations_module.__name__, '.')
            return not any(x.endswith(".py") for x in filenames if x != "__init__.py")

    def ask_not_null_addition(self, field_name, model_name):