Browse Source

Ensure packages from entry points are copied

Closes gh-21
Thomas Kluyver 11 years ago
parent
commit
08eae99c67
1 changed files with 4 additions and 0 deletions
  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,
             f.write(self.SCRIPT_TEMPLATE.format(qualifier=self.py_qualifier,
                                                 module=module, func=func))
                                                 module=module, func=func))
 
 
+        pkg = module.split('.')[0]
+        if pkg not in self.packages:
+            self.packages.append(pkg)
+
     def prepare_shortcuts(self):
     def prepare_shortcuts(self):
         """Prepare shortcut files in the build directory.
         """Prepare shortcut files in the build directory.