Add basic cron support
This commit is contained in:
+16
-3
@@ -1,15 +1,19 @@
|
||||
import importlib
|
||||
import pip
|
||||
from interruptingcow import timeout
|
||||
|
||||
class WorkersManager:
|
||||
def __init__(self, config):
|
||||
self._config = config
|
||||
self._interval_enabled_workers = {}
|
||||
self._workers = self._register_klasses()
|
||||
|
||||
def update(self, worker_name):
|
||||
messages = self.workers[worker_name].status_update()
|
||||
for message in messages:
|
||||
message.topic = '/'.join([self._config[worker_name]['topic_prefix'], message.topic])
|
||||
messages = []
|
||||
with timeout(30):
|
||||
messages = self.workers[worker_name].status_update()
|
||||
for message in messages:
|
||||
message.topic = '/'.join([self._config[worker_name]['topic_prefix'], message.topic])
|
||||
|
||||
return messages
|
||||
|
||||
@@ -17,14 +21,23 @@ class WorkersManager:
|
||||
def workers(self):
|
||||
return self._workers
|
||||
|
||||
@property
|
||||
def interval_enabled_workers(self):
|
||||
return self._interval_enabled_workers
|
||||
|
||||
def _register_klasses(self):
|
||||
arr = {}
|
||||
|
||||
for worker_klass in self._config.keys():
|
||||
moduleObj = importlib.import_module("workers.%s" % worker_klass)
|
||||
klass = getattr(moduleObj, "%sWorker" % worker_klass.title())
|
||||
|
||||
if moduleObj.REQUIREMENTS is not None:
|
||||
self._pip_install_helper(moduleObj.REQUIREMENTS)
|
||||
|
||||
if self._config[worker_klass]['update_interval'] is not None:
|
||||
self._interval_enabled_workers[worker_klass] = self._config[worker_klass]['update_interval']
|
||||
|
||||
arr[worker_klass] = klass(**self._args_for(worker_klass))
|
||||
|
||||
return arr
|
||||
|
||||
Reference in New Issue
Block a user