from __future__ import absolute_import from django.conf import settings from django.contrib.comments.models import Comment from django.contrib.contenttypes import generic from django.utils.translation import ugettext_lazy as _ from acls.api import class_permissions from common.utils import encapsulate from documents.models import Document from navigation.api import register_links, register_model_list_columns from .links import (comment_delete, comment_add, comments_for_document) from .permissions import (PERMISSION_COMMENT_CREATE, PERMISSION_COMMENT_DELETE, PERMISSION_COMMENT_VIEW) if 'django.contrib.comments' not in settings.INSTALLED_APPS: raise Exception('This app depends on the django.contrib.comments app.') register_model_list_columns(Comment, [ { 'name': _(u'date'), 'attribute': 'submit_date' }, { 'name': _(u'user'), 'attribute': encapsulate(lambda x: x.user.get_full_name() if x.user.get_full_name() else x.user) }, { 'name': _(u'comment'), 'attribute': 'comment' } ]) register_links(['comments_for_document', 'comment_add', 'comment_delete', 'comment_multiple_delete'], [comment_add], menu_name='sidebar') register_links(Comment, [comment_delete]) register_links(Document, [comments_for_document], menu_name='form_header') Document.add_to_class( 'comments', generic.GenericRelation( Comment, content_type_field='content_type', object_id_field='object_pk' ) ) class_permissions(Document, [ PERMISSION_COMMENT_CREATE, PERMISSION_COMMENT_DELETE, PERMISSION_COMMENT_VIEW ])