如何在 Python 中获取 Linux 控制台窗口宽度
- 2025-01-06 08:32:00
- admin 原创
- 84
问题描述:
在 python 中是否有一种方法可以以编程方式确定控制台的宽度?我的意思是一行中可容纳的字符数(不换行),而不是窗口的像素宽度。
编辑
寻找适用于 Linux 的解决方案
解决方案 1:
不确定它为什么在模块中shutil
,但它在 Python 3.3 中就在那里。参见:
查询输出终端的大小
>>> import shutil
>>> shutil.get_terminal_size((80, 20)) # pass fallback
os.terminal_size(columns=87, lines=23) # returns a named-tuple
os 模块中有低级实现。跨平台 — 可在 Linux、Mac OS 和 Windows 下运行,也可能支持其他类 Unix。还有一个反向移植,但已不再适用。
解决方案 2:
import os
rows, columns = os.popen('stty size', 'r').read().split()
使用 'stty size' 命令,根据python 邮件列表上的一个主题,该命令在 linux 上相当通用。它将 'stty size' 命令作为文件打开,从中“读取”,并使用简单的字符串拆分来分隔坐标。
与 os.environ["COLUMNS"] 值(尽管使用 bash 作为我的标准 shell 但我无法访问它)不同,数据也将是最新的,而我相信 os.environ["COLUMNS"] 值仅在启动 python 解释器时有效(假设用户从那时起调整了窗口大小)。
(请参阅@GringoSuave 的回答,了解如何在 Python 3.3+ 上执行此操作)
解决方案 3:
使用
import console
(width, height) = console.getTerminalSize()
print "Your terminal's width is: %d" % width
编辑:哦,对不起。这不是 Python 标准库,这是 console.py 的源代码(我不知道它来自哪里)。
该模块的工作方式似乎是这样的:如果termcap
可用,则检查是否可用。如果可用,则使用该模块;如果不可用,则检查终端是否支持特殊ioctl
调用,如果该调用也不起作用,则检查某些 shell 为此导出的环境变量。这可能仅适用于 UNIX。
def getTerminalSize():
import os
env = os.environ
def ioctl_GWINSZ(fd):
try:
import fcntl, termios, struct, os
cr = struct.unpack('hh', fcntl.ioctl(fd, termios.TIOCGWINSZ,
'1234'))
except:
return
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:
pass
if not cr:
cr = (env.get('LINES', 25), env.get('COLUMNS', 80))
### Use get(key[, default]) instead of a try/catch
#try:
# cr = (env['LINES'], env['COLUMNS'])
#except:
# cr = (25, 80)
return int(cr[1]), int(cr[0])
解决方案 4:
上面的代码在我的 Linux 上没有返回正确的结果,因为 winsize-struct 有 4 个无符号短裤,而不是 2 个有符号短裤:
def terminal_size():
import fcntl, termios, struct
h, w, hp, wp = struct.unpack('HHHH',
fcntl.ioctl(0, termios.TIOCGWINSZ,
struct.pack('HHHH', 0, 0, 0, 0)))
return w, h
hp 和 hp 应该包含像素宽度和高度,但是没有。
解决方案 5:
它可以是:
import os
columns, rows = os.get_terminal_size(0)
# or
import shutil
columns, rows = shutil.get_terminal_size()
该shutil
函数只是一个包装器os
,用于捕获一些错误并设置后备,但它有一个巨大的警告 -它在管道时会中断!,这是一个相当大的问题。
要在管道时获取终端大小,请使用os.get_terminal_size(0)
。
第一个参数0
表示应使用 stdin 文件描述符而不是默认 stdout。我们希望使用 stdin,因为 stdout 在管道传输时会自行分离,在这种情况下会引发错误。
我试图弄清楚何时使用 stdout 而不是 stdin 参数才有意义,但不知道为什么它在这里是默认设置。
解决方案 6:
我四处搜索并找到了适用于 Windows 的解决方案:
http://code.activestate.com/recipes/440694-define-size-of-console-window-on-windows/
这里有一个适用于 Linux 的解决方案。
因此这里有一个可以在 linux、os x 和 windows/cygwin 上运行的版本:
""" getTerminalSize()
- get width and height of console
- works on linux,os x,windows,cygwin(windows)
"""
__all__=['getTerminalSize']
def getTerminalSize():
import platform
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:
print "default"
tuple_xy = (80, 25) # default value
return tuple_xy
def _getTerminalSize_windows():
res=None
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)
except:
return None
if res:
import struct
(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
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:
import subprocess
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:
return None
def _getTerminalSize_linux():
def ioctl_GWINSZ(fd):
try:
import fcntl, termios, struct, os
cr = struct.unpack('hh', fcntl.ioctl(fd, termios.TIOCGWINSZ,'1234'))
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:
pass
if not cr:
try:
cr = (env['LINES'], env['COLUMNS'])
except:
return None
return int(cr[1]), int(cr[0])
if __name__ == "__main__":
sizex,sizey=getTerminalSize()
print 'width =',sizex,'height =',sizey
解决方案 7:
从 Python 3.3 开始,它很简单:
https://docs.python.org/3/library/os.html#querying-the-size-of-a-terminal
>>> import os
>>> ts = os.get_terminal_size()
>>> ts.lines
24
>>> ts.columns
80
解决方案 8:
如果调用此脚本时没有控制终端,则此处的许多 Python 2 实现都会失败。您可以检查 sys.stdout.isatty() 来确定这是否实际上是一个终端,但这将排除许多情况,因此我认为找出终端大小的最 Python 方式是使用内置 curses 包。
import curses
w = curses.initscr()
height, width = w.getmaxyx()
解决方案 9:
看起来该代码存在一些问题,约翰内斯:
getTerminalSize
需要import os
是什麼
env
?看起来像os.environ
。
另外,为什么在返回之前要切换lines
和?如果和都说那么,我说就保持原样。这让我困惑了整整 10 分钟,然后我才注意到不一致之处。cols
`TIOCGWINSZstty
lines`cols
Sridhar,当我通过管道输出时,我没有收到该错误。我非常确定它在 try-except 中被正确捕获了。
pascal,"HHHH"
在我的计算机上无法运行,但"hh"
可以。我很难找到该函数的文档。它看起来依赖于平台。
chochem 有限公司。
这是我的版本:
def getTerminalSize():
"""
returns (lines:int, cols:int)
"""
import os, struct
def ioctl_GWINSZ(fd):
import fcntl, termios
return struct.unpack("hh", fcntl.ioctl(fd, termios.TIOCGWINSZ, "1234"))
# try stdin, stdout, stderr
for fd in (0, 1, 2):
try:
return ioctl_GWINSZ(fd)
except:
pass
# try os.ctermid()
try:
fd = os.open(os.ctermid(), os.O_RDONLY)
try:
return ioctl_GWINSZ(fd)
finally:
os.close(fd)
except:
pass
# try `stty size`
try:
return tuple(int(x) for x in os.popen("stty size", "r").read().split())
except:
pass
# try environment variables
try:
return tuple(int(os.getenv(var)) for var in ("LINES", "COLUMNS"))
except:
pass
# i give up. return default.
return (25, 80)
解决方案 10:
尝试“祝福”
我正在寻找同样的东西。它非常易于使用,并提供了用于在终端中着色、样式和定位的工具。您需要的非常简单:
from blessings import Terminal
t = Terminal()
w = t.width
h = t.height
在 Linux 中运行良好。(我不确定 MacOSX 和 Windows 是否如此)
点击此处下载并记录
或者你可以使用 pip 安装它:
pip install blessings
解决方案 11:
我正在尝试从这里调用以下解决方案stty size
:
columns = int(subprocess.check_output(['stty', 'size']).split()[1])
然而,这对我来说失败了,因为我正在编写一个需要在 stdin 上重定向输入的脚本,stty
在这种情况下会抱怨“stdin 不是终端”。
我可以让它像这样工作:
with open('/dev/tty') as tty:
height, width = subprocess.check_output(['stty', 'size'], stdin=tty).split()
解决方案 12:
如果您使用的是 Python 3.3 或更高版本,我建议您使用内置的,get_terminal_size()
因为已经推荐过。但是,如果您坚持使用旧版本,并且想要一种简单的跨平台方式来执行此操作,则可以使用asciimatics。此包支持 Python 2.7 之前的版本,并使用与上面建议的选项类似的选项来获取当前终端/控制台大小。
只需构建您的Screen
类并使用该dimensions
属性即可获取高度和宽度。这已被证明可以在 Linux、OSX 和 Windows 上运行。
哦 - 这里完全披露:我是作者,所以如果您在操作过程中遇到任何问题,请随时提出新问题。
解决方案 13:
使用subprocess
,这是最方便的方法:
进口:
import subprocess
使用示例:
print(subprocess.check_output(['stty', 'size']).split())
注意:此函数返回字节,但您可以使用该
int()
函数将其转换为整数。
注意:此函数返回一个数组,即:
array[0]
=行和array[1]
=列。
输出:
[b'46', b'188']
例如,如果您需要比较控制台的宽度是否大于W
,您可以执行以下操作:
if int(subprocess.check_output(['stty', 'size']).split()[1]) > W:
...
解决方案 14:
@reannual 的答案很好,但有一个问题:os.popen
现在已被弃用。subprocess
应该改用模块,因此这里有一个 @reannual 代码版本,它使用subprocess
并直接回答了这个问题(通过将列宽直接作为int
:
import subprocess
columns = int(subprocess.check_output(['stty', 'size']).split()[1])
在 OS X 10.9 上测试
解决方案 15:
这是一个应该与 Linux 和 Solaris 兼容的版本。基于madchine的帖子和评论。需要子进程模块。
定义术语大小():
导入 shlex、子进程、re
输出 = 子进程.check_output(shlex.split('/bin/stty -a'))
m = re.search('rowsD+(?Pd+); columnsD+(?Pd+);', 输出)
如果米:
返回 m.group('rows'), m.group('columns')
引发 OSError('错误响应:%s'%(输出))
>>> 术语大小()
(‘40’,‘100’)