mirror of
https://github.com/ARM-software/workload-automation.git
synced 2025-01-19 04:21:17 +00:00
544f5f0b92
terminalsize was loaded from a location added to sys.path during bootstrap. This appeared to be causing import issues. There is no longer a good reason for terminalsize to be loaded that way, so just moved it under wlauto.utils so that it can be loaded normally.
95 lines
2.8 KiB
Python
95 lines
2.8 KiB
Python
# Adapted from
|
|
# https://gist.github.com/jtriley/1108174
|
|
# pylint: disable=bare-except,unpacking-non-sequence
|
|
import os
|
|
import shlex
|
|
import struct
|
|
import platform
|
|
import subprocess
|
|
|
|
|
|
def get_terminal_size():
|
|
""" getTerminalSize()
|
|
- get width and height of console
|
|
- works on linux,os x,windows,cygwin(windows)
|
|
originally retrieved from:
|
|
http://stackoverflow.com/questions/566746/how-to-get-console-window-width-in-python
|
|
"""
|
|
current_os = platform.system()
|
|
tuple_xy = None
|
|
if current_os == 'Windows':
|
|
tuple_xy = _get_terminal_size_windows()
|
|
if tuple_xy is None:
|
|
tuple_xy = _get_terminal_size_tput()
|
|
# needed for window's python in cygwin's xterm!
|
|
if current_os in ['Linux', 'Darwin'] or current_os.startswith('CYGWIN'):
|
|
tuple_xy = _get_terminal_size_linux()
|
|
if tuple_xy is None:
|
|
print "default"
|
|
tuple_xy = (80, 25) # default value
|
|
return tuple_xy
|
|
|
|
|
|
def _get_terminal_size_windows():
|
|
# pylint: disable=unused-variable,redefined-outer-name,too-many-locals
|
|
try:
|
|
from ctypes import windll, create_string_buffer
|
|
# stdin handle is -10
|
|
# stdout handle is -11
|
|
# stderr handle is -12
|
|
h = windll.kernel32.GetStdHandle(-12)
|
|
csbi = create_string_buffer(22)
|
|
res = windll.kernel32.GetConsoleScreenBufferInfo(h, csbi)
|
|
if res:
|
|
(bufx, bufy, curx, cury, wattr,
|
|
left, top, right, bottom,
|
|
maxx, maxy) = struct.unpack("hhhhHhhhhhh", csbi.raw)
|
|
sizex = right - left + 1
|
|
sizey = bottom - top + 1
|
|
return sizex, sizey
|
|
except:
|
|
pass
|
|
|
|
|
|
def _get_terminal_size_tput():
|
|
# get terminal width
|
|
# src: http://stackoverflow.com/questions/263890/how-do-i-find-the-width-height-of-a-terminal-window
|
|
try:
|
|
cols = int(subprocess.check_call(shlex.split('tput cols')))
|
|
rows = int(subprocess.check_call(shlex.split('tput lines')))
|
|
return (cols, rows)
|
|
except:
|
|
pass
|
|
|
|
|
|
def _get_terminal_size_linux():
|
|
def ioctl_GWINSZ(fd):
|
|
try:
|
|
import fcntl
|
|
import termios
|
|
cr = struct.unpack('hh',
|
|
fcntl.ioctl(fd, termios.TIOCGWINSZ, '1234'))
|
|
return cr
|
|
except:
|
|
pass
|
|
cr = ioctl_GWINSZ(0) or ioctl_GWINSZ(1) or ioctl_GWINSZ(2)
|
|
if not cr:
|
|
try:
|
|
fd = os.open(os.ctermid(), os.O_RDONLY)
|
|
cr = ioctl_GWINSZ(fd)
|
|
os.close(fd)
|
|
except:
|
|
pass
|
|
if not cr:
|
|
try:
|
|
cr = (os.environ['LINES'], os.environ['COLUMNS'])
|
|
except:
|
|
return None
|
|
return int(cr[1]), int(cr[0])
|
|
|
|
|
|
if __name__ == "__main__":
|
|
sizex, sizey = get_terminal_size()
|
|
print 'width =', sizex, 'height =', sizey
|
|
|