sfraint
diff --git a/‎experimental/python/wheel.bzl
Lines changed: 8 additions & 0 deletions b/‎experimental/python/wheel.bzl
Lines changed: 8 additions & 0 deletions
diff --git a/‎experimental/rules_python/wheelmaker.py
Lines changed: 24 additions & 4 deletions b/‎experimental/rules_python/wheelmaker.py
Lines changed: 24 additions & 4 deletions
@@ -135,6 +135,9 @@ def _py_wheel_impl(ctx):
     for name, ref in ctx.attr.console_scripts.items():
         args.add("--console_script", name + " = " + ref)
 
+    for name, ref in ctx.attr.plugins.items():
+        args.add("--plugin", name + " : " + ref)
+
     if ctx.attr.description_file:
         description_file = ctx.file.description_file
         args.add("--description_file", description_file)
@@ -201,6 +204,11 @@ _entrypoint_attrs = {
     "console_scripts": attr.string_dict(
         doc = """\
 console_script entry points, e.g. 'experimental.examples.wheel.main:main'.
+""",
+    ),
+    "plugins": attr.string_dict(
+        doc = """\
+plugin entry points, e.g. {'batfish_session': 'bfe = experimental.examples.wheel.main:main'}.
 """,
     ),
 }
 
@@ -92,6 +92,7 @@ def add_string(self, filename, contents):
 
     def add_file(self, package_filename, real_filename):
         """Add given file to the distribution."""
+
         def arcname_from(name):
             # Always use unix path separators.
             normalized_arcname = name.replace(os.path.sep, '/')
@@ -157,12 +158,26 @@ def add_metadata(self, extra_headers, description, classifiers, requires,
         metadata += "\n"
         self.add_string(self.distinfo_path('METADATA'), metadata)
 
-    def add_entry_points(self, console_scripts):
+    def add_entry_points(self, console_scripts, plugins):
         """Write entry_points.txt file to the distribution."""
         # https://packaging.python.org/specifications/entry-points/
-        if not console_scripts:
+        import re
+        pattern = re.compile(r'([^=:]+):([^:=]+)=([^=]+)')
+        lines = []
+        if console_scripts:
+            lines += ["[console_scripts]"] + console_scripts
+        if plugins:
+            for line in plugins:
+                match = pattern.fullmatch(line)
+                if not match:
+                    raise ValueError('{line} is not a valid entry point'.format(line=line))
+                plugin_type = match.group(1).strip()
+                plugin_name = match.group(2).strip()
+                plugin_value = match.group(3).strip()
+                lines += ['[{plugin_type}]'.format(plugin_type=plugin_type),
+                          '{plugin_name} = {plugin_value}'.format(plugin_name=plugin_name, plugin_value=plugin_value)]
+        if not lines:
             return
-        lines = ["[console_scripts]"] + console_scripts
         contents = '\n'.join(lines)
         self.add_string(self.distinfo_path('entry_points.txt'), contents)
 
@@ -246,6 +261,10 @@ def main():
         '--console_script', action='append',
         help="Defines a 'console_script' entry point. "
              "Can be supplied multiple times.")
+    contents_group.add_argument(
+        '--plugin', action='append',
+        help="Defines a 'plugin' entry_point in the format 'a:b=c'. "
+             "Can be supplied multiple times.")
 
     requirements_group = parser.add_argument_group("Package requirements")
     requirements_group.add_argument(
@@ -303,13 +322,14 @@ def main():
         requires = arguments.requires or []
         extra_headers = arguments.header or []
         console_scripts = arguments.console_script or []
+        plugins = arguments.plugin or []
 
         maker.add_metadata(extra_headers=extra_headers,
                            description=description,
                            classifiers=classifiers,
                            requires=requires,
                            extra_requires=extra_requires)
-        maker.add_entry_points(console_scripts=console_scripts)
+        maker.add_entry_points(console_scripts=console_scripts, plugins=plugins)
         maker.add_recordfile()