mirror of
https://github.com/paperless-ngx/paperless-ngx.git
synced 2025-04-19 10:19:27 -05:00
131 lines
4.0 KiB
Python
131 lines
4.0 KiB
Python
from django.http import HttpResponse, HttpResponseBadRequest
|
|
from django.views.generic import DetailView, FormView, TemplateView
|
|
from django_filters.rest_framework import DjangoFilterBackend
|
|
from paperless.db import GnuPG
|
|
from paperless.mixins import SessionOrBasicAuthMixin
|
|
from paperless.views import StandardPagination
|
|
from rest_framework.filters import OrderingFilter, SearchFilter
|
|
from rest_framework.mixins import (
|
|
DestroyModelMixin,
|
|
ListModelMixin,
|
|
RetrieveModelMixin,
|
|
UpdateModelMixin
|
|
)
|
|
from rest_framework.permissions import IsAuthenticated
|
|
from rest_framework.viewsets import (
|
|
GenericViewSet,
|
|
ModelViewSet,
|
|
ReadOnlyModelViewSet
|
|
)
|
|
|
|
from .filters import CorrespondentFilterSet, DocumentFilterSet, TagFilterSet
|
|
from .forms import UploadForm
|
|
from .models import Correspondent, Document, Log, Tag
|
|
from .serialisers import (
|
|
CorrespondentSerializer,
|
|
DocumentSerializer,
|
|
LogSerializer,
|
|
TagSerializer
|
|
)
|
|
|
|
|
|
class IndexView(TemplateView):
|
|
template_name = "documents/index.html"
|
|
|
|
|
|
class FetchView(SessionOrBasicAuthMixin, DetailView):
|
|
|
|
model = Document
|
|
|
|
def render_to_response(self, context, **response_kwargs):
|
|
"""
|
|
Override the default to return the unencrypted image/PDF as raw data.
|
|
"""
|
|
|
|
content_types = {
|
|
Document.TYPE_PDF: "application/pdf",
|
|
Document.TYPE_PNG: "image/png",
|
|
Document.TYPE_JPG: "image/jpeg",
|
|
Document.TYPE_GIF: "image/gif",
|
|
Document.TYPE_TIF: "image/tiff",
|
|
}
|
|
|
|
if self.kwargs["kind"] == "thumb":
|
|
return HttpResponse(
|
|
GnuPG.decrypted(self.object.thumbnail_file),
|
|
content_type=content_types[Document.TYPE_PNG]
|
|
)
|
|
|
|
response = HttpResponse(
|
|
GnuPG.decrypted(self.object.source_file),
|
|
content_type=content_types[self.object.file_type]
|
|
)
|
|
response["Content-Disposition"] = 'attachment; filename="{}"'.format(
|
|
self.object.file_name)
|
|
|
|
return response
|
|
|
|
|
|
class PushView(SessionOrBasicAuthMixin, FormView):
|
|
"""
|
|
A crude REST-ish API for creating documents.
|
|
"""
|
|
|
|
form_class = UploadForm
|
|
|
|
def form_valid(self, form):
|
|
form.save()
|
|
return HttpResponse("1", status=202)
|
|
|
|
def form_invalid(self, form):
|
|
return HttpResponseBadRequest(str(form.errors))
|
|
|
|
|
|
class CorrespondentViewSet(ModelViewSet):
|
|
model = Correspondent
|
|
queryset = Correspondent.objects.all()
|
|
serializer_class = CorrespondentSerializer
|
|
pagination_class = StandardPagination
|
|
permission_classes = (IsAuthenticated,)
|
|
filter_backends = (DjangoFilterBackend, OrderingFilter)
|
|
filter_class = CorrespondentFilterSet
|
|
ordering_fields = ("name", "slug")
|
|
|
|
|
|
class TagViewSet(ModelViewSet):
|
|
model = Tag
|
|
queryset = Tag.objects.all()
|
|
serializer_class = TagSerializer
|
|
pagination_class = StandardPagination
|
|
permission_classes = (IsAuthenticated,)
|
|
filter_backends = (DjangoFilterBackend, OrderingFilter)
|
|
filter_class = TagFilterSet
|
|
ordering_fields = ("name", "slug")
|
|
|
|
|
|
class DocumentViewSet(RetrieveModelMixin,
|
|
UpdateModelMixin,
|
|
DestroyModelMixin,
|
|
ListModelMixin,
|
|
GenericViewSet):
|
|
model = Document
|
|
queryset = Document.objects.all()
|
|
serializer_class = DocumentSerializer
|
|
pagination_class = StandardPagination
|
|
permission_classes = (IsAuthenticated,)
|
|
filter_backends = (DjangoFilterBackend, SearchFilter, OrderingFilter)
|
|
filter_class = DocumentFilterSet
|
|
search_fields = ("title", "correspondent__name", "content")
|
|
ordering_fields = (
|
|
"id", "title", "correspondent__name", "created", "modified")
|
|
|
|
|
|
class LogViewSet(ReadOnlyModelViewSet):
|
|
model = Log
|
|
queryset = Log.objects.all().by_group()
|
|
serializer_class = LogSerializer
|
|
pagination_class = StandardPagination
|
|
permission_classes = (IsAuthenticated,)
|
|
filter_backends = (DjangoFilterBackend, OrderingFilter)
|
|
ordering_fields = ("time",)
|