Lazy-open output files, and ensure they are always closed

pypa · Jul 2, 2024 · 11c7a5d · 11c7a5d
1 parent 8fc05cf
commit 11c7a5d
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/readme_renderer/__main__.py b/readme_renderer/__main__.py
@@ -18,7 +18,7 @@ def main(cli_args: Optional[List[str]] = None) -> None:
                         help="README format (inferred from input file name or package)")
     parser.add_argument('input', help="Input README file or package name")
     parser.add_argument('-o', '--output', help="Output file (default: stdout)",
-                        type=argparse.FileType('w'), default='-')
+                        default='-')
     args = parser.parse_args(cli_args)
 
     content_format = args.format
@@ -55,7 +55,11 @@ def main(cli_args: Optional[List[str]] = None) -> None:
                          "`rst`, or `txt`)")
     if rendered is None:
         sys.exit(1)
-    print(rendered, file=args.output)
+    if args.output == "-":
+        print(rendered, file=sys.stdout)
+    else:
+        with open(args.output, "w") as fp:
+            print(rendered, file=fp)
 
 
 if __name__ == '__main__':