from django.db import models

from filingcabinet.models import (
    AbstractDocument,
    AbstractDocumentCollection,
)


class Document(AbstractDocument):
    original = models.ForeignKey(
        'foirequest.FoiAttachment', null=True, blank=True,
        on_delete=models.SET_NULL, related_name='original_document'
    )

    foirequest = models.ForeignKey(
        'foirequest.FoiRequest', null=True, blank=True,
        on_delete=models.SET_NULL
    )
    publicbody = models.ForeignKey(
        'publicbody.PublicBody', null=True, blank=True,
        on_delete=models.SET_NULL
    )

    def is_public(self):
        return self.public


class DocumentCollection(AbstractDocumentCollection):
    pass

from django.db import models

from filingcabinet.models import (
    AbstractDocument,
    AbstractDocumentCollection,
)


class Document(AbstractDocument):
    original = models.ForeignKey(
        'foirequest.FoiAttachment', null=True, blank=True,
        on_delete=models.SET_NULL, related_name='original_document'
    )

    foirequest = models.ForeignKey(
        'foirequest.FoiRequest', null=True, blank=True,
        on_delete=models.SET_NULL
    )
    publicbody = models.ForeignKey(
        'publicbody.PublicBody', null=True, blank=True,
        on_delete=models.SET_NULL
    )

    def is_public(self):
        return self.public

    def get_serializer_class(self, detail=False):
        from .api_views import DocumentSerializer, DocumentDetailSerializer

        if detail:
            return DocumentDetailSerializer
        return DocumentSerializer


class DocumentCollection(AbstractDocumentCollection):
    pass

Add get_serializer_class to document model
Add get_serializer_class to document model
  from django.db import models
  
  from filingcabinet.models import (
      AbstractDocument,
      AbstractDocumentCollection,
  )
  
  
  class Document(AbstractDocument):
      original = models.ForeignKey(
          'foirequest.FoiAttachment', null=True, blank=True,
          on_delete=models.SET_NULL, related_name='original_document'
      )
  
      foirequest = models.ForeignKey(
          'foirequest.FoiRequest', null=True, blank=True,
          on_delete=models.SET_NULL
      )
      publicbody = models.ForeignKey(
          'publicbody.PublicBody', null=True, blank=True,
          on_delete=models.SET_NULL
      )
  
      def is_public(self):
          return self.public
  
+     def get_serializer_class(self, detail=False):
+         from .api_views import DocumentSerializer, DocumentDetailSerializer
+ 
+         if detail:
+             return DocumentDetailSerializer
+         return DocumentSerializer
+ 
  
  class DocumentCollection(AbstractDocumentCollection):
      pass
  
Add get_serializer_class to document model
## Code Before:
from django.db import models

from filingcabinet.models import (
    AbstractDocument,
    AbstractDocumentCollection,
)


class Document(AbstractDocument):
    original = models.ForeignKey(
        'foirequest.FoiAttachment', null=True, blank=True,
        on_delete=models.SET_NULL, related_name='original_document'
    )

    foirequest = models.ForeignKey(
        'foirequest.FoiRequest', null=True, blank=True,
        on_delete=models.SET_NULL
    )
    publicbody = models.ForeignKey(
        'publicbody.PublicBody', null=True, blank=True,
        on_delete=models.SET_NULL
    )

    def is_public(self):
        return self.public


class DocumentCollection(AbstractDocumentCollection):
    pass

## Instruction:
Add get_serializer_class to document model
## Code After:
from django.db import models

from filingcabinet.models import (
    AbstractDocument,
    AbstractDocumentCollection,
)


class Document(AbstractDocument):
    original = models.ForeignKey(
        'foirequest.FoiAttachment', null=True, blank=True,
        on_delete=models.SET_NULL, related_name='original_document'
    )

    foirequest = models.ForeignKey(
        'foirequest.FoiRequest', null=True, blank=True,
        on_delete=models.SET_NULL
    )
    publicbody = models.ForeignKey(
        'publicbody.PublicBody', null=True, blank=True,
        on_delete=models.SET_NULL
    )

    def is_public(self):
        return self.public

    def get_serializer_class(self, detail=False):
        from .api_views import DocumentSerializer, DocumentDetailSerializer

        if detail:
            return DocumentDetailSerializer
        return DocumentSerializer


class DocumentCollection(AbstractDocumentCollection):
    pass
