From 34e73a6b6df265d722ff563207f66adcbd2e89e2 Mon Sep 17 00:00:00 2001 From: Ehsan Azarnasab Date: Fri, 21 Aug 2015 11:31:16 -0700 Subject: [PATCH] make sure only relevant platform artifacts are packaged. avoid foreign artifacts potentially left over from VM builts. --- setup.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/setup.py b/setup.py index 9f61bd920..8e2aeaee5 100644 --- a/setup.py +++ b/setup.py @@ -294,6 +294,11 @@ class build(_build): # this is where the extension and Python examples are located out_dir = os.path.join(script_dir, "./python_examples") + # these are the created artifacts we want to package + dll_ext = ['.so'] + if sys.platform == "win32": + dll_ext = ['.pyd', '.dll'] + ext_found = False # manually copy everything to distribution folder with package hierarchy in mind names = os.listdir(out_dir) @@ -301,9 +306,11 @@ class build(_build): srcname = os.path.join(out_dir, name) dstname = os.path.join(dist_dir, name) dstextname = os.path.join(dist_dir_examples, name) - if name.endswith('.py') or name.endswith('.txt'): + + name, extension = os.path.splitext(name.lower()) + if extension in ['.py', '.txt']: copy_file(srcname, dstextname) - elif name.endswith('.dll') or name.endswith('.so') or name.endswith('.pyd'): + elif extension in dll_ext: if name.startswith('dlib'): ext_found = True copy_file(srcname, dstname)