瀏覽代碼

Ensure packages from entry points are copied

Closes gh-21
Thomas Kluyver 11 年之前
父節點
當前提交
08eae99c67
共有 1 個文件被更改,包括 4 次插入0 次删除
  1. 4 0
      nsist/__init__.py

+ 4 - 0
nsist/__init__.py

@@ -150,6 +150,10 @@ from {module} import {func}
             f.write(self.SCRIPT_TEMPLATE.format(qualifier=self.py_qualifier,
                                                 module=module, func=func))
 
+        pkg = module.split('.')[0]
+        if pkg not in self.packages:
+            self.packages.append(pkg)
+
     def prepare_shortcuts(self):
         """Prepare shortcut files in the build directory.