3
���h�B � @ s� d Z ddlZddlmZmZmZmZmZmZm Z m
Z
m Z m Z m
Z
ddlmZ ddlmZ e
eeef ee f Ze d�Ze d�Zeee ed�d d
�Zd0ee eeee d �d
d�Zd1e
ee e ed�dd�Zd2eee e ed�dd�Zd3eee e eee e ed�dd�Z d4eee d�dd �Z edd �d d �Z d5ee e ed �d!d"�Z d6eee e eeed$�d%d&�Z!ee
e ee
e e ee ed'f ee ed'f f d(�d)d*�Z"d7e
eee ee
eef f d+�d,d-�Z#G d.d/� d/eeef �Z$dS )8z
Printing tools.
� N) �Any�Callable�Dict�Iterable�List�Mapping�Optional�Sequence�Tuple�TypeVar�Union)�
get_option)� is_sequence�_KT�_VT)�space�lists�returnc
s� |j dt��|j dt�}g }g }� �fdd�|dd � D �}|jttt|d
��� ttt|��}xPt|�D ]D\}} || || dd�}
|
jd || g|t| � � |j|
� qlW t|� } x | D ]} |jd
j | �� q�W d j |�S )a�
Glues together two sets of strings using the amount of space requested.
The idea is to prettify.
----------
space : int
number of spaces for padding
lists : str
list of str which being joined
strlen : callable
function used to calculate the length of each str. Needed for unicode
handling.
justfunc : callable
function used to justify str. Needed for unicode handling.
�strlen�justfuncc s g | ]}t t�|��� �qS � )�max�map)�.0�x)r r r �</tmp/pip-build-5_djhm0z/pandas/pandas/io/formats/printing.py�
<listcomp>2 s zadjoin.<locals>.<listcomp>N� �left)�mode� � �
���r# )
�pop�len�justify�appendr r � enumerate�extend�zip�join)
r r �kwargsr Z out_linesZnewLists�lengths�maxLen�i�lst�nlZtoJoin�linesr )r r r �adjoin s
r3 �right)�texts�max_lenr r c sJ |dkr� fdd�| D �S |dkr4� fdd�| D �S � fdd�| D �S dS )zB
Perform ljust, center, rjust against string or list-like
r c s g | ]}|j � ��qS r )�ljust)r r )r6 r r r E s zjustify.<locals>.<listcomp>�centerc s g | ]}|j � ��qS r )r8 )r r )r6 r r r G s c s g | ]}|j � ��qS r )�rjust)r r )r6 r r r I s Nr )r5 r6 r r )r6 r r& @ s
r& )�seq� _nest_lvl�
max_seq_itemsr c s� t | t�rd}nt| d�r dnd}�dkr4t| �}n�pFtd�pFt| �}t| ��� ���fdd�tt|t| ���D �}d j|�}|t| �k r�|d
7 }n t | t �r�t| �d kr�|d 7 }|j
|d
�S )z�
internal. pprinter for iterables. you should probably use pprint_thing()
rather than calling this directly.
bounds length of printed sequence, depending on options
z
{{{body}}}� __setitem__z[{body}]z({body})Fr< c s* g | ]"}t t��� d fd�i����qS )r r< )� pprint_thing�next)r r/ )r; �kwdsr< �sr r r z s z _pprint_seq.<locals>.<listcomp>z, z, ...r �,)�body) �
isinstance�set�hasattrr% r
�iter�range�minr+ �tuple�format)r: r; r< r@ �fmt�nitems�rrC r )r; r@ r< rA r � _pprint_seqd s
rO c
K s� d}g }d}|dkr t | �}n|p0td�p0t | �}x`t| j� �d|� D ]H\}} |j|jt||d fd|i|��t| |d fd|i|��d�� qHW |t | �k r�|jdj|�d d
�S |jdj|�d
�S dS ) zy
internal. pprinter for iterables. you should probably use pprint_thing()
rather than calling this directly.
z {{{things}}}z {key}: {val}Fr< Nr )�key�valz, z, ...)Zthings)r% r
�list�itemsr' rK r> r+ )
r: r; r< r@ rL �pairsZpfmtrM �k�vr r r � _pprint_dict� s
$ rW F)�thingr; � escape_chars�default_escapes�
quote_stringsr< r c s� |ft tt td�� fdd�
}t| d�r0t| �S t| t�rX|td�k rXt| |d|d�}nPt | �r�|td�k r�t
| ||||d�}n(t| t�r�|r�d || �� d �}n|| �}|S )
a
This function is the sanctioned way of converting objects
to a string representation and properly handles nested sequences.
Parameters
----------
thing : anything to be formatted
_nest_lvl : internal use only. pprint_thing() is mutually-recursive
with pprint_sequence, this argument is used to keep track of the
current nesting level, and limit it.
escape_chars : list or dict, optional
Characters to escape. If a dict is passed the values are the
replacements
default_escapes : bool, default False
Whether the input escape characters replaces or adds to the defaults
max_seq_items : int or None, default None
Pass through to other pretty printers to limit sequence printing
Returns
-------
str
)rX rY r c sl dddd�}t |t�r8� r&|j|� n|}t|j� �}n
|p@t� }t| �}x |D ]}|j||| �}qPW |S )Nz\tz\nz\r)� r" �
)rD �dict�updaterR �keysrJ �str�replace)rX rY � translate�result�c)rZ r r �as_escaped_string� s
z'pprint_thing.<locals>.as_escaped_string�__next__zdisplay.pprint_nest_depthT)r[ r< )rY r[ r< �') r r � EscapeCharsra rF rD r^ r
rW r rO )rX r; rY rZ r[ r< rf rd r )rZ r r> � s&
r> �utf-8rb )�encoding�errorsr c C s t | �}|j||�S )N)r> �encode)�objectrk rl �valuer r r �pprint_thing_encoded� s rp )�enabler c C s� dt jkrd S ddlm} |� }|d kr,d S |jj}d}| rv||krjddlm} G dd� d|�}|� ||< d|| _n||kr�d || _d S )
N�IPythonr )� get_ipythonz!application/vnd.dataresource+json)�
BaseFormatterc @ s e Zd ZdZefZdS )z=_enable_data_resource_formatter.<locals>.TableSchemaFormatterZ_repr_data_resource_N)�__name__�
__module__� __qualname__Z print_methodr^ Z _return_typer r r r �TableSchemaFormatter s rx TF) �sys�modulesrr rs Zdisplay_formatter�
formattersZIPython.core.formattersrt Zenabled)rq rs �ipr{ �mimetypert rx r r r � _enable_data_resource_formatter� s
r~ )rX r< r c C s t | dd|d�S )Nr\ r] r" T)rY r[ r< )r\ r] r" )r> )rX r< r r r �default_pprint s
r T)� formatter�
is_justify�name�indent_for_name�line_break_each_valuer c s
ddl m} ddlm} |� \}} |dkr6td�p4d}|dkrHt| �j}|rzt|�}
dd|
d � �} dd|
d
� �} nd} d } t| �}
|r�d dt|� }nd
}td�p�|
}|
|k}|� � tttt tt
ttf d�� fdd� }t t t d�� fdd� }d}|
dk�rd|� �}�n�|
d k�rH| �rH�| d �}d|� d|� �}�n�|
d
k�r�| �r��| d �}�| d% �}d|� d|� d|� �}�nz|
|k�r�t |d
d�}
�fdd�| d|
� D �}�fd d�| |
d� D �}ng }�fd d�| D �}|�rt|�rt
||�\}}nf|�s>tdj|��|k �o8tdj|��|k �rtt||�||����fd d�|D �}�fd d�|D �}|�r�|t| � }|d }x<ttd t|�d ��D ]"�t|�d �}t|�|k �r�P �q�W �fd!d�|D �}�fd"d�|D �}d#}| } x:tt|��D ]*�|� | d } ||| | || �\}} �qW |�rR|| j� | d$ 7 }| } x>tt|�d �D ]*�|� | d } ||| | || �\}} �qdW ||| |d& |d
| �\}} || 7 }d|jd� }||7 }t|�|k�s�|�r�|| 7 }n|d7 }d|t| �d� }|S )'a
Return the formatted obj as a unicode string
Parameters
----------
obj : object
must be iterable and support __getitem__
formatter : callable
string formatter for an element
is_justify : boolean
should justify the display
name : name, optional
defaults to the class name of the obj
indent_for_name : bool, default True
Whether subsequent lines should be be indented to
align with the name.
line_break_each_value : bool, default False
If True, inserts a line break for each value of ``obj``.
If False, only break lines when the a line of values gets wider
than the display width.
.. versionadded:: 0.25.0
Returns
-------
summary string
r )�get_console_size)�_get_adjustmentNz
display.width�P r" r r � z
z,
rB zdisplay.max_seq_items)rA �linero �
display_width�next_line_prefixr c s@ � j |j� �� j |j� � |kr0| |j� 7 } |}||7 }| |fS )N)r% �rstrip)rA r� ro r� r� )�adjr r � _extend_linec s
z+format_object_summary.<locals>._extend_line)�valuesr c s" | rt � fdd�| D ��S dS d S )Nc 3 s | ]}� j |�V qd S )N)r% )r r )r� r r � <genexpr>o s z:format_object_summary.<locals>.best_len.<locals>.<genexpr>r )r )r� )r� r r �best_lenm s z'format_object_summary.<locals>.best_lenz, z[]�[�]�
c s g | ] }� |��qS r r )r r )r� r r r � s z)format_object_summary.<locals>.<listcomp>c s g | ] }� |��qS r r )r r )r� r r r � s c s g | ] }� |��qS r r )r r )r� r r r � s c s g | ]}|j � ��qS r )r9 )r r )�
max_lengthr r r � s c s g | ]}|j � ��qS r )r9 )r r )r� r r r � s )r< c s g | ]}t |� d ��qS ))r< )rO )r r )� max_itemsr r r � s c s g | ]}t |� d ��qS ))r< )rO )r r )r� r r r � s r! z...r# r# )Zpandas.io.formats.consoler� Zpandas.io.formats.formatr� r
�typeru r% ra �intr
r rI �_justifyr+ r �reversedrH rO r� ) �objr� r� r� r� r� r� r� r� �_Zname_lenZspace1Zspace2�n�sepr< Z is_truncatedr� r� �close�summary�first�last�head�tailZ max_spacero Z pprinted_seqr� �wordr )r� r� r� r� r �format_object_summary s� #
r� .)r� r� r c sv | | }dgt |d � � x.|D ]&}dd� |D �}dd� t� |�D �� q W � fdd�| D �} � fdd�|D �}| |fS )a�
Justify items in head and tail, so they are right-aligned when stacked.
Parameters
----------
head : list-like of list-likes of strings
tail : list-like of list-likes of strings
Returns
-------
tuple of list of tuples of strings
Same as head and tail, but items are right aligned when stacked
vertically.
Examples
--------
>>> _justify([['a', 'b']], [['abc', 'abcd']])
([(' a', ' b')], [('abc', 'abcd')])
r c S s g | ] }t |��qS r )r% )r �itemr r r r � s z _justify.<locals>.<listcomp>c S s g | ]\}}t ||��qS r )r )r r �yr r r r � s c s$ g | ] }t d d� t|� �D ���qS )c s s | ]\}}|j |�V qd S )N)r9 )r r r6 r r r r� � s z&_justify.<locals>.<listcomp>.<genexpr>)rJ r* )r r: )r� r r r � s c s$ g | ] }t d d� t|� �D ���qS )c s s | ]\}}|j |�V qd S )N)r9 )r r r6 r r r r� � s z&_justify.<locals>.<listcomp>.<genexpr>)rJ r* )r r: )r� r r r � s )r% r* )r� r� ZcombinedZ inner_seq�lengthr )r� r r� � s
r� )r� �
include_dtyper c C s� g }t | d�r*|r*|jdd| j� d�f� t| dd�dk rP|jdt| j�f� n.t| dd�dk r~t| j�r~|jdt| j�f� td�p�t | �}t | �|kr�|jdt | �f� |S )a:
Return a list of tuples of the (attr, formatted_value)
for common attrs, including dtype, name, length
Parameters
----------
obj : object
must be iterable
include_dtype : bool
If False, dtype won't be in the returned list
Returns
-------
list of 2-tuple
�dtyperh r� N�nameszdisplay.max_seq_itemsr� )
rF r' r� �getattrr r� �anyr� r
r% )r� r� �attrsr< r r r �format_object_attrs� s r� c @ s e Zd ZdZed�dd�ZdS )�
PrettyDictz.Dict extension to support abbreviated __repr__)r c C s t | �S )N)r> )�selfr r r �__repr__ s zPrettyDict.__repr__N)ru rv rw �__doc__ra r� r r r r r� s r� )r4 )r N)r N)r NFFN)rj rb )N)TNTF)T)%r� ry Ztypingr r r r r r r r r
r r Zpandas._configr
Z pandas.core.dtypes.inferencer ra ri r r r� r3 r& rO rW �boolr> �bytesrp r~ r r� r� r� r� r r r r �<module> sD 4 # % " C! '.,"" |