sɕ d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZmZm Z m
Z
d dl m Z ej
e j Ze ej d Zed Zed Zed Zed Zed Zed
Zd Zd Zd
Zd Z d Z d Z d Z d Z d Z!ddZ"e#dk r e" dS dS ) N)backup_backendshelperslogrestore)BackendNonApplicableErrorc > t j fd}|S )Nc dj vr-|r+| dd | z d| |d| d S )N
extra_argszunrecognized arguments: )parserr
)argserrorjoin)r r
kwargsfuncspecs I/opt/imunify360/venv/lib/python3.11/site-packages/restore_infected/cli.pywrapperzaction..wrapper sd ty ( (Z ( LL*SXXj-A-AA
z action_init..6 s# GGGUh1F1F51F1F1Fr zinit arguments required: r zinit unknown keys: N)
r r initr parse_extra_args#_get_values_for_envvars_init_actionfill_args_from_envvalidate_paramsr r ) backendr
r _r parsed_args
parsed_kwargs init_args init_kwargsmissingunknownr s @r action_initr- % s
!', / /D!(!9*!E!E K 27<FFH $ 7 I{ .tY LLGWGGGGgGGGGw F 0388G3D3DDEEE @ *SXXg->->>???GL)+{+++++r c H t j | j fd| D } | j |fi |}|rCt t j d |D d S t j
d d S )Nc . i | ]\ }}|j v ||S r
)
kwonlyargs)r kvr s r
z action_list..G s+ D D DTQqDO/C/CAq/C/C/Cr c , g | ]}t | S r
)str)r bs r r z action_list..K s $=$=$=SVV$=$=$=r zNo backups found) r r backupsitemsloggerinfooslinesepr sysexit)r% untilr r r7 r s @r action_listr@ D s !'/ 2 2D D D D DV\\^^ D D DD goe,,t,,G % BJOO$=$=W$=$=$=>>????? #$$$$$r c t j | || \ }}|rQt d t t j | |rSt d t t j | d S d S )NzRestore success:zRestore failed:)r restore_infectedr9 r: r; r< r )r% filesr? r& successfaileds r action_restorerF P s .wuEEOGV . &''' BJOOG,,---
- %&&& BJOOF++,,,,,- -r c . | d S N)cleanup)r% r& s r action_cleanuprJ ] s OOr c | }|r.t t j | d S d S rH )r: r9 yaml safe_dump)r% r& r: s r action_inforN b sC <<>>D * DN4(()))))* *r c t | |d }t |dd s(| d | || }|r t | d S d S )NextraFzinvalid extra action: {0})getattrr formatr9 r: )r% extra_actionr
r r&
extra_funcresults r action_extrarV i s ,55J :w . . G 077 EEFFF
Z
$F
F r c f | dd |j z d S )Nz%one of the action must be specified: r )r r choices)r
subparsersr& s r action_nonerZ t s9
LL/#((:;M2N2NN r c 2 | | |g S )Nparents)
add_parser)namerY parents r r^ r^ z s x 8 88r c D t j d } | ddddd | d d
d d d
t j | g }| t | ddt
j d || fS )NF)add_helpz-vz --verbosecountr zfPrint more messages depending on flag count: v=verbose, vv=verbose with timestamps, vvv=debug messages)r defaulthelpz-oz--outputoutputPATHzWrite log to specified file)destmetavarre r\ )r r% BACKENDzBackup backend to use)ri typere )argparseArgumentParser add_argument set_defaultsrZ r r% )r` r s r
setup_parsersrp ~ s
$e
4
4
4F
k'1 2 3 3 3
jx : < < <
$fX
6
6
6F
{+++
9 , 4 4 6 6 6 6>r c > | }t d|| }| t d t d|| }| t d | ddd dt j d
t d || }| t d | d
ddd | ddd dt j d
t d|| }| t d t d|| }| t d t d|| }| t d | dd |S )Nr Init)r action_reprlistzGetting backup listz-uz--untilDATEz!Dig backups not earlier this date)rd ri rk re r RestorerC +FILEzList files to restore)nargsri re rI Cleanupr: InforP z
Extra actionsrS zExtra action for backend)re ) add_subparsersr^ ro r- r@ rn r DateTimerF rJ rN rV ) r r` rY parser_init parser_listparser_restoreparser_cleanup parser_info parser_extras r setup_subparsersr s &&((J VZ88K KV DDD VZ88K K)> @ @ @ T9dF")"2"E G G G :v>>N ~9 MMM sF%< > > > iv%,%5%H J J J :v>>N ~9 MMM VZ88K KV DDD gz6::L \ OOO n3M NNN r c l t j | j j }|j }|j pi }t
j }|D ]\}d|z }i }||v r(|| } | |d< t | t rd|d< nd|d< |j
|fd| i| ]| | S )Nz--%srd rw ry Trequiredri )
r r r% r7 r0 kwonlydefaultsrl rm
isinstancetuplern upperparse_known_args)
namespacer r keysdefaultsr keyr_ argument_kwrd s
r list_custom_args_parser s !)"3"; < >Iz;&& 4Y
K K &JJi JDMMJJ %I I%I r c | s|st j d S t j | t j | d S rH )r log_only_cli_modulelog_to_console log_to_file) verbosityrf s r enable_logsr sL V !!!y!!!OFr c | di | d S # t j $ rJ}|j r t |j t j |j Y d }~d S d }~wt $ r`}t d |d t | t j d Y d }~d S d }~ww xY w)Nz {} error: %srs r
) r ActionErrormessager9 r r= r> code ExceptionrR r5 )r r es r apply_actionr s v 9 $ LL # # # ^**6-+@AA3q66JJJ s! C
?A
C
-ACC
c V t | di }d | D fdt |j |j rt |j ndd |j r|j ng D fd| D }i |S )Nfrom_envc 4 g | ]\ }}|t j v|S r
r; environ)r argenv_names r r z7_get_values_for_envvars_init_action.. s/ XXRZ5O5O5O5O5Or c $ i | ] \ }}|v ||
S r
r
)r r1 r2 r+ s r r3 z7_get_values_for_envvars_init_action.. s4 Aq
<<
1
<. sG C h 8rz#9#9 RZ
!#9#9#9r )rQ r8 zipr r len)r r r values_from_envr r+ s @@r r" r" s tZ,,H !) !1!1 G
IT]As4=))))BB C !] 2DMM
H& %^^ - - O +h */ **r c n | t j dd } t \ }}t || } t ||| }n+# t
$ r }t j | Y d }~nd }~ww xY wt |j |j t |j i t | d S )Nr r )
r= argvrp r r r r> r verboserf r r r )r
arg_parser
parent_parsersubsnserrs r mainr s |x| - J
J
6 6D
Dt
4
4
4 $
BI&&&LtBxxL)))))s A
A4A//A4__main__rH )$rl r loggingr; r= rL rB r r r r $restore_infected.backup_backends_libr getLogger
CLI_LOGGERr9 setLevelDEBUGr r- r@ rF rJ rN rV rZ r^ rp r r r r r r" r __name__r
r r r s
C C C C C C C C C C C C J J J J J J 3> * *
, , ,< % % % - - - * * * 9 9 9 " @) ) )0 $+ $+ $+N* * * *" zDFFFFF r