Move swag from devtools/ to library/python/testing/

1 files changed, 26 insertions, 0 deletions

diff --git a/library/python/testing/swag/pathutil.py b/library/python/testing/swag/pathutil.py
new file mode 100644
index 00000000000..3f7910eff3e
--- /dev/null
+++ b/library/python/testing/swag/pathutil.py
@@ -0,0 +1,26 @@
+import os
+import tempfile
+
+
+def get_valid_filename(filename, dirname):
+    current_file, counter = filename, 0
+    while os.path.exists(os.path.join(dirname, current_file)):
+        current_file = "%s_%d" % (filename, counter)
+        counter += 1
+    valid_path = os.path.join(dirname, current_file)
+    os.mknod(valid_path)
+    return valid_path
+
+
+def get_valid_tmpdir(name, tmp_dir):
+    current_dir, counter = name, 0
+    while os.path.exists(os.path.join(tmp_dir, current_dir)):
+        current_dir = "%s_%d" % (name, counter)
+        counter += 1
+    os.mkdir(os.path.join(tmp_dir, current_dir))
+    return os.path.join(tmp_dir, current_dir)
+
+
+def get_base_tmpdir(name):
+    tmppath = tempfile.gettempdir()
+    return get_valid_tmpdir(name, tmppath)