"""AttributetypeAdmin for Gstudio"""
from datetime import datetime
from django.forms import Media
from django.contrib import admin
from django.contrib.auth.models import User
from django.utils.html import strip_tags
from django.utils.text import truncate_words
from django.conf.urls.defaults import url
from django.conf.urls.defaults import patterns
from django.conf import settings as project_settings
from django.utils.translation import ugettext_lazy as _
from django.core.urlresolvers import reverse, NoReverseMatch
from tagging.models import Tag
import reversion
from gstudio import settings
from gstudio.managers import HIDDEN
from gstudio.managers import PUBLISHED
from gstudio.ping import DirectoryPinger
from gstudio.admin.forms import AttributetypeAdminForm
from gstudio.settings import GSTUDIO_VERSIONING
from markitup.widgets import AdminMarkItUpWidget
if GSTUDIO_VERSIONING == True:
parent_class = reversion.VersionAdmin
else:
parent_class = admin.ModelAdmin
class AttributetypeAdmin(parent_class):
"""Admin for Attributetype model"""
form = AttributetypeAdminForm
date_hierarchy = 'creation_date'
fieldsets = ((_('Attribute Definiton'), {'fields': ('title','altnames','subjecttype','applicable_node\
types','dataType','verbose_name','null','blank','help_text','max_digits','decimal_places','auto_now','auto_now_add','upload_to','path','verify_exists','parent','slug','status') }),
(_('Content'), {'fields': ('content', 'image',),
'classes': ('collapse', 'collapse-closed')}),
(_('Dependency'), {'fields': ('prior_nodes', 'posterior_nodes',),
'classes': ('collapse', 'collapse-closed')}),
(_('Options'), {'fields': ('featured', 'excerpt', 'template',
'authors',
'creation_date',
'start_publication',
'end_publication'),
'classes': ('collapse', 'collapse-closed')}),
(_('Privacy'), {'fields': ('password', 'login_required',),
'classes': ('collapse', 'collapse-closed')}),
(_('Publication'), {'fields': ('tags',
'sites')}))
list_filter = ('parent','metatypes', 'authors', 'status', 'featured',
'login_required', 'comment_enabled', 'pingback_enabled',
'creation_date', 'start_publication',
'end_publication', 'sites')
list_display = ('get_title', 'get_authors', 'get_metatypes',
'get_tags', 'get_sites',
'get_comments_are_open', 'pingback_enabled',
'get_is_actual', 'get_is_visible', 'get_link',
'get_short_url', 'creation_date')
radio_fields = {'template': admin.VERTICAL}
filter_horizontal = ('metatypes', 'authors')
prepopulated_fields = {'slug': ('title', )}
search_fields = ('title', 'excerpt', 'content', 'tags')
actions = ['make_mine', 'make_published', 'make_hidden',
'close_comments', 'close_pingbacks',
'ping_directories', 'make_tweet', 'put_on_top']
actions_on_top = True
actions_on_bottom = True
def __init__(self, model, admin_site):
self.form.admin_site = admin_site
super(AttributetypeAdmin, self).__init__(model, admin_site)
def formfield_for_dbfield(self, db_field, **kwargs):
if db_field.name == 'content':
kwargs['widget'] = AdminMarkItUpWidget()
return super(AttributetypeAdmin, self).formfield_for_dbfield(db_field, **kwargs)
# Custom Display
def get_title(self, attributetype):
"""Return the title with word count and number of comments"""
title = _('%(title)s (%(word_count)i words)') % \
{'title': attributetype.title, 'word_count': attributetype.word_count}
comments = attributetype.comments.count()
if comments:
return _('%(title)s (%(comments)i comments)') % \
{'title': title, 'comments': comments}
return title
get_title.short_description = _('title')
def get_authors(self, attributetype):
"""Return the authors in HTML"""
try:
authors = ['%s' %
(reverse('gstudio_author_detail',
args=[author.username]),
author.username) for author in attributetype.authors.all()]
except NoReverseMatch:
authors = [author.username for author in attributetype.authors.all()]
return ', '.join(authors)
get_authors.allow_tags = True
get_authors.short_description = _('author(s)')
def get_metatypes(self, attributetype):
"""Return the metatypes linked in HTML"""
try:
metatypes = ['%s' %
(metatype.get_absolute_url(), metatype.title)
for metatype in attributetype.metatypes.all()]
except NoReverseMatch:
metatypes = [metatype.title for metatype in
attributetype.metatypes.all()]
return ', '.join(metatypes)
get_metatypes.allow_tags = True
get_metatypes.short_description = _('metatype(s)')
def get_tags(self, attributetype):
"""Return the tags linked in HTML"""
try:
return ', '.join(['%s' %
(reverse('gstudio_tag_detail',
args=[tag.name]), tag.name)
for tag in Tag.attributes.get_for_attribute(attributetype)])
except NoReverseMatch:
return attributetype.tags
get_tags.allow_tags = True
get_tags.short_description = _('tag(s)')
def get_sites(self, attributetype):
"""Return the sites linked in HTML"""
return ', '.join(
['%(name)s' %
site.__dict__ for site in attributetype.sites.all()])
get_sites.allow_tags = True
get_sites.short_description = _('site(s)')
def get_comments_are_open(self, attributetype):
"""Admin wrapper for attributetype.comments_are_open"""
return attributetype.comments_are_open
get_comments_are_open.boolean = True
get_comments_are_open.short_description = _('comment enabled')
def get_is_actual(self, attributetype):
"""Admin wrapper for attributetype.is_actual"""
return attributetype.is_actual
get_is_actual.boolean = True
get_is_actual.short_description = _('is actual')
def get_is_visible(self, attributetype):
"""Admin wrapper for attributetype.is_visible"""
return attributetype.is_visible
get_is_visible.boolean = True
get_is_visible.short_description = _('is visible')
def get_link(self, attributetype):
"""Return a formated link to the attributetype"""
return u'%s' % (
attributetype.get_absolute_url(), _('View'))
get_link.allow_tags = True
get_link.short_description = _('View on site')
def get_short_url(self, attributetype):
"""Return the short url in HTML"""
short_url = attributetype.short_url
if not short_url:
return _('Unavailable')
return '%(url)s' % \
{'url': short_url}
get_short_url.allow_tags = True
get_short_url.short_description = _('short url')
# Custom Methods
def save_model(self, request, attributetype, form, change):
"""Save the authors, update time, make an excerpt"""
if not form.cleaned_data.get('excerpt') and attributetype.status == PUBLISHED:
attributetype.excerpt = truncate_words(strip_tags(attributetype.content), 50)
if attributetype.pk and not request.user.has_perm('gstudio.can_change_author'):
form.cleaned_data['authors'] = attributetype.authors.all()
if not form.cleaned_data.get('authors'):
form.cleaned_data['authors'].append(request.user)
attributetype.last_update = datetime.now()
attributetype.save()
def queryset(self, request):
"""Make special filtering by user permissions"""
queryset = super(AttributetypeAdmin, self).queryset(request)
if request.user.has_perm('gstudio.can_view_all'):
return queryset
return request.user.attributetypes.all()
def formfield_for_manytomany(self, db_field, request, **kwargs):
"""Filters the disposable authors"""
if db_field.name == 'authors':
if request.user.has_perm('gstudio.can_change_author'):
kwargs['queryset'] = User.objects.filter(is_staff=True)
else:
kwargs['queryset'] = User.objects.filter(pk=request.user.pk)
return super(AttributetypeAdmin, self).formfield_for_manytomany(
db_field, request, **kwargs)
def get_actions(self, request):
"""Define user actions by permissions"""
actions = super(AttributetypeAdmin, self).get_actions(request)
if not request.user.has_perm('gstudio.can_change_author') \
or not request.user.has_perm('gstudio.can_view_all'):
del actions['make_mine']
if not settings.PING_DIRECTORIES:
del actions['ping_directories']
if not settings.USE_TWITTER:
del actions['make_tweet']
return actions
# Custom Actions
def make_mine(self, request, queryset):
"""Set the attributetypes to the user"""
for attributetype in queryset:
if request.user not in attributetype.authors.all():
attributetype.authors.add(request.user)
self.message_user(
request, _('The selected attributetypes now belong to you.'))
make_mine.short_description = _('Set the attributetypes to the user')
def make_published(self, request, queryset):
"""Set attributetypes selected as published"""
queryset.update(status=PUBLISHED)
self.ping_directories(request, queryset, messages=False)
self.message_user(
request, _('The selected attributetypes are now marked as published.'))
make_published.short_description = _('Set attributetypes selected as published')
def make_hidden(self, request, queryset):
"""Set attributetypes selected as hidden"""
queryset.update(status=HIDDEN)
self.message_user(
request, _('The selected attributetypes are now marked as hidden.'))
make_hidden.short_description = _('Set attributetypes selected as hidden')
def make_tweet(self, request, queryset):
"""Post an update on Twitter"""
import tweepy
auth = tweepy.OAuthHandler(settings.TWITTER_CONSUMER_KEY,
settings.TWITTER_CONSUMER_SECRET)
auth.set_access_token(settings.TWITTER_ACCESS_KEY,
settings.TWITTER_ACCESS_SECRET)
api = tweepy.API(auth)
for attributetype in queryset:
short_url = attributetype.short_url
message = '%s %s' % (attributetype.title[:139 - len(short_url)], short_url)
api.update_status(message)
self.message_user(
request, _('The selected attributetypes have been tweeted.'))
make_tweet.short_description = _('Tweet attributetypes selected')
def close_comments(self, request, queryset):
"""Close the comments for selected attributetypes"""
queryset.update(comment_enabled=False)
self.message_user(
request, _('Comments are now closed for selected attributetypes.'))
close_comments.short_description = _('Close the comments for '\
'selected attributetypes')
def close_pingbacks(self, request, queryset):
"""Close the pingbacks for selected attributetypes"""
queryset.update(pingback_enabled=False)
self.message_user(
request, _('Linkbacks are now closed for selected attributetypes.'))
close_pingbacks.short_description = _(
'Close the linkbacks for selected attributetypes')
def put_on_top(self, request, queryset):
"""Put the selected attributetypes on top at the current date"""
queryset.update(creation_date=datetime.now())
self.ping_directories(request, queryset, messages=False)
self.message_user(request, _(
'The selected attributetypes are now set at the current date.'))
put_on_top.short_description = _(
'Put the selected attributetypes on top at the current date')
def ping_directories(self, request, queryset, messages=True):
"""Ping Directories for selected attributetypes"""
for directory in settings.PING_DIRECTORIES:
pinger = DirectoryPinger(directory, queryset)
pinger.join()
if messages:
success = 0
for result in pinger.results:
if not result.get('flerror', True):
success += 1
else:
self.message_user(request,
'%s : %s' % (directory,
result['message']))
if success:
self.message_user(
request,
_('%(directory)s directory succesfully ' \
'pinged %(success)d attributetypes.') %
{'directory': directory, 'success': success})
ping_directories.short_description = _(
'Ping Directories for selected attributetypes')
def get_urls(self):
attributetype_admin_urls = super(AttributetypeAdmin, self).get_urls()
urls = patterns(
'django.views.generic.simple',
url(r'^autocomplete_tags/$', 'direct_to_template',
{'template': 'admin/gstudio/attributetype/autocomplete_tags.js',
'mimetype': 'application/javascript'},
name='gstudio_attributetype_autocomplete_tags'),
url(r'^wymeditor/$', 'direct_to_template',
{'template': 'admin/gstudio/attributetype/wymeditor.js',
'mimetype': 'application/javascript'},
name='gstudio_attributetype_wymeditor'),
url(r'^markitup/$', 'direct_to_template',
{'template': 'admin/gstudio/attributetype/markitup.js',
'mimetype': 'application/javascript'},
name='gstudio_attributetype_markitup'),)
return urls + attributetype_admin_urls
def _media(self):
STATIC_URL = '%sgstudio/' % project_settings.STATIC_URL
media = super(AttributetypeAdmin, self).media + Media(
css={'all': ('%scss/jquery.autocomplete.css' % STATIC_URL,)},
js=('%sjs/jquery.js' % STATIC_URL,
'%sjs/jquery.bgiframe.js' % STATIC_URL,
'%sjs/jquery.autocomplete.js' % STATIC_URL,
reverse('admin:gstudio_attributetype_autocomplete_tags'),))
if settings.WYSIWYG == 'wymeditor':
media += Media(
js=('%sjs/wymeditor/jquery.wymeditor.pack.js' % STATIC_URL,
'%sjs/wymeditor/plugins/hovertools/'
'jquery.wymeditor.hovertools.js' % STATIC_URL,
reverse('admin:gstudio_attributetype_wymeditor')))
elif settings.WYSIWYG == 'tinymce':
from tinymce.widgets import TinyMCE
media += TinyMCE().media + Media(
js=(reverse('tinymce-js', args=('admin/gstudio/attributetype',)),))
elif settings.WYSIWYG == 'markitup':
media += Media(
js=('%sjs/markitup/jquery.markitup.js' % STATIC_URL,
'%sjs/markitup/sets/%s/set.js' % (
STATIC_URL, settings.MARKUP_LANGUAGE),
reverse('admin:gstudio_attributetype_markitup')),
css={'all': (
'%sjs/markitup/skins/django/style.css' % STATIC_URL,
'%sjs/markitup/sets/%s/style.css' % (
STATIC_URL, settings.MARKUP_LANGUAGE))})
return media
media = property(_media)