amdi-odoo
c2f337779b
Add a helper method to know when the user calendar is fully synced to Google Calendar, meaning when API calls can correctly be performed. The user calendar is considered fully synced if the Google API is correctly configured in the settings (Client ID / Secret) and the user has synced its calendar with its Google Account. Task-3285601 Part-of: odoo/odoo#133921
159 lines
8.0 KiB
Python
159 lines
8.0 KiB
Python
# -*- coding: utf-8 -*-
|
|
# Part of Odoo. See LICENSE file for full copyright and licensing details.
|
|
|
|
import logging
|
|
|
|
|
|
from odoo import api, fields, models, Command
|
|
from odoo.addons.google_calendar.utils.google_calendar import GoogleCalendarService, InvalidSyncToken
|
|
from odoo.addons.google_calendar.models.google_sync import google_calendar_token
|
|
from odoo.loglevels import exception_to_unicode
|
|
from odoo.tools import str2bool
|
|
|
|
_logger = logging.getLogger(__name__)
|
|
|
|
class User(models.Model):
|
|
_inherit = 'res.users'
|
|
|
|
google_calendar_account_id = fields.Many2one('google.calendar.credentials')
|
|
google_calendar_rtoken = fields.Char(related='google_calendar_account_id.calendar_rtoken', groups="base.group_system")
|
|
google_calendar_token = fields.Char(related='google_calendar_account_id.calendar_token')
|
|
google_calendar_token_validity = fields.Datetime(related='google_calendar_account_id.calendar_token_validity')
|
|
google_calendar_sync_token = fields.Char(related='google_calendar_account_id.calendar_sync_token')
|
|
google_calendar_cal_id = fields.Char(related='google_calendar_account_id.calendar_cal_id')
|
|
google_synchronization_stopped = fields.Boolean(related='google_calendar_account_id.synchronization_stopped', readonly=False)
|
|
|
|
_sql_constraints = [
|
|
('google_token_uniq', 'unique (google_calendar_account_id)', "The user has already a google account"),
|
|
]
|
|
|
|
|
|
@property
|
|
def SELF_READABLE_FIELDS(self):
|
|
return super().SELF_READABLE_FIELDS + ['google_synchronization_stopped', 'google_calendar_account_id']
|
|
|
|
@property
|
|
def SELF_WRITEABLE_FIELDS(self):
|
|
return super().SELF_WRITEABLE_FIELDS + ['google_synchronization_stopped', 'google_calendar_account_id']
|
|
|
|
def _get_google_calendar_token(self):
|
|
self.ensure_one()
|
|
if self.google_calendar_account_id.calendar_rtoken and not self.google_calendar_account_id._is_google_calendar_valid():
|
|
self.sudo().google_calendar_account_id._refresh_google_calendar_token()
|
|
return self.google_calendar_account_id.calendar_token
|
|
|
|
def _get_google_sync_status(self):
|
|
""" Returns the calendar synchronization status (active, paused or stopped). """
|
|
status = "sync_active"
|
|
if str2bool(self.env['ir.config_parameter'].sudo().get_param("google_calendar_sync_paused"), default=False):
|
|
status = "sync_paused"
|
|
elif self.google_synchronization_stopped:
|
|
status = "sync_stopped"
|
|
return status
|
|
|
|
def _sync_google_calendar(self, calendar_service: GoogleCalendarService):
|
|
self.ensure_one()
|
|
results = self._sync_request(calendar_service)
|
|
if not results or not results.get('events'):
|
|
return False
|
|
events, default_reminders, full_sync = results.values()
|
|
# Google -> Odoo
|
|
send_updates = not full_sync
|
|
events.clear_type_ambiguity(self.env)
|
|
recurrences = events.filter(lambda e: e.is_recurrence())
|
|
synced_recurrences = self.env['calendar.recurrence']._sync_google2odoo(recurrences)
|
|
synced_events = self.env['calendar.event']._sync_google2odoo(events - recurrences, default_reminders=default_reminders)
|
|
|
|
# Odoo -> Google
|
|
recurrences = self.env['calendar.recurrence']._get_records_to_sync(full_sync=full_sync)
|
|
recurrences -= synced_recurrences
|
|
recurrences.with_context(send_updates=send_updates)._sync_odoo2google(calendar_service)
|
|
synced_events |= recurrences.calendar_event_ids - recurrences._get_outliers()
|
|
synced_events |= synced_recurrences.calendar_event_ids - synced_recurrences._get_outliers()
|
|
events = self.env['calendar.event']._get_records_to_sync(full_sync=full_sync)
|
|
(events - synced_events).with_context(send_updates=send_updates)._sync_odoo2google(calendar_service)
|
|
|
|
return bool(events | synced_events) or bool(recurrences | synced_recurrences)
|
|
|
|
def _sync_single_event(self, calendar_service: GoogleCalendarService, odoo_event, event_id):
|
|
self.ensure_one()
|
|
results = self._sync_request(calendar_service, event_id)
|
|
if not results or not results.get('events'):
|
|
return False
|
|
event, default_reminders, full_sync = results.values()
|
|
# Google -> Odoo
|
|
send_updates = not full_sync
|
|
event.clear_type_ambiguity(self.env)
|
|
synced_events = self.env['calendar.event']._sync_google2odoo(event, default_reminders=default_reminders)
|
|
# Odoo -> Google
|
|
odoo_event.with_context(send_updates=send_updates)._sync_odoo2google(calendar_service)
|
|
return bool(odoo_event | synced_events)
|
|
|
|
def _sync_request(self, calendar_service, event_id=None):
|
|
if self._get_google_sync_status() != "sync_active":
|
|
return False
|
|
# don't attempt to sync when another sync is already in progress, as we wouldn't be
|
|
# able to commit the transaction anyway (row is locked)
|
|
self.env.cr.execute("""SELECT id FROM res_users WHERE id = %s FOR NO KEY UPDATE SKIP LOCKED""", [self.id])
|
|
if not self.env.cr.rowcount:
|
|
_logger.info("skipping calendar sync, locked user %s", self.login)
|
|
return False
|
|
|
|
full_sync = not bool(self.google_calendar_sync_token)
|
|
with google_calendar_token(self) as token:
|
|
try:
|
|
if not event_id:
|
|
events, next_sync_token, default_reminders = calendar_service.get_events(self.google_calendar_account_id.calendar_sync_token, token=token)
|
|
else:
|
|
# We force the sync_token parameter to avoid doing a full sync.
|
|
# Other events are fetched when the calendar view is displayed.
|
|
events, next_sync_token, default_reminders = calendar_service.get_events(sync_token=token, token=token, event_id=event_id)
|
|
except InvalidSyncToken:
|
|
events, next_sync_token, default_reminders = calendar_service.get_events(token=token)
|
|
full_sync = True
|
|
if next_sync_token:
|
|
self.google_calendar_account_id.calendar_sync_token = next_sync_token
|
|
return {
|
|
'events': events,
|
|
'default_reminders': default_reminders,
|
|
'full_sync': full_sync,
|
|
}
|
|
|
|
@api.model
|
|
def _sync_all_google_calendar(self):
|
|
""" Cron job """
|
|
users = self.env['res.users'].search([('google_calendar_rtoken', '!=', False), ('google_synchronization_stopped', '=', False)])
|
|
google = GoogleCalendarService(self.env['google.service'])
|
|
for user in users:
|
|
_logger.info("Calendar Synchro - Starting synchronization for %s", user)
|
|
try:
|
|
user.with_user(user).sudo()._sync_google_calendar(google)
|
|
self.env.cr.commit()
|
|
except Exception as e:
|
|
_logger.exception("[%s] Calendar Synchro - Exception : %s!", user, exception_to_unicode(e))
|
|
self.env.cr.rollback()
|
|
|
|
def is_google_calendar_synced(self):
|
|
""" True if Google Calendar settings are filled (Client ID / Secret) and user calendar is synced
|
|
meaning we can make API calls, false otherwise."""
|
|
self.ensure_one()
|
|
return self.google_calendar_token and self._get_google_sync_status() == 'sync_active'
|
|
|
|
def stop_google_synchronization(self):
|
|
self.ensure_one()
|
|
self.google_synchronization_stopped = True
|
|
|
|
def restart_google_synchronization(self):
|
|
self.ensure_one()
|
|
if not self.google_calendar_account_id:
|
|
self.google_calendar_account_id = self.env['google.calendar.credentials'].sudo().create([{'user_ids': [Command.set(self.ids)]}])
|
|
self.google_synchronization_stopped = False
|
|
self.env['calendar.recurrence']._restart_google_sync()
|
|
self.env['calendar.event']._restart_google_sync()
|
|
|
|
def unpause_google_synchronization(self):
|
|
self.env['ir.config_parameter'].sudo().set_param("google_calendar_sync_paused", False)
|
|
|
|
def pause_google_synchronization(self):
|
|
self.env['ir.config_parameter'].sudo().set_param("google_calendar_sync_paused", True)
|