Leonardo Pavan Rocha
ca66b8b9bd
Currently the onboarding of calendar is set if you simply open the credential configuration page and close it. Ideally we just want to close it if the credentials have been set. This commit implements this change. task-3570053 Part-of: odoo/odoo#139736
168 lines
8.3 KiB
Python
168 lines
8.3 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)
|
|
|
|
@api.model
|
|
def check_calendar_credentials(self):
|
|
res = super().check_calendar_credentials()
|
|
get_param = self.env['ir.config_parameter'].sudo().get_param
|
|
client_id = get_param('google_calendar_client_id')
|
|
client_secret = get_param('google_calendar_client_secret')
|
|
res['google_calendar'] = bool(client_id and client_secret)
|
|
return res
|