Change notificationworker to use a data interface

workers/notificationworker/models_pre_oci.py

@@ -0,0 +1,46 @@
+import json
+
+from data import model
+from workers.notificationworker.models_interface import (
+  NotificationWorkerDataInterface, Notification, Repository)
+
+
+class PreOCIModel(NotificationWorkerDataInterface):
+  def get_enabled_notification(self, notification_uuid):
+    try:
+      notification_row = model.notification.get_enabled_notification(notification_uuid)
+    except model.InvalidNotificationException:
+      return None
+
+    return Notification(uuid=notification_uuid, event_name=notification_row.event.name,
+                        method_name=notification_row.method.name,
+                        event_config_dict=json.loads(notification_row.event_config_json),
+                        method_config_dict=json.loads(notification_row.config_json),
+                        repository=Repository(notification_row.repository.namespace_user.username,
+                                              notification_row.repository.name))
+
+  def reset_number_of_failures_to_zero(self, notification):
+    model.notification.reset_notification_number_of_failures(
+      notification.repository.namespace_name, notification.repository.name, notification.uuid)
+
+  def increment_notification_failure_count(self, notification):
+    model.notification.increment_notification_failure_count(notification.uuid)
+
+  def create_notification_for_testing(self, target_username):
+    repo = model.repository.get_repository('devtable', 'simple')
+    method_data = {
+      'target': {
+        'kind': 'user',
+        'name': target_username,
+      }
+    }
+    notification = model.notification.create_repo_notification(repo, 'build_success',
+                                                               'quay_notification', method_data, {})
+    return notification.uuid
+
+  def user_has_local_notifications(self, target_username):
+    user = model.user.get_namespace_user(target_username)
+    return bool(list(model.notification.list_notifications(user)))
+
+
+pre_oci_model = PreOCIModel()