File: //home/arjun/.local/lib/python3.10/site-packages/tqdm/__pycache__/utils.cpython-310.pyc
o
���g-. � @ s� d Z ddlZddlZddlZddlmZmZmZ ddlm Z ddl
m Z ddl m
Z
ddlmZ eeeef\ZZZZejZedd � d
D ��Zed d � d D ��Ze� d
�Z z er\ddl Z ne �W n e yj dZ Y nw ze j dd� W n
e!y� e � � Y nw d6dd�Z"G dd� de#�Z$G dd� de#�Z%G dd� de#�Z&G dd� de&�Z'G dd� de&�Z(G d d � d e&�Z)d d � Z*d d!� Z+d"d#� Z,d$d%� Z-d&d'� Z.d(d)� Z/d*d+� Z0d,d-� Z1d.d/� Z2d0d1� Z3d2d3� Z4d4d5� Z5dS )7z*
General helpers required for `tqdm.std`.
� N)�partial�
partialmethod�wraps)� signature��east_asian_width)�warn��proxyc c � � | ]}t �|�V qd S �N��CUR_OS�
startswith��.0�i� r �=/home/arjun/.local/lib/python3.10/site-packages/tqdm/utils.py� <genexpr> � � r )�win32�cygwinc c r r r
r r r r r r )�aix�linux�darwin�freebsdz\x1b\[[;\d]*[A-Za-z]F)�stripc sN �du ri �t �����fdd�tj�� D �� |rtnt�� ��fdd�}|S )a
Override parameter defaults via `os.environ[prefix + param_name]`.
Maps UPPER_CASE env vars map to lower_case param names.
camelCase isn't supported (because Windows ignores case).
Precedence (highest first):
- call (`foo(a=3)`)
- environ (`FOO_A=2`)
- signature (`def foo(a=1)`)
Parameters
----------
prefix : str
Env var prefix, e.g. "FOO_"
types : dict, optional
Fallback mappings `{'param_name': type, ...}` if types cannot be
inferred from function signature.
Consider using `types=collections.defaultdict(lambda: ast.literal_eval)`.
is_method : bool, optional
Whether to use `functools.partialmethod`. If (default: False) use `functools.partial`.
Examples
--------
```
$ cat foo.py
from tqdm.utils import envwrap
@envwrap("FOO_")
def test(a=1, b=2, c=3):
print(f"received: a={a}, b={b}, c={c}")
$ FOO_A=42 FOO_C=1337 python -c 'import foo; foo.test(c=99)'
received: a=42, b=2, c=99
```
Nc s, i | ]\}}|� ��r|� d � �� |�qS r )r �lower�r �k�v)r �prefixr r �
<dictcomp>I s , zenvwrap.<locals>.<dictcomp>c
s� t | �j� � fdd���� D �}|D ]S}� | }|j|jur?t|jd|jf�D ]}z
||| �||< W n ty< Y q'w q|jd urPt|j�|| �||< qz �| || �||< W q t ye Y qw �| fi |��S )Nc s i | ] \}}|� v r||�qS r r r ��paramsr r r# O � z)envwrap.<locals>.wrap.<locals>.<dictcomp>�__args__)
r �
parameters�items�
annotation�empty�getattr� Exception�default�type�KeyError)�func� overridesr �param�typ)�
env_overrides�part�typesr$ r �wrapL s*
��
�zenvwrap.<locals>.wrap)�len�os�environr) r r )r" r7 � is_methodr8 r )r5 r r6 r" r7 r �envwrap" s $ r= c @ s" e Zd ZdZddd�Zdd� ZdS ) �
FormatReplacezN
>>> a = FormatReplace('something')
>>> f"{a:5d}"
'something'
� c C s || _ d| _d S �Nr )�replace�
format_called)�selfrA r r r �__init__l s
zFormatReplace.__init__c C s | j d7 _ | jS )N� )rB rA )rC �_r r r �
__format__p s zFormatReplace.__format__N)r? )�__name__�
__module__� __qualname__�__doc__rD rG r r r r r> f s
r> c @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dS )�
Comparablez1Assumes child has self._comparable attr/@propertyc C s | j |j k S r �� _comparable�rC �otherr r r �__lt__w � zComparable.__lt__c C s | |k p| |kS r r rO r r r �__le__z s zComparable.__le__c C s | j |j kS r rM rO r r r �__eq__} rR zComparable.__eq__c C s
| |k S r r rO r r r �__ne__� �
zComparable.__ne__c C s
| |k S r r rO r r r �__gt__� rV zComparable.__gt__c C s
| |k S r r rO r r r �__ge__� rV zComparable.__ge__N)
rH rI rJ rK rQ rS rT rU rW rX r r r r rL u s rL c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �
ObjectWrapperc C s t | j|�S r )r, �_wrapped�rC �namer r r � __getattr__� rR zObjectWrapper.__getattr__c C s t | j||�S r )�setattrrZ �rC r\ �valuer r r � __setattr__� s zObjectWrapper.__setattr__c C s, zt �| |�W S ty t| |� Y S w )z7Actual `self.getattr` rather than self._wrapped.getattr)�objectr] �AttributeErrorr, r[ r r r �wrapper_getattr� s
�z ObjectWrapper.wrapper_getattrc C s t �| ||�S )z7Actual `self.setattr` rather than self._wrapped.setattr)rb ra r_ r r r �wrapper_setattr� s z ObjectWrapper.wrapper_setattrc C s | � d|� dS )z4
Thin wrapper around a given object
rZ N)re )rC �wrappedr r r rD � s zObjectWrapper.__init__N)rH rI rJ r] ra rd re rD r r r r rY � s rY c s0 e Zd ZdZ� fdd�Zdd� Zdd� Z� ZS )�SimpleTextIOWrapperz�
Change only `.write()` of the wrapped object by encoding the passed
value and passing the result to the wrapped object's `.write()` method.
c s t � �|� | �d|� d S )N�encoding)�superrD re )rC rf rh �� __class__r r rD � s z SimpleTextIOWrapper.__init__c C s | j �|�| �d���S )zP
Encode `s` and pass to the wrapped object's `.write()` method.
rh )rZ �write�encoderd )rC �sr r r rl � s zSimpleTextIOWrapper.writec C � | j t|d|�kS �NrZ �rZ r, rO r r r rT � � zSimpleTextIOWrapper.__eq__)rH rI rJ rK rD rl rT �
__classcell__r r rj r rg � s
rg c s4 e Zd ZdZedd� �Z� fdd�Zdd� Z� ZS )�DisableOnWriteErrorzO
Disable the given `tqdm_instance` upon `write()` or `flush()` errors.
c s t ���� �fdd�}|S )zV
Quietly set `tqdm_instance.miniters=inf` if `func` raises `errno=5`.
c s� z� | i |��W S t y9 } z%|jdkr� ztd��_W n ty& Y nw W Y d }~d S W Y d }~d S d }~w tyj } z&dt|�vrG� ztd��_W n tyW Y nw W Y d }~d S W Y d }~d S d }~ww )N� �inf�closed)�OSError�errno�float�miniters�ReferenceError�
ValueError�str)�args�kwargs�e�r1 �
tqdm_instancer r �inner� s2
��� ����z7DisableOnWriteError.disable_on_exception.<locals>.innerr )r� r1 r� r r� r �disable_on_exception� s z(DisableOnWriteError.disable_on_exceptionc sT t � �|� t|d�r| �d| �||j�� t|d�r(| �d| �||j�� d S d S )Nrl �flush)ri rD �hasattrre r� rl r� )rC rf r� rj r r rD � s
�
��z DisableOnWriteError.__init__c C ro rp rq rO r r r rT � rr zDisableOnWriteError.__eq__) rH rI rJ rK � staticmethodr� rD rT rs r r rj r rt � s
rt c s e Zd Zd� fdd� Z� ZS )�CallbackIOWrapper�readc sz t � �|� t||��|dkr"t��� �fdd��}| �d|� dS |dkr9t��� �fdd��}| �d|� dS td��) z{
Wrap a given `file`-like object's `read()` or `write()` to report
lengths to the given `callback`
rl c s& �| g|�R i |��}� t | �� |S r �r9 )�datar r� �res��callbackr1 r r rl � s z)CallbackIOWrapper.__init__.<locals>.writer� c s �| i |��}� t |�� |S r r� )r r� r� r� r r r� � s z(CallbackIOWrapper.__init__.<locals>.readz Can only wrap read/write methodsN)ri rD r, r re r0 )rC r� �stream�methodrl r� rj r� r rD � s
zCallbackIOWrapper.__init__)r� )rH rI rJ rD rs r r rj r r� � s r� c C sf zd� | � W dS ty Y dS ty2 z| �� �d�p#d| kW Y S ty1 Y Y dS w w )Nu █▉Fzutf-�U8T)rm �UnicodeEncodeErrorr- r r )rh r r r �_is_utf� s � ��r� c C s" zt | j�W S ty Y dS w �NF)r� rh rc ��fpr r r �_supports_unicode
s
�r� c C s2 t | t�r| D ] }t|�dkr dS qdS t| �S )N� FT)�
isinstancer~ �ordr� )rn �cr r r � _is_ascii s
�r� c C s$ d} t r t} | du r t} trt} | S )zy
Return a function which returns console dimensions (width, height).
Supported: linux, osx, windows, cygwin.
N)�IS_WIN�_screen_shape_windows�_screen_shape_tput�IS_NIX�_screen_shape_linux)�
_screen_shaper r r �_screen_shape_wrapper s r� c C s� zTdd l }ddlm}m} ddlm}m} d}| |kr d}n| |kr$d}|j�|�}|d�}|j� ||�} | rR|�
d|j �\ }
} } }
}}}}}}}|| || fW S W d S t y^ Y d S w )
Nr )�create_string_buffer�windll)�stdin�stdouti�i����i����� � hhhhHhhhhhh�NN)
�struct�ctypesr� r� �sysr� r� �kernel32� GetStdHandle�GetConsoleScreenBufferInfo�unpack�rawr- )r� r� r� r� r� r� � io_handle�h�csbir� �_bufx�_bufy�_curx�_cury�_wattr�left�top�right�bottom�_maxx�_maxyr r r r� ) s. �� ��r� c s@ zddl �ddlm� � �fdd�dD �W S ty Y dS w )zcygwin xterm (windows)r N)�
check_callc s& g | ]}t � ��d | ���d �qS )ztput rE )�int�splitr �r� �shlexr r �
<listcomp>F s �z&_screen_shape_tput.<locals>.<listcomp>)�cols�linesr� )r� �
subprocessr� r- )rF r r� r r� A s � �r� c C s� zddl m } ddlm} ddlm} W n
ty Y dS w z|d|| |d��d d� \}}||fW S tyR z
d d
� d D �W Y S ttfyQ Y Y dS w w ) Nr )�array)�ioctl)�
TIOCGWINSZr� r� z � c S s g | ] }t tj| �d �qS )rE )r� r: r; r r r r r� [ r& z'_screen_shape_linux.<locals>.<listcomp>)�COLUMNS�LINES) r� �fcntlr� �termiosr� � ImportErrorr- r0 r} )r� r� r� r� �rowsr� r r r r� M s" �
��r� c s4 t dtdd� t� � � sdS t� �� fdd��} | S )zd
Return a function which returns console width.
Supported: linux, osx, windows, cygwin.
zQUse `_screen_shape_wrapper()(file)[0]` instead of `_environ_cols_wrapper()(file)`r� )�
stacklevelNc s � | �d S r@ r r� ��shaper r r� k s z$_environ_cols_wrapper.<locals>.inner)r �DeprecationWarningr� r )r� r r� r �_environ_cols_wrapper` s �r� c C s t jdkr td u r dS dS )N�ntr? z[A)r: r\ �coloramar r r r �
_term_move_upr s r� c C s t dd� t| �D ��S )Nc s s$ � | ]
}t |�d v r dndV qdS )�FWr� rE Nr )r �chr r r r w s �" z _text_width.<locals>.<genexpr>)�sumr~ )rn r r r � _text_widthv s r� c C s t t�d| ��S )zp
Returns the real on-screen length of a string which may contain
ANSI control codes and wide chars.
r? )r� �RE_ANSI�sub)r� r r r �disp_lenz s r� c C sz t | �t| �kr| d|� S tt�| ��}t| �|kr'| dd� } t| �|ks|r;tt�| ��r;| �d�r7| S | d S | S )zB
Trim a string which may contain ANSI control characters.
N���z[0m)r9 r� �boolr� �search�endswith)r� �length� ansi_presentr r r � disp_trim� s �r� r� )6rK r: �rer� � functoolsr r r �inspectr � unicodedatar �warningsr �weakrefr
�range�chrr~ �_range�_unich�_unicode� _basestring�platformr �anyr� r� �compiler� r� r� �init� TypeErrorr= rb r> rL rY rg rt r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r �<module> s\
� � �
D+