Das deutsche Python-Forum

Code: Alles auswählen

def get_column_widths(items, spacing=2, max_line_width=80):
    column_widths = []
    filled_width = 0
    column_index = 0
    for item in items:
        item_width = len(item)
        if item_width > max_line_width:
            # Item would fill more than the whole line length
            raise ValueError('Item width too long for line')
        if column_index == len(column_widths):
            # Current index after last column
            if (filled_width + item_width) <= max_line_width:
                # New column for item will fit in line
                # => Add new column
                # XXX: Setting zero-length is needed to see
                #      the offset in following code (ugly!)
                column_widths.append(0)
            else:
                # New column would be too large for line
                # => Fallback to first column
                column_index = 0
        offset = item_width - column_widths[column_index]
        if offset > 0:
            # Item wider than column => Enlarge column
            column_widths[column_index] += offset
            # Enlarge total length with respect to spacing
            filled_width += offset + spacing
        # Increase index used for next iteration
        column_index += 1
    return column_widths

Code: Alles auswählen

>>> import columnizer
>>> columnizer.get_column_widths([76 * 'x', 2 * 'x', 78 * 'x'])
[78, 2]

Code: Alles auswählen

>>> columnizer.get_column_widths([50 * 'x', 25 * 'x', 60 * 'x'])
[60, 25]

Code: Alles auswählen

from collections import namedtuple
from itertools import count

LineSpec = namedtuple('LineSpec', ['spacing', 'column_widths'])

def get_line_spec(items, spacing=2, max_width=80):
    for max_lines in count(1):
        column_widths = []
        current_line = 1
        for item in items:
            item_width = len(item)
            if item_width > max_width:
                raise ValueError('item width exceeds max_width')
            if not column_widths or current_line > max_lines:
                # append column
                column_widths.append(item_width)
                current_line = 2
            else:
                # put item into current column
                column_widths[-1] = max(column_widths[-1], item_width)
                current_line += 1
            total_spacing = spacing * (len(column_widths) - 1)
            line_width = sum(column_widths) + total_spacing
            if line_width > max_width:
                # restart with max_lines increased by 1
                # (next iteration in outer loop)
                break
        else:
            return LineSpec(spacing, column_widths)

Code: Alles auswählen

from ctypes import c_ushort, sizeof, Structure
from termios import TIOCGWINSZ
from fcntl import ioctl
import sys


class WinSize(Structure):
    '''To get the terminal size information via the UNIX systemcall
    'ioctl()'. This is needed cause the control sequence '\033[18t'
    only work proper in xterm-like terminals but not without running
    X-Server.

    The 'WinSize.from_file()'-method returns the result of 'ioctl()'
    and is wrapped by 'window_size()'. Use this function instead.

    Thanks to 'BlackJack' from 'python-forum.de' for this
    peace of code...! '''
    _fields_ = [
        ('rows', c_ushort),
        ('columns', c_ushort),
        ('x_pixels', c_ushort), # Unused.
        ('y_pixels', c_ushort), # Unused.
    ]

    @classmethod
    def from_file(cls, tty_file):
        result = ioctl(tty_file, TIOCGWINSZ, '\0' * sizeof(cls))
        return cls.from_buffer_copy(result)


def window_size():
    '''window_size() -> (rows, columns) '''
    size = WinSize.from_file(sys.stdout)
    return size.rows, size.columns