From 4aa17934bcc17f7f5fb0d619c0d97925d0cc77c2 Mon Sep 17 00:00:00 2001
From: Barnaby Shearer <b@zi.is>
Date: Sun, 31 Oct 2021 18:29:56 +0000
Subject: [PATCH] Bugfix: docs (#4)

---
 docs/efm8_read.rst | 7 +------
 docs/u2fzero.rst   | 2 +-
 efm8/__main__.py   | 7 ++++---
 setup.cfg          | 5 +++++
 4 files changed, 11 insertions(+), 10 deletions(-)

diff --git a/docs/efm8_read.rst b/docs/efm8_read.rst
index 35c7dfc..bcacaa2 100644
--- a/docs/efm8_read.rst
+++ b/docs/efm8_read.rst
@@ -1,7 +1,2 @@
-.. autoprogram:: efm8.__main__:_parser()
+.. autoprogram:: efm8.__main__:_parser(True)
    :prog: efm8_read
-
-----------
-
-.. automodule:: efm8_read
-   :members:
diff --git a/docs/u2fzero.rst b/docs/u2fzero.rst
index 3757e20..7fec010 100644
--- a/docs/u2fzero.rst
+++ b/docs/u2fzero.rst
@@ -3,5 +3,5 @@
 
 ----------
 
-.. automodule:: u2fzero
+.. automodule:: efm8.u2fzero
    :members:
diff --git a/efm8/__main__.py b/efm8/__main__.py
index ead8a9c..d6ffdd3 100755
--- a/efm8/__main__.py
+++ b/efm8/__main__.py
@@ -32,13 +32,15 @@
 import efm8
 
 
-def _parser():
+def _parser(read=False):
     parser = argparse.ArgumentParser(description=__doc__)
     parser.add_argument(
         "-p", "--product", help="USB Product ID of device to program", default="EAC9"
     )
     parser.add_argument("-s", "--serial", help="Serial number of device to program")
     parser.add_argument("firmware", help="Intel Hex format file to flash")
+    if read:
+        parser.add_argument("-l", "--length", help="Length to read", default="0x4000")
     return parser
 
 
@@ -55,8 +57,7 @@ def main():
 
 def read():
     """Command line."""
-    parser = _parser()
-    parser.add_argument("-l", "--length", help="Length to read", default="0x4000")
+    parser = _parser(True)
     args = parser.parse_args()
     efm8.write_hex(
         efm8.read_flash(
diff --git a/setup.cfg b/setup.cfg
index 4db2ada..d126586 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -24,3 +24,8 @@ include_package_data = True
 [options.package_data]
 * = py.typed
 
+[options.entry_points]
+console_scripts =
+    efm8 = efm8.__main__:main
+    efm8_read = efm8.__main__:read
+    u2fzero = efm8.u2fzero:main