如何在Python中获取Linux控制台窗口宽度

时间:2009-02-19 19:17:19

标签: python linux console terminal width

python中有没有办法以编程方式确定控制台的宽度?我指的是没有包装的一行中的字符数,而不是窗口的像素宽度。

修改

寻找适用于Linux的解决方案

14 个答案:

答案 0 :(得分:238)

import os
rows, columns = os.popen('stty size', 'r').read().split()

使用'stty size'命令,根据a thread on the python mailing list,这个命令在linux上是合理通用的。它将'stty size'命令作为文件打开,从中读取',并使用简单的字符串拆分来分隔坐标。

与os.environ [“COLUMNS”]值(尽管使用bash作为我的标准shell无法访问)不同,数据也是最新的,而我相信os.environ [“ COLUMNS“]值仅在python解释器启动时有效(假设用户从那时起调整了窗口大小)。

答案 1 :(得分:213)

不确定为什么它出现在模块shutil中,但它落在Python 3.3中,Querying the size of the output terminal

>>> import shutil
>>> shutil.get_terminal_size((80, 20))  # pass fallback
os.terminal_size(columns=87, lines=23)  # returns a named-tuple

os模块中有一个低级实现。

现在可以在Python 3.2及以下版本中使用backport:

答案 2 :(得分:65)

使用

import console
(width, height) = console.getTerminalSize()

print "Your terminal's width is: %d" % width

编辑:哦,对不起。这不是一个python标准的lib,这里是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])

答案 3 :(得分:53)

上面的代码没有在我的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应包含像素宽度和高度,但不包含。

答案 4 :(得分:38)

我四处寻找并找到了一个Windows解决方案:

http://code.activestate.com/recipes/440694-determine-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

答案 5 :(得分:15)

从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

答案 6 :(得分:12)

它是:

.roots

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)是一个参数,指示应该使用stdin文件描述符而不是默认的stdout。我们想要使用stdin,因为stdout在被管道传输时会自行分离,在这种情况下会引发错误。
我试图弄清楚何时使用stdout而不是stdin参数是有意义的,并且不知道为什么它是默认值。

答案 7 :(得分:6)

看起来该代码存在一些问题,Johannes:

  • getTerminalSize需要import os
  • 什么是env?看起来像os.environ

另外,为什么在返回之前切换linescols?如果TIOCGWINSZstty都说lines然后cols,我就这么说吧。在我注意到不一致之前,这让我困惑了10分钟。

Sridhar,当我输出输出时,我没有收到错误。我很确定它在try-except中被正确捕获。

pascal,"HHHH"无法在我的计算机上运行,​​但"hh"可以。我很难找到该功能的文档。看起来它依赖于平台。

chochem,inc。

这是我的版本:

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)

答案 8 :(得分:6)

如果在调用此脚本时没有控制终端,则此处的许多Python 2实现都将失败。您可以检查sys.stdout.isatty()以确定它是否实际上是一个终端,但这将排除一堆案例,因此我认为找出终端大小的最pythonic方法是使用builtin curses包。

import curses
w = curses.initscr()
height, width = w.getmaxyx()

答案 9 :(得分:2)

我正在尝试从此处调出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()

答案 10 :(得分:1)

@ reannual的答案效果很好,但问题是:os.popen is now deprecated。应该使用subprocess模块,因此这里是使用subprocess并直接回答问题的@ reannual代码的版本(通过将列宽直接指定为{{ 1}}:

int

在OS X 10.9上测试

答案 11 :(得分:1)

尝试"祝福"

我一直在寻找同样的事情。它非常易于使用,并提供终端着色,造型和定位工具。您需要的就是:

from blessings import Terminal

t = Terminal()

w = t.width
h = t.height

就像Linux中的魅力一样。 (我对MacOSX和Windows不确定)

下载并记录here

或者您可以使用pip安装它:

pip install blessings

答案 12 :(得分:1)

如果您使用的是Python 3.3或更高版本,我建议您使用已建议的内置get_terminal_size()。但是,如果您遇到旧版本并希望采用简单的跨平台方式,则可以使用asciimatics。该软件包支持Python版本2.7,并使用与上面建议的类似的选项来获取当前的终端/控制台大小。

只需构建您的Screen类并使用dimensions属性来获取高度和宽度。事实证明,这可以在Linux,OSX和Windows上使用。

哦 - 并且在此完全披露:我是作者,所以如果你有任何问题,请随时打开一个新问题。

答案 13 :(得分:0)

这是一个应该兼容Linux和Solaris的版本。基于madchine的帖子和内容。需要子进程模块。

def termsize():
    import shlex, subprocess, re
    output = subprocess.check_output(shlex.split('/bin/stty -a'))
    m = re.search('rows\D+(?P\d+); columns\D+(?P\d+);', output)
    if m:
        return m.group('rows'), m.group('columns')
    raise OSError('Bad response: %s' % (output))
>>> termsize()
('40', '100')