fast entrypoints

1 files changed, 21 insertions, 0 deletions

diff --git a/setup.py b/setup.py
index 42cc6586..f380d85b 100755
--- a/setup.py
+++ b/setup.py
@@ -21,11 +21,26 @@
 # ========================================================================== #
 
 
+import os
+import textwrap
+
 from setuptools import setup
+from setuptools.command.easy_install import ScriptWriter
 
 
 # =====
 def main() -> None:
+    ScriptWriter.template = textwrap.dedent("""
+        # EASY-INSTALL-ENTRY-SCRIPT: %(spec)r,%(group)r,%(name)r
+
+        __requires__ = %(spec)r
+
+        from kvmd.apps.%(name)r import main
+
+        if __name__ == "__main__":
+            main()
+    """).strip()
+
     setup(
         name="kvmd",
         version="0.158",
@@ -53,6 +68,12 @@ def main() -> None:
             "kvmd": ["data/*.yaml"],
         },
 
+        scripts=[
+            os.path.join("scripts", name)
+            for name in os.listdir("scripts")
+            if not name.startswith(".")
+        ],
+
         entry_points={
             "console_scripts": [
                 "kvmd = kvmd.apps.kvmd:main",