U
S��W� � @ sX d Z ddlmZ G dd� de�Ze� Ze� Ze� Ze� Z e� Z
e� Z G dd� de�ZdS )z.
Provide django-style hooks for model events.
� )�Modelc @ sB e Zd Zdd� Zddd�Zddd�Zddd �Zd
d � Zd d
� ZdS )�Signalc C s | � � d S �N)�_flush��self� r � /signals.py�__init__ s zSignal.__init__Nc C s@ |p|j }|| jkr0||f| j|< | j�|� n td| ��d S )Nz#receiver named %s already connected)�__name__�
_receivers�_receiver_list�append�
ValueError)r �receiver�name�senderr r r �connect s
zSignal.connectc C s0 |r
|j }|r$| j|= | j�|� ntd��d S )Nz%a receiver or a name must be provided)r r r
�remover )r r r r r r �
disconnect s zSignal.disconnectc s � ��fdd�}|S )Nc s �� | � �� | S r )r )�fn�r r r r r � decorator s z"Signal.__call__.<locals>.decoratorr )r r r r r r r �__call__ s zSignal.__call__c O sX t |�}g }| jD ]@}| j| \}}|d ks6t||�r|�||||f|�|�f� q|S r )�typer
r �
isinstancer ) r �instance�args�kwargsr Z responsesr �r�sr r r �send" s
z Signal.sendc C s i | _ g | _d S r )r r
r r r r r + s z
Signal._flush)NN)NN)NN) r �
__module__� __qualname__r
r r r r! r r r r r r s
r c s@ e Zd Z� fdd�Z� fdd�Z� fdd�Z� fdd�Z� ZS ) r c s t t| �j||� t�| � d S r )�superr r
�pre_initr! )r r r �� __class__r r r
9 s zModel.__init__c s t t| ��� t�| � d S r )r$ r �prepared� post_initr! r r& r r r( = s zModel.preparedc sP | � � }|�dd�p t|� }tj| |d� tt| �j||�}tj| |d� |S )NZ force_insertF)�created) Z
_get_pk_value�get�bool�pre_saver! r$ r �save� post_save)r r r Zpk_valuer* �retr& r r r. A s z
Model.savec s* t �| � tt| �j||�}t�| � |S r )�
pre_deleter! r$ r �delete_instance� post_delete)r r r r0 r& r r r2 I s
zModel.delete_instance)r r" r# r
r( r. r2 �
__classcell__r r r& r r 8 s r N) �__doc__Zpeeweer Z_Model�objectr r- r/ r1 r3 r% r) r r r r �<module> s )