Files
mayan-edms/mayan/apps/document_comments/views.py
Roberto Rosario a4ef6b3e8a Small code cleanups
Add keyword arguments.

Replace get_queryset with get_object_list.

Signed-off-by: Roberto Rosario <Roberto.Rosario@mayan-edms.com>
2019-02-05 05:49:47 -04:00

99 lines
3.1 KiB
Python

from __future__ import absolute_import, unicode_literals
from django.template import RequestContext
from django.urls import reverse
from django.utils.translation import ugettext_lazy as _
from mayan.apps.common.generics import (
SingleObjectCreateView, SingleObjectDeleteView, SingleObjectListView
)
from mayan.apps.common.mixins import ExternalObjectMixin
from mayan.apps.documents.models import Document
from .icons import icon_comments_for_document
from .links import link_comment_add
from .models import Comment
from .permissions import (
permission_comment_create, permission_comment_delete,
permission_comment_view
)
class DocumentCommentCreateView(ExternalObjectMixin, SingleObjectCreateView):
external_object_class = Document
external_object_permission = permission_comment_create
external_object_pk_url_kwarg = 'document_id'
fields = ('comment',)
model = Comment
def get_extra_context(self):
return {
'object': self.external_object,
'title': _('Add comment to document: %s') % self.external_object,
}
def get_instance_extra_data(self):
return {
'document': self.external_object, 'user': self.request.user,
}
def get_post_action_redirect(self):
return reverse(
viewname='comments:comments_for_document', kwargs={
'document_id': self.kwargs['document_id']
}
)
def get_save_extra_data(self):
return {
'_user': self.request.user,
}
class DocumentCommentDeleteView(SingleObjectDeleteView):
model = Comment
pk_url_kwarg = 'comment_id'
object_permission = permission_comment_delete
def get_delete_extra_data(self):
return {'_user': self.request.user}
def get_extra_context(self):
return {
'object': self.object.document,
'title': _('Delete comment: %s?') % self.object,
}
def get_post_action_redirect(self):
return reverse(
viewname='comments:comments_for_document', kwargs={
'document_id': self.object.document.pk
}
)
class DocumentCommentListView(ExternalObjectMixin, SingleObjectListView):
external_object_class = Document
external_object_permission = permission_comment_view
external_object_pk_url_kwarg = 'document_id'
def get_extra_context(self):
return {
'hide_link': True,
'hide_object': True,
'no_results_icon': icon_comments_for_document,
'no_results_external_link': link_comment_add.resolve(
RequestContext(self.request, {'object': self.external_object})
),
'no_results_text': _(
'Document comments are timestamped text entries from users. '
'They are great for collaboration.'
),
'no_results_title': _('There are no comments'),
'object': self.external_object,
'title': _('Comments for document: %s') % self.external_object,
}
def get_source_queryset(self):
return self.external_object.comments.all()