You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
248 lines
8.5 KiB
248 lines
8.5 KiB
# -*- coding: utf-8 -*-
|
|
|
|
from __future__ import unicode_literals
|
|
import base64
|
|
import urlparse
|
|
import aldryn_common
|
|
import warnings
|
|
import vobject
|
|
|
|
from django.core.urlresolvers import reverse
|
|
from django.conf import settings
|
|
from django.db import models
|
|
from django.utils.encoding import python_2_unicode_compatible
|
|
from django.utils.translation import ugettext_lazy as _
|
|
try:
|
|
from django.utils.text import slugify
|
|
except ImportError:
|
|
# In Djano <= 1.4 slugify is here
|
|
from django.template.defaultfilters import slugify
|
|
|
|
from cms.models.pluginmodel import CMSPlugin
|
|
from djangocms_text_ckeditor.fields import HTMLField
|
|
from filer.fields.image import FilerImageField
|
|
from parler.models import TranslatableModel, TranslatedFields
|
|
import aldryn_common.admin_fields.sortedm2m
|
|
|
|
from .utils import get_additional_styles
|
|
|
|
|
|
@python_2_unicode_compatible
|
|
class Group(TranslatableModel):
|
|
translations = TranslatedFields(
|
|
name=models.CharField(_('name'), max_length=255),
|
|
description=HTMLField(_('description'), blank=True),
|
|
address=models.TextField(
|
|
verbose_name=_('address'), blank=True),
|
|
city=models.CharField(
|
|
verbose_name=_('city'), max_length=255, blank=True)
|
|
)
|
|
postal_code = models.CharField(
|
|
verbose_name=_('postal code'), max_length=20, blank=True)
|
|
phone = models.CharField(
|
|
verbose_name=_('phone'), null=True, blank=True, max_length=100)
|
|
fax = models.CharField(
|
|
verbose_name=_('fax'), null=True, blank=True, max_length=100)
|
|
email = models.EmailField(
|
|
verbose_name=_('email'), blank=True, default='')
|
|
website = models.URLField(
|
|
verbose_name=_('website'), null=True, blank=True)
|
|
|
|
@property
|
|
def company_name(self):
|
|
warnings.warn(
|
|
'"Group.company_name" has been refactored to "Group.name"',
|
|
DeprecationWarning
|
|
)
|
|
return self.safe_translation_getter('name')
|
|
|
|
@property
|
|
def company_description(self):
|
|
warnings.warn(
|
|
'"Group.company_description" has been refactored to '
|
|
'"Group.description"',
|
|
DeprecationWarning
|
|
)
|
|
return self.safe_translation_getter('description')
|
|
|
|
def __str__(self):
|
|
return self.safe_translation_getter('name', str(self.pk))
|
|
|
|
class Meta:
|
|
verbose_name = _('Group')
|
|
verbose_name_plural = _('Groups')
|
|
|
|
|
|
@python_2_unicode_compatible
|
|
class Person(TranslatableModel):
|
|
translations = TranslatedFields(
|
|
function=models.CharField(
|
|
_('function'), max_length=255, blank=True, default=''),
|
|
description=HTMLField(
|
|
_('Description'), blank=True, default=''),
|
|
name=models.CharField(
|
|
verbose_name=_('name'), max_length=255)
|
|
)
|
|
phone = models.CharField(
|
|
verbose_name=_('phone'), null=True, blank=True, max_length=100)
|
|
mobile = models.CharField(
|
|
verbose_name=_('mobile'), null=True, blank=True, max_length=100)
|
|
fax = models.CharField(
|
|
verbose_name=_('fax'), null=True, blank=True, max_length=100)
|
|
email = models.EmailField(
|
|
verbose_name=_("email"), blank=True, default='')
|
|
website = models.URLField(
|
|
verbose_name=_('website'), null=True, blank=True)
|
|
group = models.ForeignKey(
|
|
Group, verbose_name=_('group'), blank=True, null=True)
|
|
visual = FilerImageField(
|
|
null=True, blank=True, default=None, on_delete=models.SET_NULL, related_name='visual')
|
|
visual_portrait = FilerImageField(
|
|
null=True, blank=True, default=None, on_delete=models.SET_NULL, related_name='visual_portrait')
|
|
slug = models.CharField(
|
|
verbose_name=_('unique slug'), max_length=255, blank=True, null=True,
|
|
unique=True)
|
|
vcard_enabled = models.BooleanField(
|
|
verbose_name=_('enable vCard download'), default=True)
|
|
user = models.ForeignKey(
|
|
getattr(settings, 'AUTH_USER_MODEL', 'auth.User'),
|
|
null=True, blank=True, unique=True)
|
|
|
|
class Meta:
|
|
verbose_name = _('Person')
|
|
verbose_name_plural = _('People')
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
@property
|
|
def comment(self):
|
|
return self.safe_translation_getter('description', '')
|
|
|
|
def get_absolute_url(self):
|
|
if self.slug:
|
|
kwargs = {'slug': self.slug}
|
|
else:
|
|
kwargs = {'pk': self.pk}
|
|
return reverse('aldryn_people:person-detail', kwargs=kwargs)
|
|
|
|
def get_vcard(self, request=None):
|
|
if self.group:
|
|
group_name = self.group.safe_translation_getter('name')
|
|
else:
|
|
group_name = ''
|
|
function = self.safe_translation_getter('function')
|
|
|
|
vcard = vobject.vCard()
|
|
vcard.add('n').value = vobject.vcard.Name(given=self.name)
|
|
vcard.add('fn').value = self.name
|
|
|
|
if self.visual:
|
|
try:
|
|
with open(self.visual.path, 'rb') as f:
|
|
photo = vcard.add('photo')
|
|
photo.type_param = self.visual.extension.upper()
|
|
photo.value = base64.b64encode(f.read())
|
|
photo.encoded = True
|
|
photo.encoding_param = 'B'
|
|
except IOError:
|
|
if request:
|
|
photo = vcard.add('photo')
|
|
photo.type_param = self.visual.extension.upper()
|
|
photo.value = urlparse.urljoin(
|
|
request.build_absolute_uri(), self.visual.url)
|
|
|
|
if self.email:
|
|
vcard.add('email').value = self.email
|
|
if function:
|
|
vcard.add('title').value = function
|
|
if self.phone:
|
|
tel = vcard.add('tel')
|
|
tel.value = unicode(self.phone)
|
|
tel.type_param = 'WORK'
|
|
if self.mobile:
|
|
tel = vcard.add('tel')
|
|
tel.value = unicode(self.mobile)
|
|
tel.type_param = 'CELL'
|
|
if self.fax:
|
|
fax = vcard.add('tel')
|
|
fax.value = unicode(self.fax)
|
|
fax.type_param = 'FAX'
|
|
if self.website:
|
|
website = vcard.add('url')
|
|
website.value = unicode(self.website)
|
|
|
|
if self.group:
|
|
if group_name:
|
|
vcard.add('org').value = [group_name]
|
|
if self.group.address or self.group.city or self.group.postal_code:
|
|
vcard.add('adr')
|
|
vcard.adr.type_param = 'WORK'
|
|
vcard.adr.value = vobject.vcard.Address()
|
|
if self.group.address:
|
|
vcard.adr.value.street = self.group.address
|
|
if self.group.city:
|
|
vcard.adr.value.city = self.group.city
|
|
if self.group.postal_code:
|
|
vcard.adr.value.code = self.group.postal_code
|
|
if self.group.phone:
|
|
tel = vcard.add('tel')
|
|
tel.value = unicode(self.group.phone)
|
|
tel.type_param = 'WORK'
|
|
if self.group.fax:
|
|
fax = vcard.add('tel')
|
|
fax.value = unicode(self.group.fax)
|
|
fax.type_param = 'FAX'
|
|
if self.group.website:
|
|
website = vcard.add('url')
|
|
website.value = unicode(self.group.website)
|
|
|
|
return vcard.serialize()
|
|
|
|
def save(self, **kwargs):
|
|
name = self.safe_translation_getter('name')
|
|
if not self.slug:
|
|
self.slug = slugify(name)
|
|
return super(Person, self).save(**kwargs)
|
|
|
|
|
|
@python_2_unicode_compatible
|
|
class BasePeoplePlugin(CMSPlugin):
|
|
|
|
STYLE_CHOICES = [
|
|
('standard', _('Standard')),
|
|
('feature', _('Feature'))
|
|
] + get_additional_styles()
|
|
|
|
style = models.CharField(
|
|
_('Style'), choices=STYLE_CHOICES,
|
|
default=STYLE_CHOICES[0][0], max_length=50)
|
|
people = aldryn_common.admin_fields.sortedm2m.SortedM2MModelField(
|
|
Person, blank=True, null=True)
|
|
group_by_group = models.BooleanField(
|
|
verbose_name=_('group by group'),
|
|
default=True,
|
|
help_text=_('when checked, people are grouped by their group')
|
|
)
|
|
show_links = models.BooleanField(
|
|
verbose_name=_('Show links to Detail Page'), default=False)
|
|
show_vcard = models.BooleanField(
|
|
verbose_name=_('Show links to download vCard'), default=False)
|
|
|
|
class Meta:
|
|
abstract = True
|
|
|
|
def __str__(self):
|
|
return unicode(self.pk)
|
|
|
|
def copy_relations(self, oldinstance):
|
|
self.people = oldinstance.people.all()
|
|
|
|
def get_selected_people(self):
|
|
return self.people.select_related('group', 'visual')
|
|
|
|
|
|
class PeoplePlugin(BasePeoplePlugin):
|
|
|
|
class Meta:
|
|
abstract = False
|
|
|