add CLI completion

microsoft · Dec 20, 2024 · 1ba0a8e · 1ba0a8e
1 parent 6ea36f9
commit 1ba0a8e
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -35,6 +35,7 @@ dependencies = [
   "pdfminer.six",
   "puremagic",
   "pydub",
+  "shtab",
   "youtube-transcript-api",
   "SpeechRecognition",
   "pathvalidate",

diff --git a/src/markitdown/__main__.py b/src/markitdown/__main__.py
@@ -4,6 +4,7 @@
 import sys
 import argparse
 from textwrap import dedent
+import shtab
 from ._markitdown import MarkItDown
 
 
@@ -22,7 +23,8 @@ def main():
 
     parser.add_argument(
         "filename", nargs="?", help="if unspecified, defaults to stdin"
-    )
+    ).complete = shtab.FILE
+    shtab.add_argument_to(parser)
     args = parser.parse_args()
     markitdown = MarkItDown()
     result = markitdown.convert(args.filename or sys.stdin.buffer)