[IMP] import_ggole : imprt calendar ebent
bzr revid: jam@tinyerp.com-20110607123228-d2mbz2klb5a48u08
This commit is contained in:
parent
28dd15a694
commit
568f796a4e
|
@ -37,7 +37,7 @@ class crm_installer(osv.osv_memory):
|
|||
'thunderbird': fields.boolean('Thunderbird', help="Allows you to link your e-mail to OpenERP's documents. You can attach it to any existing one in OpenERP or create a new one."),
|
||||
'outlook': fields.boolean('MS-Outlook', help="Allows you to link your e-mail to OpenERP's documents. You can attach it to any existing one in OpenERP or create a new one."),
|
||||
'wiki_sale_faq': fields.boolean('Sale FAQ', help="Helps you manage wiki pages for Frequently Asked Questions on Sales Application."),
|
||||
'import_google_contact': fields.boolean('Google Contacts', help="Imports contacts from your google account."),
|
||||
'import_google': fields.boolean('Google Contacts', help="Imports contacts from your google account."),
|
||||
}
|
||||
|
||||
def fields_view_get(self, cr, uid, view_id=None, view_type='form', context=None, toolbar=False, submenu=False):
|
||||
|
|
|
@ -37,7 +37,7 @@
|
|||
<separator string="Synchronization" colspan="4" />
|
||||
<field name="crm_caldav" />
|
||||
<field name="fetchmail" />
|
||||
<field name="import_google_contact"/>
|
||||
<field name="import_google"/>
|
||||
</group>
|
||||
<group colspan="2" col="2">
|
||||
<separator string="Plug-In" colspan="4" />
|
||||
|
|
|
@ -37,7 +37,7 @@
|
|||
'test': [
|
||||
# 'test/test_sync_google_contact_import_partner.yml',
|
||||
# 'test/test_sync_google_contact_import_address.yml',
|
||||
# 'test/test_sync_google_calendar.yml',
|
||||
'test/test_sync_google_calendar.yml',
|
||||
],
|
||||
'installable': True,
|
||||
'active': False,
|
||||
|
|
|
@ -5,7 +5,7 @@
|
|||
I create a record for the gmail account for which I want to import the contacts.
|
||||
-
|
||||
!record {model: google.login, id: google_login_contact_id0}:
|
||||
user: testmail.openerp
|
||||
user: testmail.openerp@gmail.com
|
||||
password: openerptiny
|
||||
- |
|
||||
I login into that account.
|
||||
|
@ -33,4 +33,4 @@
|
|||
model_ids = model_obj.search(cr, uid, [('res_id','in',meeting_ids),('model','=','crm.meeting'),('module','=','sync_google_calendar')])
|
||||
assert model_ids, 'Meetings not created !'
|
||||
|
||||
|
||||
|
||||
|
|
|
@ -19,12 +19,6 @@
|
|||
#
|
||||
##############################################################################
|
||||
import datetime
|
||||
import dateutil
|
||||
from dateutil.parser import *
|
||||
from pytz import timezone
|
||||
import time
|
||||
import re
|
||||
import urllib
|
||||
|
||||
try:
|
||||
import gdata
|
||||
|
@ -35,17 +29,16 @@ except ImportError:
|
|||
from osv import fields
|
||||
from osv import osv
|
||||
from tools.translate import _
|
||||
import tools
|
||||
from import_google import import_contact
|
||||
|
||||
class google_login_contact(osv.osv_memory):
|
||||
_inherit = 'google.login'
|
||||
_name = 'google.login.contact'
|
||||
|
||||
|
||||
def _get_next_action(self, cr, uid, context):
|
||||
data_obj = self.pool.get('ir.model.data')
|
||||
data_id = data_obj._get_id(cr, uid, 'import_google', 'view_synchronize_google_contact_import_form')
|
||||
|
||||
|
||||
view_id = False
|
||||
if data_id:
|
||||
view_id = data_obj.browse(cr, uid, data_id, context=context).res_id
|
||||
|
@ -122,7 +115,7 @@ class synchronize_google_contact(osv.osv_memory):
|
|||
# return partner_ids[0]
|
||||
# return False
|
||||
|
||||
def import_contact(self, cr, uid, ids, context=None):
|
||||
def import_contact_all(self, cr, uid, ids, context=None):
|
||||
tables = ['contact']
|
||||
obj = self.browse(cr, uid, ids, context=context)[0]
|
||||
user_obj = self.pool.get('res.users').browse(cr, uid, uid)
|
||||
|
@ -282,119 +275,17 @@ class synchronize_google_contact(osv.osv_memory):
|
|||
# return {'type': 'ir.actions.act_window_close'}
|
||||
|
||||
synchronize_google_contact()
|
||||
def _get_tinydates(self, stime, etime):
|
||||
stime = dateutil.parser.parse(stime)
|
||||
etime = dateutil.parser.parse(etime)
|
||||
try:
|
||||
au_dt = au_tz.normalize(stime.astimezone(au_tz))
|
||||
timestring = datetime.datetime(*au_dt.timetuple()[:6]).strftime('%Y-%m-%d %H:%M:%S')
|
||||
au_dt = au_tz.normalize(etime.astimezone(au_tz))
|
||||
timestring_end = datetime.datetime(*au_dt.timetuple()[:6]).strftime('%Y-%m-%d %H:%M:%S')
|
||||
except:
|
||||
timestring = datetime.datetime(*stime.timetuple()[:6]).strftime('%Y-%m-%d %H:%M:%S')
|
||||
timestring_end = datetime.datetime(*etime.timetuple()[:6]).strftime('%Y-%m-%d %H:%M:%S')
|
||||
return (timestring, timestring_end)
|
||||
|
||||
|
||||
def _get_rules(self, datas):
|
||||
new_val = {}
|
||||
if datas['FREQ'] == 'WEEKLY' and datas.get('BYDAY'):
|
||||
for day in datas['BYDAY'].split(','):
|
||||
new_val[day.lower()] = True
|
||||
datas.pop('BYDAY')
|
||||
|
||||
if datas.get('UNTIL'):
|
||||
until = parser.parse(''.join((re.compile('\d')).findall(datas.get('UNTIL'))))
|
||||
new_val['end_date'] = until.strftime('%Y-%m-%d')
|
||||
new_val['end_type'] = 'end_date'
|
||||
datas.pop('UNTIL')
|
||||
|
||||
if datas.get('COUNT'):
|
||||
new_val['count'] = datas.get('COUNT')
|
||||
new_val['end_type'] = 'count'
|
||||
datas.pop('COUNT')
|
||||
|
||||
if datas.get('INTERVAL'):
|
||||
new_val['interval'] = datas.get('INTERVAL')
|
||||
else:
|
||||
new_val['interval'] = 1
|
||||
|
||||
if datas.get('BYMONTHDAY'):
|
||||
new_val['day'] = datas.get('BYMONTHDAY')
|
||||
datas.pop('BYMONTHDAY')
|
||||
new_val['select1'] = 'date'
|
||||
|
||||
if datas.get('BYDAY'):
|
||||
d = datas.get('BYDAY')
|
||||
if '-' in d:
|
||||
new_val['byday'] = d[:2]
|
||||
new_val['week_list'] = d[2:4].upper()
|
||||
else:
|
||||
new_val['byday'] = d[:1]
|
||||
new_val['week_list'] = d[1:3].upper()
|
||||
new_val['select1'] = 'day'
|
||||
|
||||
if datas.get('BYMONTH'):
|
||||
new_val['month_list'] = datas.get('BYMONTH')
|
||||
datas.pop('bymonth')
|
||||
return new_val
|
||||
|
||||
|
||||
def _get_repeat_status(self, str_google):
|
||||
rrule = str_google[str_google.find('FREQ'):str_google.find('\nBEGIN')]
|
||||
status = {}
|
||||
for rule in rrule.split(';'):
|
||||
status[rule.split('=')[0]] = rule.split('=')[-1:] and rule.split('=')[-1:][0] or ''
|
||||
rules = _get_rules(self, status)
|
||||
if status.get('FREQ') == 'WEEKLY':
|
||||
status.update({'rrule_type': 'weekly'})
|
||||
status.pop('FREQ')
|
||||
elif status.get('FREQ') == 'DAILY':
|
||||
status.update({'rrule_type': 'daily'})
|
||||
status.pop('FREQ')
|
||||
elif status.get('FREQ') == 'MONTHLY':
|
||||
status.update({'rrule_type': 'monthly'})
|
||||
status.pop('FREQ')
|
||||
elif status.get('FREQ') == 'YEARLY':
|
||||
status.update({'rrule_type': 'yearly'})
|
||||
status.pop('FREQ')
|
||||
status.update(rules)
|
||||
return status
|
||||
|
||||
|
||||
def _get_repeat_dates(self, x):
|
||||
if len(x) > 4:
|
||||
if x[3].startswith('BY'):
|
||||
zone_time = x[4].split('+')[-1:][0].split(':')[0][:4]
|
||||
else:
|
||||
zone_time = x[3].split('+')[-1:][0].split(':')[0][:4]
|
||||
else:
|
||||
zone_time = x[2].split('+')[-1:][0].split(':')[0][:4]
|
||||
tz_format = zone_time[:2]+':'+zone_time[2:]
|
||||
repeat_start = x[1].split('\n')[0].split(':')[1]
|
||||
repeat_end = x[2].split('\n')[0].split(':')[1]
|
||||
o = repeat_start.split('T')
|
||||
repeat_start = str(o[0][:4]) + '-' + str(o[0][4:6]) + '-' + str(o[0][6:8])
|
||||
if len(o) == 2:
|
||||
repeat_start += ' ' + str(o[1][:2]) + ':' + str(o[1][2:4]) + ':' + str(o[1][4:6])
|
||||
else:
|
||||
repeat_start += ' ' + '00' + ':' + '00' + ':' + '00'
|
||||
p = repeat_end.split('T')
|
||||
repeat_end = str(p[0][:4]) + '-' + str(p[0][4:6]) + '-' + str(p[0][6:8])
|
||||
if len(p) == 2:
|
||||
repeat_end += ' ' + str(p[1][:2]) + ':' + str(p[1][2:4]) + ':' + str(p[1][4:6])
|
||||
else:
|
||||
repeat_end += ' ' + '00' + ':' + '00' + ':' + '00'
|
||||
return (repeat_start, repeat_end, tz_format)
|
||||
|
||||
class google_login_calendar(osv.osv_memory):
|
||||
"""Gdata Login Object for Google Calendar Import"""
|
||||
_inherit = 'google.login'
|
||||
_name = 'google.login.calendar'
|
||||
|
||||
|
||||
def _get_next_action(self, cr, uid, context):
|
||||
data_obj = self.pool.get('ir.model.data')
|
||||
data_id = data_obj._get_id(cr, uid, 'import_google', 'view_synchronize_google_calendar_import_form')
|
||||
|
||||
data_id = data_obj._get_id(cr, uid, 'import_google', 'view_synchronize_google_calendar_import_form')
|
||||
|
||||
view_id = False
|
||||
if data_id:
|
||||
view_id = data_obj.browse(cr, uid, data_id, context=context).res_id
|
||||
|
@ -412,7 +303,11 @@ class google_login_calendar(osv.osv_memory):
|
|||
return value
|
||||
google_login_calendar()
|
||||
|
||||
|
||||
class synchronize_google_calendar_events(osv.osv_memory):
|
||||
"""
|
||||
Wizard to initiate import specific calendar or all calendars
|
||||
"""
|
||||
_name = 'synchronize.google.calendar'
|
||||
|
||||
def _get_calendars(self, cr, uid, context=None):
|
||||
|
@ -421,9 +316,7 @@ class synchronize_google_calendar_events(osv.osv_memory):
|
|||
res = []
|
||||
try:
|
||||
gd_client = google.google_login(user_obj.gmail_user, user_obj.gmail_password, type='calendar')
|
||||
|
||||
calendars = gd_client.GetAllCalendarsFeed()
|
||||
|
||||
for cal in calendars.entry:
|
||||
res.append((cal.id.text, cal.title.text))
|
||||
except Exception, e:
|
||||
|
@ -439,105 +332,32 @@ class synchronize_google_calendar_events(osv.osv_memory):
|
|||
'calendar_name': 'all',
|
||||
}
|
||||
|
||||
def get_events(self, cr, uid, event_feed, context=None):
|
||||
meeting_obj = self.pool.get('crm.meeting')
|
||||
categ_obj = self.pool.get('crm.case.categ')
|
||||
model_obj = self.pool.get('ir.model.data')
|
||||
object_id = self.pool.get('ir.model').search(cr, uid, [('model', '=', 'crm.meeting')])
|
||||
meeting_ids = []
|
||||
categ_id = categ_obj.search(cr, uid, [('name', '=', event_feed.title.text), ('object_id', '=', object_id and object_id[0]), ('user_id', '=', uid)])
|
||||
if not categ_id:
|
||||
categ_id.append(categ_obj.create(cr, uid, {'name': event_feed.title.text,
|
||||
'object_id': object_id and object_id[0],
|
||||
'user_id': uid}))
|
||||
if 'tz' in context and context['tz']:
|
||||
time_zone = context['tz']
|
||||
else:
|
||||
time_zone = tools.get_server_timezone()
|
||||
au_tz = timezone(time_zone)
|
||||
|
||||
for feed in event_feed.entry:
|
||||
google_id = feed.id.text
|
||||
model_data = {
|
||||
'name': google_id,
|
||||
'model': 'crm.meeting',
|
||||
'module': 'sync_google_calendar',
|
||||
'noupdate': True,
|
||||
}
|
||||
vals = {
|
||||
'name': feed.title.text or '(No title)',
|
||||
'description': feed.content.text,
|
||||
'categ_id': categ_id and categ_id[0],
|
||||
}
|
||||
if feed.when:
|
||||
timestring, timestring_end = _get_tinydates(self, feed.when[0].start_time, feed.when[0].end_time)
|
||||
else:
|
||||
x = feed.recurrence.text.split(';')
|
||||
repeat_status = _get_repeat_status(self, feed.recurrence.text)
|
||||
repeat_start, repeat_end, zone_time = _get_repeat_dates(self, x)
|
||||
timestring = time.strftime('%Y-%m-%d %H:%M:%S', time.strptime(repeat_start, "%Y-%m-%d %H:%M:%S"))
|
||||
timestring_end = time.strftime('%Y-%m-%d %H:%M:%S', time.strptime(repeat_end, "%Y-%m-%d %H:%M:%S"))
|
||||
if repeat_status:
|
||||
repeat_status.update({'recurrency': True})
|
||||
vals.update(repeat_status)
|
||||
|
||||
vals.update({'date': timestring, 'date_deadline': timestring_end})
|
||||
data_ids = model_obj.search(cr, uid, [('model', '=', 'crm.meeting'), ('name', '=', google_id)])
|
||||
if data_ids:
|
||||
res_id = model_obj.browse(cr, uid, data_ids[0], context=context).res_id
|
||||
meeting = meeting_obj.browse(cr, uid, res_id, context=context)
|
||||
google_updated = feed.updated.text
|
||||
utime = dateutil.parser.parse(google_updated)
|
||||
au_dt = au_tz.normalize(utime.astimezone(au_tz))
|
||||
updated_dt = datetime.datetime(*au_dt.timetuple()[:6]).strftime('%Y-%m-%d %H:%M:%S')
|
||||
if meeting.write_date < updated_dt:
|
||||
meeting_ids.append(res_id)
|
||||
meeting_obj.write(cr, uid, [res_id], vals, context=context)
|
||||
else:
|
||||
res_id = meeting_obj.create(cr, uid, vals, context=context)
|
||||
meeting_ids.append(res_id)
|
||||
model_data.update({'res_id': res_id})
|
||||
model_obj.create(cr, uid, model_data, context=context)
|
||||
return meeting_ids
|
||||
|
||||
def import_calendar_events(self, cr, uid, ids, context=None):
|
||||
obj = self.browse(cr, uid, ids, context=context)[0]
|
||||
if context == None:
|
||||
context = {}
|
||||
if not ids:
|
||||
return {'type': 'ir.actions.act_window_close'}
|
||||
|
||||
table = ['Events']
|
||||
user_obj = self.pool.get('res.users').browse(cr, uid, uid)
|
||||
gmail_user = user_obj.gmail_user
|
||||
gamil_pwd = user_obj.gmail_password
|
||||
|
||||
google = self.pool.get('google.login.calendar')
|
||||
gd_client = google.google_login(gmail_user, gamil_pwd, type='calendar')
|
||||
|
||||
if not gmail_user or not gamil_pwd:
|
||||
raise osv.except_osv(_('Error'), _("Please specify the user and password !"))
|
||||
|
||||
meetings = []
|
||||
if obj.calendar_name != 'all':
|
||||
events_query = gdata.calendar.service.CalendarEventQuery(user=urllib.unquote(obj.calendar_name.split('/')[~0]))
|
||||
events_query.start_index = 1
|
||||
events_query.max_results = 1000
|
||||
event_feed = gd_client.GetCalendarEventFeed(events_query.ToUri())
|
||||
meetings.append(self.get_events(cr, uid, event_feed, context=context))
|
||||
else:
|
||||
calendars = map(lambda x: x[0], [cal for cal in self._get_calendars(cr, uid, context) if cal[0] != 'all'])
|
||||
for cal in calendars:
|
||||
events_query = gdata.calendar.service.CalendarEventQuery(user=urllib.unquote(cal.split('/')[~0]))
|
||||
events_query.start_index = 1
|
||||
events_query.max_results = 1000
|
||||
event_feed = gd_client.GetCalendarEventFeed(events_query.ToUri())
|
||||
meetings.append(self.get_events(cr, uid, event_feed, context=context))
|
||||
|
||||
meeting_ids = []
|
||||
for meeting in meetings:
|
||||
meeting_ids += meeting
|
||||
|
||||
return {
|
||||
'type': 'ir.actions.act_window_close',
|
||||
}
|
||||
gmail_pwd = user_obj.gmail_password
|
||||
if not gmail_user or not gmail_pwd:
|
||||
raise osv.except_osv(_('Error'), _("Invalid login detail !\n Specify Username/Password."))
|
||||
current_rec = self.browse(cr, uid, ids, context=context)
|
||||
calendars = False
|
||||
for rec in current_rec:
|
||||
if rec.calendar_name != 'all':
|
||||
calendars = [rec.calendar_name]
|
||||
else:
|
||||
calendars = map(lambda x: x[0], [cal for cal in self._get_calendars(cr, uid, context) if cal[0] != 'all'])
|
||||
context.update({'user': gmail_user,
|
||||
'password': gmail_pwd,
|
||||
'calendars': calendars,
|
||||
'events': True})
|
||||
imp = import_contact(self, cr, uid, 'import_google', "import_google_calendar", [gmail_user], context)
|
||||
imp.set_table_list(table)
|
||||
imp.start()
|
||||
return {}
|
||||
|
||||
synchronize_google_calendar_events()
|
||||
# vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4:
|
||||
|
|
|
@ -1,88 +1,257 @@
|
|||
# -*- coding: utf-8 -*-
|
||||
##############################################################################
|
||||
#
|
||||
# OpenERP, Open Source Management Solution
|
||||
# Copyright (C) 2004-2010 Tiny SPRL (<http://tiny.be>).
|
||||
#
|
||||
# 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 re
|
||||
import urllib
|
||||
import dateutil
|
||||
from dateutil import *
|
||||
from pytz import timezone
|
||||
from datetime import datetime
|
||||
import time
|
||||
try:
|
||||
import gdata
|
||||
import gdata.contacts.service
|
||||
import gdata.calendar.service
|
||||
import gdata.contacts
|
||||
except ImportError:
|
||||
raise osv.except_osv(_('Google Contacts Import Error!'), _('Please install gdata-python-client from http://code.google.com/p/gdata-python-client/downloads/list'))
|
||||
from import_base.import_framework import *
|
||||
from import_base.mapper import *
|
||||
|
||||
|
||||
class import_contact(import_framework):
|
||||
|
||||
gd_client = False
|
||||
gd_calendar_client = False
|
||||
calendars = False
|
||||
DATETIME_FORMAT = "%Y-%m-%d %H:%M:%S"
|
||||
TABLE_CONTACT = 'contact'
|
||||
TABLE_MEETING = 'Event'
|
||||
TABLE_EVENT = 'Events'
|
||||
|
||||
def initialize(self):
|
||||
self.gd_client = gdata.contacts.service.ContactsService()
|
||||
self.gd_client.ClientLogin(self.context.get('user', False),self.context.get('password', False))
|
||||
|
||||
if 'events' in self.context and self.context.get('events'):
|
||||
self.gd_calendar_client = gdata.calendar.service.CalendarService()
|
||||
self.gd_calendar_client.ClientLogin(self.context.get('user', False),self.context.get('password', False))
|
||||
self.calendars = self.context.get('calendars')
|
||||
|
||||
def get_mapping(self):
|
||||
return {
|
||||
self.TABLE_CONTACT: self.get_contact_mapping(),
|
||||
self.TABLE_EVENT: self.get_event_mapping(),
|
||||
}
|
||||
def _retreive_data(self,entry):
|
||||
if entry:
|
||||
data = {}
|
||||
data['id'] = entry.id.text
|
||||
name = tools.ustr(entry.title.text)
|
||||
if name == "None":
|
||||
name = entry.email[0].address
|
||||
data['name'] = name
|
||||
emails = ','.join(email.address for email in entry.email)
|
||||
data['email'] = emails
|
||||
if entry.organization:
|
||||
if entry.organization.org_name:
|
||||
data.update({'company': entry.organization.org_name.text})
|
||||
if entry.organization.org_title:
|
||||
data.update ({'function': entry.organization.org_title.text})
|
||||
|
||||
if entry.phone_number:
|
||||
for phone in entry.phone_number:
|
||||
if phone.rel == gdata.contacts.REL_WORK:
|
||||
data['phone'] = phone.text
|
||||
else:
|
||||
data['phone'] = False
|
||||
if phone.rel == gdata.contacts.PHONE_MOBILE:
|
||||
data['mobile'] = phone.text
|
||||
else :
|
||||
data['mobile'] = False
|
||||
if phone.rel == gdata.contacts.PHONE_WORK_FAX:
|
||||
data['fax'] = phone.text
|
||||
else :
|
||||
data['fax'] = False
|
||||
|
||||
data.update({
|
||||
'mobile': data.has_key('mobile') and data['mobile'] or False,
|
||||
'phone':data.has_key('phone') and data['phone'] or False,
|
||||
'fax':data.has_key('fax') and data['fax'] or False,
|
||||
'type':'contact',
|
||||
'id_new':data['id'] + '_data_'+ name,
|
||||
})
|
||||
|
||||
|
||||
address = {
|
||||
'name': 'name',
|
||||
'type': 'type',
|
||||
'phone': 'phone',
|
||||
'mobile': 'mobile',
|
||||
'email': 'email',
|
||||
'fax': 'fax',
|
||||
}
|
||||
return self.import_object_mapping(address,data, 'res.partner.address', 'res.partner.address',data['id_new'], self.DO_NOT_FIND_DOMAIN)
|
||||
|
||||
def get_data(self, table):
|
||||
val = {
|
||||
self.TABLE_EVENT: self.get_events(),
|
||||
}
|
||||
return val.get(table)
|
||||
|
||||
|
||||
def get_contact_mapping(self):
|
||||
contact = self.gd_client.GetContactsFeed()
|
||||
while contact:
|
||||
val = []
|
||||
for entry in contact.entry:
|
||||
val = self._retreive_data(entry)
|
||||
#val.append(self._retreive_data(entry))
|
||||
return {
|
||||
'model': 'res.partner.address',
|
||||
'import' : False,
|
||||
'dependencies': [],
|
||||
#'hook': self._retreive_data,
|
||||
'map': val
|
||||
}
|
||||
def _get_tinydates(self, stime, etime):
|
||||
stime = dateutil.parser.parse(stime)
|
||||
etime = dateutil.parser.parse(etime)
|
||||
try:
|
||||
au_dt = au_tz.normalize(stime.astimezone(au_tz))
|
||||
timestring = datetime.datetime(*au_dt.timetuple()[:6]).strftime('%Y-%m-%d %H:%M:%S')
|
||||
au_dt = au_tz.normalize(etime.astimezone(au_tz))
|
||||
timestring_end = datetime.datetime(*au_dt.timetuple()[:6]).strftime('%Y-%m-%d %H:%M:%S')
|
||||
except:
|
||||
timestring = datetime.datetime(*stime.timetuple()[:6]).strftime('%Y-%m-%d %H:%M:%S')
|
||||
timestring_end = datetime.datetime(*etime.timetuple()[:6]).strftime('%Y-%m-%d %H:%M:%S')
|
||||
return (timestring, timestring_end)
|
||||
|
||||
def _get_rules(self, datas):
|
||||
new_val = {}
|
||||
if datas['FREQ'] == 'WEEKLY' and datas.get('BYDAY'):
|
||||
for day in datas['BYDAY'].split(','):
|
||||
new_val[day.lower()] = True
|
||||
datas.pop('BYDAY')
|
||||
|
||||
if datas.get('UNTIL'):
|
||||
until = parser.parse(''.join((re.compile('\d')).findall(datas.get('UNTIL'))))
|
||||
new_val['end_date'] = until.strftime('%Y-%m-%d')
|
||||
new_val['end_type'] = 'end_date'
|
||||
datas.pop('UNTIL')
|
||||
|
||||
if datas.get('COUNT'):
|
||||
new_val['count'] = datas.get('COUNT')
|
||||
new_val['end_type'] = 'count'
|
||||
datas.pop('COUNT')
|
||||
|
||||
if datas.get('INTERVAL'):
|
||||
new_val['interval'] = datas.get('INTERVAL')
|
||||
else:
|
||||
new_val['interval'] = 1
|
||||
|
||||
if datas.get('BYMONTHDAY'):
|
||||
new_val['day'] = datas.get('BYMONTHDAY')
|
||||
datas.pop('BYMONTHDAY')
|
||||
new_val['select1'] = 'date'
|
||||
|
||||
if datas.get('BYDAY'):
|
||||
d = datas.get('BYDAY')
|
||||
if '-' in d:
|
||||
new_val['byday'] = d[:2]
|
||||
new_val['week_list'] = d[2:4].upper()
|
||||
else:
|
||||
new_val['byday'] = d[:1]
|
||||
new_val['week_list'] = d[1:3].upper()
|
||||
new_val['select1'] = 'day'
|
||||
|
||||
if datas.get('BYMONTH'):
|
||||
new_val['month_list'] = datas.get('BYMONTH')
|
||||
datas.pop('bymonth')
|
||||
return new_val
|
||||
|
||||
|
||||
def _get_repeat_status(self, str_google):
|
||||
rrule = str_google[str_google.find('FREQ'):str_google.find('\nBEGIN')]
|
||||
status = {}
|
||||
for rule in rrule.split(';'):
|
||||
status[rule.split('=')[0]] = rule.split('=')[-1:] and rule.split('=')[-1:][0] or ''
|
||||
rules = self._get_rules(status)
|
||||
if status.get('FREQ') == 'WEEKLY':
|
||||
status.update({'rrule_type': 'weekly'})
|
||||
status.pop('FREQ')
|
||||
elif status.get('FREQ') == 'DAILY':
|
||||
status.update({'rrule_type': 'daily'})
|
||||
status.pop('FREQ')
|
||||
elif status.get('FREQ') == 'MONTHLY':
|
||||
status.update({'rrule_type': 'monthly'})
|
||||
status.pop('FREQ')
|
||||
elif status.get('FREQ') == 'YEARLY':
|
||||
status.update({'rrule_type': 'yearly'})
|
||||
status.pop('FREQ')
|
||||
status.update(rules)
|
||||
return status
|
||||
|
||||
|
||||
def _get_repeat_dates(self, x):
|
||||
if len(x) > 4:
|
||||
if x[3].startswith('BY'):
|
||||
zone_time = x[4].split('+')[-1:][0].split(':')[0][:4]
|
||||
else:
|
||||
zone_time = x[3].split('+')[-1:][0].split(':')[0][:4]
|
||||
else:
|
||||
zone_time = x[2].split('+')[-1:][0].split(':')[0][:4]
|
||||
tz_format = zone_time[:2]+':'+zone_time[2:]
|
||||
repeat_start = x[1].split('\n')[0].split(':')[1]
|
||||
repeat_end = x[2].split('\n')[0].split(':')[1]
|
||||
o = repeat_start.split('T')
|
||||
repeat_start = str(o[0][:4]) + '-' + str(o[0][4:6]) + '-' + str(o[0][6:8])
|
||||
if len(o) == 2:
|
||||
repeat_start += ' ' + str(o[1][:2]) + ':' + str(o[1][2:4]) + ':' + str(o[1][4:6])
|
||||
else:
|
||||
repeat_start += ' ' + '00' + ':' + '00' + ':' + '00'
|
||||
p = repeat_end.split('T')
|
||||
repeat_end = str(p[0][:4]) + '-' + str(p[0][4:6]) + '-' + str(p[0][6:8])
|
||||
if len(p) == 2:
|
||||
repeat_end += ' ' + str(p[1][:2]) + ':' + str(p[1][2:4]) + ':' + str(p[1][4:6])
|
||||
else:
|
||||
repeat_end += ' ' + '00' + ':' + '00' + ':' + '00'
|
||||
return (repeat_start, repeat_end, tz_format)
|
||||
|
||||
def get_events(self):
|
||||
if 'tz' in self.context and self.context['tz']:
|
||||
time_zone = self.context['tz']
|
||||
else:
|
||||
time_zone = tools.get_server_timezone()
|
||||
au_tz = timezone(time_zone)
|
||||
event_vals = []
|
||||
for cal in self.calendars:
|
||||
events_query = gdata.calendar.service.CalendarEventQuery(user=urllib.unquote(cal.split('/')[~0]))
|
||||
events_query.start_index = 1
|
||||
events_query.max_results = 1000
|
||||
event_feed = self.gd_calendar_client.GetCalendarEventFeed(events_query.ToUri())
|
||||
for feed in event_feed.entry:
|
||||
event = {
|
||||
'recurrency': "0",
|
||||
'end_date' : False,
|
||||
'end_type' : False,
|
||||
'byday': 0,
|
||||
'count' : 0,
|
||||
'interval': 1,
|
||||
'day': False,
|
||||
'select1': False,
|
||||
'week_list': "",
|
||||
'month_list': False,
|
||||
'rrule_type': False,
|
||||
}
|
||||
|
||||
timestring = timestring_end = datetime.datetime.now().strftime(self.DATETIME_FORMAT)
|
||||
if feed.when:
|
||||
timestring, timestring_end = self._get_tinydates(feed.when[0].start_time, feed.when[0].end_time)
|
||||
else:
|
||||
x = feed.recurrence.text.split(';')
|
||||
repeat_status = self._get_repeat_status(feed.recurrence.text)
|
||||
repeat_start, repeat_end, zone_time = self._get_repeat_dates(x)
|
||||
timestring = time.strftime('%Y-%m-%d %H:%M:%S', time.strptime(repeat_start, "%Y-%m-%d %H:%M:%S"))
|
||||
timestring_end = time.strftime('%Y-%m-%d %H:%M:%S', time.strptime(repeat_end, "%Y-%m-%d %H:%M:%S"))
|
||||
if repeat_status:
|
||||
repeat_status.update({'recurrency': True})
|
||||
event.update(repeat_status)
|
||||
|
||||
event.update({'id' : feed.id.text,
|
||||
'DateStart': timestring,
|
||||
'DateEnd': timestring_end,
|
||||
'Category':event_feed.title.text,
|
||||
'Name': feed.title.text or 'No title',
|
||||
'Description': feed.content.text,
|
||||
})
|
||||
event_vals.append(event)
|
||||
return event_vals
|
||||
|
||||
|
||||
def get_event_category(self, val, name):
|
||||
fields = ['name', 'object_id']
|
||||
nameid = 'event_category_'+name
|
||||
data = [name, 'crm.meeting']
|
||||
return self.import_object(fields, data, 'crm.case.categ', "crm_case_categ", nameid, [('name', 'ilike', name)])
|
||||
|
||||
def get_event(self, val):
|
||||
if val.get("recurrency"):
|
||||
val.update({"recurrency": "1"})
|
||||
return val
|
||||
|
||||
def get_event_mapping(self):
|
||||
return {
|
||||
'model': 'crm.meeting',
|
||||
'hook': self.get_event,
|
||||
'map': {
|
||||
'id': 'id',
|
||||
'name': 'Name',
|
||||
'description': 'Description',
|
||||
'email_from': 'Email',
|
||||
'date': 'DateStart',
|
||||
'date_deadline': 'DateEnd',
|
||||
'categ_id/id': call(self.get_event_category, value('Category')),
|
||||
# 'allday': map_val('IsAllDayEvent', self.boolean_map),
|
||||
'recurrency': 'recurrency',
|
||||
'end_date' : 'end_date',
|
||||
'end_type' : 'end_type',
|
||||
'byday':'byday',
|
||||
'count' : 'count',
|
||||
'interval': 'interval',
|
||||
'day': 'day',
|
||||
'select1': 'date',
|
||||
'week_list': 'week_list',
|
||||
'month_list':'month_list',
|
||||
'rrule_type': 'rrule_type',
|
||||
}
|
||||
}
|
||||
|
||||
|
|
Loading…
Reference in New Issue