71 lines
No EOL
3.3 KiB
Python
71 lines
No EOL
3.3 KiB
Python
import json
|
|
|
|
from app import notification_queue
|
|
from data import model
|
|
from data.model import InvalidNotificationException
|
|
from endpoints.api.repositorynotification_models_interface import RepoNotificationInterface, RepositoryNotification
|
|
from endpoints.notificationevent import NotificationEvent
|
|
from endpoints.notificationhelper import build_notification_data
|
|
|
|
|
|
class RepoNotificationPreOCIModel(RepoNotificationInterface):
|
|
|
|
def create_repo_notification(self, namespace_name, repository_name, event_name, method_name, method_config, event_config, title=None):
|
|
repository = model.repository.get_repository(namespace_name, repository_name)
|
|
return self._notification(model.notification.create_repo_notification(repository,
|
|
event_name,
|
|
method_name,
|
|
method_config,
|
|
event_config,
|
|
title))
|
|
|
|
def list_repo_notifications(self, namespace_name, repository_name, event_name=None):
|
|
return [self._notification(n)
|
|
for n in model.notification.list_repo_notifications(namespace_name, repository_name, event_name)]
|
|
|
|
def get_repo_notification(self, uuid):
|
|
try:
|
|
found = model.notification.get_repo_notification(uuid)
|
|
except InvalidNotificationException:
|
|
return None
|
|
return self._notification(found)
|
|
|
|
def delete_repo_notification(self, namespace_name, repository_name, uuid):
|
|
try:
|
|
found = model.notification.delete_repo_notification(namespace_name, repository_name, uuid)
|
|
except InvalidNotificationException:
|
|
return None
|
|
return self._notification(found)
|
|
|
|
def reset_notification_number_of_failures(self, namespace_name, repository_name, uuid):
|
|
return self._notification(
|
|
model.notification.reset_notification_number_of_failures(namespace_name, repository_name, uuid))
|
|
|
|
def queue_test_notification(self, uuid):
|
|
try:
|
|
notification = model.notification.get_repo_notification(uuid)
|
|
except InvalidNotificationException:
|
|
return None
|
|
|
|
event_info = NotificationEvent.get_event(notification.event.name)
|
|
sample_data = event_info.get_sample_data(notification)
|
|
notification_data = build_notification_data(notification, sample_data)
|
|
notification_queue.put([notification.repository.namespace_user.username, notification.uuid,
|
|
notification.event.name], json.dumps(notification_data))
|
|
return self._notification(notification)
|
|
|
|
|
|
def _notification(self, notification):
|
|
if not notification:
|
|
return None
|
|
return RepositoryNotification(uuid=notification.uuid,
|
|
title=notification.title,
|
|
event_name=notification.event.name,
|
|
method_name=notification.method.name,
|
|
config_json=notification.config_json,
|
|
event_config_json=notification.event_config_json,
|
|
number_of_failures=notification.number_of_failures)
|
|
|
|
|
|
|
|
pre_oci_model = RepoNotificationPreOCIModel() |