관리-도구
편집 파일: fw_transaction.cpython-36.pyc
3 @)�f� � @ 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