3 Y�j��@sJdZdgZddlZddlmZddlmZddlmZGdd�de �Z dS)z!Transaction classes for firewalld�FirewallTransaction�N)�log)�errors)� FirewallErrorc@s�eZdZdd�Zdd�Zdd�Zdd�Zd d �Zdd�Zd d�Z dd�Z dd�Zdd�Zdd�Z dd�Zdd�Zdd�Zdd�Zdd �Zd!d"�Zd#S)$rcCs(||_i|_g|_g|_g|_g|_dS)N)�fw�rules� pre_funcs� post_funcs� fail_funcs�modules)�selfr�r �$/usr/lib/python3.6/fw_transaction.py�__init__!szFirewallTransaction.__init__cCs2|jj�|jdd�=|jdd�=|jdd�=dS)N)r�clearrr r )rr r rr)s zFirewallTransaction.clearcCs|jj|jg�j|�dS)N)r� setdefault�name�append)r�backend�ruler r r�add_rule/szFirewallTransaction.add_rulecCsx|D]}|j||�qWdS)N)r)rrrrr r r� add_rules2s zFirewallTransaction.add_rulescCs|j|jko||j|jkS)N)rr)rrrr r r� query_rule6szFirewallTransaction.query_rulecCs2|j|jkr.||j|jkr.|j|jj|�dS)N)rr�remove)rrrr r r�remove_rule9szFirewallTransaction.remove_rulecGs|jj||f�dS)N)rr)r�func�argsr r r�add_pre=szFirewallTransaction.add_precGs|jj||f�dS)N)r r)rrrr r r�add_post@szFirewallTransaction.add_postcGs|jj||f�dS)N)r r)rrrr r r�add_failCszFirewallTransaction.add_failcCs||jkr|jj|�dS)N)rr)r�moduler r r� add_moduleFs zFirewallTransaction.add_modulecCs||jkr|jj|�dS)N)rr)rr r r r� remove_moduleJs z!FirewallTransaction.remove_modulecCsx|D]}|j|�qWdS)N)r!)rrr r r r�add_modulesNs zFirewallTransaction.add_modulescCsx|D]}|j|�qWdS)N)r")rrr r r r�remove_modulesRs z"FirewallTransaction.remove_modulescCs�tjdt|�|df�i}|sjxp|jD]<}x6t|j|�D]$}|j|g�j|jj|�j |��qX|j|�q>W|s�|jj ||�} | r�| \} }| r�tj|�|�ri}xH|D]@}g||<x2t||�D]"}||j|jj|�j|��qWq�Wxb|D]Z}y|jj|||�Wn<tk �r�}ztjt j ��tj|�WYdd}~XnX�q0Wxh|jD]^\} }y| |�WnFtk �r�}z(tjt j ��tjd| ||f�WYdd}~XnX�q�Wttj|��|j�dS)Nz%s.execute(%s)F�Tz#Calling fail func %s(%s) failed: %s)rr%r&r-�prerr� Exception�debug1� traceback� format_exc�errorrZhandle_modulesr'r(r)r rrZCOMMAND_FAILED�post)rr+rrr4ZerrorMsg�doner,�msgZ module_returnZstatusZ undo_rulesrrrr r r�executefsV "&zFirewallTransaction.executecCs|tjdt|��xd|jD]Z\}}y||�Wqtk rr}z(tjtj��tjd|||f�WYdd}~XqXqWdS)Nz%s.pre()z"Calling pre func %s(%s) failed: %s) rr%r&rr0r1r2r3r4)rrrr7r r rr/�szFirewallTransaction.precCs|tjdt|��xd|jD]Z\}}y||�Wqtk rr}z(tjtj��tjd|||f�WYdd}~XqXqWdS)Nz %s.post()z#Calling post func %s(%s) failed: %s) rr%r&r r0r1r2r3r4)rrrr7r r rr5�szFirewallTransaction.postN)�__name__� __module__�__qualname__rrrrrrrrrr!r"r#r$r-r8r/r5r r r rr s"@)�__doc__�__all__r2Zfirewall.core.loggerrZfirewallrZfirewall.errorsr�objectrr r r r�s