From 3bc6db1e8cc886642660b20547ff5d6af739a3c2 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Daniel=20Ekl=C3=B6f?= Date: Sun, 8 Mar 2020 13:52:59 +0100 Subject: [PATCH] scripts/generate-alt-random-writes: use TIOCGWINSZ instead of shutil --- scripts/generate-alt-random-writes.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/scripts/generate-alt-random-writes.py b/scripts/generate-alt-random-writes.py index f7afbad4..9af73cfd 100755 --- a/scripts/generate-alt-random-writes.py +++ b/scripts/generate-alt-random-writes.py @@ -1,8 +1,10 @@ #!/usr/bin/env python3 import argparse import enum -import shutil +import fcntl +import struct import sys +import termios class ColorVariant(enum.IntEnum): @@ -25,9 +27,11 @@ def main(): opts = parser.parse_args() out = opts.out if opts.out is not None else sys.stdout - term_size = shutil.get_terminal_size() - lines = term_size.lines - cols = term_size.columns + lines, cols, _, _ = struct.unpack( + 'HHHH', + fcntl.ioctl(sys.stdout.fileno(), + termios.TIOCGWINSZ, + struct.pack('HHHH', 0, 0, 0, 0))) # Number of characters to write to screen count = 256 * 1024**1