luciano/markitdown
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

refactor(CLI): use save function from DocumentConverterResult

Browse source
This commit is contained in:
abdeladim-s 2024-12-21 13:33:55 -05:00
parent f8e40df008
commit 57ccae421b
1 changed files with 7 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
src/markitdown/__main__.py
View file

@ -57,6 +57,12 @@ def main():
"--output", "--output",
help="Output file name. If not provided, output is written to stdout.", 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() args = parser.parse_args()
if args.filename is None: if args.filename is None:
@ -72,8 +78,7 @@ def main():
def _handle_output(args, result: DocumentConverterResult): def _handle_output(args, result: DocumentConverterResult):
"""Handle output to stdout or file""" """Handle output to stdout or file"""
if args.output: if args.output:
with open(args.output, "w", encoding="utf-8") as f: result.save(args.output, encoding=args.encoding)
f.write(result.text_content)
else: else:
print(result.text_content) print(result.text_content)