Source code for geonode.documents.api.serializers

#########################################################################
#
# Copyright (C) 2020 OSGeo
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
#
#########################################################################
import logging

from dynamic_rest.fields.fields import DynamicComputedField
from geonode.base.api.serializers import ResourceBaseSerializer
from geonode.documents.models import Document


[docs]
logger = logging.getLogger(__name__)




[docs]
class GeonodeFilePathField(DynamicComputedField):

[docs]
    def get_attribute(self, instance):
        return instance.files





[docs]
class DocumentFieldField(DynamicComputedField):

[docs]
    def get_attribute(self, instance):
        return instance.files





[docs]
class DocumentSerializer(ResourceBaseSerializer):
    def __init__(self, *args, **kwargs):
        # Instantiate the superclass normally
        super().__init__(*args, **kwargs)


[docs]
    file_path = GeonodeFilePathField(required=False)


[docs]
    doc_file = DocumentFieldField(required=False)



[docs]
    class Meta:

[docs]
        model = Document


[docs]
        name = "document"


[docs]
        view_name = "documents-list"


[docs]
        fields = (
            "pk",
            "uuid",
            "name",
            "href",
            "subtype",
            "extension",
            "mime_type",
            "executions",
            "file_path",
            "doc_file",
            "doc_url",
            "metadata",
        )




[docs]
    def to_representation(self, obj):
        _doc = super(DocumentSerializer, self).to_representation(obj)
        # better to hide internal server file path
        _doc.pop("file_path")
        _doc.pop("doc_file")
        return _doc