#277 Make running of ViewSet checks flexible to allow dynamic document definition (#278)

Author: barseghyanartur

CHANGELOG.rst

@@ -15,14 +15,20 @@ are used for versioning (schema follows below):
   0.3.4 to 0.4).
 - All backwards incompatible changes are mentioned in this document.
 
+0.22.3
+------
+2022-01-24
+
+- Make running of ViewSet checks flexible (to allow dynamic document definition).
+
 0.22.2
-----
+------
 2021-08-29
 
 - Tested against Django 3.2.
 
 0.22.1
-----
+------
 2021-05-01
 
 - `DictionaryProxy` gets an optional `meta` argument, which will hold meta

setup.py

@@ -2,7 +2,7 @@
 
 from setuptools import find_packages, setup
 
-version = '0.22.2'
+version = '0.22.3'
 
 DOCS_TRANSFORMATIONS = (
     (

src/django_elasticsearch_dsl_drf/__init__.py

@@ -3,7 +3,7 @@
 """
 
 __title__ = 'django-elasticsearch-dsl-drf'
-__version__ = '0.22.2'
+__version__ = '0.22.3'
 __author__ = 'Artur Barseghyan <[email protected]>'
 __copyright__ = '2017-2021 Artur Barseghyan'
 __license__ = 'GPL 2.0/LGPL 2.1'

src/django_elasticsearch_dsl_drf/viewsets.py

@@ -154,7 +154,7 @@ class BaseDocumentViewSet(ReadOnlyModelViewSet):
     ignore = []
 
     def __init__(self, *args, **kwargs):
-        assert self.document is not None
+        self.run_checks()
 
         self.client = connections.get_connection(
             self.document._get_using()
@@ -168,6 +168,9 @@ def __init__(self, *args, **kwargs):
         )
         super(BaseDocumentViewSet, self).__init__(*args, **kwargs)
 
+    def run_checks(self):
+        assert self.document is not None
+
     def get_queryset(self):
         """Get queryset."""
         queryset = self.search.query()