from __future__ import absolute_import, unicode_literals from django.contrib.comments.models import Comment from django.contrib.contenttypes import generic from django.utils.translation import ugettext_lazy as _ from acls import ModelPermission from common import MayanAppConfig, menu_facet, menu_object, menu_sidebar from common.classes import ModelAttribute from common.utils import encapsulate from documents.models import Document from navigation import SourceColumn from .links import ( link_comment_add, link_comment_delete, link_comments_for_document ) from .permissions import ( permission_comment_create, permission_comment_delete, permission_comment_view ) class DocumentCommentsApp(MayanAppConfig): app_namespace = 'comments' app_url = 'comments' name = 'document_comments' verbose_name = _('Document comments') def ready(self): super(DocumentCommentsApp, self).ready() Document.add_to_class( 'comments', generic.GenericRelation( Comment, content_type_field='content_type', object_id_field='object_pk' ) ) ModelAttribute( Document, label=_('Comments'), name='comments', type_name='related' ) SourceColumn(source=Comment, label=_('Date'), attribute='submit_date') SourceColumn( source=Comment, label=_('User'), attribute=encapsulate(lambda x: x.user.get_full_name() if x.user.get_full_name() else x.user) ) SourceColumn(source=Comment, label=_('Comment'), attribute='comment') ModelPermission.register( model=Document, permissions=( permission_comment_create, permission_comment_delete, permission_comment_view ) ) menu_sidebar.bind_links( links=[link_comment_add], sources=[ 'comments:comments_for_document', 'comments:comment_add', 'comments:comment_delete', 'comments:comment_multiple_delete' ] ) menu_object.bind_links(links=[link_comment_delete], sources=[Comment]) menu_facet.bind_links( links=[link_comments_for_document], sources=[Document] )