##// END OF EJS Templates
Added tag v4.27.1 for changeset 1e0ab770108a
Added tag v4.27.1 for changeset 1e0ab770108a

File last commit:

r4306:09801de9 default
r4784:d07c3d05 stable
Show More
scheduler.py
60 lines | 2.2 KiB | text/x-python | PythonLexer
# -*- coding: utf-8 -*-
# Copyright (C) 2010-2020 RhodeCode GmbH
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License, version 3
# (only), as published by the Free Software Foundation.
#
# 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 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/>.
#
# This program is dual-licensed. If you wish to learn more about the
# RhodeCode Enterprise Edition, including its added features, Support services,
# and proprietary license terms, please see https://rhodecode.com/licenses/
import logging
import importlib
from celery.beat import (
PersistentScheduler, ScheduleEntry as CeleryScheduleEntry)
log = logging.getLogger(__name__)
class FileScheduleEntry(CeleryScheduleEntry):
def __init__(self, name=None, task=None, last_run_at=None,
total_run_count=None, schedule=None, args=(), kwargs=None,
options=None, relative=False, app=None, **_kwargs):
kwargs = kwargs or {}
options = options or {}
# because our custom loader passes in some variables that the original
# function doesn't expect, we have this thin wrapper
super(FileScheduleEntry, self).__init__(
name=name, task=task, last_run_at=last_run_at,
total_run_count=total_run_count, schedule=schedule, args=args,
kwargs=kwargs, options=options, relative=relative, app=app)
class FileScheduler(PersistentScheduler):
"""CE base scheduler"""
Entry = FileScheduleEntry
def setup_schedule(self):
log.info("setup_schedule called")
super(FileScheduler, self).setup_schedule()
try:
# try if we have EE scheduler available
module = importlib.import_module('rc_ee.lib.celerylib.scheduler')
RcScheduler = module.RcScheduler
except ImportError:
# fallback to CE scheduler
RcScheduler = FileScheduler