o
K��^E � @ s� d Z ddlmZ ddlZddlmZmZmZmZ ddl m
Z
ddl Z ddl Z e�
d�Ze jG dd� d��Zd d
� ZG d d � d e�Zd
d� ZG dd� de�ZG dd� de�ZG dd� de�Zedkr�ddlmZ ed� eee �d�de jddd �fe �d �g�ee �d �e jd dd �e �d �e �d!�g�ged"d#d$�ed"d%d&�d'�d(d)�Zed*efe � d+�g�Zd,d-� Z e j ee d.�Z e � � dS dS )/aI
A table widget.
One of the principles of urwid is that widgets get their size from
their container rather than deciding it for themselves. At times (as
in stretchy.py) this does not make for the best UI. This module
defines TablePile and TableListBox widgets that by default only take
up as much horizontal space as needed for their cells. If the table
wants more horizontal space than is present, there is a degree of
customization available as to what to do: you can tell which column to
allow to shrink, and to omit another column to try to keep the
shrinking column above a given threshold.
You can also let columns take all available space, as is the urwid
default.
Other features include cells that span multiple columns and binding
tables together so that they use the same widths for their columns.
There is not a lot of care about how the various features interact, so
be careful, or rather do not be surprised when things break. Gotchas
that have occurred to me during implementation:
1. This code needs to know the "natural width" of anything you put
into a table. Don't be surprised if widget_width needs extending.
2. Having a cell that spans multiple columns span a column that can
shrink or be omitted will be confusing.
3. Binding tables together that have different column options will
similarly not do anything sensible.
4. You can wrap table rows in decorators that do not affect their size
(AttrMap, etc) but do not use ones that do affect size (Padding,
etc) or things will get confusing.
5. I haven't tested this code with more than one column that can
shrink or more than one column that can be omitted.
Example:
```
v = TablePile([
TableRow([
urwid.Text("aa"),
(2, urwid.Text("0123456789"*5, wrap='clip')),
urwid.Text('eeee')]),
TableRow([
urwid.Text("ccc"),
urwid.Text("0123456789"*4, wrap='clip'),
urwid.Text('fff'*10), urwid.Text('g')]),
], {
0: ColSpec(omittable=True),
1: ColSpec(can_shrink=True, min_width=10),
}, spacing=4)
```
� )� defaultdictN)�Columns�ListBox�Pile�
WidgetWrap)� widget_widthzsubiquitycore.ui.tablec @ sL e Zd ZdZejdd�Zejdd�Zejdd�Zejdd�Z ejdd�Z
dS )�ColSpeczDetails about a column.T)�defaultFr N) �__name__�
__module__� __qualname__�__doc__�attr�ib�pack�
can_shrink� min_width� omittable�rpad� r r �./usr/share/subiquity/subiquitycore/ui/table.pyr Y s r c s> t |�}t� fdd�|d d� D ��}|� d|d 7 }|S )Nc 3 s0 � | ]}� d | � � d | d d� V qdS )� � r N)�get)�.0�i��widthsr r � <genexpr>x s �. z_width.<locals>.<genexpr>���r )�sorted�sum)r � user_indices�rr r r �_widthr s r$ c sH e Zd ZdZ� fdd�Zdd� Zdd� Zdd � Zd
d � Zd d
� Z � Z
S )�TableRowz]A row in a table.
A wrapper around a Columns. The widths will be set when rendered.
c s� g | _ g }|D ](}d}t|t�r|\}}|dksJ �| j �||f� |�|� |�t�d�� q|d= t|�| _t� � | j� dS )zlcells is a list of [widget] or [(colspan, widget)].
colspan is assumed to be 1 if omitted.
r r � r N)
�cells�
isinstance�tuple�append�urwid�Textr �columns�super�__init__)�selfr' �cols�cell�colspan�� __class__r r r/ � s
zTableRow.__init__c C s$ | j jD ] \}}|�� r dS qdS )NTF)�_w�contents�
selectable)r0 �w�_r r r r8 � s
�zTableRow.selectablec c s6 � d}| j D ]\}}t||| �|fV ||7 }qdS )z=Yield the user indices each cell spans and the cell.
r N)r' �range)r0 �user_ir3 r2 r r r �_user_indices_cells� s �
�z TableRow._user_indices_cellsc C sF i }| � � D ]\}}t|�dkr |d |vr t|�|d|d < q|S )z�Return a mapping {underlying-index:natural-width}.
Cells spanning multiple columns are ignored (handled in
adjust_for_spanning_cells).
r r r )r= �lenr )r0 �
unpacked_colsr r" r2 r r r �get_natural_widths� s �zTableRow.get_natural_widthsc
C s� | � � D ]M\}}t|�|@ rqt|�dkrqt||�}t|�}||k rQt|�|@ }|r-|}t|| t|��\} }
t|�D ]\} } |d| | t| |
k � 7 < q<qdS )z�Make sure columns are wide enough for cells with colspan > 1.
This very roughly follows the approach in
https://www.w3.org/TR/CSS2/tables.html#width-layout.
r r N)r= �setr> r$ r �divmod� enumerate�int)
r0 �unpacked_user_indices�no_inherent_sizer r" r2 � cur_width�
cell_width�unsized�div�modr �user_jr r r �adjust_for_spanning_cells� s
"��z"TableRow.adjust_for_spanning_cellsc C s� g }| � � D ]L\}}zt||�}W n ty! | j�dd�}Y n
w |dkr'q| j�d|�}|�||f� |�dt|� d d�}|rR|�t� d�| j�d|�f� q|| jj
dd�< dS )z�Configure row to given widths.
`widths` is a mapping {underlying-index:width}. An index being
missing means let the column shrink, a width being 0 means omit
the column entirely.
�weightr r �givenr r& N) r= r$ �KeyErrorr- �optionsr* r �maxr+ r, r7 )r0 r r1 r" r2 �width�opt�nr r r �
set_widths� s � �zTableRow.set_widths) r
r r r
r/ r8 r= r@ rM rV �
__classcell__r r r4 r r% } s r% c sF dd� |� � D �}dd� |� � D �}|D ] }|j�|�}|� � D ]\}} t| |�|d��||< q qt|�� �d d }
|rFt|
t|�d �}
t� } t|
�D ]} | |vrh|�d| d�dkrh| �| � d|d| < qM|� � fdd�t|
d �D �� |� d d� |� � D �� |�
� }
|D ]
}|j� || |� q�t |�
� �}|| ks�|�r t|
�D ]} || js�d| |v r�|
d| |d| < q�t|
�D ]V} || js�|| j�sd| |v r�|d| = || j�r | t |�
� � }||| jkr�n't|
�D ] }|| js�q�|�d| ��rd|d| < d|d| d < nq�nq�q�| }||t|�fS )
z6Return {cell-index:width} and total width for a table.c S s h | ] \}}|j s|�qS r �r �r r< �csr r r � <setcomp>� s
��z+_compute_widths_for_size.<locals>.<setcomp>c S s" i | ]
\}}|j rd | |j�qS )r )r r )r r rZ r r r �
<dictcomp>� s " z,_compute_widths_for_size.<locals>.<dictcomp>r r r c s i | ] }d | d � �qS )r r r )r r< ��default_spacingr r r\ � c S s* i | ]\}}|j d urd| d |j �qS )Nr r )r rY r r r r\ s
�)�items� base_widgetr@ rR r �keysrA r; �add�update�copyrM r! �valuesr r r r �bool)�maxcol�
table_rows�colspecsr^ rE r �row�
row_widths� underlying_ir9 �colcountrF r< �
widths_before� total_width� remainingrL r r] r �_compute_widths_for_size� st � �
���� �
���
rr c sp e Zd Zd� fdd� Zdd� Zdd � Zd
d � Zd d
� Z� fdd�Z� fdd�Z e
dd� �Z e j dd� �Z � Z
S )�
AbstractTableNr �leftc s\ � fdd�|D �| _ |du ri }tt|�| _|| _t� �| �| j �� d| _t | g�| _
dS )z�Create a Table.
`rows` - a list of possibly-decorated TableRows
`colspecs` - a mapping {column-index:ColSpec}
'spacing` - how much space to put between cells.
c s g | ] }t j|� d ��qS )��align�r+ �Padding�r rk ru r r �
<listcomp>F r_ z*AbstractTable.__init__.<locals>.<listcomp>N) ri r r rj �spacingr. r/ �_make�
_last_sizerA �group)r0 �rowsrj r{ rv r4 ru r r/ ? s zAbstractTable.__init__c C s | j |j B }|D ]}||_ qdS )z�Bind two tables such that they will use the same column widths.
Don't expect anything good to happen if the two tables do not
use the same colspecs.
N)r~ )r0 � other_table� new_group�tabler r r �bindP s �zAbstractTable.bindc C s
d | _ d S �N)r} �r0 r r r �
invalidateZ s
zAbstractTable.invalidatec C s~ | j |krd S g }| jD ]}|�|j� q t|d || j| j�\}}}| jD ]}||_ |jD ]
}|s5||_|j� |� q.q&d S )Nr )
r} r~ �extendri rr rj r{ rS ra rV )r0 �sizer r� r rp � has_unpackedrk r r r rr ] s
�
��z&AbstractTable._compute_widths_for_sizec C s8 g }| j D ]}|�|j� qtd|| j| j�\}}}|S )Ni�� )r~ r� ri rr rj r{ )r0 r r� r rp r� r r r �get_natural_widthm s
�z AbstractTable.get_natural_widthc � | � |� t� �||�S r� )rr r. r �r0 r� �focusr4 r r r u �
zAbstractTable.rowsc r� r� )rr r. �renderr� r4 r r r� y r� zAbstractTable.renderc C s
| j jjS r� �r6 ra �focus_positionr� r r r r� } s
z AbstractTable.focus_positionc C s || j j_d S r� r� )r0 �valr r r r� � s )Nr rt )r
r r r/ r� r� rr r� r r� �propertyr� �setterrW r r r4 r rs < s
rs c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
� TablePilec C s t dd� |D ��S )Nc S s g | ]}d |f�qS rX r )r r# r r r rz � s z#TablePile._make.<locals>.<listcomp>)r �r0 r r r r r| � s zTablePile._makec s8 � � � |� j||�< � fdd�|D �� jj||�< d S )Nc s" g | ]
}t �|�� j�d �f�qS rX )r+ rx r6 rQ )r r9 r� r r rz � s �z)TablePile.insert_rows.<locals>.<listcomp>)r� ri r6 r7 )r0 �index�new_rowsr r� r � insert_rows� s
�zTablePile.insert_rowsc C s� | � � t| jj�|| }| jj|k}| j||�= | jj||�= |r)| �� d S | jj�� s?| j jd7 _| jj�� r/d S d S )Nr ) r� r> r6 r7 r� ri �_select_last_selectabler� r8 )r0 �start�end�
new_length�refocusr r r � remove_rows� s �zTablePile.remove_rowsc sv � � � dd� |D �}|� _t� jj�dk}� fdd�|D �� jjdd�< t� jj�dk}|r7|s9� �� dS dS dS )�Update the list of rows. c S � g | ]}t �|��qS r rw ry r r r rz � � z*TablePile.set_contents.<locals>.<listcomp>r c s g | ]
}|� j �d �f�qS rX )r6 rQ ry r� r r rz � s N)r� ri r> r6 r7 �_select_first_selectable)r0 r � empty_before� empty_afterr r� r � set_contents� s �zTablePile.set_contentsN)r
r r r| r� r� r� r r r r r� � s
r� c @ s e Zd Zdd� Zdd� ZdS )� TableListBoxc C s t |�S r� )r r� r r r r| � s zTableListBox._makec C sf | � � dd� |D �}|| _| jjj}t|�dk}||dd�< t|�dk}|r/|s1| �� dS dS dS )r� c S r� r rw ry r r r rz � r� z-TableListBox.set_contents.<locals>.<listcomp>r N)r� ri r6 ra �bodyr> r� )r0 r r� r� r� r r r r� � s
�zTableListBox.set_contentsN)r
r r r| r� r r r r r� � s r� �__main__)� setup_loggerz
.subiquity�aar �201234567890123456789012345678901234567890123456789�clip)�wrap�eeee�ccc�(0123456789012345678901234567890123456789� ffffffffffffffffffffffffffffff�gT�
)r r r )r r )r r � )r{ r �xc G s t j�r� )r+ � ExitMainLoop)�argsr r r �unhandled_input� s r� )r� )!r
� collectionsr �logging�subiquitycore.ui.containerr r r r �subiquitycore.ui.widthr r r+ � getLogger�log�sr r$ r% rr rs r� r� r
�subiquitycore.logr� r, �v� SolidFillr� �MainLoop�loop�runr r r r �<module> sX 6
pOJ+ � ��
��
� �