o
K��^�) � @ s~ d Z ddlmZ ddlZddlmZmZmZmZm Z m
Z m Z m
Z
mZmZ ddlmZmZ ddlmZmZmZ ddlmZ ddlmZ dd lm Z dd
l m Z e� d �Z d d
� Z!e!G dd� d��Z
e"g d��Z#dd� Z$e$G dd� d��Z%dd� e#e"dg�B D �Z&dd� Z'dd� Z(dd � Z) d0d d �Z*G d!d"� d"e�Z+G d#d$� d$e �Z, & 'd1d(d)�Z-d*d+� Z.G d,d-� d-e�Z/G d.d/� d/e�Z0dS )2z UI utilities � )�
partialmethodN)
�ACTIVATE�AttrMap�CompositeCanvas�connect_signal�LineBox�Padding�SelectableIcon�Text�WidgetDecoration�
WidgetDisable)�
cancel_btn� other_btn)�ListBox�Pile�
WidgetWrap)�Spinner)�Stretchy)�TableRow)� widget_widthzsubiquitycore.ui.utilsc
C s� d}t d|�D ][}t| d�|�tt|d�� t| d�|�tt|d�� t| d�|�ttd||d �� t| d
�|�ttdd |fd �� t| d
�|�ttdd |fd �� t| d�|�ttdd |fd �� q| S )aT Decorator for generating useful padding methods
Loops through and generates methods like:
Padding.push_1(Widget)
Sets the left padding attribute by 1
Padding.pull_24(Widget)
Sets right padding attribute by 24.
Padding.center_50(Widget)
Provides center padding with a relative width of 50
�d � zpush_{})�leftzpull_{})�rightzfixed_{}�center)�align�width� min_widthz center_{}�relative)r r zleft_{}r zright_{}r )�range�setattr�formatr �_Padding)�cls�
padding_count�i� r& �./usr/share/subiquity/subiquitycore/ui/utils.py�
apply_padders3 s2 �� �� �� ��r( c @ s e Zd ZdZee�ZdS )r au Padding methods
.. py:meth:: push_X(:class:`urwid.Widget`)
This method supports padding the left side of the widget
from 1-99, for example:
.. code::
Padding.push_20(Text("This will be indented 20 columns")
.. py:meth:: pull_X(:class:`urwid.Widget`)
This method supports padding the right side of the widget
from 1-99, for example:
.. code::
Padding.pull_20(Text("This will be right indented 20 columns")
.. py:meth:: fixed_X(:class:`urwid.Widget`)
This method supports padding the widget to a fixed size and
centering it.
from 1-99, for example:
.. code::
Padding.fixed_20(Text("This will be centered and fixed sized
of 20 columns"))
.. py:meth:: center_X(:class:`urwid.Widget`)
This method centers a widget with X being the relative width of
the widget.
.. code::
Padding.center_10(Text("This will be centered with a "
"width of 10 columns"))
.. py:meth:: left_X(:class:`urwid.Widget`)
This method aligns a widget left with X being the relative width of
the widget.
.. code::
Padding.left_10(Text("This will be left aligned with a "
"width of 10 columns"))
.. py:meth:: right_X(:class:`urwid.Widget`)
This method right aligns a widget with X being the relative width of
the widget.
.. code::
Padding.right_10(Text("This will be right aligned with a "
"width of 10 columns"))
N)�__name__�
__module__� __qualname__�__doc__r r
�
line_breakr& r& r& r' r X s >r )�bodyzdanger_button focus�
danger_buttonzdone_button focus� done_buttonzframe_button focus� frame_button� frame_header�frame_header_fringe�
info_error�
info_minor� info_primary�menu_button focus� menu_buttonzother_button focus� other_button�progress_complete�progress_incompletezscrollbar focus� scrollbarzstring_input focus� string_inputc C sJ t D ] }|d }|t v rt| |tt||d�� qt| |tt|d�� q| S )z� Applies AttrMap attributes to Color class
Eg:
Color.frame_header(Text("I'm text in the Orange frame header"))
Color.body(Text("Im text in wrapped with the body color"))
z focus)�attr_map� focus_map)r> )� STYLE_NAMESr r r )r# �k�kfr& r& r' �apply_style_map� s rC c @ s e Zd ZdZdS )�ColoraZ Partial methods for :class:`~subiquity.palette.STYLES`
.. py:meth:: frame_header(:class:`urwid.Widget`)
This method colors widget based on the style map used.
.. code::
Color.frame_header(Text("This will use foreground and background "
"defined from the STYLES attribute"))
N)r) r* r+ r, r& r& r& r' rD � s rD c C s i | ]}|d �qS )r5 r& )�.0rA r& r& r' �
<dictcomp>� s rF c C s t t| t��S �N)r r �_disable_everything_map��wr& r& r' �disabled� � rK c C s$ t | t�r| j} t | t�r| j} | S rG )�
isinstancer �original_widgetr rI r& r& r' �
undisabled� s
rO c C s0 d}| D ] }t t|�|�}qtt| �||dd�S )N� r )r r r )�maxr r" r )�buttonsr �buttonr& r& r' � button_pile� s �rT TFc C s� t | t�r t| �} |rt�| �} |du rd}n t |t�r t|�}g }|dur2dt|�fdtd�fg}| dtd�fg}|durK|�d|fdtd�fg� t|| �}|rZt |�d |_
tj |dd�S )a_ Helper to create a common screen layout.
The commonest screen layout in subiquity is:
[ 1 line padding (optional) ]
excerpt (optional)
[ 1 line padding ]
Box widget (usually a ListBox)
[ 1 line padding ]
a button_pile
[ 1 line padding ]
This helper makes creating this a 1-liner.
NF�pack� � �L )r ) rM �listr r � center_63rT r
�extendr �len�focus_position� center_79)�rowsrR �
focus_buttons�excerpt� narrow_rows� excerpt_rowsr. �piler& r& r' �screen� s2
�
�
� re c sH e Zd ZdZdZd� fdd� Zdd� Zdd � Zd
d � Zdd
d�Z � Z
S )�CursorOverridezKDecoration to override where the cursor goes when a widget is focused.
Tr c s t � �|� || _d S rG )�super�__init__�cursor_x)�selfrJ ri �� __class__r& r' rh $ s
zCursorOverride.__init__c C s
| j dfS )Nr �ri )rj �sizer& r& r' �get_cursor_coords( s
z CursorOverride.get_cursor_coordsc C � | j �||�S rG )�_original_widgetr_ �rj rn �focusr& r& r' r_ + rL zCursorOverride.rowsc C rp rG )rq �keypressrr r& r& r' rt . rL zCursorOverride.keypressFc C s* | j �||�}|rt|�}| �|�|_|S rG )rq �renderr ro �cursor)rj rn rs �cr& r& r' ru 1 s
zCursorOverride.render)r )F) r) r* r+ r, �has_original_widthrh ro r_ rt ru �
__classcell__r& r& rk r' rf s rf c @ s e Zd ZdZdgZdd� ZdS )�
ClickableIconzLike Button, but simpler. �clickc C s | j | tkr |S | �d� d S )Nr{ )� _command_mapr �_emit)rj rn �keyr& r& r' rt > s zClickableIcon.keypressN)r) r* r+ r, �signalsrt r& r& r& r' rz : s rz r8 r7 rW c sp t | �}t�t�s
d �i�t�t�sd �i�tt||d����� t|d� �fdd�� t|d� �fdd�� � S )Nrm �openc �
� � ��S rG �� set_attr_map��menu)�amr? r& r' �<lambda>O �
z&make_action_menu_row.<locals>.<lambda>�closec r� rG r� r� )r� r> r& r' r� P r� )r rM �dictr rf r )�cellsr� r> r? ri �rowr& )r� r> r? r' �make_action_menu_rowD s
r� c C s" | � d�}d�dd� |D ���� S )Nz
c S s g | ]}|� d d��qS )�
� )�replace)rE �pr& r& r' �
<listcomp>V s zrewrap.<locals>.<listcomp>)�split�join�strip)�text�parasr& r& r' �rewrapT s
r� c s$ e Zd Z� fdd�Zdd� Z� ZS )�SomethingFailedc sX || _ ttd�| jd�}t|�td�t|�d��td�t|g�g}t� �d|dd� d S )N�Close��label�on_pressrV r� rW � ) �parentr �_r� r
r� rT rg rh )rj r� �msg�stderr�ok�widgetsrk r& r' rh Z s ��zSomethingFailed.__init__c C s | j �� d S rG )r� �remove_overlay�rj �senderr& r& r' r� i rL zSomethingFailed.close�r) r* r+ rh r� ry r& r& rk r' r� Y s r� c s& e Zd Z� fdd�Zddd�Z� ZS )�
LoadingDialogc s� || _ t|dd�| _| j�� d| _t|�d | _dtd| �fd| jfg}|d ur@|| _t t
d�| j d�}|� dt
|g�f� t� �tt|��� d S ) N�dots)�styleFr� rU r� �Cancelr� )r� r �spinner�start�closedr\ r r
�task_to_cancelr
r� r� �appendrT rg rh r r )rj r� �aio_loopr� r� r� �cancelrk r& r' rh o s
�zLoadingDialog.__init__Nc C s: | j rd S |d ur| j�� d| _ | j�� | j�� d S )NT)r� r� r� r� �stopr� r� r� r& r& r' r� � s
zLoadingDialog.closerG r� r& r& rk r' r� m s r� )NTNF)r8 r7 rW )1r, � functoolsr �logging�urwidr r r r r r r" r r
r r �subiquitycore.ui.buttonsr
r �subiquitycore.ui.containerr r r �subiquitycore.ui.spinnerr �subiquitycore.ui.stretchyr �subiquitycore.ui.tabler �subiquitycore.ui.widthr � getLogger�logr( �setr@ rC rD rH rK rO rT re rf rz r� r� r� r� r& r& r& r' �<module> sB 0
% E
�-
�