Utilities - logging.

Author: gudrutis

SCRAM/SCRAM/Utilities/AddDir.py

@@ -1,5 +1,10 @@
 import shutil
 from os import makedirs, walk, path
+import sys
+import logging
+
+
+log = logging.getLogger(__name__)
 
 
 def fixpath(malformed_path):
@@ -11,8 +16,12 @@ def fixpath(malformed_path):
 
 
 def adddir(path_to_dir):
-    makedirs(fixpath(path_to_dir), 755)
-    return
+    try:
+        makedirs(fixpath(path_to_dir), 755)
+    except IOError as e:
+        print ("ERROR: failed to create directory directory {0}"
+               .format(path_to_dir), e)
+        sys.exit(1)
 
 
 def copydir(src, dst):
@@ -21,11 +30,21 @@ def copydir(src, dst):
     :param dst: destination directory. It shouldn't exists.
     :return:
     """
-    shutil.copytree(src, dst, symlinks=True, ignore=None)
+    try:
+        shutil.copytree(src, dst, symlinks=True, ignore=None)
+    except IOError as e:
+        print ("ERROR: failed to copy directory from {0} to {1}. "
+               .format(src, dst), e)
+        sys.exit(1)
 
 
 def copyfile(src, dst):
-    shutil.copy2(src, dst)
+    try:
+        shutil.copy2(src, dst)
+    except IOError as e:
+        print ("ERROR: failed to copy file from {0} to {1}. "
+               .format(src, dst), e)
+        sys.exit(1)
 
 
 def getfilelist(dir_path):
@@ -35,7 +54,12 @@ def getfilelist(dir_path):
     :return:
     """
     rez_list = []
-    for root, dirs, files in walk(dir_path, topdown=False):
-        for name in files:
-            rez_list.append(path.join(root.replace(dir_path, ""), name))
+    try:
+        for root, dirs, files in walk(dir_path, topdown=False):
+            for name in files:
+                rez_list.append(path.join(root.replace(dir_path, ""), name))
+    except IOError as e:
+        print ("ERROR: failed to list path: {0}. ".format(dir_path), e)
+        sys.exit(1)
+
     return rez_list