3
��g� � @ sJ d Z dgZddlZddlmZ ddlmZ ddlmZ G dd� de �Z
dS )z!Transaction classes for firewalld�FirewallTransaction� N)�log)�errors)�
FirewallErrorc @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#S )$r c C s( || _ i | _g | _g | _g | _g | _d S )N)�fw�rules� pre_funcs�
post_funcs�
fail_funcs�modules)�selfr � r
�$/usr/lib/python3.6/fw_transaction.py�__init__! s z FirewallTransaction.__init__c C s2 | j j� | jd d �= | jd d �= | jd d �= d S )N)r �clearr r r
)r r
r
r r ) s
zFirewallTransaction.clearc C s | j j|jg �j|� d S )N)r �
setdefault�name�append)r �backend�ruler
r
r �add_rule/ s z FirewallTransaction.add_rulec C s x|D ]}| j ||� qW d S )N)r )r r r r r
r
r � add_rules2 s
z FirewallTransaction.add_rulesc C s |j | jko|| j|j kS )N)r r )r r r r
r
r �
query_rule6 s z FirewallTransaction.query_rulec C s2 |j | jkr.|| j|j kr.| j|j j|� d S )N)r r �remove)r r r r
r
r � remove_rule9 s z FirewallTransaction.remove_rulec G s | j j||f� d S )N)r r )r �func�argsr
r
r �add_pre= s zFirewallTransaction.add_prec G s | j j||f� d S )N)r r )r r r r
r
r �add_post@ s z FirewallTransaction.add_postc G s | j j||f� d S )N)r
r )r r r r
r
r �add_failC s z FirewallTransaction.add_failc C s || j kr| j j|� d S )N)r r )r �moduler
r
r �
add_moduleF s
z FirewallTransaction.add_modulec C s || j kr| j j|� d S )N)r r )r r r
r
r �
remove_moduleJ s
z!FirewallTransaction.remove_modulec C s x|D ]}| j |� qW d S )N)r! )r r r r
r
r � add_modulesN s
z FirewallTransaction.add_modulesc C s x|D ]}| j |� qW d S )N)r" )r r r r
r
r �remove_modulesR s
z"FirewallTransaction.remove_modulesc C s� t jdt| �|df � i }|sjxp| jD ]<}x6t| j| �D ]$}|j|g �j| jj|�j |�� q<W q(W n(x&| jD ] }|j|g �j
| j| � qrW || j fS )Nz%s.prepare(%s, %s)z...) r �debug4�typer �reversedr r r �get_backend_by_name� reverse_rule�extendr )r �enabler � backend_namer r
r
r �prepareV s zFirewallTransaction.preparec C s t jdt| �|f � | j|�\}}| j� d}d}g }xp|D ]h}y| jj||| � W nB tk
r� } z&d}|}t jt j
� � t j |� W Y d d }~X q>X |j |� q>W |s�| jj
||�} | r�| \}
}|
r�t j|� |�ri } xH|D ]@}g | |< x2t|| �D ]"} | | j | jj|�j| �� �q W q�W xb| D ]Z}y| jj|| | � W n< tk
�r� } z t jt j
� � t j |� W Y d d }~X nX �q0W xh| jD ]^\}
}y |
|� W nF tk
�r� } z(t jt j
� � t j d|
||f � W Y d d }~X nX �q�W ttj|��| j� d S )Nz%s.execute(%s)F� Tz#Calling fail func %s(%s) failed: %s)r r% r& r- �prer r � Exception�debug1� traceback�
format_exc�errorr Zhandle_modulesr' r( r) r
r r ZCOMMAND_FAILED�post)r r+ r r r4 ZerrorMsg�doner, �msgZ
module_returnZstatusZ
undo_rulesr r r r
r
r �executef sV
" & zFirewallTransaction.executec C s| t jdt| � � xd| jD ]Z\}}y ||� W q tk
rr } z(t jtj� � t jd|||f � W Y d d }~X qX qW d S )Nz%s.pre()z"Calling pre func %s(%s) failed: %s) r r% r& r r0 r1 r2 r3 r4 )r r r r7 r
r
r r/ � s zFirewallTransaction.prec C s| t jdt| � � xd| jD ]Z\}}y ||� W q tk
rr } z(t jtj� � t jd|||f � W Y d d }~X qX qW d S )Nz %s.post()z#Calling post func %s(%s) failed: %s) r r% r& r r0 r1 r2 r3 r4 )r r r r7 r
r
r r5 � s zFirewallTransaction.postN)�__name__�
__module__� __qualname__r r r r r r r r r r! r" r# r$ r- r8 r/ r5 r
r
r
r r s" @ ) �__doc__�__all__r2 Zfirewall.core.loggerr Zfirewallr Zfirewall.errorsr �objectr r
r
r
r �<module> s