Update build.py

clayote · clayote · commit 3179d636a96a · 2013-11-06T20:13:08.000-05:00
diff --git a/src/build.py b/src/build.py
@@ -1,6 +1,6 @@
 #!/usr/bin/env python2.7
 
-from os.path import dirname, join, isfile, realpath, relpath, split
+from os.path import basename, dirname, join, isfile, realpath, relpath, split
 from zipfile import ZipFile
 import sys
 sys.path.insert(0, 'buildlib/jinja2.egg')
@@ -164,7 +164,17 @@ def select(fn):
     for sd in source_dirs:
         sd = realpath(sd)
         compile_dir(sd)
-        files += [(x, relpath(realpath(x), sd)) for x in listfiles(sd) if select(x)]
+        for x in listdir(sd):
+            if select(x):
+                # symbolic links should go in the archive by the name of the link, not the name of what they point to
+                try:
+                    realbase = os.readlink(x)
+                    linkbase = basename(x)
+                    realp = realpath(x).replace(realbase, linkbase)
+                    nufile = (x, relpath(realp, sd))
+                except OSError:
+                    nufile = (x, relpath(realpath(x), sd))
+                files.append(nufile)
 
     # create tar.gz of thoses files
     tf = tarfile.open(tfn, 'w:gz')
