Add "--retrylock" to tct/resource_locking.py

Change-Id: I9445f87d19db642c14487c8bde11b48764f1656e
Signed-off-by: Aleksander Mistewicz <a.mistewicz@samsung.com>

1 files changed, 12 insertions, 0 deletions

diff --git a/tct/resource_locking.py b/tct/resource_locking.py
index d98d650..3bda871 100755
--- a/tct/resource_locking.py
+++ b/tct/resource_locking.py
@@ -19,6 +19,7 @@
 # @author Aleksander Mistewicz <a.mistewicz@samsung.com>
 
 import os
+import time
 import subprocess
 import argparse
 import logging
@@ -103,6 +104,11 @@ class UUIDmanager:
         logging.info("Unlocking: %s", self.target)
         return Lockfile.unlock(UUID_DIR + UUID_PREFIX + self.target) == 0
 
+    def retrylock(self):
+        # try to lock a target every 60s. finish if operation is successful
+        while not self.lock():
+            time.sleep(60)
+
 def parse_arguments():
     parser = argparse.ArgumentParser(description="Manager of locks on UUID files")
 
@@ -115,6 +121,10 @@ def parse_arguments():
             action="store_true", default=False, dest="lock",
             help="Lock a device")
 
+    group.add_argument("-r", "--retrylock",
+            action="store_true", default=False, dest="retrylock",
+            help="Try to lock a device until successful")
+
     group.add_argument("-u", "--unlock",
             action="store_true", default=False, dest="unlock",
             help="Unlock a device")
@@ -142,6 +152,8 @@ def main():
     elif args.unlock:
         if not uuid_man.unlock():
             logging.warn("File unlocking unsuccessful!")
+    elif args.retrylock:
+        uuid_man.retrylock()
     logging.debug("End")
 
 if __name__ == '__main__':