HOME

Mini Shell 1.0
DIR:/snap/core20/current/usr/share/subiquity/subiquitycore/ui/__pycache__/
Current File : //snap/core20/current/usr/share/subiquity/subiquitycore/ui/__pycache__/table.cpython-38.pyc
U

K��^E�	@s�dZddlmZddlZddlmZmZmZmZddl	m
Z
ddlZddlZe�
d�ZejGdd�d��Zd	d
�ZGdd�de�Zd
d�ZGdd�de�ZGdd�de�ZGdd�de�Zedk�r�ddlmZed�eee�d�dejddd�fe�d�g�ee�d�ejddd�e�d �e�d!�g�ged"d#d$�ed"d%d&�d'�d(d)�Zed*efe�d+�g�Zd,d-�Zejeed.�Ze� �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@sLeZdZdZejdd�Zejdd�Zejdd�Zejdd�Z	ejdd�Z
dS)�ColSpeczDetails about a column.T)�defaultFrN)�__name__�
__module__�__qualname__�__doc__�attrZib�pack�
can_shrink�	min_width�	omittable�rpad�rr�./usr/share/subiquity/subiquitycore/ui/table.pyrYsrcs>t|�}t�fdd�|dd�D��}|�d|d7}|S)Nc3s.|]&}�d|��d|dd�VqdS)��rN)�get)�.0�i��widthsrr�	<genexpr>xsz_width.<locals>.<genexpr>���r)�sorted�sum)r�user_indices�rrrr�_widthrsr#csHeZdZdZ�fdd�Zdd�Zdd�Zdd	�Zd
d�Zdd
�Z	�Z
S)�TableRowz]A row in a table.

    A wrapper around a Columns. The widths will be set when rendered.
    cs�g|_g}|D]P}d}t|t�r(|\}}|dks4t�|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.
        rr�rN)�cells�
isinstance�tuple�AssertionError�append�urwid�Textr�columns�super�__init__)�selfr&�cols�cell�colspan��	__class__rrr/�s


zTableRow.__init__cCs$|jjD]\}}|��rdSqdS)NTF)�_w�contents�
selectable)r0�w�_rrrr8�szTableRow.selectableccs4d}|jD]$\}}t|||�|fV||7}q
dS)z=Yield the user indices each cell spans and the cell.
        rN)r&�range)r0�user_ir3r2rrr�_user_indices_cells�szTableRow._user_indices_cellscCsFi}|��D]4\}}t|�dkr|d|krt|�|d|d<q|S)z�Return a mapping {underlying-index:natural-width}.

        Cells spanning multiple columns are ignored (handled in
        adjust_for_spanning_cells).
        rrr)r=�lenr)r0Z
unpacked_colsrr!r2rrr�get_natural_widths�s
zTableRow.get_natural_widthsc
	Cs�|��D]�\}}t|�|@rqt|�dkr,qt||�}t|�}||krt|�|@}|rZ|}t||t|��\}	}
t|�D](\}}|d||	t||
k�7<qxqdS)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.
        rrN)r=�setr>r#r�divmod�	enumerate�int)
r0�unpacked_user_indices�no_inherent_sizerr!r2Z	cur_widthZ
cell_widthZunsizedZdiv�modr�user_jrrr�adjust_for_spanning_cells�s
z"TableRow.adjust_for_spanning_cellsc	Cs�g}|��D]�\}}zt||�}Wn"tk
rD|j�dd�}YnX|dkrPq|j�d|�}|�||f�|�dt|�dd�}|r|�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.
        ZweightrrZgivenrr%N)r=r#�KeyErrorr-�optionsr*r�maxr+r,r7)r0rr1r!r2�widthZopt�nrrr�
set_widths�s zTableRow.set_widths)r
rrr
r/r8r=r?rHrN�
__classcell__rrr4rr$}sr$csddd�|��D�}dd�|��D�}|D]8}|j�|�}|��D]\}}	t|	|�|d��||<q@q(t|���dd}
|r�t|
t|�d�}
t�}t|
�D]6}||kr�|�d|d�dkr�|�|�d|d|<q�|�	�fdd�t|
d�D��|�	d	d�|��D��|�
�}
|D]}|j�|||��qt|�
��}||k�sJ|�rVt|
�D]4}||j�sRd||k�rR|
d||d|<�qRt|
�D]�}||j�s�||j�s�d||k�r�|d|=||j�r�|t|�
��}|||jk�r�q�t|
�D]H}||j�s�q�|�d|��r�d|d|<d|d|d<�qАq��q��qАq�|}||t|�fS)
z6Return {cell-index:width} and total width for a table.cSsh|]\}}|js|�qSr�r�rr<�csrrr�	<setcomp>�sz+_compute_widths_for_size.<locals>.<setcomp>cSs"i|]\}}|jrd||j�qS)r)rr)rrrRrrr�
<dictcomp>�sz,_compute_widths_for_size.<locals>.<dictcomp>rrrcsi|]}d|d��qS)rrr)rr<��default_spacingrrrT	s
cSs*i|]"\}}|jdk	rd|d|j�qS)Nrr)rrQrrrrTs
�
)�items�base_widgetr?rKr�keysr@r;�add�update�copyrHr �valuesrrrr�bool)Zmaxcol�
table_rows�colspecsrVrDr�rowZ
row_widthsZunderlying_ir9ZcolcountrEr<Z
widths_before�total_widthZ	remainingrGrrUr�_compute_widths_for_size�sl��
���

rccspeZdZd�fdd�	Zdd�Zdd	�Zd
d�Zdd
�Z�fdd�Z�fdd�Z	e
dd��Zejdd��Z�Z
S)�
AbstractTableNr�leftcs\�fdd�|D�|_|dkr i}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.
        csg|]}tj|�d��qS)��align�r+�Padding�rrarfrr�
<listcomp>Fsz*AbstractTable.__init__.<locals>.<listcomp>N)r_rrr`�spacingr.r/�_make�
_last_sizer@�group)r0�rowsr`rlrgr4rfrr/?szAbstractTable.__init__cCs |j|jB}|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)ro)r0Zother_tableZ	new_group�tablerrr�bindPszAbstractTable.bindcCs
d|_dS�N)rn�r0rrr�
invalidateZszAbstractTable.invalidatecCs~|j|krdSg}|jD]}|�|j�qt|d||j|j�\}}}|jD],}||_|jD]}|sj||_|j�	|�q\qLdS)Nr)
rnro�extendr_rcr`rlrLrXrN)r0�sizerprqrrb�has_unpackedrarrrrc]s"

�


z&AbstractTable._compute_widths_for_sizecCs8g}|jD]}|�|j�q
td||j|j�\}}}|S)Ni��)rorvr_rcr`rl)r0rprqrrbrxrrr�get_natural_widthms
�
zAbstractTable.get_natural_widthcs|�|�t��||�Srs)rcr.rp�r0rw�focusr4rrrpus
zAbstractTable.rowscs|�|�t��||�Srs)rcr.�renderrzr4rrr|ys
zAbstractTable.rendercCs
|jjjSrs�r6rX�focus_positionrtrrrr~}szAbstractTable.focus_positioncCs||jj_dSrsr})r0�valrrrr~�s)Nrre)r
rrr/rrrurcryrpr|�propertyr~�setterrOrrr4rrd<s

rdc@s,eZdZdd�Zdd�Zdd�Zdd�Zd	S)
�	TablePilecCstdd�|D��S)NcSsg|]}d|f�qSrPr)rr"rrrrk�sz#TablePile._make.<locals>.<listcomp>)r�r0rprrrrm�szTablePile._makecs8���|�j||�<�fdd�|D��jj||�<dS)Ncs"g|]}t�|��j�d�f�qSrP)r+rir6rJ)rr9rtrrrk�sz)TablePile.insert_rows.<locals>.<listcomp>)rur_r6r7)r0�indexZnew_rowsrrtr�insert_rows�s

�zTablePile.insert_rowscCsr|��t|jj�||}|jj|k}|j||�=|jj||�=|rP|��n|jj��sn|jjd7_qPdS)Nr)	rur>r6r7r~r_Z_select_last_selectabler{r8)r0�start�endZ
new_lengthZrefocusrrr�remove_rows�s
zTablePile.remove_rowscsn���dd�|D�}|�_t�jj�dk}�fdd�|D��jjdd�<t�jj�dk}|rj|sj���dS)�Update the list of rows. cSsg|]}t�|��qSrrhrjrrrrk�sz*TablePile.set_contents.<locals>.<listcomp>rcsg|]}|�j�d�f�qSrP)r6rJrjrtrrrk�sN)rur_r>r6r7�_select_first_selectable)r0rp�empty_before�empty_afterrrtr�set_contents�szTablePile.set_contentsN)r
rrrmr�r�r�rrrrr��sr�c@seZdZdd�Zdd�ZdS)�TableListBoxcCst|�Srs)rr�rrrrm�szTableListBox._makecCs^|��dd�|D�}||_|jjj}t|�dk}||dd�<t|�dk}|rZ|sZ|��dS)r�cSsg|]}t�|��qSrrhrjrrrrk�sz-TableListBox.set_contents.<locals>.<listcomp>rN)rur_r6rX�bodyr>r�)r0rpr�r�r�rrrr��s
zTableListBox.set_contentsN)r
rrrmr�rrrrr��sr��__main__)�setup_loggerz
.subiquityZaarZ201234567890123456789012345678901234567890123456789Zclip)ZwrapZeeeeZcccZ(0123456789012345678901234567890123456789Zffffffffffffffffffffffffffffff�gT�
)rrr)rr)rr�)rlr�xcGs
tj�dSrs)r+ZExitMainLoop)�argsrrr�unhandled_input�sr�)r�)!r
�collectionsrZloggingZsubiquitycore.ui.containerrrrrZsubiquitycore.ui.widthrrr+Z	getLogger�log�srr#r$rcrdr�r�r
Zsubiquitycore.logr�r,�vZ	SolidFillr�ZMainLoopZloop�runrrrr�<module>sV6
pOJ+
���


��
�