merge master

src/wagtail_factories/factories.py

@@ -2,10 +2,28 @@
 
 import factory
 from django.utils.text import slugify
-
+from factory import errors, utils
+from factory.declarations import ParameteredAttribute
+from wagtail import VERSION as WAGTAIL_VERSION
 from wagtail.models import Collection, Page, Site
 from wagtail.images import get_image_model
 
+try:
+    from factory.django import DjangoModelFactory
+except ImportError:
+    from factory import DjangoModelFactory
+
+if WAGTAIL_VERSION >= (2, 8):
+    from wagtail.documents import get_document_model
+else:
+    from wagtail.documents.models import get_document_model
+
+__all__ = ["CollectionFactory", "ImageFactory", "PageFactory", "SiteFactory", "DocumentFactory"]
+logger = logging.getLogger(__file__)
+
+
+class ParentNodeFactory(ParameteredAttribute):
+
     EXTEND_CONTAINERS = True
     FORCE_SEQUENCE = False
     UNROLL_CONTEXT_BEFORE_EVALUATION = False