mirror of
https://github.com/saltstack/salt.git
synced 2025-04-10 14:51:40 +00:00
117 lines
3.1 KiB
Python
117 lines
3.1 KiB
Python
"""
|
|
getTerminalSize()
|
|
- get width and height of console
|
|
- works on linux,os x,windows,cygwin(windows)
|
|
- taken from http://stackoverflow.com/questions/566746/how-to-get-console-window-width-in-python
|
|
"""
|
|
|
|
|
|
import ctypes
|
|
import fcntl
|
|
import os
|
|
import platform
|
|
import struct
|
|
import subprocess
|
|
import termios
|
|
|
|
__all__ = ["getTerminalSize"]
|
|
|
|
|
|
def getTerminalSize():
|
|
current_os = platform.system()
|
|
tuple_xy = None
|
|
if current_os == "Windows":
|
|
tuple_xy = _getTerminalSize_windows()
|
|
if tuple_xy is None:
|
|
tuple_xy = _getTerminalSize_tput()
|
|
# needed for window's python in cygwin's xterm!
|
|
if (
|
|
current_os == "Linux"
|
|
or current_os == "Darwin"
|
|
or current_os.startswith("CYGWIN")
|
|
):
|
|
tuple_xy = _getTerminalSize_linux()
|
|
if tuple_xy is None:
|
|
tuple_xy = (80, 25) # default value
|
|
return tuple_xy
|
|
|
|
|
|
def _getTerminalSize_windows():
|
|
res = None
|
|
try:
|
|
# stdin handle is -10
|
|
# stdout handle is -11
|
|
# stderr handle is -12
|
|
|
|
h = ctypes.windll.kernel32.GetStdHandle(-12)
|
|
csbi = ctypes.create_string_buffer(22)
|
|
res = ctypes.windll.kernel32.GetConsoleScreenBufferInfo(h, csbi)
|
|
except Exception: # pylint: disable=broad-except
|
|
return None
|
|
if res:
|
|
(
|
|
bufx,
|
|
bufy,
|
|
curx,
|
|
cury,
|
|
wattr,
|
|
left,
|
|
top,
|
|
right,
|
|
bottom,
|
|
maxx,
|
|
maxy,
|
|
) = struct.unpack(b"hhhhHhhhhhh", csbi.raw)
|
|
sizex = right - left + 1
|
|
sizey = bottom - top + 1
|
|
return sizex, sizey
|
|
else:
|
|
return None
|
|
|
|
|
|
def _getTerminalSize_tput():
|
|
# get terminal width
|
|
# src: http://stackoverflow.com/questions/263890/how-do-i-find-the-width-height-of-a-terminal-window
|
|
try:
|
|
proc = subprocess.Popen(
|
|
["tput", "cols"], stdin=subprocess.PIPE, stdout=subprocess.PIPE
|
|
)
|
|
output = proc.communicate(input=None)
|
|
cols = int(output[0])
|
|
proc = subprocess.Popen(
|
|
["tput", "lines"], stdin=subprocess.PIPE, stdout=subprocess.PIPE
|
|
)
|
|
output = proc.communicate(input=None)
|
|
rows = int(output[0])
|
|
return (cols, rows)
|
|
except Exception: # pylint: disable=broad-except
|
|
return None
|
|
|
|
|
|
def _getTerminalSize_linux():
|
|
def ioctl_GWINSZ(fd):
|
|
try:
|
|
cr = struct.unpack(b"hh", fcntl.ioctl(fd, termios.TIOCGWINSZ, "1234"))
|
|
except Exception: # pylint: disable=broad-except
|
|
return None
|
|
return cr
|
|
|
|
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 Exception: # pylint: disable=broad-except
|
|
pass
|
|
if not cr:
|
|
try:
|
|
cr = (os.environ["LINES"], os.environ["COLUMNS"])
|
|
except Exception: # pylint: disable=broad-except
|
|
return None
|
|
return int(cr[1]), int(cr[0])
|
|
|
|
|
|
if __name__ == "__main__":
|
|
sizex, sizey = getTerminalSize()
|
|
print(f"width = {sizex} height = {sizey}")
|