Call ioctl more defensively

jktr · Sep 8, 2023 · 919f6ff · 919f6ff
1 parent c52b17b
commit 919f6ff
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/matplotlib-backend-kitty/__init__.py b/matplotlib-backend-kitty/__init__.py
@@ -10,6 +10,7 @@
 import termios
 import tty
 from base64 import standard_b64encode
+from contextlib import suppress
 from subprocess import run
 
 from matplotlib import interactive, is_interactive
@@ -23,10 +24,14 @@
     interactive(True)
 
 def term_size_px():
+    width_px = height_px = 0
+
     # try to get terminal size from ioctl
-    buf = array.array("H", [0, 0, 0, 0])
-    fcntl.ioctl(sys.stdout, termios.TIOCGWINSZ, buf)
-    _, _, width_px, height_px = buf
+    with suppress(OSError):
+        buf = array.array('H', [0, 0, 0, 0])
+        fcntl.ioctl(sys.stdout, termios.TIOCGWINSZ, buf)
+        _, _, width_px, height_px = buf
+
     if width_px != 0 and height_px != 0:
         return height_px, width_px