a
«°ië ã @ 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.9/site-packages/firewall/core/fw_transaction.pyÚ__init__! s z FirewallTransaction.__init__c C s2 | j ¡ | jd d …= | jd d …= | jd d …= d S r )r Úclearr r
r )r
r r r r ) s
zFirewallTransaction.clearc C s | j |jg ¡ |¡ d S r )r Ú
setdefaultÚnameÚappend©r
ÚbackendÚruler r r Úadd_rule/ s z FirewallTransaction.add_rulec C s |D ]}| ||¡ qd S r )r )r
r r r r r r Ú add_rules2 s z FirewallTransaction.add_rulesc C s |j | jv o|| j|j v S r )r r r r r r Ú
query_rule6 s z FirewallTransaction.query_rulec C s2 |j | jv r.|| j|j v r.| j|j |¡ d S r )r r Úremover r r r Ú remove_rule9 s z FirewallTransaction.remove_rulec G s | j ||f¡ d S r )r r ©r
ÚfuncÚargsr r r Úadd_pre= s zFirewallTransaction.add_prec G s | j ||f¡ d S r )r
r r r r r Úadd_post@ s z FirewallTransaction.add_postc G s | j ||f¡ d S r )r r r r r r Úadd_failC s z FirewallTransaction.add_failc C s || j vr| j |¡ d S r )r r ©r
Úmoduler r r Ú
add_moduleF s
z FirewallTransaction.add_modulec C s || j v r| j |¡ d S r )r r r# r r r Ú
remove_moduleJ s
z!FirewallTransaction.remove_modulec C s |D ]}| |¡ qd S r )r% ©r
r r$ r r r Ú add_modulesN s z FirewallTransaction.add_modulesc C s |D ]}| |¡ qd S r )r& r' r r r Úremove_modulesR s z"FirewallTransaction.remove_modulesc C s t dt| ƒ|df ¡ i }|sb| jD ]8}t| j| ƒD ]$}| |g ¡ | j |¡ |¡¡ q8q&n$| jD ] }| |g ¡
| j| ¡ qh|| 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 dt| ƒ|f ¡ | |¡\}}| ¡ d}d}g }|D ]j}z| j ||| ¡ W nD tyš } z,d}|}t t
¡ ¡ t |¡ W Y d }~q tyz } z$t t
¡ ¡ t |¡ W Y d }~n
d }~0 0 q"| jD ]`\}
}z |
|Ž W nH tyâ } z.t t
¡ ¡ t d|
||f ¡ W Y d }~n
d }~0 0 q†ttj|ƒ‚| ¡ d S )Nz%s.execute(%s)FÚ Tz#Calling fail func %s(%s) failed: %s)r r* r+ r2 Úprer r Ú ExceptionZdebug1Ú tracebackÚ
format_excÚerrorr Zhandle_modulesr, r- r. r r r ZCOMMAND_FAILEDÚpost)r
r0 r r r8 ZerrorMsgZdoner1 ÚmsgZ
module_returnÚstatusZ
undo_rulesr r r r r r Úexecutef sZ
ÿ$ ÿ zFirewallTransaction.executec C s. t dt| ƒ ¡ | jD ]\}}||Ž qd S )Nz%s.pre())r r* r+ r r r r r r4 ¦ s zFirewallTransaction.prec C s. t dt| ƒ ¡ | jD ]\}}||Ž qd S )Nz %s.post())r r* r+ r
r r r r r9 ¬ s zFirewallTransaction.postN)Ú__name__Ú
__module__Ú __qualname__r r r r r r r r! r" r% r& r( r) r2 r< r4 r9 r r r r r s" @) Ú__doc__Ú__all__r6 Zfirewall.core.loggerr Zfirewallr Zfirewall.errorsr Úobjectr r r r r Ú s