--- /dev/null
+# -*- coding: utf-8 -*-
+##############################################################################
+#
+# OpenERP, Open Source Business Applications
+# Copyright (c) 2011-2014 OpenERP S.A. <http://openerp.com>
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Affero General Public License as
+# published by the Free Software Foundation, either version 3 of the
+# License, or (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+# GNU Affero General Public License for more details.
+#
+# You should have received a copy of the GNU Affero General Public License
+# along with this program. If not, see <http://www.gnu.org/licenses/>.
+#
+##############################################################################
+
+import pytz
+import re
+import time
+import openerp
+import openerp.service.report
+import uuid
+from werkzeug.exceptions import BadRequest
+from datetime import datetime, timedelta
+from dateutil import parser
+from dateutil import rrule
+from dateutil.relativedelta import relativedelta
+from openerp import tools, SUPERUSER_ID
+from openerp.osv import fields, osv
+from openerp.tools import DEFAULT_SERVER_DATE_FORMAT, DEFAULT_SERVER_DATETIME_FORMAT
+from openerp.tools.translate import _
+from openerp.http import request
+from operator import itemgetter
+
+import logging
+_logger = logging.getLogger(__name__)
+
+
+def calendar_id2real_id(calendar_id=None, with_date=False):
+ """
+ Convert a "virtual/recurring event id" (type string) into a real event id (type int).
+ E.g. virtual/recurring event id is 4-20091201100000, so it will return 4.
+ @param calendar_id: id of calendar
+ @param with_date: if a value is passed to this param it will return dates based on value of withdate + calendar_id
+ @return: real event id
+ """
+ if calendar_id and isinstance(calendar_id, (str, unicode)):
+ res = calendar_id.split('-')
+ if len(res) >= 2:
+ real_id = res[0]
+ if with_date:
+ real_date = time.strftime("%Y-%m-%d %H:%M:%S", time.strptime(res[1], "%Y%m%d%H%M%S"))
+ start = datetime.strptime(real_date, "%Y-%m-%d %H:%M:%S")
+ end = start + timedelta(hours=with_date)
+ return (int(real_id), real_date, end.strftime("%Y-%m-%d %H:%M:%S"))
+ return int(real_id)
+ return calendar_id and int(calendar_id) or calendar_id
+
+
+def get_real_ids(ids):
+ if isinstance(ids, (str, int, long)):
+ return calendar_id2real_id(ids)
+
+ if isinstance(ids, (list, tuple)):
+ return [calendar_id2real_id(id) for id in ids]
+
+
+class calendar_attendee(osv.Model):
+ """
+ Calendar Attendee Information
+ """
+ _name = 'calendar.attendee'
+ _rec_name = 'cn'
+ _description = 'Attendee information'
+
+ def _compute_data(self, cr, uid, ids, name, arg, context=None):
+ """
+ Compute data on function fields for attendee values.
+ @param ids: list of calendar attendee's IDs
+ @param name: name of field
+ @return: dictionary of form {id: {'field Name': value'}}
+ """
+ name = name[0]
+ result = {}
+ for attdata in self.browse(cr, uid, ids, context=context):
+ id = attdata.id
+ result[id] = {}
+ if name == 'cn':
+ if attdata.partner_id:
+ result[id][name] = attdata.partner_id.name or False
+ else:
+ result[id][name] = attdata.email or ''
+ if name == 'event_date':
+ result[id][name] = attdata.event_id.date
+ if name == 'event_end_date':
+ result[id][name] = attdata.event_id.date_deadline
+ return result
+
+ STATE_SELECTION = [
+ ('needsAction', 'Needs Action'),
+ ('tentative', 'Uncertain'),
+ ('declined', 'Declined'),
+ ('accepted', 'Accepted'),
+ ]
+
+ _columns = {
+ 'state': fields.selection(STATE_SELECTION, 'Status', readonly=True, help="Status of the attendee's participation"),
+ 'cn': fields.function(_compute_data, string='Common name', type="char", multi='cn', store=True),
+ 'partner_id': fields.many2one('res.partner', 'Contact', readonly="True"),
+ 'email': fields.char('Email', help="Email of Invited Person"),
+ 'event_date': fields.function(_compute_data, string='Event Date', type="datetime", multi='event_date'),
+ 'event_end_date': fields.function(_compute_data, string='Event End Date', type="datetime", multi='event_end_date'),
+ 'availability': fields.selection([('free', 'Free'), ('busy', 'Busy')], 'Free/Busy', readonly="True"),
+ 'access_token': fields.char('Invitation Token'),
+ 'event_id': fields.many2one('calendar.event', 'Meeting linked'),
+ }
+ _defaults = {
+ 'state': 'needsAction',
+ }
+
+ def copy(self, cr, uid, id, default=None, context=None):
+ raise osv.except_osv(_('Warning!'), _('You cannot duplicate a calendar attendee.'))
+
+ def onchange_partner_id(self, cr, uid, ids, partner_id, context=None):
+ """
+ Make entry on email and availability on change of partner_id field.
+ @param partner_id: changed value of partner id
+ """
+ if not partner_id:
+ return {'value': {'email': ''}}
+ partner = self.pool['res.partner'].browse(cr, uid, partner_id, context=context)
+ return {'value': {'email': partner.email}}
+
+ def get_ics_file(self, cr, uid, event_obj, context=None):
+ """
+ Returns iCalendar file for the event invitation.
+ @param event_obj: event object (browse record)
+ @return: .ics file content
+ """
+ res = None
+
+ def ics_datetime(idate, short=False):
+ if idate:
+ return datetime.strptime(idate.split('.')[0], '%Y-%m-%d %H:%M:%S').replace(tzinfo=pytz.timezone('UTC'))
+ return False
+
+ try:
+ # FIXME: why isn't this in CalDAV?
+ import vobject
+ except ImportError:
+ return res
+
+ cal = vobject.iCalendar()
+ event = cal.add('vevent')
+ if not event_obj.date_deadline or not event_obj.date:
+ raise osv.except_osv(_('Warning!'), _("First you have to specify the date of the invitation."))
+ event.add('created').value = ics_datetime(time.strftime('%Y-%m-%d %H:%M:%S'))
+ event.add('dtstart').value = ics_datetime(event_obj.date)
+ event.add('dtend').value = ics_datetime(event_obj.date_deadline)
+ event.add('summary').value = event_obj.name
+ if event_obj.description:
+ event.add('description').value = event_obj.description
+ if event_obj.location:
+ event.add('location').value = event_obj.location
+ if event_obj.rrule:
+ event.add('rrule').value = event_obj.rrule
+
+ if event_obj.alarm_ids:
+ for alarm in event_obj.alarm_ids:
+ valarm = event.add('valarm')
+ interval = alarm.interval
+ duration = alarm.duration
+ trigger = valarm.add('TRIGGER')
+ trigger.params['related'] = ["START"]
+ if interval == 'days':
+ delta = timedelta(days=duration)
+ elif interval == 'hours':
+ delta = timedelta(hours=duration)
+ elif interval == 'minutes':
+ delta = timedelta(minutes=duration)
+ trigger.value = delta
+ valarm.add('DESCRIPTION').value = alarm.name or 'OpenERP'
+ for attendee in event_obj.attendee_ids:
+ attendee_add = event.add('attendee')
+ attendee_add.value = 'MAILTO:' + (attendee.email or '')
+ res = cal.serialize()
+ return res
+
+ def _send_mail_to_attendees(self, cr, uid, ids, email_from=tools.config.get('email_from', False), template_xmlid='calendar_template_meeting_invitation', context=None):
+ """
+ Send mail for event invitation to event attendees.
+ @param email_from: email address for user sending the mail
+ """
+ res = False
+ mail_ids = []
+ data_pool = self.pool['ir.model.data']
+ mailmess_pool = self.pool['mail.message']
+ mail_pool = self.pool['mail.mail']
+ template_pool = self.pool['email.template']
+ local_context = context.copy()
+ color = {
+ 'needsAction': 'grey',
+ 'accepted': 'green',
+ 'tentative': '#FFFF00',
+ 'declined': 'red'
+ }
+
+ if not isinstance(ids, (tuple, list)):
+ ids = [ids]
+
+ dummy, template_id = data_pool.get_object_reference(cr, uid, 'calendar', template_xmlid)
+ dummy, act_id = data_pool.get_object_reference(cr, uid, 'calendar', "view_calendar_event_calendar")
+ local_context.update({
+ 'color': color,
+ 'action_id': self.pool['ir.actions.act_window'].search(cr, uid, [('view_id', '=', act_id)], context=context)[0],
+ 'dbname': cr.dbname,
+ 'base_url': self.pool['ir.config_parameter'].get_param(cr, uid, 'web.base.url', default='http://localhost:8069', context=context)
+ })
+
+ for attendee in self.browse(cr, uid, ids, context=context):
+ if attendee.email and email_from and attendee.email != email_from:
+ ics_file = self.get_ics_file(cr, uid, attendee.event_id, context=context)
+ mail_id = template_pool.send_mail(cr, uid, template_id, attendee.id, context=local_context)
+
+ vals = {}
+ if ics_file:
+ vals['attachment_ids'] = [(0, 0, {'name': 'invitation.ics',
+ 'datas_fname': 'invitation.ics',
+ 'datas': str(ics_file).encode('base64')})]
+ vals['model'] = None # We don't want to have the mail in the tchatter while in queue!
+ the_mailmess = mail_pool.browse(cr, uid, mail_id, context=context).mail_message_id
+ mailmess_pool.write(cr, uid, [the_mailmess.id], vals, context=context)
+ mail_ids.append(mail_id)
+
+ if mail_ids:
+ res = mail_pool.send(cr, uid, mail_ids, context=context)
+
+ return res
+
+ def onchange_user_id(self, cr, uid, ids, user_id, *args, **argv):
+ """
+ Make entry on email and availability on change of user_id field.
+ @param ids: list of attendee's IDs
+ @param user_id: changed value of User id
+ @return: dictionary of values which put value in email and availability fields
+ """
+ if not user_id:
+ return {'value': {'email': ''}}
+
+ user = self.pool['res.users'].browse(cr, uid, user_id, *args)
+ return {'value': {'email': user.email, 'availability': user.availability}}
+
+ def do_tentative(self, cr, uid, ids, context=None, *args):
+ """
+ Makes event invitation as Tentative.
+ @param ids: list of attendee's IDs
+ """
+ return self.write(cr, uid, ids, {'state': 'tentative'}, context)
+
+ def do_accept(self, cr, uid, ids, context=None, *args):
+ """
+ Marks event invitation as Accepted.
+ @param ids: list of attendee's IDs
+ """
+ if context is None:
+ context = {}
+ meeting_obj = self.pool['calendar.event']
+ res = self.write(cr, uid, ids, {'state': 'accepted'}, context)
+ for attendee in self.browse(cr, uid, ids, context=context):
+ meeting_obj.message_post(cr, uid, attendee.event_id.id, body=_(("%s has accepted invitation") % (attendee.cn)), subtype="calendar.subtype_invitation", context=context)
+
+ return res
+
+ def do_decline(self, cr, uid, ids, context=None, *args):
+ """
+ Marks event invitation as Declined.
+ @param ids: list of calendar attendee's IDs
+ """
+ if context is None:
+ context = {}
+ meeting_obj = self.pool['calendar.event']
+ res = self.write(cr, uid, ids, {'state': 'declined'}, context)
+ for attendee in self.browse(cr, uid, ids, context=context):
+ meeting_obj.message_post(cr, uid, attendee.event_id.id, body=_(("%s has declined invitation") % (attendee.cn)), subtype="calendar.subtype_invitation", context=context)
+ return res
+
+ def create(self, cr, uid, vals, context=None):
+ if context is None:
+ context = {}
+ if not vals.get("email") and vals.get("cn"):
+ cnval = vals.get("cn").split(':')
+ email = filter(lambda x: x.__contains__('@'), cnval)
+ vals['email'] = email and email[0] or ''
+ vals['cn'] = vals.get("cn")
+ res = super(calendar_attendee, self).create(cr, uid, vals, context=context)
+ return res
+
+
+class res_partner(osv.Model):
+ _inherit = 'res.partner'
+ _columns = {
+ 'calendar_last_notif_ack': fields.datetime('Last notification marked as read from base Calendar'),
+ }
+
+ def get_attendee_detail(self, cr, uid, ids, meeting_id, context=None):
+ datas = []
+ meeting = False
+ if meeting_id:
+ meeting = self.pool['calendar.event'].browse(cr, uid, get_real_ids(meeting_id), context=context)
+ for partner in self.browse(cr, uid, ids, context=context):
+ data = self.name_get(cr, uid, [partner.id], context)[0]
+ if meeting:
+ for attendee in meeting.attendee_ids:
+ if attendee.partner_id.id == partner.id:
+ data = (data[0], data[1], attendee.state)
+ datas.append(data)
+ return datas
+
+ def calendar_last_notif_ack(self, cr, uid, context=None):
+ partner = self.pool['res.users'].browse(cr, uid, uid, context=context).partner_id
+ self.write(cr, uid, partner.id, {'calendar_last_notif_ack': datetime.now()}, context=context)
+ return
+
+
+class calendar_alarm_manager(osv.AbstractModel):
+ _name = 'calendar.alarm_manager'
+
+ def get_next_potential_limit_alarm(self, cr, uid, seconds, notif=True, mail=True, partner_id=None, context=None):
+ res = {}
+ base_request = """
+ SELECT
+ cal.id,
+ cal.date - interval '1' minute * calcul_delta.max_delta AS first_alarm,
+ CASE
+ WHEN cal.recurrency THEN cal.end_date - interval '1' minute * calcul_delta.min_delta
+ ELSE cal.date_deadline - interval '1' minute * calcul_delta.min_delta
+ END as last_alarm,
+ cal.date as first_event_date,
+ CASE
+ WHEN cal.recurrency THEN cal.end_date
+ ELSE cal.date_deadline
+ END as last_event_date,
+ calcul_delta.min_delta,
+ calcul_delta.max_delta,
+ cal.rrule AS rule
+ FROM
+ calendar_event AS cal
+ RIGHT JOIN
+ (
+ SELECT
+ rel.calendar_event_id, max(alarm.duration_minutes) AS max_delta,min(alarm.duration_minutes) AS min_delta
+ FROM
+ calendar_alarm_calendar_event_rel AS rel
+ LEFT JOIN calendar_alarm AS alarm ON alarm.id = rel.calendar_alarm_id
+ WHERE alarm.type in %s
+ GROUP BY rel.calendar_event_id
+ ) AS calcul_delta ON calcul_delta.calendar_event_id = cal.id
+ """
+
+ filter_user = """
+ RIGHT JOIN calendar_event_res_partner_rel AS part_rel ON part_rel.calendar_event_id = cal.id
+ AND part_rel.res_partner_id = %s
+ """
+
+ #Add filter on type
+ type_to_read = ()
+ if notif:
+ type_to_read += ('notification',)
+ if mail:
+ type_to_read += ('email',)
+
+ tuple_params = (type_to_read,)
+
+ #ADD FILTER ON PARTNER_ID
+ if partner_id:
+ base_request += filter_user
+ tuple_params += (partner_id, )
+
+ #Add filter on hours
+ tuple_params += (seconds, seconds,)
+
+ cr.execute("""SELECT *
+ FROM ( %s ) AS ALL_EVENTS
+ WHERE ALL_EVENTS.first_alarm < (now() at time zone 'utc' + interval '%%s' second )
+ AND ALL_EVENTS.last_alarm > (now() at time zone 'utc' - interval '%%s' second )
+ """ % base_request, tuple_params)
+
+ for event_id, first_alarm, last_alarm, first_meeting, last_meeting, min_duration, max_duration, rule in cr.fetchall():
+ res[event_id] = {
+ 'event_id': event_id,
+ 'first_alarm': first_alarm,
+ 'last_alarm': last_alarm,
+ 'first_meeting': first_meeting,
+ 'last_meeting': last_meeting,
+ 'min_duration': min_duration,
+ 'max_duration': max_duration,
+ 'rrule': rule
+ }
+
+ return res
+
+ def do_check_alarm_for_one_date(self, cr, uid, one_date, event, event_maxdelta, in_the_next_X_seconds, after=False, notif=True, mail=True, context=None):
+ res = []
+ alarm_type = []
+
+ if notif:
+ alarm_type.append('notification')
+ if mail:
+ alarm_type.append('email')
+
+ if one_date - timedelta(minutes=event_maxdelta) < datetime.now() + timedelta(seconds=in_the_next_X_seconds): # if an alarm is possible for this date
+ for alarm in event.alarm_ids:
+ if alarm.type in alarm_type and \
+ one_date - timedelta(minutes=alarm.duration_minutes) < datetime.now() + timedelta(seconds=in_the_next_X_seconds) and \
+ (not after or one_date - timedelta(minutes=alarm.duration_minutes) > datetime.strptime(after.split('.')[0], "%Y-%m-%d %H:%M:%S")):
+ alert = {
+ 'alarm_id': alarm.id,
+ 'event_id': event.id,
+ 'notify_at': one_date - timedelta(minutes=alarm.duration_minutes),
+ }
+ res.append(alert)
+ return res
+
+ def get_next_mail(self, cr, uid, context=None):
+ cron = self.pool.get('ir.cron').search(cr, uid, [('model', 'ilike', self._name)], context=context)
+ if cron and len(cron) == 1:
+ cron = self.pool.get('ir.cron').browse(cr, uid, cron[0], context=context)
+ else:
+ raise ("Cron for " + self._name + " not identified :( !")
+
+ if cron.interval_type == "weeks":
+ cron_interval = cron.interval_number * 7 * 24 * 60 * 60
+ elif cron.interval_type == "days":
+ cron_interval = cron.interval_number * 24 * 60 * 60
+ elif cron.interval_type == "hours":
+ cron_interval = cron.interval_number * 60 * 60
+ elif cron.interval_type == "minutes":
+ cron_interval = cron.interval_number * 60
+ elif cron.interval_type == "seconds":
+ cron_interval = cron.interval_number
+
+ if not cron_interval:
+ raise ("Cron delay for " + self._name + " can not be calculated :( !")
+
+ all_events = self.get_next_potential_limit_alarm(cr, uid, cron_interval, notif=False, context=context)
+
+ for event in all_events: # .values()
+ max_delta = all_events[event]['max_duration']
+ curEvent = self.pool.get('calendar.event').browse(cr, uid, event, context=context)
+ if curEvent.recurrency:
+ bFound = False
+ LastFound = False
+ for one_date in self.pool.get('calendar.event').get_recurrent_date_by_event(cr, uid, curEvent, context=context):
+ in_date_format = datetime.strptime(one_date, '%Y-%m-%d %H:%M:%S')
+ LastFound = self.do_check_alarm_for_one_date(cr, uid, in_date_format, curEvent, max_delta, cron_interval, notif=False, context=context)
+ if LastFound:
+ for alert in LastFound:
+ self.do_mail_reminder(cr, uid, alert, context=context)
+
+ if not bFound: # if it's the first alarm for this recurrent event
+ bFound = True
+ if bFound and not LastFound: # if the precedent event had an alarm but not this one, we can stop the search for this event
+ break
+ else:
+ in_date_format = datetime.strptime(curEvent.date, '%Y-%m-%d %H:%M:%S')
+ LastFound = self.do_check_alarm_for_one_date(cr, uid, in_date_format, curEvent, max_delta, cron_interval, notif=False, context=context)
+ if LastFound:
+ for alert in LastFound:
+ self.do_mail_reminder(cr, uid, alert, context=context)
+
+ def get_next_notif(self, cr, uid, context=None):
+ ajax_check_every_seconds = 300
+ partner = self.pool.get('res.users').browse(cr, uid, uid, context=context).partner_id
+ all_notif = []
+
+ if not partner:
+ return []
+
+ all_events = self.get_next_potential_limit_alarm(cr, uid, ajax_check_every_seconds, partner_id=partner.id, mail=False, context=context)
+
+ for event in all_events: # .values()
+ max_delta = all_events[event]['max_duration']
+ curEvent = self.pool.get('calendar.event').browse(cr, uid, event, context=context)
+ if curEvent.recurrency:
+ bFound = False
+ LastFound = False
+ for one_date in self.pool.get("calendar.event").get_recurrent_date_by_event(cr, uid, curEvent, context=context):
+ in_date_format = datetime.strptime(one_date, '%Y-%m-%d %H:%M:%S')
+ LastFound = self.do_check_alarm_for_one_date(cr, uid, in_date_format, curEvent, max_delta, ajax_check_every_seconds, after=partner.calendar_last_notif_ack, mail=False, context=context)
+ if LastFound:
+ for alert in LastFound:
+ all_notif.append(self.do_notif_reminder(cr, uid, alert, context=context))
+ if not bFound: # if it's the first alarm for this recurrent event
+ bFound = True
+ if bFound and not LastFound: # if the precedent event had alarm but not this one, we can stop the search fot this event
+ break
+ else:
+ in_date_format = datetime.strptime(curEvent.date, '%Y-%m-%d %H:%M:%S')
+ LastFound = self.do_check_alarm_for_one_date(cr, uid, in_date_format, curEvent, max_delta, ajax_check_every_seconds, partner.calendar_last_notif_ack, mail=False, context=context)
+ if LastFound:
+ for alert in LastFound:
+ all_notif.append(self.do_notif_reminder(cr, uid, alert, context=context))
+ return all_notif
+
+ def do_mail_reminder(self, cr, uid, alert, context=None):
+ if context is None:
+ context = {}
+ res = False
+
+ event = self.pool['calendar.event'].browse(cr, uid, alert['event_id'], context=context)
+ alarm = self.pool['calendar.alarm'].browse(cr, uid, alert['alarm_id'], context=context)
+
+ if alarm.type == 'email':
+ res = self.pool['calendar.attendee']._send_mail_to_attendees(cr, uid, [att.id for att in event.attendee_ids], template_xmlid='calendar_template_meeting_reminder', context=context)
+
+ return res
+
+ def do_notif_reminder(self, cr, uid, alert, context=None):
+ alarm = self.pool['calendar.alarm'].browse(cr, uid, alert['alarm_id'], context=context)
+ event = self.pool['calendar.event'].browse(cr, uid, alert['event_id'], context=context)
+
+ if alarm.type == 'notification':
+ message = event.display_time
+
+ delta = alert['notify_at'] - datetime.now()
+ delta = delta.seconds + delta.days * 3600 * 24
+
+ return {
+ 'event_id': event.id,
+ 'title': event.name,
+ 'message': message,
+ 'timer': delta,
+ 'notify_at': alert['notify_at'].strftime("%Y-%m-%d %H:%M:%S"),
+ }
+
+
+class calendar_alarm(osv.Model):
+ _name = 'calendar.alarm'
+ _description = 'Event alarm'
+
+ def _get_duration(self, cr, uid, ids, field_name, arg, context=None):
+ res = {}
+ for alarm in self.browse(cr, uid, ids, context=context):
+ if alarm.interval == "minutes":
+ res[alarm.id] = alarm.duration
+ elif alarm.interval == "hours":
+ res[alarm.id] = alarm.duration * 60
+ elif alarm.interval == "days":
+ res[alarm.id] = alarm.duration * 60 * 24
+ else:
+ res[alarm.id] = 0
+ return res
+
+ _columns = {
+ 'name': fields.char('Name', required=True), # fields function
+ 'type': fields.selection([('notification', 'Notification'), ('email', 'Email')], 'Type', required=True),
+ 'duration': fields.integer('Amount', required=True),
+ 'interval': fields.selection([('minutes', 'Minutes'), ('hours', 'Hours'), ('days', 'Days')], 'Unit', required=True),
+ 'duration_minutes': fields.function(_get_duration, type='integer', string='duration_minutes', store=True),
+ }
+
+ _defaults = {
+ 'type': 'notification',
+ 'duration': 1,
+ 'interval': 'hours',
+ }
+
+
+class ir_values(osv.Model):
+ _inherit = 'ir.values'
+
+ def set(self, cr, uid, key, key2, name, models, value, replace=True, isobject=False, meta=False, preserve_user=False, company=False):
+
+ new_model = []
+ for data in models:
+ if type(data) in (list, tuple):
+ new_model.append((data[0], calendar_id2real_id(data[1])))
+ else:
+ new_model.append(data)
+ return super(ir_values, self).set(cr, uid, key, key2, name, new_model,
+ value, replace, isobject, meta, preserve_user, company)
+
+ def get(self, cr, uid, key, key2, models, meta=False, context=None, res_id_req=False, without_user=True, key2_req=True):
+ if context is None:
+ context = {}
+ new_model = []
+ for data in models:
+ if type(data) in (list, tuple):
+ new_model.append((data[0], calendar_id2real_id(data[1])))
+ else:
+ new_model.append(data)
+ return super(ir_values, self).get(cr, uid, key, key2, new_model,
+ meta, context, res_id_req, without_user, key2_req)
+
+
+class ir_model(osv.Model):
+
+ _inherit = 'ir.model'
+
+ def read(self, cr, uid, ids, fields=None, context=None, load='_classic_read'):
+ new_ids = isinstance(ids, (str, int, long)) and [ids] or ids
+ if context is None:
+ context = {}
+ data = super(ir_model, self).read(cr, uid, new_ids, fields=fields, context=context, load=load)
+ if data:
+ for val in data:
+ val['id'] = calendar_id2real_id(val['id'])
+ return isinstance(ids, (str, int, long)) and data[0] or data
+
+
+original_exp_report = openerp.service.report.exp_report
+
+
+def exp_report(db, uid, object, ids, data=None, context=None):
+ """
+ Export Report
+ """
+ if object == 'printscreen.list':
+ original_exp_report(db, uid, object, ids, data, context)
+ new_ids = []
+ for id in ids:
+ new_ids.append(calendar_id2real_id(id))
+ if data.get('id', False):
+ data['id'] = calendar_id2real_id(data['id'])
+ return original_exp_report(db, uid, object, new_ids, data, context)
+
+
+openerp.service.report.exp_report = exp_report
+
+
+class calendar_event_type(osv.Model):
+ _name = 'calendar.event.type'
+ _description = 'Meeting Type'
+ _columns = {
+ 'name': fields.char('Name', required=True, translate=True),
+ }
+
+
+class calendar_event(osv.Model):
+ """ Model for Calendar Event """
+ _name = 'calendar.event'
+ _description = "Meeting"
+ _order = "id desc"
+ _inherit = ["mail.thread", "ir.needaction_mixin"]
+
+ def do_run_scheduler(self, cr, uid, id, context=None):
+ self.pool['calendar.alarm_manager'].do_run_scheduler(cr, uid, context=context)
+
+ def get_recurrent_date_by_event(self, cr, uid, event, context=None):
+ """Get recurrent dates based on Rule string and all event where recurrent_id is child
+ """
+
+ def todate(date):
+ val = parser.parse(''.join((re.compile('\d')).findall(date)))
+ ## Dates are localized to saved timezone if any, else current timezone.
+ if not val.tzinfo:
+ val = pytz.UTC.localize(val)
+ return val.astimezone(timezone)
+
+ timezone = pytz.timezone(event.vtimezone or context.get('tz') or 'UTC')
+ startdate = pytz.UTC.localize(datetime.strptime(event.date, "%Y-%m-%d %H:%M:%S")) # Add "+hh:mm" timezone
+ if not startdate:
+ startdate = datetime.now()
+
+ ## Convert the start date to saved timezone (or context tz) as it'll
+ ## define the correct hour/day asked by the user to repeat for recurrence.
+ startdate = startdate.astimezone(timezone) # transform "+hh:mm" timezone
+ rset1 = rrule.rrulestr(str(event.rrule), dtstart=startdate, forceset=True)
+ ids_depending = self.search(cr, uid, [('recurrent_id', '=', event.id), '|', ('active', '=', False), ('active', '=', True)], context=context)
+ all_events = self.browse(cr, uid, ids_depending, context=context)
+
+ for ev in all_events:
+ rset1._exdate.append(todate(ev.recurrent_id_date))
+
+ return [d.astimezone(pytz.UTC) for d in rset1]
+
+ def _get_recurrency_end_date(self, data, context=None):
- if data.get('recurrency') and data.get('end_type') in ('count', unicode('count')):
- data_date_deadline = datetime.strptime(data.get('date_deadline'), '%Y-%m-%d %H:%M:%S')
- if data.get('rrule_type') in ('daily', unicode('count')):
- rel_date = relativedelta(days=data.get('count') + 1)
- elif data.get('rrule_type') in ('weekly', unicode('weekly')):
- rel_date = relativedelta(days=(data.get('count') + 1) * 7)
- elif data.get('rrule_type') in ('monthly', unicode('monthly')):
- rel_date = relativedelta(months=data.get('count') + 1)
- elif data.get('rrule_type') in ('yearly', unicode('yearly')):
- rel_date = relativedelta(years=data.get('count') + 1)
- end_date = data_date_deadline + rel_date
- else:
- end_date = data.get('end_date')
++ if not data.get('recurrency'):
++ return False
++
++ end_type = data.get('end_type')
++ end_date = data.get('end_date')
++
++ if end_type == 'count' and all(data.get(key) for key in ['count', 'rrule_type', 'date_deadline']):
++ count = data['count'] + 1
++ delay, mult = {
++ 'daily': ('days', 1),
++ 'weekly': ('days', 7),
++ 'monthly': ('months', 1),
++ 'yearly': ('years', 1),
++ }[data['rrule_type']]
++
++ deadline = datetime.strptime(data['date_deadline'], tools.DEFAULT_SERVER_DATETIME_FORMAT)
++ return deadline + relativedelta(**{delay: count * mult})
+ return end_date
+
+ def _find_my_attendee(self, cr, uid, meeting_ids, context=None):
+ """
+ Return the first attendee where the user connected has been invited from all the meeting_ids in parameters
+ """
+ user = self.pool['res.users'].browse(cr, uid, uid, context=context)
+ for meeting_id in meeting_ids:
+ for attendee in self.browse(cr, uid, meeting_id, context).attendee_ids:
+ if user.partner_id.id == attendee.partner_id.id:
+ return attendee
+ return False
+
+ def _get_display_time(self, cr, uid, meeting_id, context=None):
+ """
+ Return date and time (from to from) based on duration with timezone in string :
+ eg.
+ 1) if user add duration for 2 hours, return : August-23-2013 at ( 04-30 To 06-30) (Europe/Brussels)
+ 2) if event all day ,return : AllDay, July-31-2013
+ """
+ if context is None:
+ context = {}
+
+ tz = context.get('tz', False)
+ if not tz: # tz can have a value False, so dont do it in the default value of get !
+ tz = pytz.timezone('UTC')
+
+ meeting = self.browse(cr, uid, meeting_id, context=context)
+ date = fields.datetime.context_timestamp(cr, uid, datetime.strptime(meeting.date, tools.DEFAULT_SERVER_DATETIME_FORMAT), context=context)
+ date_deadline = fields.datetime.context_timestamp(cr, uid, datetime.strptime(meeting.date_deadline, tools.DEFAULT_SERVER_DATETIME_FORMAT), context=context)
+ event_date = date.strftime('%B-%d-%Y')
+ display_time = date.strftime('%H-%M')
+ if meeting.allday:
+ time = _("AllDay , %s") % (event_date)
+ elif meeting.duration < 24:
+ duration = date + timedelta(hours=meeting.duration)
+ time = ("%s at ( %s To %s) (%s)") % (event_date, display_time, duration.strftime('%H-%M'), tz)
+ else:
+ time = ("%s at %s To\n %s at %s (%s)") % (event_date, display_time, date_deadline.strftime('%B-%d-%Y'), date_deadline.strftime('%H-%M'), tz)
+ return time
+
+ def _compute(self, cr, uid, ids, fields, arg, context=None):
+ res = {}
+ for meeting_id in ids:
+ res[meeting_id] = {}
+ attendee = self._find_my_attendee(cr, uid, [meeting_id], context)
+ for field in fields:
+ if field == 'is_attendee':
+ res[meeting_id][field] = True if attendee else False
+ elif field == 'attendee_status':
+ res[meeting_id][field] = attendee.state if attendee else 'needsAction'
+ elif field == 'display_time':
+ res[meeting_id][field] = self._get_display_time(cr, uid, meeting_id, context=context)
+ return res
+
+ def _get_rulestring(self, cr, uid, ids, name, arg, context=None):
+ """
+ Gets Recurrence rule string according to value type RECUR of iCalendar from the values given.
+ @return: dictionary of rrule value.
+ """
+ result = {}
+ if not isinstance(ids, list):
+ ids = [ids]
+
+ for id in ids:
+ #read these fields as SUPERUSER because if the record is private a normal search could return False and raise an error
+ data = self.browse(cr, SUPERUSER_ID, id, context=context)
+
+ if data.interval and data.interval < 0:
+ raise osv.except_osv(_('Warning!'), _('Interval cannot be negative.'))
+ if data.count and data.count <= 0:
+ raise osv.except_osv(_('Warning!'), _('Count cannot be negative or 0.'))
+
+ data = self.read(cr, uid, id, ['id', 'byday', 'recurrency', 'month_list', 'end_date', 'rrule_type', 'month_by', 'interval', 'count', 'end_type', 'mo', 'tu', 'we', 'th', 'fr', 'sa', 'su', 'day', 'week_list'], context=context)
+ event = data['id']
+ if data['recurrency']:
+ result[event] = self.compute_rule_string(data)
+ else:
+ result[event] = ""
+ return result
+
+ # retro compatibility function
+ def _rrule_write(self, cr, uid, ids, field_name, field_value, args, context=None):
+ return self._set_rulestring(self, cr, uid, ids, field_name, field_value, args, context=context)
+
+ def _set_rulestring(self, cr, uid, ids, field_name, field_value, args, context=None):
+ if not isinstance(ids, list):
+ ids = [ids]
+ data = self._get_empty_rrule_data()
+ if field_value:
+ data['recurrency'] = True
+ for event in self.browse(cr, uid, ids, context=context):
+ rdate = event.date
+ update_data = self._parse_rrule(field_value, dict(data), rdate)
+ data.update(update_data)
+ self.write(cr, uid, ids, data, context=context)
+ return True
+
+ def _tz_get(self, cr, uid, context=None):
+ return [(x.lower(), x) for x in pytz.all_timezones]
+
+ _track = {
+ 'location': {
+ 'calendar.subtype_invitation': lambda self, cr, uid, obj, ctx=None: True,
+ },
+ 'date': {
+ 'calendar.subtype_invitation': lambda self, cr, uid, obj, ctx=None: True,
+ },
+ }
+ _columns = {
+ 'id': fields.integer('ID', readonly=True),
+ 'state': fields.selection([('draft', 'Unconfirmed'), ('open', 'Confirmed')], string='Status', readonly=True, track_visibility='onchange'),
+ 'name': fields.char('Meeting Subject', required=True, states={'done': [('readonly', True)]}),
+ 'is_attendee': fields.function(_compute, string='Attendee', type="boolean", multi='attendee'),
+ 'attendee_status': fields.function(_compute, string='Attendee Status', type="selection", selection=calendar_attendee.STATE_SELECTION, multi='attendee'),
+ 'display_time': fields.function(_compute, string='Event Time', type="char", multi='attendee'),
+ 'date': fields.datetime('Date', states={'done': [('readonly', True)]}, required=True, track_visibility='onchange'),
+ 'date_deadline': fields.datetime('End Date', states={'done': [('readonly', True)]}, required=True,),
+ 'duration': fields.float('Duration', states={'done': [('readonly', True)]}),
+ 'description': fields.text('Description', states={'done': [('readonly', True)]}),
+ 'class': fields.selection([('public', 'Public'), ('private', 'Private'), ('confidential', 'Public for Employees')], 'Privacy', states={'done': [('readonly', True)]}),
+ 'location': fields.char('Location', help="Location of Event", track_visibility='onchange', states={'done': [('readonly', True)]}),
+ 'show_as': fields.selection([('free', 'Free'), ('busy', 'Busy')], 'Show Time as', states={'done': [('readonly', True)]}),
+ 'rrule': fields.function(_get_rulestring, type='char', fnct_inv=_set_rulestring, store=True, string='Recurrent Rule'),
+ 'rrule_type': fields.selection([('daily', 'Day(s)'), ('weekly', 'Week(s)'), ('monthly', 'Month(s)'), ('yearly', 'Year(s)')], 'Recurrency', states={'done': [('readonly', True)]}, help="Let the event automatically repeat at that interval"),
+ 'recurrency': fields.boolean('Recurrent', help="Recurrent Meeting"),
+ 'recurrent_id': fields.integer('Recurrent ID'),
+ 'recurrent_id_date': fields.datetime('Recurrent ID date'),
+ 'vtimezone': fields.selection(_tz_get, string='Timezone'),
+ 'end_type': fields.selection([('count', 'Number of repetitions'), ('end_date', 'End date')], 'Recurrence Termination'),
+ 'interval': fields.integer('Repeat Every', help="Repeat every (Days/Week/Month/Year)"),
+ 'count': fields.integer('Repeat', help="Repeat x times"),
+ 'mo': fields.boolean('Mon'),
+ 'tu': fields.boolean('Tue'),
+ 'we': fields.boolean('Wed'),
+ 'th': fields.boolean('Thu'),
+ 'fr': fields.boolean('Fri'),
+ 'sa': fields.boolean('Sat'),
+ 'su': fields.boolean('Sun'),
+ 'month_by': fields.selection([('date', 'Date of month'), ('day', 'Day of month')], 'Option', oldname='select1'),
+ 'day': fields.integer('Date of month'),
+ 'week_list': fields.selection([('MO', 'Monday'), ('TU', 'Tuesday'), ('WE', 'Wednesday'), ('TH', 'Thursday'), ('FR', 'Friday'), ('SA', 'Saturday'), ('SU', 'Sunday')], 'Weekday'),
+ 'byday': fields.selection([('1', 'First'), ('2', 'Second'), ('3', 'Third'), ('4', 'Fourth'), ('5', 'Fifth'), ('-1', 'Last')], 'By day'),
+ 'end_date': fields.date('Repeat Until'),
+ 'allday': fields.boolean('All Day', states={'done': [('readonly', True)]}),
+ 'user_id': fields.many2one('res.users', 'Responsible', states={'done': [('readonly', True)]}),
+ 'color_partner_id': fields.related('user_id', 'partner_id', 'id', type="integer", string="colorize", store=False), # Color of creator
+ 'active': fields.boolean('Active', help="If the active field is set to true, it will allow you to hide the event alarm information without removing it."),
+ 'categ_ids': fields.many2many('calendar.event.type', 'meeting_category_rel', 'event_id', 'type_id', 'Tags'),
+ 'attendee_ids': fields.one2many('calendar.attendee', 'event_id', 'Attendees', ondelete='cascade'),
+ 'partner_ids': fields.many2many('res.partner', string='Attendees', states={'done': [('readonly', True)]}),
+ 'alarm_ids': fields.many2many('calendar.alarm', string='Reminders', ondelete="restrict"),
+
+ }
+ _defaults = {
+ 'end_type': 'count',
+ 'count': 1,
+ 'rrule_type': False,
+ 'state': 'draft',
+ 'class': 'public',
+ 'show_as': 'busy',
+ 'month_by': 'date',
+ 'interval': 1,
+ 'active': 1,
+ 'user_id': lambda self, cr, uid, ctx: uid,
+ 'partner_ids': lambda self, cr, uid, ctx: [self.pool['res.users'].browse(cr, uid, [uid], context=ctx)[0].partner_id.id]
+ }
+
+ def _check_closing_date(self, cr, uid, ids, context=None):
+ for event in self.browse(cr, uid, ids, context=context):
+ if event.date_deadline < event.date:
+ return False
+ return True
+
+ _constraints = [
+ (_check_closing_date, 'Error ! End date cannot be set before start date.', ['date_deadline']),
+ ]
+
+ def onchange_dates(self, cr, uid, ids, start_date, duration=False, end_date=False, allday=False, context=None):
+
+ """Returns duration and/or end date based on values passed
+ @param ids: List of calendar event's IDs.
+ """
+ if context is None:
+ context = {}
+
+ value = {}
+ if not start_date:
+ return value
+
+ if not end_date and not duration:
+ duration = 1.00
+ value['duration'] = duration
+
+ if allday: # For all day event
+ start = datetime.strptime(start_date, "%Y-%m-%d %H:%M:%S")
+ user = self.pool['res.users'].browse(cr, uid, uid)
+ tz = pytz.timezone(user.tz) if user.tz else pytz.utc
+ start = pytz.utc.localize(start).astimezone(tz) # convert start in user's timezone
+ start = start.astimezone(pytz.utc) # convert start back to utc
+
+ value['duration'] = 24.0
+ value['date'] = datetime.strftime(start, "%Y-%m-%d %H:%M:%S")
+ else:
+ start = datetime.strptime(start_date, "%Y-%m-%d %H:%M:%S")
+
+ if end_date and not duration:
+ end = datetime.strptime(end_date, "%Y-%m-%d %H:%M:%S")
+ diff = end - start
+ duration = float(diff.days) * 24 + (float(diff.seconds) / 3600)
+ value['duration'] = round(duration, 2)
+ elif not end_date:
+ end = start + timedelta(hours=duration)
+ value['date_deadline'] = end.strftime("%Y-%m-%d %H:%M:%S")
+ elif end_date and duration and not allday:
+ # we have both, keep them synchronized:
+ # set duration based on end_date (arbitrary decision: this avoid
+ # getting dates like 06:31:48 instead of 06:32:00)
+ end = datetime.strptime(end_date, "%Y-%m-%d %H:%M:%S")
+ diff = end - start
+ duration = float(diff.days) * 24 + (float(diff.seconds) / 3600)
+ value['duration'] = round(duration, 2)
+
+ return {'value': value}
+
+ def new_invitation_token(self, cr, uid, record, partner_id):
+ return uuid.uuid4().hex
+
+ def create_attendees(self, cr, uid, ids, context):
+ user_obj = self.pool['res.users']
+ current_user = user_obj.browse(cr, uid, uid, context=context)
+ res = {}
+ for event in self.browse(cr, uid, ids, context):
+ attendees = {}
+ for att in event.attendee_ids:
+ attendees[att.partner_id.id] = True
+ new_attendees = []
+ new_att_partner_ids = []
+ for partner in event.partner_ids:
+ if partner.id in attendees:
+ continue
+ access_token = self.new_invitation_token(cr, uid, event, partner.id)
+ values = {
+ 'partner_id': partner.id,
+ 'event_id': event.id,
+ 'access_token': access_token,
+ 'email': partner.email,
+ }
+
+ if partner.id == current_user.partner_id.id:
+ values['state'] = 'accepted'
+
+ att_id = self.pool['calendar.attendee'].create(cr, uid, values, context=context)
+ new_attendees.append(att_id)
+ new_att_partner_ids.append(partner.id)
+
+ if not current_user.email or current_user.email != partner.email:
+ mail_from = current_user.email or tools.config.get('email_from', False)
+ if self.pool['calendar.attendee']._send_mail_to_attendees(cr, uid, att_id, email_from=mail_from, context=context):
+ self.message_post(cr, uid, event.id, body=_("An invitation email has been sent to attendee %s") % (partner.name,), subtype="calendar.subtype_invitation", context=context)
+
+ if new_attendees:
+ self.write(cr, uid, [event.id], {'attendee_ids': [(4, att) for att in new_attendees]}, context=context)
+ if new_att_partner_ids:
+ self.message_subscribe(cr, uid, [event.id], new_att_partner_ids, context=context)
+
+ # We remove old attendees who are not in partner_ids now.
+ all_partner_ids = [part.id for part in event.partner_ids]
+ all_part_attendee_ids = [att.partner_id.id for att in event.attendee_ids]
+ all_attendee_ids = [att.id for att in event.attendee_ids]
+ partner_ids_to_remove = map(lambda x: x, set(all_part_attendee_ids + new_att_partner_ids) - set(all_partner_ids))
+
+ attendee_ids_to_remove = []
+
+ if partner_ids_to_remove:
+ attendee_ids_to_remove = self.pool["calendar.attendee"].search(cr, uid, [('partner_id.id', 'in', partner_ids_to_remove), ('event_id.id', '=', event.id)], context=context)
+ if attendee_ids_to_remove:
+ self.pool['calendar.attendee'].unlink(cr, uid, attendee_ids_to_remove, context)
+
+ res[event.id] = {
+ 'new_attendee_ids': new_attendees,
+ 'old_attendee_ids': all_attendee_ids,
+ 'removed_attendee_ids': attendee_ids_to_remove
+ }
+ return res
+
+ def get_search_fields(self, browse_event, order_fields, r_date=None):
+ sort_fields = {}
+ for ord in order_fields:
+ if ord == 'id' and r_date:
+ sort_fields[ord] = '%s-%s' % (browse_event[ord], r_date.strftime("%Y%m%d%H%M%S"))
+ else:
+ sort_fields[ord] = browse_event[ord]
+ 'If we sort on FK, we obtain a browse_record, so we need to sort on name_get'
+ if type(browse_event[ord]) is openerp.osv.orm.browse_record:
+ name_get = browse_event[ord].name_get()
+ if len(name_get) and len(name_get[0]) >= 2:
+ sort_fields[ord] = name_get[0][1]
+
+ return sort_fields
+
+ def get_recurrent_ids(self, cr, uid, event_id, domain, order=None, context=None):
+
+ """Gives virtual event ids for recurring events
+ This method gives ids of dates that comes between start date and end date of calendar views
+
+ @param order: The fields (comma separated, format "FIELD {DESC|ASC}") on which the events should be sorted
+ """
+
+ if not context:
+ context = {}
+
+ if isinstance(event_id, (str, int, long)):
+ ids_to_browse = [event_id] # keep select for return
+ else:
+ ids_to_browse = event_id
+
+ if order:
+ order_fields = [field.split()[0] for field in order.split(',')]
+ else:
+ # fallback on self._order defined on the model
+ order_fields = [field.split()[0] for field in self._order.split(',')]
+
+ if 'id' not in order_fields:
+ order_fields.append('id')
+
+ result_data = []
+ result = []
+ for ev in self.browse(cr, uid, ids_to_browse, context=context):
+ if not ev.recurrency or not ev.rrule:
+ result.append(ev.id)
+ result_data.append(self.get_search_fields(ev, order_fields))
+ continue
+
+ rdates = self.get_recurrent_date_by_event(cr, uid, ev, context=context)
+
+ for r_date in rdates:
+ # fix domain evaluation
+ # step 1: check date and replace expression by True or False, replace other expressions by True
+ # step 2: evaluation of & and |
+ # check if there are one False
+ pile = []
+ ok = True
+ for arg in domain:
+ if str(arg[0]) in (str('date'), str('date_deadline'), str('end_date')):
+ if (arg[1] == '='):
+ ok = r_date.strftime('%Y-%m-%d') == arg[2]
+ if (arg[1] == '>'):
+ ok = r_date.strftime('%Y-%m-%d') > arg[2]
+ if (arg[1] == '<'):
+ ok = r_date.strftime('%Y-%m-%d') < arg[2]
+ if (arg[1] == '>='):
+ ok = r_date.strftime('%Y-%m-%d') >= arg[2]
+ if (arg[1] == '<='):
+ ok = r_date.strftime('%Y-%m-%d') <= arg[2]
+ pile.append(ok)
+ elif str(arg) == str('&') or str(arg) == str('|'):
+ pile.append(arg)
+ else:
+ pile.append(True)
+ pile.reverse()
+ new_pile = []
+ for item in pile:
+ if not isinstance(item, basestring):
+ res = item
+ elif str(item) == str('&'):
+ first = new_pile.pop()
+ second = new_pile.pop()
+ res = first and second
+ elif str(item) == str('|'):
+ first = new_pile.pop()
+ second = new_pile.pop()
+ res = first or second
+ new_pile.append(res)
+
+ if [True for item in new_pile if not item]:
+ continue
+ result_data.append(self.get_search_fields(ev, order_fields, r_date=r_date))
+
+ if order_fields:
+ def comparer(left, right):
+ for fn, mult in comparers:
+ result = cmp(fn(left), fn(right))
+ if result:
+ return mult * result
+ return 0
+
+ sort_params = [key.split()[0] if key[-4:].lower() != 'desc' else '-%s' % key.split()[0] for key in (order or self._order).split(',')]
+ comparers = [((itemgetter(col[1:]), -1) if col[0] == '-' else (itemgetter(col), 1)) for col in sort_params]
+ ids = [r['id'] for r in sorted(result_data, cmp=comparer)]
+
+ if isinstance(event_id, (str, int, long)):
+ return ids and ids[0] or False
+ else:
+ return ids
+
+ def compute_rule_string(self, data):
+ """
+ Compute rule string according to value type RECUR of iCalendar from the values given.
+ @param self: the object pointer
+ @param data: dictionary of freq and interval value
+ @return: string containing recurring rule (empty if no rule)
+ """
+ def get_week_string(freq, data):
+ weekdays = ['mo', 'tu', 'we', 'th', 'fr', 'sa', 'su']
+ if freq == 'weekly':
+ byday = map(lambda x: x.upper(), filter(lambda x: data.get(x) and x in weekdays, data))
+ # byday = map(lambda x: x.upper(),[data[day] for day in weekdays if data[day]])
+
+ if byday:
+ return ';BYDAY=' + ','.join(byday)
+ return ''
+
+ def get_month_string(freq, data):
+ if freq == 'monthly':
+ if data.get('month_by') == 'date' and (data.get('day') < 1 or data.get('day') > 31):
+ raise osv.except_osv(_('Error!'), ("Please select a proper day of the month."))
+
+ if data.get('month_by') == 'day': # Eg : Second Monday of the month
+ return ';BYDAY=' + data.get('byday') + data.get('week_list')
+ elif data.get('month_by') == 'date': # Eg : 16th of the month
+ return ';BYMONTHDAY=' + str(data.get('day'))
+ return ''
+
+ def get_end_date(data):
+ if data.get('end_date'):
+ data['end_date_new'] = ''.join((re.compile('\d')).findall(data.get('end_date'))) + 'T235959Z'
+
+ return (data.get('end_type') == 'count' and (';COUNT=' + str(data.get('count'))) or '') +\
+ ((data.get('end_date_new') and data.get('end_type') == 'end_date' and (';UNTIL=' + data.get('end_date_new'))) or '')
+
+ freq = data.get('rrule_type', False) # day/week/month/year
+ res = ''
+ if freq:
+ interval_srting = data.get('interval') and (';INTERVAL=' + str(data.get('interval'))) or ''
+ res = 'FREQ=' + freq.upper() + get_week_string(freq, data) + interval_srting + get_end_date(data) + get_month_string(freq, data)
+
+ return res
+
+ def _get_empty_rrule_data(self):
+ return {
+ 'byday': False,
+ 'recurrency': False,
+ 'end_date': False,
+ 'rrule_type': False,
+ 'month_by': False,
+ 'interval': 0,
+ 'count': False,
+ 'end_type': False,
+ 'mo': False,
+ 'tu': False,
+ 'we': False,
+ 'th': False,
+ 'fr': False,
+ 'sa': False,
+ 'su': False,
+ 'day': False,
+ 'week_list': False
+ }
+
+ def _parse_rrule(self, rule, data, date_start):
+ day_list = ['mo', 'tu', 'we', 'th', 'fr', 'sa', 'su']
+ rrule_type = ['yearly', 'monthly', 'weekly', 'daily']
+ r = rrule.rrulestr(rule, dtstart=datetime.strptime(date_start, "%Y-%m-%d %H:%M:%S"))
+
+ if r._freq > 0 and r._freq < 4:
+ data['rrule_type'] = rrule_type[r._freq]
+
+ data['count'] = r._count
+ data['interval'] = r._interval
+ data['end_date'] = r._until and r._until.strftime("%Y-%m-%d %H:%M:%S")
+ #repeat weekly
+ if r._byweekday:
+ for i in xrange(0, 7):
+ if i in r._byweekday:
+ data[day_list[i]] = True
+ data['rrule_type'] = 'weekly'
+ #repeat monthly by nweekday ((weekday, weeknumber), )
+ if r._bynweekday:
+ data['week_list'] = day_list[r._bynweekday[0][0]].upper()
+ data['byday'] = str(r._bynweekday[0][1])
+ data['month_by'] = 'day'
+ data['rrule_type'] = 'monthly'
+
+ if r._bymonthday:
+ data['day'] = r._bymonthday[0]
+ data['month_by'] = 'date'
+ data['rrule_type'] = 'monthly'
+
+ #repeat yearly but for openerp it's monthly, take same information as monthly but interval is 12 times
+ if r._bymonth:
+ data['interval'] = data['interval'] * 12
+
+ #FIXEME handle forever case
+ #end of recurrence
+ #in case of repeat for ever that we do not support right now
+ if not (data.get('count') or data.get('end_date')):
+ data['count'] = 100
+ if data.get('count'):
+ data['end_type'] = 'count'
+ else:
+ data['end_type'] = 'end_date'
+ return data
+
+ def message_get_subscription_data(self, cr, uid, ids, user_pid=None, context=None):
+ res = {}
+ for virtual_id in ids:
+ real_id = calendar_id2real_id(virtual_id)
+ result = super(calendar_event, self).message_get_subscription_data(cr, uid, [real_id], user_pid=None, context=context)
+ res[virtual_id] = result[real_id]
+ return res
+
+ def onchange_partner_ids(self, cr, uid, ids, value, context=None):
+ """ The basic purpose of this method is to check that destination partners
+ effectively have email addresses. Otherwise a warning is thrown.
+ :param value: value format: [[6, 0, [3, 4]]]
+ """
+ res = {'value': {}}
+
+ if not value or not value[0] or not value[0][0] == 6:
+ return
+
+ res.update(self.check_partners_email(cr, uid, value[0][2], context=context))
+ return res
+
+ def onchange_rec_day(self, cr, uid, id, date, mo, tu, we, th, fr, sa, su):
+ """ set the start date according to the first occurence of rrule"""
+ rrule_obj = self._get_empty_rrule_data()
+ rrule_obj.update({
+ 'byday': True,
+ 'rrule_type': 'weekly',
+ 'mo': mo,
+ 'tu': tu,
+ 'we': we,
+ 'th': th,
+ 'fr': fr,
+ 'sa': sa,
+ 'su': su,
+ 'interval': 1
+ })
+ str_rrule = self.compute_rule_string(rrule_obj)
+ first_occurence = list(rrule.rrulestr(str_rrule + ";COUNT=1", dtstart=datetime.strptime(date, "%Y-%m-%d %H:%M:%S"), forceset=True))[0]
+ return {'value': {'date': first_occurence.strftime("%Y-%m-%d") + ' 00:00:00'}}
+
+ def check_partners_email(self, cr, uid, partner_ids, context=None):
+ """ Verify that selected partner_ids have an email_address defined.
+ Otherwise throw a warning. """
+ partner_wo_email_lst = []
+ for partner in self.pool['res.partner'].browse(cr, uid, partner_ids, context=context):
+ if not partner.email:
+ partner_wo_email_lst.append(partner)
+ if not partner_wo_email_lst:
+ return {}
+ warning_msg = _('The following contacts have no email address :')
+ for partner in partner_wo_email_lst:
+ warning_msg += '\n- %s' % (partner.name)
+ return {'warning': {
+ 'title': _('Email addresses not found'),
+ 'message': warning_msg,
+ }}
+
+ # ----------------------------------------
+ # OpenChatter
+ # ----------------------------------------
+
+ # shows events of the day for this user
+
+ def _needaction_domain_get(self, cr, uid, context=None):
+ return [('end_date', '>=', time.strftime(DEFAULT_SERVER_DATE_FORMAT + ' 23:59:59')), ('date', '>=', time.strftime(DEFAULT_SERVER_DATE_FORMAT + ' 23:59:59')), ('user_id', '=', uid)]
+
+ def message_post(self, cr, uid, thread_id, body='', subject=None, type='notification', subtype=None, parent_id=False, attachments=None, context=None, **kwargs):
+ if isinstance(thread_id, str):
+ thread_id = get_real_ids(thread_id)
+ if context.get('default_date'):
+ del context['default_date']
+ return super(calendar_event, self).message_post(cr, uid, thread_id, body=body, subject=subject, type=type, subtype=subtype, parent_id=parent_id, attachments=attachments, context=context, **kwargs)
+
+ def do_sendmail(self, cr, uid, ids, context=None):
+ for event in self.browse(cr, uid, ids, context):
+ current_user = self.pool['res.users'].browse(cr, uid, uid, context=context)
+
+ if current_user.email:
+ if self.pool['calendar.attendee']._send_mail_to_attendees(cr, uid, [att.id for att in event.attendee_ids], email_from=current_user.email, context=context):
+ self.message_post(cr, uid, event.id, body=_("An invitation email has been sent to attendee(s)"), subtype="calendar.subtype_invitation", context=context)
+ return
+
+ def get_attendee(self, cr, uid, meeting_id, context=None):
+ # Used for view in controller
+ invitation = {'meeting': {}, 'attendee': []}
+
+ meeting = self.browse(cr, uid, int(meeting_id), context)
+ invitation['meeting'] = {
+ 'event': meeting.name,
+ 'where': meeting.location,
+ 'when': meeting.display_time
+ }
+
+ for attendee in meeting.attendee_ids:
+ invitation['attendee'].append({'name': attendee.cn, 'status': attendee.state})
+ return invitation
+
+ def get_interval(self, cr, uid, ids, date, interval, context=None):
+ #Function used only in calendar_event_data.xml for email template
+ date = datetime.strptime(date.split('.')[0], DEFAULT_SERVER_DATETIME_FORMAT)
+ if interval == 'day':
+ res = str(date.day)
+ elif interval == 'month':
+ res = date.strftime('%B') + " " + str(date.year)
+ elif interval == 'dayname':
+ res = date.strftime('%A')
+ elif interval == 'time':
+ res = date.strftime('%I:%M %p')
+ return res
+
+ def search(self, cr, uid, args, offset=0, limit=0, order=None, context=None, count=False):
+ if context is None:
+ context = {}
+
+ if context.get('mymeetings', False):
+ partner_id = self.pool['res.users'].browse(cr, uid, uid, context).partner_id.id
+ args += ['|', ('partner_ids', 'in', [partner_id]), ('user_id', '=', uid)]
+
+ new_args = []
+ for arg in args:
+ new_arg = arg
+
+ if arg[0] in ('date', unicode('date')) and arg[1] == ">=":
+ if context.get('virtual_id', True):
+ new_args += ['|', '&', ('recurrency', '=', 1), ('end_date', arg[1], arg[2])]
+ elif arg[0] == "id":
+ new_id = get_real_ids(arg[2])
+ new_arg = (arg[0], arg[1], new_id)
+ new_args.append(new_arg)
+
+ if not context.get('virtual_id', True):
+ return super(calendar_event, self).search(cr, uid, new_args, offset=offset, limit=limit, order=order, context=context, count=count)
+
+ # offset, limit, order and count must be treated separately as we may need to deal with virtual ids
+ res = super(calendar_event, self).search(cr, uid, new_args, offset=0, limit=0, order=None, context=context, count=False)
+ res = self.get_recurrent_ids(cr, uid, res, args, order=order, context=context)
+ if count:
+ return len(res)
+ elif limit:
+ return res[offset: offset + limit]
+ return res
+
+ def copy(self, cr, uid, id, default=None, context=None):
+ if context is None:
+ context = {}
+
+ default = default or {}
+ default['attendee_ids'] = False
+
+ res = super(calendar_event, self).copy(cr, uid, calendar_id2real_id(id), default, context)
+ return res
+
+ def _detach_one_event(self, cr, uid, id, values=dict(), context=None):
+ real_event_id = calendar_id2real_id(id)
+ data = self.read(cr, uid, id, ['date', 'date_deadline', 'rrule', 'duration'])
+
+ if data.get('rrule'):
+ data.update(
+ values,
+ recurrent_id=real_event_id,
+ recurrent_id_date=data.get('date'),
+ rrule_type=False,
+ rrule='',
+ recurrency=False,
+ end_date=datetime.strptime(values.get('date', False) or data.get('date'), "%Y-%m-%d %H:%M:%S") + timedelta(hours=values.get('duration', False) or data.get('duration'))
+ )
+
+ #do not copy the id
+ if data.get('id'):
+ del(data['id'])
+ new_id = self.copy(cr, uid, real_event_id, default=data, context=context)
+ return new_id
+
+ def open_after_detach_event(self, cr, uid, ids, context=None):
+ if context is None:
+ context = {}
+
+ new_id = self._detach_one_event(cr, uid, ids[0], context=context)
+ return {
+ 'type': 'ir.actions.act_window',
+ 'res_model': 'calendar.event',
+ 'view_mode': 'form',
+ 'res_id': new_id,
+ 'target': 'current',
+ 'flags': {'form': {'action_buttons': True, 'options': {'mode': 'edit'}}}
+ }
+
+ def write(self, cr, uid, ids, values, context=None):
+ def _only_changes_to_apply_on_real_ids(field_names):
+ ''' return True if changes are only to be made on the real ids'''
+ for field in field_names:
+ if field in ['date', 'active']:
+ return True
+ return False
+
+ context = context or {}
+
+ if isinstance(ids, (str, int, long)):
+ if len(str(ids).split('-')) == 1:
+ ids = [int(ids)]
+ else:
+ ids = [ids]
+
+ res = False
+ new_id = False
+
+ # Special write of complex IDS
+ for event_id in ids:
+ if len(str(event_id).split('-')) == 1:
+ continue
+
+ ids.remove(event_id)
+ real_event_id = calendar_id2real_id(event_id)
+
+ # if we are setting the recurrency flag to False or if we are only changing fields that
+ # should be only updated on the real ID and not on the virtual (like message_follower_ids):
+ # then set real ids to be updated.
+ if not values.get('recurrency', True) or not _only_changes_to_apply_on_real_ids(values.keys()):
+ ids.append(real_event_id)
+ continue
+ else:
+ data = self.read(cr, uid, event_id, ['date', 'date_deadline', 'rrule', 'duration'])
+ if data.get('rrule'):
+ new_id = self._detach_one_event(cr, uid, event_id, values, context=None)
+
+ res = super(calendar_event, self).write(cr, uid, ids, values, context=context)
+
+ # set end_date for calendar searching
+ if values.get('recurrency', True) and values.get('end_type', 'count') in ('count', unicode('count')) and \
+ (values.get('rrule_type') or values.get('count') or values.get('date') or values.get('date_deadline')):
- for data in self.read(cr, uid, ids, ['date', 'date_deadline', 'recurrency', 'rrule_type', 'count', 'end_type'], context=context):
++ for data in self.read(cr, uid, ids, ['end_date', 'date_deadline', 'recurrency', 'rrule_type', 'count', 'end_type'], context=context):
+ end_date = self._get_recurrency_end_date(data, context=context)
+ super(calendar_event, self).write(cr, uid, [data['id']], {'end_date': end_date}, context=context)
+
+ attendees_create = False
+ if values.get('partner_ids', False):
+ attendees_create = self.create_attendees(cr, uid, ids, context)
+
+ if values.get('date', False) and values.get('active', True):
+ the_id = new_id or (ids and int(ids[0]))
+
+ if attendees_create:
+ attendees_create = attendees_create[the_id]
+ mail_to_ids = list(set(attendees_create['old_attendee_ids']) - set(attendees_create['removed_attendee_ids']))
+ else:
+ mail_to_ids = [att.id for att in self.browse(cr, uid, the_id, context=context).attendee_ids]
+
+ if mail_to_ids:
+ current_user = self.pool['res.users'].browse(cr, uid, uid, context=context)
+ if self.pool['calendar.attendee']._send_mail_to_attendees(cr, uid, mail_to_ids, template_xmlid='calendar_template_meeting_changedate', email_from=current_user.email, context=context):
+ self.message_post(cr, uid, the_id, body=_("A email has been send to specify that the date has been changed !"), subtype="calendar.subtype_invitation", context=context)
+
+ return res or True and False
+
+ def create(self, cr, uid, vals, context=None):
+ if context is None:
+ context = {}
+
+ if not 'user_id' in vals: # Else bug with quick_create when we are filter on an other user
+ vals['user_id'] = uid
+
- if vals.get('recurrency', True) and vals.get('end_type', 'count') in ('count', unicode('count')) and \
- (vals.get('rrule_type') or vals.get('count') or vals.get('date') or vals.get('date_deadline')):
- vals['end_date'] = self._get_recurrency_end_date(vals, context=context)
-
+ res = super(calendar_event, self).create(cr, uid, vals, context=context)
++
++ data = self.read(cr, uid, [res], ['end_date', 'date_deadline', 'recurrency', 'rrule_type', 'count', 'end_type'], context=context)[0]
++ end_date = self._get_recurrency_end_date(data, context=context)
++ self.write(cr, uid, [res], {'end_date': end_date}, context=context)
++
+ self.create_attendees(cr, uid, [res], context=context)
+ return res
+
+ def read_group(self, cr, uid, domain, fields, groupby, offset=0, limit=None, context=None, orderby=False):
+ if not context:
+ context = {}
+
+ if 'date' in groupby:
+ raise osv.except_osv(_('Warning!'), _('Group by date is not supported, use the calendar view instead.'))
+ virtual_id = context.get('virtual_id', True)
+ context.update({'virtual_id': False})
+ res = super(calendar_event, self).read_group(cr, uid, domain, fields, groupby, offset=offset, limit=limit, context=context, orderby=orderby)
+ for result in res:
+ #remove the count, since the value is not consistent with the result of the search when expand the group
+ for groupname in groupby:
+ if result.get(groupname + "_count"):
+ del result[groupname + "_count"]
+ result.get('__context', {}).update({'virtual_id': virtual_id})
+ return res
+
+ def read(self, cr, uid, ids, fields=None, context=None, load='_classic_read'):
+ if context is None:
+ context = {}
+ fields2 = fields and fields[:] or None
+ EXTRAFIELDS = ('class', 'user_id', 'duration', 'date', 'rrule', 'vtimezone')
+ for f in EXTRAFIELDS:
+ if fields and (f not in fields):
+ fields2.append(f)
+
+ if isinstance(ids, (str, int, long)):
+ select = [ids]
+ else:
+ select = ids
+
+ select = map(lambda x: (x, calendar_id2real_id(x)), select)
+ result = []
+
+ real_data = super(calendar_event, self).read(cr, uid, [real_id for calendar_id, real_id in select], fields=fields2, context=context, load=load)
+ real_data = dict(zip([x['id'] for x in real_data], real_data))
+
+ for calendar_id, real_id in select:
+ res = real_data[real_id].copy()
+ ls = calendar_id2real_id(calendar_id, with_date=res and res.get('duration', 0) or 0)
+ if not isinstance(ls, (str, int, long)) and len(ls) >= 2:
+ res['date'] = ls[1]
+ res['date_deadline'] = ls[2]
+ res['id'] = calendar_id
+ result.append(res)
+
+ for r in result:
+ if r['user_id']:
+ user_id = type(r['user_id']) in (tuple, list) and r['user_id'][0] or r['user_id']
+ if user_id == uid:
+ continue
+ if r['class'] == 'private':
+ for f in r.keys():
+ if f not in ('id', 'date', 'date_deadline', 'duration', 'user_id', 'state', 'interval', 'count', 'recurrent_id_date'):
+ if isinstance(r[f], list):
+ r[f] = []
+ else:
+ r[f] = False
+ if f == 'name':
+ r[f] = _('Busy')
+
+ for r in result:
+ for k in EXTRAFIELDS:
+ if (k in r) and (fields and (k not in fields)):
+ del r[k]
+
+ if isinstance(ids, (str, int, long)):
+ return result and result[0] or False
+ return result
+
+ def unlink(self, cr, uid, ids, unlink_level=0, context=None):
+ if not isinstance(ids, list):
+ ids = [ids]
+ res = False
+
+ ids_to_exclure = []
+ ids_to_unlink = []
+
+ # One time moved to google_Calendar, we can specify, if not in google, and not rec or get_inst = 0, we delete it
+ for event_id in ids:
+ if unlink_level == 1 and len(str(event_id).split('-')) == 1: # if ID REAL
+ if self.browse(cr, uid, event_id).recurrent_id:
+ ids_to_exclure.append(event_id)
+ else:
+ ids_to_unlink.append(event_id)
+ else:
+ ids_to_exclure.append(event_id)
+
+ if ids_to_unlink:
+ res = super(calendar_event, self).unlink(cr, uid, ids_to_unlink, context=context)
+
+ if ids_to_exclure:
+ for id_to_exclure in ids_to_exclure:
+ res = self.write(cr, uid, id_to_exclure, {'active': False}, context=context)
+
+ return res
+
+
+class mail_message(osv.Model):
+ _inherit = "mail.message"
+
+ def search(self, cr, uid, args, offset=0, limit=0, order=None, context=None, count=False):
+ '''
+ convert the search on real ids in the case it was asked on virtual ids, then call super()
+ '''
+ for index in range(len(args)):
+ if args[index][0] == "res_id" and isinstance(args[index][2], str):
+ args[index][2] = get_real_ids(args[index][2])
+ return super(mail_message, self).search(cr, uid, args, offset=offset, limit=limit, order=order, context=context, count=count)
+
+ def _find_allowed_model_wise(self, cr, uid, doc_model, doc_dict, context=None):
+ if doc_model == 'calendar.event':
+ order = context.get('order', self._order)
+ for virtual_id in self.pool[doc_model].get_recurrent_ids(cr, uid, doc_dict.keys(), [], order=order, context=context):
+ doc_dict.setdefault(virtual_id, doc_dict[get_real_ids(virtual_id)])
+ return super(mail_message, self)._find_allowed_model_wise(cr, uid, doc_model, doc_dict, context=context)
+
+
+class ir_attachment(osv.Model):
+ _inherit = "ir.attachment"
+
+ def search(self, cr, uid, args, offset=0, limit=0, order=None, context=None, count=False):
+ '''
+ convert the search on real ids in the case it was asked on virtual ids, then call super()
+ '''
+ for index in range(len(args)):
+ if args[index][0] == "res_id" and isinstance(args[index][2], str):
+ args[index][2] = get_real_ids(args[index][2])
+ return super(ir_attachment, self).search(cr, uid, args, offset=offset, limit=limit, order=order, context=context, count=count)
+
+ def write(self, cr, uid, ids, vals, context=None):
+ '''
+ when posting an attachment (new or not), convert the virtual ids in real ids.
+ '''
+ if isinstance(vals.get('res_id'), str):
+ vals['res_id'] = get_real_ids(vals.get('res_id'))
+ return super(ir_attachment, self).write(cr, uid, ids, vals, context=context)
+
+
+class ir_http(osv.AbstractModel):
+ _inherit = 'ir.http'
+
+ def _auth_method_calendar(self):
+ token = request.params['token']
+ db = request.params['db']
+
+ registry = openerp.modules.registry.RegistryManager.get(db)
+ attendee_pool = registry.get('calendar.attendee')
+ error_message = False
+ with registry.cursor() as cr:
+ attendee_id = attendee_pool.search(cr, openerp.SUPERUSER_ID, [('access_token', '=', token)])
+ if not attendee_id:
+ error_message = """Invalid Invitation Token."""
+ elif request.session.uid and request.session.login != 'anonymous':
+ # if valid session but user is not match
+ attendee = attendee_pool.browse(cr, openerp.SUPERUSER_ID, attendee_id[0])
+ user = registry.get('res.users').browse(cr, openerp.SUPERUSER_ID, request.session.uid)
+ if attendee.partner_id.id != user.partner_id.id:
+ error_message = """Invitation cannot be forwarded via email. This event/meeting belongs to %s and you are logged in as %s. Please ask organizer to add you.""" % (attendee.email, user.email)
+
+ if error_message:
+ raise BadRequest(error_message)
+ return True
+
+
+class invite_wizard(osv.osv_memory):
+ _inherit = 'mail.wizard.invite'
+
+ def default_get(self, cr, uid, fields, context=None):
+ '''
+ in case someone clicked on 'invite others' wizard in the followers widget, transform virtual ids in real ids
+ '''
+ result = super(invite_wizard, self).default_get(cr, uid, fields, context=context)
+ if 'res_id' in result:
+ result['res_id'] = get_real_ids(result['res_id'])
+ return result