fix console entries for bdist_wheel command

Author: fafhrd91

CHANGES.rst

@@ -1,6 +1,12 @@
 CHANGES
 ^^^^^^^
 
+0.8.2 (2017-09-08)
+------------------
+
+- Fix script generation for bdist_wheel
+
+
 0.8.1 (2017-09-08)
 ------------------
 

setup.py

@@ -1,6 +1,6 @@
 from setuptools import setup
 
-version = '0.8.1'
+version = '0.8.2'
 
 
 setup(

setuptools_rust/patch.py

@@ -4,6 +4,12 @@
 from distutils.dist import Distribution as DistDistribution
 from setuptools.dist import Distribution
 
+try:
+    from wheel.bdist_wheel import bdist_wheel
+    wheel = True
+except:
+    wheel = False
+
 
 def monkey_patch_dist(build_ext):
     # allow to use 'rust_extensions' parameter for setup() call
@@ -57,7 +63,9 @@ def finalize_options(self):
                     ep_scripts = self.distribution.entry_points.get(
                         'console_scripts')
                     if ep_scripts:
-                        ep_scripts.extend(scripts)
+                        for script in scripts:
+                            if script not in ep_scripts:
+                                ep_scripts.append(scripts)
                     else:
                         ep_scripts = scripts
 
@@ -73,6 +81,37 @@ def finalize_options(self):
 
     install.finalize_options = finalize_options
 
+    if wheel:
+        # this is for console entries
+        bdist_wheel.orig_finalize_options = bdist_wheel.finalize_options
+
+        def finalize_options(self):
+            scripts = []
+            for ext in self.distribution.rust_extensions:
+                scripts.extend(ext.entry_points())
+
+            if scripts:
+                if not self.distribution.entry_points:
+                    self.distribution.entry_points = {
+                        'console_scripts': scripts,
+                    }
+                else:
+                    ep_scripts = self.distribution.entry_points.get(
+                        'console_scripts')
+                    if ep_scripts:
+                        for script in scripts:
+                            if script not in ep_scripts:
+                                ep_scripts.append(scripts)
+                    else:
+                        ep_scripts = scripts
+
+                    self.distribution.entry_points[
+                        'console_scripts'] = ep_scripts
+
+            self.orig_finalize_options()
+
+        bdist_wheel.finalize_options = finalize_options
+
     # clean rust project
     def run_clean(self):
         self.orig_run()