Only delete jobs that match the pattern

Signed-off-by: Zack Cerza <zack.cerza@inktank.com>

diff --git a/teuthology/beanstalk.py b/teuthology/beanstalk.py
index a9967d8d0a18ac2e1b1f1cea0926d0b6306e322f..6776300b60c07b265f77e5bbb2c2af97089346da 100644 (file)
--- a/teuthology/beanstalk.py
+++ b/teuthology/beanstalk.py
@@ -124,6 +124,15 @@ class RunPrinter(JobProcessor):
 
 
 class JobDeleter(JobProcessor):
+    def __init__(self, pattern):
+        self.pattern = pattern
+        super(JobDeleter, self).__init__()
+
+    def add_job(self, job_id, job_config, job_obj=None):
+        job_name = job_config['name']
+        if self.pattern in job_name:
+            super(JobDeleter, self).add_job(job_id, job_config, job_obj)
+
     def process_job(self, job_id):
         job_config = self.jobs[job_id]['job_config']
         job_name = job_config['name']
@@ -147,7 +156,7 @@ def main(args):
         watch_tube(connection, machine_type)
         if delete:
             walk_jobs(connection, machine_type,
-                      JobDeleter())
+                      JobDeleter(delete))
         elif runs:
             walk_jobs(connection, machine_type,
                       RunPrinter())