From 57ccae421b19d384787a5ff4f23c4c2a98f433e1 Mon Sep 17 00:00:00 2001 From: abdeladim-s Date: Sat, 21 Dec 2024 13:33:55 -0500 Subject: [PATCH] refactor(CLI): use save function from DocumentConverterResult --- src/markitdown/__main__.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/src/markitdown/__main__.py b/src/markitdown/__main__.py index b6cf963..7f662e4 100644 --- a/src/markitdown/__main__.py +++ b/src/markitdown/__main__.py @@ -57,6 +57,12 @@ def main(): "--output", help="Output file name. If not provided, output is written to stdout.", ) + parser.add_argument( + "-e", + "--encoding", + help="Encoding of the output file. Defaults to utf-8.", + default="utf-8", + ) args = parser.parse_args() if args.filename is None: @@ -72,8 +78,7 @@ def main(): def _handle_output(args, result: DocumentConverterResult): """Handle output to stdout or file""" if args.output: - with open(args.output, "w", encoding="utf-8") as f: - f.write(result.text_content) + result.save(args.output, encoding=args.encoding) else: print(result.text_content)