Add shell completion

dschep · Mar 20, 2019 · 93b4f0e · 93b4f0e
1 parent 1f6721c
commit 93b4f0e
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/ntfy/cli.py b/ntfy/cli.py
@@ -26,6 +26,11 @@
 except ImportError:
     psutil = None
 
+try:
+    import argcomplete
+except ImportError:
+    argcomplete = None
+
 try:
     from .terminal import is_focused
 except ImportError:
@@ -294,6 +299,8 @@ def default_sender(args):
 
 
 def main(cli_args=None):
+    if argcomplete is not None:
+        argcomplete.autocomplete(parser)
     if cli_args is not None:
         args = parser.parse_args(cli_args)
     else:

diff --git a/setup.py b/setup.py
@@ -1,3 +1,4 @@
+# PYTHON_ARGCOMPLETE_OK
 from sys import version_info
 
 from setuptools import find_packages, setup