def __init__(self, api_session_manager, api_name, wrapped_name, settings):
self.api_session_manager = api_session_manager
self.api_name = api_name
self.wrapped_name = wrapped_name
self.name = (
settings.get('unique_name')
or '%s:%s.%s' % (self.application_name, self.api_name, wrapped_name))
self.active = False
self.next_date = None
self.updating = False
self.last_called_date = None
self.domain_name = settings.get('domain_name', None)
self.title = settings.get('title', None)
cron_settings = {}
for key in DATES_RANGES.keys():
if key in settings:
cron_settings[key] = settings[key]
self.cron = Cron(**cron_settings)
self.enable()
class APIJob(object):
def __init__(self, api_session_manager, api_name, wrapped_name, settings):
self.api_session_manager = api_session_manager
self.api_name = api_name
self.wrapped_name = wrapped_name
self.name = (
settings.get('unique_name')
or '%s:%s.%s' % (self.application_name, self.api_name, wrapped_name))
self.active = False
self.next_date = None
self.updating = False
self.last_called_date = None
self.domain_name = settings.get('domain_name', None)
self.title = settings.get('title', None)
cron_settings = {}
for key in DATES_RANGES.keys():
if key in settings:
cron_settings[key] = settings[key]
self.cron = Cron(**cron_settings)
self.enable()
def __eq__(self, other):
return isinstance(other, APIJob) and self.name == other.name
def __hash__(self):
return hash(self.name)
def __repr__(self):
return '%s (%s)' % (self.name, self.next_date)
@property
def application_name(self):
return self.api_session_manager.config.application_name
def disable(self):
if self.active:
self.active = False
self.next_date = None
def enable(self):
if not self.domain_name:
self.active = True
elif self.domain_name == DOMAIN_NAME:
self.active = True
else:
self.active = False
self.next_date = None
if self.active and not self.next_date:
self.find_next()
def find_next(self):
if self.active:
try:
self.next_date = self.cron.find_next(NOW())
except NoMoreDates:
self.next_date = None
else:
self.next_date = None
def will_run(self):
return bool(self.active and not self.updating and self.next_date)
def __call__(self):
if self.will_run():
api_session = self.api_session_manager.system_session(self)
lock_key = JOBS_LOCK_KEY(self.name)
if lock_key not in self.api_session_manager.config.cache.lockme:
try:
self.api_session_manager.config.cache.lock(lock_key, timeout=1)
except LockTimeout:
api_session.logging.log_error('job_locked', u('Job already running.'))
else:
self.updating = True
self.last_called_date = NOW()
session = getattr(api_session, self.api_name)
try:
getattr(session, self.wrapped_name)()
except (BaseException, Exception) as error:
api_session.logging.log_critical('jobs_error', str(error))
else:
jobs_session = getattr(api_session, self.api_session_manager.__api_name__)
jobs_session.after_job_running()
finally:
self.updating = False
self.api_session_manager.config.cache.unlock(lock_key)
self.find_next()