def create_notifications(pipeline_):
if notifies := pipeline_.config.get("notifications"):
pipeline_.notifications = PiplineNotifications()
if no_data_notification := notifies.get('no_data'):
pipeline_.notifications.no_data_notification = NoDataNotifications(
pipeline_,
no_data_notification,
)
Session.add(pipeline_.notifications.no_data_notification)
class SessionManager:
def __init__(self, entity):
self.entity = entity
self.session = Session()
if not self.session.object_session(self.entity):
self.session.add(self.entity)
def __enter__(self):
return self.session
def __exit__(self, exc_type, exc_value, exc_traceback):
if exc_value:
self.session.expunge(self.entity)
return False
self.session.commit()
return True
from agent.pipeline.notifications import PiplineNotifications, NoDataNotifications
from agent.modules.db import Session
def create_notifications(pipeline_):
if notifies := pipeline_.config.get("notifications"):
pipeline_.notifications = PiplineNotifications()
if no_data_notification := notifies.get('no_data'):
pipeline_.notifications.no_data_notification = NoDataNotifications(
pipeline_,
no_data_notification,
)
Session.add(pipeline_.notifications.no_data_notification)
Session.add(pipeline_.notifications)
def save(streamsets: StreamSets):
if not Session.object_session(streamsets):
Session.add(streamsets)
Session.commit()
def save(source_: Source):
if not Session.object_session(source_):
Session.add(source_)
Session.commit()
def save_auth_token(token: AuthenticationToken):
if not Session.object_session(token):
Session.add(token)
Session.commit()
def save(destination_: HttpDestination):
if not Session.object_session(destination_):
Session.add(destination_)
Session.commit()
def save(entity: Entity):
if not Session.object_session(entity):
Session.add(entity)
Session.commit()
