3
�Pf� � @ s d dl mZ G dd� de�ZdS )� )�BytesIOc @ sB e Zd ZdZdd� Zdd� Zdd� Zdd � Zdd d �Zd
d� Z d
S )�CallbackFileWrapperav
Small wrapper around a fp object which will tee everything read into a
buffer, and when that file is closed it will execute a callback with the
contents of that buffer.
All attributes are proxied to the underlying file object.
This class uses members with a double underscore (__) leading prefix so as
not to accidentally shadow an attribute.
c C s t � | _|| _|| _d S )N)r �_CallbackFileWrapper__buf�_CallbackFileWrapper__fp� _CallbackFileWrapper__callback)�self�fp�callback� r
�!/usr/lib/python3.6/filewrapper.py�__init__ s z CallbackFileWrapper.__init__c C s | j d�}t||�S )Nr )�__getattribute__�getattr)r �namer r
r
r � __getattr__ s
z CallbackFileWrapper.__getattr__c C sD y | j jd kS tk
r Y nX y| j jS tk
r> Y nX dS )NF)r r �AttributeError�closed)r r
r
r Z__is_fp_closed! s z"CallbackFileWrapper.__is_fp_closedc C s | j r| j | jj� � d | _ d S )N)r r �getvalue)r r
r
r �_close0 s zCallbackFileWrapper._closeNc C s, | j j|�}| jj|� | j� r(| j� |S )N)r �readr �write�"_CallbackFileWrapper__is_fp_closedr )r �amt�datar
r
r r ; s
zCallbackFileWrapper.readc C s@ | j j|�}|dkr |dkr |S | jj|� | j� r<| j� |S )N� s
)r �
_safe_readr r r r )r r r r
r
r r C s z CallbackFileWrapper._safe_read)N)
�__name__�
__module__� __qualname__�__doc__r r r r r r r
r
r
r r s
r N)�ior �objectr r
r
r
r � s