edge-case evaluation

lnxpy · Jun 1, 2024 · 68aac0f · 68aac0f
1 parent 3aa4d93
commit 68aac0f
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/pyaction/io.py b/pyaction/io.py
@@ -1,6 +1,7 @@
 from __future__ import annotations
 
 import os
+from contextlib import nullcontext
 from io import TextIOWrapper
 
 from rich.console import Console
@@ -36,7 +37,9 @@ def write(
             )
         console.print(table)
     else:
-        with open(stream, "+w") as streamline:
+        with nullcontext(stream) if isinstance(stream, TextIOWrapper) else open(
+            stream, "w+"
+        ) as streamline:
             for var, val in context.items():
                 if "\n" in val:
                     streamline.write(MULTILINE_OUTPUT.format(variable=var, value=val))