像列输出一样 - python库

Question

Debian的apt工具输出导致统一宽度的列。例如，尝试运行“aptitude search svn”..并且所有名称都出现在同一宽度的第一列中。

现在，如果您调整终端大小，则会相应地调整列宽。

有没有一个Python库，使人能做到这一点？请注意，库必须知道终端宽度并将表作为输入 - 例如，它可以是[('rapidsvn', 'A GUI client for subversion'), ...] ..并且您还可以为第一列（或任何列）指定最大宽度。另请注意，如果超过终端宽度，下面第二列中的字符串将被修剪。因此，不会引入不需要的第二行。

$ aptitude search svn 
[...] 
p python-svn-dbg     - A(nother) Python interface to Subversion (d 
v python2.5-svn      -            
v python2.6-svn      -            
p rapidsvn       - A GUI client for subversion     
p statsvn       - SVN repository statistics     
p svn-arch-mirror     - one-way mirroring from Subversion to Arch r 
p svn-autoreleasedeb    - Automatically release/upload debian package 
p svn-buildpackage     - helper programs to maintain Debian packages 
p svn-load       - An enhanced import facility for Subversion 
p svn-workbench      - A Workbench for Subversion     
p svnmailer       - extensible Subversion commit notification t 
p websvn       - interface for subversion repositories writt 
$ 

编辑：（响应下面Alex的答案）...输出将类似于在1“资质搜索”），只有最后一列（这是最长的串唯一的列在一行中）将被修剪，2）通常只有2-4列，但最后一列（“描述”）预计至少需要一半的终端宽度。 3）所有行包含相同数量的列，4）所有条目仅包含字符串

Answer 1

我不认为有一个通用的，跨平台的方式来“获取终端的宽度” - 绝对不是“查看COLUMNS环境变量”（请参阅​​我对该问题的评论）。在Linux和Mac OS X上（我预计所有现代Unix版本），

curses.wrapper(lambda _: curses.tigetnum('cols')) 

返回列数;但我不知道wcurses是否支持Windows。一旦你有（如果你坚持，或通过诅咒，或从甲骨文，或默认80，或任何其他方式，你喜欢os.environ ['COLUMNS']）所需的输出宽度，其余的是 比较可行。这是一项非常精简的工作，有很多错误的机会，并且非常容易受到很多您不清楚的详细规格的影响，例如：哪些列被剪切以避免包装 - 它总是最后一个，还是......？如果根据你的问题，你如何在样本输出中显示3列，只有两列通过...？如果不是所有行都具有相同数量的列，那么应该发生什么？表中的所有条目是否必须是字符串？以及许多其他许多此类之谜。

因此，对所有没有表达的规格进行一些随意的猜测，一种方法可能是......：

import sys 

def colprint(totwidth, table): 
    numcols = max(len(row) for row in table) 
    # ensure all rows have >= numcols columns, maybe empty 
    padded = [row+numcols*('',) for row in table] 
    # compute col widths, including separating space (except for last one) 
    widths = [ 1 + max(len(x) for x in column) for column in zip(*padded)] 
    widths[-1] -= 1 
    # drop or truncate columns from the right in order to fit 
    while sum(widths) > totwidth: 
    mustlose = sum(widths) - totwidth 
    if widths[-1] <= mustlose: 
     del widths[-1] 
    else: 
     widths[-1] -= mustlose 
     break 
    # and finally, the output phase! 
    for row in padded: 
    for w, i in zip(widths, row): 
     sys.stdout.write('%*s' % (-w, i[:w])) 
    sys.stdout.write('\n') 

Answer 2

那么，aptitude使用cwidget来格式化纯文本显示中的列。你可以调用cwidget为它编写一个python扩展，但我认为这不值得麻烦......你可以使用你的首选方法来获得字符的实际水平尺寸并计算自己。

Answer 3

首先，使用ioctl得到TTY的大小：

import termios, fcntl, struct, sys 

def get_tty_size(): 
    s = struct.pack("HHHH", 0, 0, 0, 0) 
    fd_stdout = sys.stdout.fileno() 
    size = fcntl.ioctl(fd_stdout, termios.TIOCGWINSZ, s) 
    return struct.unpack("HHHH", size)[:2] 

print get_tty_size() 

然后使用这样的函数，使列：

pad = lambda s, n=20: "%s%s" % (s,' '*(n-len(s))) 

把这些放在一起，你已经有了调整列为控制台！

Answer 4

更新：该colprint程序现已在applib Python库hosted in GitHub可用。

下面是对于那些你感兴趣的完整的程序：

# This function was written by Alex Martelli 
# http://stackoverflow.com/questions/1396820/ 
def colprint(table, totwidth=None): 
    """Print the table in terminal taking care of wrapping/alignment 

    - `table`: A table of strings. Elements must not be `None` 
    - `totwidth`: If None, console width is used 
    """ 
    if not table: return 
    if totwidth is None: 
     totwidth = find_console_width() 
     totwidth -= 1 # for not printing an extra empty line on windows 
    numcols = max(len(row) for row in table) 
    # ensure all rows have >= numcols columns, maybe empty 
    padded = [row+numcols*('',) for row in table] 
    # compute col widths, including separating space (except for last one) 
    widths = [ 1 + max(len(x) for x in column) for column in zip(*padded)] 
    widths[-1] -= 1 
    # drop or truncate columns from the right in order to fit 
    while sum(widths) > totwidth: 
     mustlose = sum(widths) - totwidth 
     if widths[-1] <= mustlose: 
      del widths[-1] 
     else: 
      widths[-1] -= mustlose 
      break 
    # and finally, the output phase! 
    for row in padded: 
     print(''.join([u'%*s' % (-w, i[:w]) 
         for w, i in zip(widths, row)])) 

def find_console_width(): 
    if sys.platform.startswith('win'): 
     return _find_windows_console_width() 
    else: 
     return _find_unix_console_width() 
def _find_unix_console_width(): 
    """Return the width of the Unix terminal 

    If `stdout` is not a real terminal, return the default value (80) 
    """ 
    import termios, fcntl, struct, sys 

    # fcntl.ioctl will fail if stdout is not a tty 
    if not sys.stdout.isatty(): 
     return 80 

    s = struct.pack("HHHH", 0, 0, 0, 0) 
    fd_stdout = sys.stdout.fileno() 
    size = fcntl.ioctl(fd_stdout, termios.TIOCGWINSZ, s) 
    height, width = struct.unpack("HHHH", size)[:2] 
    return width 
def _find_windows_console_width(): 
    """Return the width of the Windows console 

    If the width cannot be determined, return the default value (80) 
    """ 
    # http://code.activestate.com/recipes/440694/ 
    from ctypes import windll, create_string_buffer 
    STDIN, STDOUT, STDERR = -10, -11, -12 

    h = windll.kernel32.GetStdHandle(STDERR) 
    csbi = create_string_buffer(22) 
    res = windll.kernel32.GetConsoleScreenBufferInfo(h, csbi) 

    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 
    else: 
     sizex, sizey = 80, 25 

    return sizex