Source code for plone.app.event.dx.behaviors

"""Behaviors to enable calendarish event extension to dexterity content types.
"""
from Products.CMFCore.utils import getToolByName
from Products.CMFPlone.utils import safe_unicode
from datetime import timedelta
from datetime import tzinfo
from plone.app.dexterity.behaviors.metadata import ICategorization
from plone.app.event import messageFactory as _
from plone.app.event.base import DT
from plone.app.event.base import default_end as default_end_dt
from plone.app.event.base import default_start as default_start_dt
from plone.app.event.base import default_timezone
from plone.app.event.base import dt_end_of_day
from plone.app.event.base import dt_start_of_day
from plone.app.event.base import first_weekday
from plone.app.event.base import wkday_to_mon1
from plone.app.event.dx.interfaces import IDXEvent
from plone.app.textfield import RichText
from plone.app.textfield.value import RichTextValue
from plone.autoform import directives as form
from plone.autoform.interfaces import IFormFieldProvider
from plone.event.interfaces import IEventAccessor
from plone.event.utils import tzdel, utc, dt_to_zone
from plone.formwidget.datetime.z3cform.widget import DatetimeFieldWidget
from plone.formwidget.recurrence.z3cform.field import RecurrenceField
from plone.formwidget.recurrence.z3cform.widget import RecurrenceFieldWidget
from plone.indexer import indexer
from plone.supermodel import model
from plone.uuid.interfaces import IUUID
from z3c.form.widget import ComputedWidgetAttribute
from zope import schema
from zope.component import adapts
from zope.component import provideAdapter
from zope.event import notify
from zope.interface import Invalid
from zope.interface import alsoProvides
from zope.interface import implements
from zope.interface import invariant
from zope.lifecycleevent import ObjectModifiedEvent

import pytz


# TODO: altern., for backwards compat., we could import from plone.z3cform
from z3c.form.browser.textlines import TextLinesFieldWidget


def first_weekday_sun0():
    return wkday_to_mon1(first_weekday())


[docs]class StartBeforeEnd(Invalid): __doc__ = _("error_invalid_date", default=u"Invalid start or end date")
class IEventBasic(model.Schema): """ Basic event schema. """ form.widget('start', DatetimeFieldWidget, first_day=first_weekday_sun0) form.widget('end', DatetimeFieldWidget, first_day=first_weekday_sun0) model.fieldset('dates', fields=['timezone']) start = schema.Datetime( title=_( u'label_event_start', default=u'Event Starts' ), description=_( u'help_event_start', default=u'Date and Time, when the event begins.' ), required=True ) end = schema.Datetime( title=_( u'label_event_end', default=u'Event Ends' ), description=_( u'help_event_end', default=u'Date and Time, when the event ends.' ), required=True ) whole_day = schema.Bool( title=_( u'label_event_whole_day', default=u'Whole Day' ), description=_( u'help_event_whole_day', default=u'Event lasts whole day.' ), required=False ) open_end = schema.Bool( title=_( u'label_event_open_end', default=u'Open End' ), description=_( u'help_event_open_end', default=u"This event is open ended." ), required=False ) # TODO: form.order_before(timezone="IPublication.effective") timezone = schema.Choice( title=_( u'label_event_timezone', default=u'Timezone' ), description=_( u'help_event_timezone', default=u'Select the Timezone, where this event happens.' ), required=True, vocabulary="plone.app.event.AvailableTimezones" ) @invariant def validate_start_end(data): if data.start > data.end: raise StartBeforeEnd( _("error_end_must_be_after_start_date", default=u"End date must be after start date.") ) def default_start(data): return default_start_dt(data.context) provideAdapter(ComputedWidgetAttribute( default_start, field=IEventBasic['start']), name='default') def default_end(data): return default_end_dt(data.context) provideAdapter(ComputedWidgetAttribute( default_end, field=IEventBasic['end']), name='default') def default_tz(data): return default_timezone() provideAdapter(ComputedWidgetAttribute( default_tz, field=IEventBasic['timezone']), name='default') class IEventRecurrence(model.Schema): """ Recurring Event Schema. """ # Please note: If you create a new behavior with superclasses IEventBasic # and IRecurrence, then you have to reconfigure the dotted path value of # the start_field parameter for the RecurrenceFieldWidget to the new # behavior name, like: IMyNewBehaviorName.start. form.widget( 'recurrence', RecurrenceFieldWidget, start_field='IEventBasic.start', first_day=first_weekday_sun0 ) recurrence = RecurrenceField( title=_( u'label_event_recurrence', default=u'Recurrence' ), description=_( u'help_event_recurrence', default=u'Define the event recurrence rule.' ), required=False ) class IEventLocation(model.Schema): """ Event Location Schema. """ location = schema.TextLine( title=_( u'label_event_location', default=u'Location' ), description=_( u'help_event_location', default=u'Location of the event.' ), required=False ) class IEventAttendees(model.Schema): """ Event Attendees Schema. """ attendees = schema.Tuple( title=_( u'label_event_attendees', default=u'Attendees' ), description=_( u'help_event_attendees', default=u'List of attendees.' ), value_type=schema.TextLine(), required=False, missing_value=(), ) form.widget(attendees=TextLinesFieldWidget) class IEventContact(model.Schema): """ Event Contact Schema. """ contact_name = schema.TextLine( title=_( u'label_event_contact_name', default=u'Contact Name' ), description=_( u'help_event_contact_name', default=u'Name of a person to contact about this event.' ), required=False ) contact_email = schema.TextLine( title=_( u'label_event_contact_email', default=u'Contact E-mail' ), description=_( u'help_event_contact_email', default=u'Email address to contact about this event.' ), required=False ) contact_phone = schema.TextLine( title=_( u'label_event_contact_phone', default=u'Contact Phone' ), description=_( u'help_event_contact_phone', default=u'Phone number to contact about this event.' ), required=False ) event_url = schema.TextLine( title=_( u'label_event_url', default=u'Event URL' ), description=_( u'help_event_url', default=u'Web address with more info about the event. ' u'Add http:// for external links.' ), required=False ) class IEventSummary(model.Schema): """Event summary (body text) schema.""" text = RichText( title=_( u'label_event_announcement', default=u'Event body text' ), description=_( u'help_event_announcement', default=u'' ), required=False, ) # Mark these interfaces as form field providers alsoProvides(IEventBasic, IFormFieldProvider) alsoProvides(IEventRecurrence, IFormFieldProvider) alsoProvides(IEventLocation, IFormFieldProvider) alsoProvides(IEventAttendees, IFormFieldProvider) alsoProvides(IEventContact, IFormFieldProvider) alsoProvides(IEventSummary, IFormFieldProvider)
[docs]class FakeZone(tzinfo): """Fake timezone to be applied to EventBasic start and end dates before data_postprocessing event handler sets the correct one. """ def utcoffset(self, dt): return timedelta(0) def tzname(self, dt): return "FAKEZONE" def dst(self, dt): return timedelta(0)
class EventBasic(object): def __init__(self, context): self.context = context @property def start(self): return self._prepare_dt_get(self.context.start) @start.setter def start(self, value): self.context.start = self._prepare_dt_set(value) @property def end(self): return self._prepare_dt_get(self.context.end) @end.setter def end(self, value): self.context.end = self._prepare_dt_set(value) @property def timezone(self): return getattr(self.context, 'timezone', None) @timezone.setter def timezone(self, value): if self.timezone: # The event is edited and not newly created, otherwise the timezone # info wouldn't exist. # In order to treat user datetime input as localized, so that the # values aren't converted to the target timezone, we have to set # start and end too. Then that a temporary fake zone is applied and # the data_postprocessing event subscriber can do it's job. self.start = self.start self.end = self.end self.context.timezone = value # TODO: whole day - and other attributes - might not be set at this time! # TODO: how to provide default values? @property def whole_day(self): return getattr(self.context, 'whole_day', False) @whole_day.setter def whole_day(self, value): self.context.whole_day = value @property def open_end(self): return getattr(self.context, 'open_end', False) @open_end.setter def open_end(self, value): self.context.open_end = value @property def duration(self): return self.context.end - self.context.start def _prepare_dt_get(self, dt): # always get the date in event's timezone return dt_to_zone(dt, self.context.timezone) def _prepare_dt_set(self, dt): # Dates are always set in UTC, saving the actual timezone in another # field. But since the timezone value isn't known at time of saving the # form, we have to save it with a fake zone first and replace it with # the target zone afterwards. So, it's not timezone naive and can be # compared to timezone aware Dates. # return with fake zone and without microseconds return dt.replace(microsecond=0, tzinfo=FakeZone()) class EventRecurrence(object): def __init__(self, context): self.context = context @property def recurrence(self): return self.context.recurrence @recurrence.setter def recurrence(self, value): self.context.recurrence = value ## Event handlers def data_postprocessing(obj, event): # newly created object, without start/end/timezone (e.g. invokeFactory() # called without data from add form), ignore event; it will be notified # again later: if getattr(obj, 'start', None) is None: return # We handle date inputs as floating dates without timezones and apply # timezones afterwards. def _fix_zone(dt, zone): if dt.tzinfo is not None and isinstance(dt.tzinfo, FakeZone): # Delete the tzinfo only, if it was set by IEventBasic setter. # Only in this case the start value on the object itself is what # was entered by the user. After running this event subscriber, # it's in UTC then. # If tzinfo it's not available at all, a naive datetime was set # probably by invokeFactory in tests. dt = tzdel(dt) if dt.tzinfo is None: # In case the tzinfo was deleted above or was not present, we can # localize the dt value to the target timezone. dt = tz.localize(dt) else: # In this case, no changes to start, end or the timezone were made. # Just return the object's datetime (which is in UTC) localized to # the target timezone. dt = dt.astimezone(tz) return dt.replace(microsecond=0) behavior = IEventBasic(obj) tz = pytz.timezone(behavior.timezone) # Fix zones start = _fix_zone(obj.start, tz) end = _fix_zone(obj.end, tz) # Adapt for whole day if behavior.whole_day: start = dt_start_of_day(start) if behavior.open_end: end = start # Open end events end on same day if behavior.open_end or behavior.whole_day: end = dt_end_of_day(end) # Save back obj.start = utc(start) obj.end = utc(end) # Reindex obj.reindexObject() ## Attribute indexer # Start indexer @indexer(IDXEvent) def start_indexer(obj): event = IEventBasic(obj) if event.start is None: return None return DT(event.start) # End indexer @indexer(IDXEvent) def end_indexer(obj): event = IEventBasic(obj) if event.end is None: return None return DT(event.end) # Body text indexing @indexer(IDXEvent) def searchable_text_indexer(obj): acc = IEventAccessor(obj) text = u'' text += u'%s\n' % acc.title text += u'%s\n' % acc.description behavior = IEventSummary(obj, None) if behavior is None or behavior.text is None: return text.encode('utf-8') output = behavior.text.output transforms = getToolByName(obj, 'portal_transforms') body_plain = transforms.convertTo( 'text/plain', output, mimetype='text/html', ).getData().strip() if isinstance(body_plain, str): body_plain = body_plain.decode('utf-8') text += body_plain return text.strip().encode('utf-8') # Object adapters
[docs]class EventAccessor(object): """ Generic event accessor adapter implementation for Dexterity content objects. """ implements(IEventAccessor) adapts(IDXEvent) event_type = 'plone.app.event.dx.event' # If you use a custom type, # override this. # Unified create method via Accessor @classmethod def create(cls, container, content_id, title, description=None, start=None, end=None, timezone=None, whole_day=None, open_end=None, **kwargs): container.invokeFactory(cls.event_type, id=content_id, title=title, description=description, start=start, end=end, whole_day=whole_day, open_end=open_end, timezone=timezone) content = container[content_id] acc = IEventAccessor(content) acc.edit(**kwargs) return acc def edit(self, **kwargs): for key, value in kwargs.items(): setattr(self, key, value) notify(ObjectModifiedEvent(self.context)) def __init__(self, context): object.__setattr__(self, 'context', context) bm = dict( start=IEventBasic, end=IEventBasic, whole_day=IEventBasic, open_end=IEventBasic, timezone=IEventBasic, recurrence=IEventRecurrence, location=IEventLocation, attendees=IEventAttendees, contact_name=IEventContact, contact_email=IEventContact, contact_phone=IEventContact, event_url=IEventContact, subjects=ICategorization, text=IEventSummary, ) object.__setattr__(self, '_behavior_map', bm) def __getattr__(self, name): bm = self._behavior_map if name in bm: # adapt object with behavior and return the attribute behavior = bm[name](self.context, None) if behavior: return safe_unicode(getattr(behavior, name, None)) return None def __setattr__(self, name, value): bm = self._behavior_map if name in bm: # set the attributes on behaviors behavior = bm[name](self.context, None) if behavior: setattr(behavior, name, safe_unicode(value)) def __delattr__(self, name): bm = self._behavior_map if name in bm: behavior = bm[name](self.context, None) if behavior: delattr(behavior, name) # ro properties @property def uid(self): return IUUID(self.context, None) @property def url(self): return safe_unicode(self.context.absolute_url()) @property def created(self): return utc(self.context.creation_date) @property def last_modified(self): return utc(self.context.modification_date) @property def duration(self): return self.end - self.start # rw properties not in behaviors (yet) # TODO revisit @property def title(self): return safe_unicode(getattr(self.context, 'title', None)) @title.setter def title(self, value): setattr(self.context, 'title', safe_unicode(value)) @property def description(self): return safe_unicode(getattr(self.context, 'description', None)) @description.setter def description(self, value): setattr(self.context, 'description', safe_unicode(value)) @property def text(self): behavior = IEventSummary(self.context) textvalue = getattr(behavior, 'text', None) if textvalue is None: return u'' return safe_unicode(textvalue.output) @text.setter def text(self, value): behavior = IEventSummary(self.context) behavior.text = RichTextValue(raw=safe_unicode(value))