Infra
/
lemur
5
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

creating a new celery task to sync sources with destinations. This is as a measure to make sure important new destinations are also present as sources.

This commit is contained in:
Hossein Shafagh 2019-02-05 10:48:52 -08:00
parent 7f4f4ffded
commit 6d1ef933c4
1 changed files with 19 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
lemur/common/celery.py
View File

@ -20,6 +20,9 @@ from lemur.notifications.messaging import send_pending_failure_notification
from lemur.pending_certificates import service as pending_certificate_service from lemur.pending_certificates import service as pending_certificate_service
from lemur.plugins.base import plugins from lemur.plugins.base import plugins
from lemur.sources.cli import clean, sync, validate_sources from lemur.sources.cli import clean, sync, validate_sources
from lemur.destinations import service as destinations_service
from lemur.sources import service as sources_service
if current_app: if current_app:
flask_app = current_app flask_app = current_app
@ -226,3 +229,19 @@ def sync_source(source):
""" """
current_app.logger.debug("Syncing source {}".format(source)) current_app.logger.debug("Syncing source {}".format(source))
sync([source]) sync([source])
@celery.task()
def sync_source_destination():
"""
This celery task will sync destination and source, to make sure all new destinations are also present in source.
Some destinations do not qualify as sources, and hence should be excluded from being added as sources
"""
current_app.logger.debug("Syncing source and destination")
for dst in destinations_service.get_all():
if dst.plugin_name == 'aws-destination' and not sources_service.get_by_label(dst.label):
sources_service.create(label=dst.label,
plugin_name='aws-source',
options=dst.options,
description=dst.description)
current_app.logger.info("Source: %s added", dst.label)