Merge pull request #134 from nicholasjstock/master

allow for copying symlinks when copying tree to create zip

Author: jarosser06

lambda_uploader/utils.py

@@ -51,7 +51,11 @@ def copy_tree(src, dest, ignore=None, include_parent=False):
                 os.makedirs(pkg_path)
 
             LOG.debug("Copying %s to %s" % (path, pkg_path))
-            shutil.copy(path, pkg_path)
+            if os.path.islink(path):
+                linkto = os.readlink(path)
+                os.symlink(linkto.replace(src, dest, 1), os.path.join(pkg_path, filename))
+            else:
+                shutil.copy(path, pkg_path)
 
 
 # Iterate through every item in ignore

tests/test_utils.py

@@ -52,6 +52,25 @@ def test_copy_tree():
     rmtree(TESTING_TEMP_DIR)
     rmtree(copy_dir)
 
+def test_copy_tree_with_symlink():
+    os.mkdir(TESTING_TEMP_DIR)
+    filename = 'foo.py'
+    symlink_filename = "sym-{}".format(filename)
+    with open(path.join(TESTING_TEMP_DIR, filename), 'w') as tfile:
+        tfile.write(filename)
+    os.symlink(path.join(TESTING_TEMP_DIR,filename), path.join(TESTING_TEMP_DIR,symlink_filename))
+    copy_dir = '.copy_of_test'
+
+    utils.copy_tree(TESTING_TEMP_DIR, copy_dir)
+
+    assert os.path.islink(path.join(copy_dir,symlink_filename))
+    linkto = os.readlink(path.join(copy_dir,symlink_filename))
+    assert linkto == path.join(copy_dir, filename)
+
+    rmtree(TESTING_TEMP_DIR)
+    rmtree(copy_dir)
+
+
 
 def test_ignore_file():
     ignored = utils._ignore_file('ignore/foo.py', TEST_IGNORE)