File: //var/opt/nydus/ops/primordial/__pycache__/sizes.cpython-39.pyc
a
�,�h
1 � @ s� d dl Z d dlmZmZmZ g d�ZG dd� d�ZG dd� de�ZG dd � d e�ZG d
d � d e�Z G d d
� d
e�Z
G dd� d�Z dS )� N)�List�Optional�Tuple) �B�KiB�MiB�GiB�TiB�PiB�EiBZZiBZYiBc
@ s� e Zd ZdZdTeeeeeeedd�dd�Zeeeef d�dd �Z ed�d
d �Z
ed�d d
�Z ed�dd�Z ed�dd�Z
ed�dd�Zeed�dd��Zeed�dd��Zeed�dd��Zeed�dd��Zeed�d d ��Zeed�d d ��Zeed�d d!��Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Z d,d-� Z d.d/� Z d0d1� Z d2d3� Z d4d5� Z!d6d7� Z"d8d9� Z#d:d;� Z$d<d=� Z%d>d?� Z&d@dA� Z'dBdC� Z(dDdE� Z)dFdG� Z*dHdI� Z+dJdK� Z,dLdM� Z-dNdO� Z.dPdQ� Z/dRdS� Z0dS )U�ByteSizez*Encapsulation for tracking sizes in bytes.r N)�Bytes�KiBytes�MiBytes�GiBytes�TiBytes�PiBytes�EiBytes�returnc C sV t |�dt |�dt |�dt |�dt |�dt |�dt |� | _d S )N� )�int�_bytes)�selfr
r r r r r r � r �A/opt/nydus/tmp/pip-target-wkfpz8uv/lib/python/primordial/sizes.py�__init__
s
������zByteSize.__init__�r c C sN g }| j }tD ]4}|d@ }|r.|�d||f� |dL }|dkr qDq|pLdgS )Ni� r �
)r � )r �
_sizeUnits�insert)r �sresZcsize�sizeZsvalr r r �
_breakdown s zByteSize._breakdownc C s d� dd� | �� D ��S )N�,c S s g | ] }d | �qS ��%d%sr ��.0Zeachr r r �
<listcomp>* � z$ByteSize.__str__.<locals>.<listcomp>��joinr# �r r r r �__str__) s zByteSize.__str__c C s$ d� | jjd�dd� | �� D ���S )Nz{}({})r$ c S s g | ]\}}d � |pd|��qS )z {}ytes={}r )�format)r( �V�Ur r r r) / r* z%ByteSize.__repr__.<locals>.<listcomp>)r/ � __class__�__name__r, r# r- r r r �__repr__, s �zByteSize.__repr__c C s@ | � � }dt|�dkr"|d d n|d d d |d d f S )z-Return the size in simple form (int ceiling).r& � r )r# �len)r �bdr r r �
simpleSizeStr2 s zByteSize.simpleSizeStrc C s� | � � }t|�dksRt�|d d �t�|d d �d kr^t�|d d �dkr^d|d S |d d d |d d d }d| }|�d�r�|dd � |d d S ||d d S )
zAReturn the size in floating point form to two significant digits.r5 r r r& � �@z%.2fz.00N���)r# r6 r �index�endswith)r r7 ZfvZrspr r r �
approxSizeStr7 s ��
zByteSize.approxSizeStrc C s d� dd� | �� D ��S )z%Return the size in full detail units.r$ c S s g | ] }d | �qS r% r r' r r r r) E r* z(ByteSize.fullSizeStr.<locals>.<listcomp>r+ r- r r r � fullSizeStrC s zByteSize.fullSizeStrc C s | j S �N�r r- r r r r
G s zByteSize.Bytesc C s
| j d S �Nr9 �r
r- r r r r K s zByteSize.KiBytesc C s
| j d S rA )r r- r r r r O s zByteSize.MiBytesc C s
| j d S rA )r r- r r r r S s zByteSize.GiBytesc C s
| j d S rA )r r- r r r r W s zByteSize.TiBytesc C s
| j d S rA )r r- r r r r [ s zByteSize.PiBytesc C s
| j d S rA )r r- r r r r _ s zByteSize.EiBytesc C s | j S r? r@ r- r r r � __trunc__c s zByteSize.__trunc__c C s | j S r? r@ r- r r r � __index__f s zByteSize.__index__c C sB |d u r| j dkS z| j |j kW S ty< | j |k Y S 0 d S �Nr �r �AttributeError�r �or r r �__eq__j s
zByteSize.__eq__c C sB |d u r| j dkS z| j |j kW S ty< | j |k Y S 0 d S rE rF rH r r r �__ne__r s
zByteSize.__ne__c C sB |d u r| j dkS z| j |j kW S ty< | j |k Y S 0 d S rE rF rH r r r �__gt__z s
zByteSize.__gt__c C sB |d u r| j dkS z| j |j kW S ty< | j |k Y S 0 d S rE rF rH r r r �__ge__� s
zByteSize.__ge__c C s< |d u r dS z| j |j k W S ty6 | j |k Y S 0 d S )NFrF rH r r r �__lt__� s zByteSize.__lt__c C sB |d u r| j dkS z| j |j kW S ty< | j |k Y S 0 d S rE rF rH r r r �__le__� s
zByteSize.__le__c C s, t |d�r | �| j|j �S | �| j| �S �Nr )�hasattrr2 r rH r r r �__add__� s
zByteSize.__add__c C s. t |d�r | j|j7 _n| j|7 _| S rP )rQ r rH r r r �__iadd__� s
zByteSize.__iadd__c C s< t |d�r$| jtd| j|j �d�S | jtd| j| �d�S )Nr r rB )rQ r2 �maxr rH r r r �__sub__� s
zByteSize.__sub__c C s4 t |d�r"| jt|jd�8 _n| j|8 _| S )Nr r )rQ r rT rH r r r �__isub__� s
zByteSize.__isub__c C s | j | j| d�S �NrB �r2 r rH r r r �__mul__� s zByteSize.__mul__c C s | j | j| d�S rW rX rH r r r �__rmul__� s zByteSize.__rmul__c C s | j |9 _ | S r? r@ rH r r r �__imul__� s zByteSize.__imul__c C s* t |t�r| j|j S | �t| j| ��S r? ��
isinstancer r r2 r rH r r r � __floordiv__� s
zByteSize.__floordiv__c C s | � | jt|t�r|jn| �S r? )r2 r r] r rH r r r �__mod__� s zByteSize.__mod__c C s | | | | fS r? r rH r r r �
__divmod__� s zByteSize.__divmod__c C s. t |t�r| jd |j S | �t| j| ��S )Ng �?r\ rH r r r � __truediv__� s
zByteSize.__truediv__c C s | j | _ t| j �| _ | S r? �r r rH r r r �__idiv__� s zByteSize.__idiv__c C s
| � |�S r? )rc rH r r r � __itruediv__� s zByteSize.__itruediv__c C s | j | _ t| j �| _ | S r? rb rH r r r �
__ifloordiv__� s zByteSize.__ifloordiv__c C s | j |; _ t| j �| _ | S r? rb rH r r r �__imod__� s zByteSize.__imod__c C s
t | j�S r? ��boolr r- r r r � __nonzero__� s zByteSize.__nonzero__c C s
t | j�S r? rg r- r r r �__bool__� s zByteSize.__bool__)r r r r r r r )1r3 �
__module__� __qualname__�__doc__r r r r �strr# r. r4 r8 r= r> �propertyr
�floatr r r r r r rC rD rJ rK rL rM rN rO rR rS rU rV rY rZ r[ r^ r_ r` ra rc rd re rf ri rj r r r r r s| ��
r c @ s e Zd ZdZdS )�MemSizez+Encapsulation for tracking amount of memoryN�r3 rk rl rm r r r r rq � s rq c @ s e Zd ZdZdS )�DiskSizez<Encapsulation for tracking size of persistent storage (disk)Nrr r r r r rs � s rs c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �InvalidMemSizec C s
|| _ d S r? ��_invVal�r Z
invalid_valuer r r r � s zInvalidMemSize.__init__c C s dt | j� S )NzInvalidMemSize "%s"�rn rv r- r r r r. � s zInvalidMemSize.__str__c C s dt | j� S )NzInvalidMemSize(%s)��reprrv r- r r r r4 � s zInvalidMemSize.__repr__c C s
t |t�S r? �r] rt rH r r r rJ s zInvalidMemSize.__eq__c C s t |t� S r? r{ rH r r r rK s zInvalidMemSize.__ne__N�r3 rk rl r r. r4 rJ rK r r r r rt � s
rt c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �InvalidDiskSizec C s
|| _ d S r? ru rw r r r r
s zInvalidDiskSize.__init__c C s dt | j� S )NzInvalidDiskSize "%s"rx r- r r r r.
s zInvalidDiskSize.__str__c C s dt | j� S )NzInvalidDiskSize(%s)ry r- r r r r4 s zInvalidDiskSize.__repr__c C s
t |t�S r? �r] r} rH r r r rJ s zInvalidDiskSize.__eq__c C s t |t� S r? r~ rH r r r rK s zInvalidDiskSize.__ne__Nr| r r r r r} s
r} c @ s� e Zd ZdZd edd�dd�Zdd� Zed �d
d �Zee d �d d
��Z
ee d �dd��Z ee d �dd��Z ed �dd�Z
ed �dd�Zd ee ed�dd�Zd!ee eed�dd �Zd"ee eed�d d �ZdS )#� ByteSizesz�Encapsulation for an aggregation of byte size values.
The naturalUnits indicates which units to report sizes in if the size is
not indicated or is a close approximation of those units (naturalUnits
should be one of _sizeUnits)
r N)� naturalUnitsr c C s0 t d�| _d | _t d�| _d| _|| _d | _d S rE )r �_accum�_minSize�_maxSize� _numSizes�_natUnit� _sumUnits)r r� r r r r " s
zByteSizes.__init__c C sd t |t�st|d�}| j|7 _| jd7 _| jd u sD|| jk rJ|| _|| jkrZ|| _d | _| S )NrB r5 )r] r r� r� r� r� r� rH r r r rS * s
zByteSizes.__iadd__r c C s
t | j�S r? )rn r� r- r r r r. 6 s zByteSizes.__str__c C s | j d u rtd�S | j S rE )r� r r- r r r � minimumSize9 s zByteSizes.minimumSizec C s | j S r? )r� r- r r r � maximumSize= s zByteSizes.maximumSizec C s | j dkrtd�S | j| j S rE )r� r r� r- r r r � averageSizeA s zByteSizes.averageSizec C s| t t�d }tt t�d dd�D ]T}td| d�}|| jkr>q |d t�| j�krj| j| dkrj| j S t| S | jS )Nr5 ���r rB �d )r6 r �ranger r� r; r� )r ZsuIZbaseSizer r r �_calcSummaryUnitsE s
"
zByteSizes._calcSummaryUnitsc C s | j d u r| �� | _ | j S r? )r� r� r- r r r � summaryUnitsP s
zByteSizes.summaryUnits)�ofSizer c C s |du r| j �� S |�� S )zrReturn the specified size (or total size) in full detail units.
:param: The size to be displayed
N)r� r> )r r� r r r r> U s
zByteSizes.fullSizeStrF)r� � withUnitsr c C sf |du r| j �� S t�|j|j|j|j|j|j|j |j
d�| � � �}|rZd|| � � f S t t
|��S )ag Returns the specified size (or total size) in simple form (int ceiling) based on the summary Units.
The withUnits can be used to enable or suppress reporting of units in the output (units are summaryUnits() in
either case).
:param ofSize: The size to simplify
:param withUnits: Whether to include units in the output
N�r r r r r r r
r r& )r� r8 �math�ceilr
r r r r r r r� rn r �r r� r� �valr r r r8 ^ s
��zByteSizes.simpleSizeStrc
C sd |du r| j �� S t|j|j|j|j|j|j|j|j d�| �
� d �d }d||r\| �
� ndf S )a� Give a string representation of a close approximation of the size.
Returns the specified size (or total size) in floating point form to two significant digits, based on the on
the summary Units. The withUnits can be used to enable or suppress reporting of units in the output (units are
summaryUnits() in either case).
:param ofSize: The size to be represented
:param withUnits: Whether to include units in the output
Nr� r9 z%.2f%sr ) r� r= r r
r r r r r r r� r� r r r r= u s$
����zByteSizes.approxSizeStr)r )N)NF)NF)r3 rk rl rm rn r rS r. ro r r� r� r� r� r� r r> rh r8 r= r r r r r s r ) r� �typingr r r r r rq rs rt r} r r r r r �<module> s c