keyscan_check(): don't require ctx to be passed

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

diff --git a/teuthology/lock.py b/teuthology/lock.py
index 2faff2f274afa7fd8d6ccd6b658b5be1a2f7f1b8..5c17fe005cbb17cfff671e349041b0a70376fdb6 100644 (file)
--- a/teuthology/lock.py
+++ b/teuthology/lock.py
@@ -123,7 +123,7 @@ def update_lock(ctx, name, description=None, status=None, ssh_pub_key=None):
         keyscan_out = ''
         while not keyscan_out:
             time.sleep(10)
-            keyscan_out, _ = keyscan_check(ctx, [name])
+            keyscan_out, _ = keyscan_check([name])
     updated = {}
     if description is not None:
         updated['description'] = description
@@ -347,15 +347,14 @@ def updatekeys(ctx):
     return scan_for_locks(ctx, machines)
 
 
-def keyscan_check(ctx, machines):
+def keyscan_check(machines):
     locks = list_locks()
     current_locks = {}
     for lock in locks:
         current_locks[lock['name']] = lock
 
-    if hasattr(ctx, 'all'):
-        if ctx.all:
-            machines = current_locks.keys()
+    if len(machines) == 0:
+        machines = current_locks.keys()
 
     for i, machine in enumerate(machines):
         if '@' in machine:
@@ -387,7 +386,7 @@ def update_keys(ctx, out, current_locks):
 
 
 def scan_for_locks(ctx, machines):
-    out, current_locks = keyscan_check(ctx, machines)
+    out, current_locks = keyscan_check(machines)
     return update_keys(ctx, out, current_locks)
 
 

diff --git a/teuthology/task/internal.py b/teuthology/task/internal.py
index e584f1d7a131efc8d02dabe37413bf0e7af70388..393d25bf5216ffaa83e3007e37121e5fb72bd3d7 100644 (file)
--- a/teuthology/task/internal.py
+++ b/teuthology/task/internal.py
@@ -114,8 +114,7 @@ def lock_machines(ctx, config):
                 while len(keyscan_out.splitlines()) != len(vmlist):
                     loopcount += 1
                     time.sleep(10)
-                    keyscan_out, current_locks = lock.keyscan_check(ctx,
-                                                                    vmlist)
+                    keyscan_out, current_locks = lock.keyscan_check(vmlist)
                     log.info('virtual machine is still unavailable')
                     if loopcount == 40:
                         loopcount = 0