mirror of
https://github.com/valitydev/salt.git
synced 2024-11-07 17:09:03 +00:00
f60183b95c
https://github.com/saltstack/salt/pull/45687 and https://github.com/saltstack/salt/pull/45725 fixed problems with older Python 2 versions interacting with the struct module, since unicode_literals turned the format strings to unicode and older Python 2 releases can't accept a unicode format string. Since Python 2 and 3 both support using bytestrings for the format strings, this commit makes sure we are using bytestrings everywhere else we are using struct.pack/unpack.
109 lines
3.0 KiB
Python
109 lines
3.0 KiB
Python
# -*- coding: utf-8 -*-
|
|
# vim: sw=4 ts=4 fenc=utf-8
|
|
'''
|
|
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 python libs
|
|
from __future__ import absolute_import, print_function
|
|
import os
|
|
import platform
|
|
import struct
|
|
import ctypes
|
|
import subprocess
|
|
import fcntl
|
|
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:
|
|
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:
|
|
return None
|
|
|
|
|
|
def _getTerminalSize_linux():
|
|
def ioctl_GWINSZ(fd):
|
|
try:
|
|
cr = struct.unpack(
|
|
b'hh', fcntl.ioctl(fd, termios.TIOCGWINSZ, '1234')
|
|
)
|
|
except Exception:
|
|
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:
|
|
pass
|
|
if not cr:
|
|
try:
|
|
cr = (os.environ['LINES'], os.environ['COLUMNS'])
|
|
except Exception:
|
|
return None
|
|
return int(cr[1]), int(cr[0])
|
|
|
|
|
|
if __name__ == '__main__':
|
|
sizex, sizey = getTerminalSize()
|
|
print('width = {0} height = {1}'.format(sizex, sizey))
|