Adding in cancel notifications

buildman/jobutil/buildjob.py

@@ -14,6 +14,7 @@ class BuildJobLoadException(Exception):
   """ Exception raised if a build job could not be instantiated for some reason. """
   pass
 
+
 class BuildJob(object):
   """ Represents a single in-progress build job. """
   def __init__(self, job_item):
@@ -21,6 +22,7 @@ class BuildJob(object):
 
     try:
       self.job_details = json.loads(job_item.body)
+      self.build_notifier = BuildJobNotifier(self.build_uuid)
     except ValueError:
       raise BuildJobLoadException(
           'Could not parse build queue item config with ID %s' % self.job_details['build_uuid']
@@ -34,35 +36,7 @@ class BuildJob(object):
     return self.job_item.retries_remaining > 0
 
   def send_notification(self, kind, error_message=None, image_id=None, manifest_digests=None):
-    tags = self.build_config.get('docker_tags', ['latest'])
-    event_data = {
-      'build_id': self.repo_build.uuid,
-      'build_name': self.repo_build.display_name,
-      'docker_tags': tags,
-      'trigger_id': self.repo_build.trigger.uuid,
-      'trigger_kind': self.repo_build.trigger.service.name,
-      'trigger_metadata': self.build_config.get('trigger_metadata', {})
-    }
-
-    if image_id is not None:
-      event_data['image_id'] = image_id
-
-    if manifest_digests:
-      event_data['manifest_digests'] = manifest_digests
-
-    if error_message is not None:
-      event_data['error_message'] = error_message
-
-    # TODO(jzelinskie): remove when more endpoints have been converted to using
-    # interfaces
-    repo = AttrDict({
-      'namespace_name': self.repo_build.repository.namespace_user.username,
-      'name': self.repo_build.repository.name,
-    })
-    spawn_notification(repo, kind, event_data,
-                       subpage='build/%s' % self.repo_build.uuid,
-                       pathargs=['build', self.repo_build.uuid])
-
+    self.build_notifier.send_notification(kind, error_message, image_id, manifest_digests)
 
   @lru_cache(maxsize=1)
   def _load_repo_build(self):
@@ -182,3 +156,61 @@ class BuildJob(object):
       return list(cached_tags)[0]
 
     return None
+
+
+class BuildJobNotifier(object):
+  """ A class for sending notifications to a job that only relies on the build_uuid """
+
+  def __init__(self, build_uuid):
+    self.build_uuid = build_uuid
+
+  @property
+  def repo_build(self):
+    return self._load_repo_build()
+
+  @lru_cache(maxsize=1)
+  def _load_repo_build(self):
+    try:
+      return model.build.get_repository_build(self.build_uuid)
+    except model.InvalidRepositoryBuildException:
+      raise BuildJobLoadException(
+        'Could not load repository build with ID %s' % self.build_uuid)
+
+  @property
+  def build_config(self):
+    try:
+      return json.loads(self.repo_build.job_config)
+    except ValueError:
+      raise BuildJobLoadException(
+        'Could not parse repository build job config with ID %s' % self.repo_build.uuid
+      )
+
+  def send_notification(self, kind, error_message=None, image_id=None, manifest_digests=None):
+    tags = self.build_config.get('docker_tags', ['latest'])
+    event_data = {
+      'build_id': self.repo_build.uuid,
+      'build_name': self.repo_build.display_name,
+      'docker_tags': tags,
+      'trigger_id': self.repo_build.trigger.uuid,
+      'trigger_kind': self.repo_build.trigger.service.name,
+      'trigger_metadata': self.build_config.get('trigger_metadata', {})
+    }
+
+    if image_id is not None:
+      event_data['image_id'] = image_id
+
+    if manifest_digests:
+      event_data['manifest_digests'] = manifest_digests
+
+    if error_message is not None:
+      event_data['error_message'] = error_message
+
+    # TODO(jzelinskie): remove when more endpoints have been converted to using
+    # interfaces
+    repo = AttrDict({
+      'namespace_name': self.repo_build.repository.namespace_user.username,
+      'name': self.repo_build.repository.name,
+    })
+    spawn_notification(repo, kind, event_data,
+                       subpage='build/%s' % self.repo_build.uuid,
+                       pathargs=['build', self.repo_build.uuid])