a

U��Zt�@sVdZd
dl
Z
zd
dlmZ
Wney:


d
dlmZ
Yn0Gdd�d�Zdd�ZdS)	z*
This module implements the SmartIO class
�N)
�StringIO)
�BytesIOc@s:eZ
dZd
Zddd�
Zdd�Zdd	�Zd
d�Zdd
�ZdS)�SmartIOa�

    The SmartIO class allows one to put a cap on the memory consumption.
    StringIO objects are very fast, because they are stored in memory, but
    if they are too big the memory footprint becomes noticeable.
    The write method of a SmartIO determines if the data that is to be added
    to the (initially) StrintIO object does not exceed a certain threshold; if
    it does, it switches the storage to a temporary disk file
    �@r
cCs$|
|_t
�|_|rd
|_nd|_dS)N�
r
)�
_max_mem_sizer�_io�_fixed)�self�max_mem_sizeZ	force_mem�r�//usr/lib/python3.9/site-packages/rhn/SmartIO.py�__init__s



zSmartIO.__init__cCs
|
|_dS�
N�
r)r
rrrr
�set_max_mem_size's
zSmartIO.set_max_mem_sizec


Cs|jSrr)
r
rrr
�get_max_mem_size*s
zSmartIO.get_max_mem_sizecCsP|js@t
|
�
|j��|jkr@t�}|�|j���

d
|_||_|j�|
�

dS)Nr)r	�lenr�tellr�	_tempfile�write�getvalue)r
�dataZtmpfilerrr
r-s



z
SmartIO.writecCst|j
|
�Sr)�getattrr)r
�namerrr
�__getattr__9s
zSmartIO.__getattr__N)rr
)	�__name__�
__module__�__qualname__�__doc__rrrrrrrrr
rs

rcCs6d
dl}|j
dt��d�
\}
}t�|�

t�|
d�S)Nr
z_rhn_transports-%d-)
�prefixzwb+)�tempfileZmkstemp�os�getpid�unlink�fdopen)r!�fd�fnamerrr
r=s


�

r)	rr"Z	cStringIOr�ImportError�iorrrrrrr
�<module>s



+