From 62e01746a4b785420fde377926c8a5681a47f107 Mon Sep 17 00:00:00 2001
From: Jb DOYON <jb@jiby.tech>
Date: Tue, 12 May 2020 19:03:24 +0100
Subject: [PATCH] Add command line support using scripts/generate.py

Ship a poetry script (package-exported executable command) from the
scripts/generate.py file.

Refactor the main into a function for entrypoint purposes
---
 pyproject.toml      | 3 +++
 scripts/generate.py | 6 ++++--
 2 files changed, 7 insertions(+), 2 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index 7411e4e7..2c63459b 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -24,3 +24,6 @@ isort = "^4.3"
 
 [tool.black]
 line-length = 120
+
+[tool.poetry.scripts]
+diagrams = 'scripts.generate:main'
diff --git a/scripts/generate.py b/scripts/generate.py
index 4e3ef425..785909c5 100644
--- a/scripts/generate.py
+++ b/scripts/generate.py
@@ -94,9 +94,11 @@ def generate(pvd: str) -> None:
     apidoc = gen_apidoc(pvd, typ_paths)
     make_apidoc(pvd, apidoc)
 
-
-if __name__ == "__main__":
+def main():
     pvd = sys.argv[1]
     if pvd not in cfg.PROVIDERS:
         sys.exit()
     generate(pvd)
+
+if __name__ == "__main__":
+    main()