Skip to content

Commit 7c4f1f2

Browse files
committed
Do not auto-close STDOUT on exit
1 parent 1bc8562 commit 7c4f1f2

File tree

2 files changed

+5
-2
lines changed

2 files changed

+5
-2
lines changed

graphtage/printer.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -600,6 +600,7 @@ def __init__(self, *args, title: Optional[str] = None, **kwargs):
600600
self.indents += 1
601601

602602
def close(self):
603+
super().flush()
603604
if self.indents != 3:
604605
log.warning(f"Mismatched indent; expected 3 but got {self.indents}")
605606
self.indents -= 1
@@ -609,6 +610,7 @@ def close(self):
609610
super().newline()
610611
self.raw_write("</html>")
611612
super().newline()
613+
super().close()
612614

613615
def html_element(self, element_name, inline=False, **kwargs) -> 'HTMLPrinter':
614616
"""A convenience function for printing an element."""

graphtage/progress.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -98,9 +98,10 @@ def write(self, text: str) -> int:
9898
self.flush()
9999
return len(text)
100100

101-
def close(self) -> None:
101+
def close(self):
102102
self.flush(final=True)
103-
return self.status_stream.close()
103+
if self.status_stream is not sys.stdout and self.status_stream is not sys.stderr:
104+
self.status_stream.close()
104105

105106
def fileno(self) -> int:
106107
return self.status_stream.fileno()

0 commit comments

Comments
 (0)