4 $ d dl Z d dlZd dlZd dlmZmZ d dlmZ d dlm Z m
Z
m Z m Z m
Z
mZ d dlmZ d dlmZ d dlmZ d dlmZ d d lmZ G d
d Zd Z G d
d Z d Z d Z d Z d Z d Z!d Z" ed d Z#d Z$dS ) N) lru_cachepartial)chain)AnyDictIterableIteratorMappingTuple)app)Core)prepare_schema) RpcClient)EXITCODE_NOT_FOUNDc e Zd Zeeeef Zd Ze defd Z
e d Z defdZ defdZ
defdZdefdZdefd Zd
Zd S ) SchemaToArgparsec || _ | d | _ | d | _ | dd | _ | d | _ | dd | _ | d | _ | dd | _ | d | _ d S )
NalloweddefaultenvvarFhelp
positionalrenamerequiredtype)
_argumentget_allowed_default_envvar_help _positional_rename _required_type)selfargumentoptionss R/opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/utils/parsers.py__init__zSchemaToArgparse.__init__ s &")++i"8"8
$[[ 3 3
#KK% 8 8 !++f--
!(\5!A!A #KK 1 1 &{{:u = =!++f--
returnc Z | j r| j S d| j dd z S )N--_-)r" r replacer& s r) argnamezSchemaToArgparse.argname s2 "> !dn,,S#6666r+ c t t | | | | | | }|S N)dictr choicesr r metavarnargsr )r& argparse_optionss r) r( zSchemaToArgparse.options% si
r+ c # K d}| j dk r%| j s| j r| j s|dfV d S |dfV d S | j r| j s| j
|dfV d S d S d S )Nr9 list+*?)r% r" r$ r r r& options r) r9 zSchemaToArgparse.nargs3 s : #
"
"t|
" ck!!!!! ck!!!!!
4< 4=3H#+ 3H3Hr+ c # K d| j fV d S )Nr7 )r r2 s r) r7 zSchemaToArgparse.choices? s &&&&&&r+ c # K d| j fV d S )Nr )r! r2 s r) r zSchemaToArgparse.helpB s dj r+ c # K d}| j r || j fV d S | j dk r || j fV d S d S )Nr8 r< )r# upperr% r r@ s r) r8 zSchemaToArgparse.metavarE st < 1$,,,... . . . . .
Z6
!
!$...000 0 0 0 0 0 "
!r+ c # l K | j &| j s!| j dk s| j sd| j fV d S d S d S d S )Nr< r )r r r% r" r2 s r) r zSchemaToArgparse.defaultL sZ M %L
&v%%T-=%T]* * * * * *
& % % %%%r+ c # b K | j r | j dk r| j s| j s dV d S d S d S d S d S )Nr< )r T)r$ r% r r" r2 s r) r zSchemaToArgparse.requiredT sd N #
f$$L % $ %
# " " " " " # #$$$$$$r+ N)__name__
__module__ __qualname__r r strr
OptionTyper* propertyr3 r( r9 r7 r r8 r r r+ r) r r s
%S/*J
.
.
. 7 7 7 7 X7
X
z
' ' ' ' '!j ! ! ! !1 1 1 1 1+ + + + +# # # # #r+ r c | d dk r| d o| dd }| | }| d| dd z |d
| d | dd z |d
|j di || d i d S t || } | j |j fi |j d S )Nr booleanr r F)r r. r/ r0
store_true)destactionz--no- store_falser rN )r add_mutually_exclusive_group add_argumentr1 set_defaultsr r3 r( )parserr' r( r bool_parser converters r) schema_to_argparser[ _ s6 {{6i'';;z**O7;;x3O3O/O 9989LL 8##C-- - !
h&&sC00 0 !
! FFHgkk).D.D#EFFFFF$Xw77 I -CC1BCCCCCr+ c e Zd Zedefd Zed Zededee de
e e f fd Z ed Z d S )
EnvParserenvvar_parameter_optionsc | sdS d d d fd| D S )N c @ d| v r| d d| d S | d S )Nr r z rN )r( s r)
format_argz)EnvParser.format_help..format_argy s5 !(+BB BBB8$ $r+ z
environment variables:
{}z
c 3 . K | ]} | V d S r5 rN ).0r( rb s r) z(EnvParser.format_help.. sA
7## r+ )formatjoinvalues)r^ rb s @r) format_helpzEnvParser.format_helpt s| ' 2 % % %
177 KK 7 > > @ @
r+ c p d|v r1 | d n# t $ r
d| d| dfcY S w xY wd S )Nisasciiasciizerror: =z must only contain ascii symbols)encodeUnicodeEncodeError)r valuer( s r) _validatezEnvParser._validate ss
W%%%%%
NfNNuNNN
ts 33environexcluder, c l i }| D ]\ }}||v r |d } || x} ||< | || | x}
rK| |||
} t | t j t j t # t $ r d|v r
|d ||< Y | d sY | ||d
| } t | t j t j t Y w xY w|S )Nr )filer r z-error: environment variable {} is not defined) itemsrq
_format_errorprintsysstderrexitr KeyErrorr rf ) clsrr commandr^ rs kwargs parameterr( envvar_namerp errmsgs r) parsezEnvParser.parse sz ":"@"@"B"B 1 1 IwG##!(+K
1,3K,@@y)" -- UGDDD3 1 ++ !93 C #CJ////H /000+
-
-
- ''(/ (:F9% H {{:.. H'',CJJ# c
+++++,,,,,
-,
s
B D18D1A D10D1c ~ d d | | | | S )Nz {command}:
{help}
{message} )r~ r message)rf rg ri )r} r~ r^ r s r) rw zEnvParser._format_error sA 077HHW%% !9:: 8
r+ N)
rH rI rJ staticmethodr
ri rq classmethodr rK r r rw rN r+ r) r] r] s s
g
\
\ $$
# $
c3h
$ $ $ [$L
[
r+ r] c ^ t j | n# t j $ r Y dS w xY wdS )NFT) ipaddress IPv4AddressAddressValueError)addrs r) is_valid_ipv4_addrr sE d#### & uu 4s **c # K | D ]7\ }}|| di dg v r||fV 8d S )Ncliusers)rv r )schemauserkeyrh s r) _filter_userr se ||~~ V 6::eR((,,Wb99 9 9v+ r+ c > t | j | di |S )N)require_svc_is_runningrN )r cmd)r~ require_rpcparamss r) rpc_endpointr s9 F <9K 8 8 8 . sT '\Bg>S\\>1B&B r+ Available commandsr )r r T)r r zparser is not definedr r envvar_onlyFr r r z--jsonrQ zreturn data in JSON format)rS r z --verbosez-vcount)rS r r running)r )endpointr r^ r~ rN )sortedkeys
isinstancetupler< enumerater
add_parserr argparseRawDescriptionHelpFormatteranyadd_subparsersrv updater[ r] ri epilogrV rW r r r )
subparsersr _subparserscommandsrh rX subparserir~ hashableexists_subparserr^ r' r( r r s @r) apply_parserr sy Kfkkmm$$H P
P
'E4=11111 #G,, " 1 " 1JAwCLL1$$$"-- F++$,$H .
% ,2+@+@ 1 ,A , ,K( !1q5!122#.??8#<#< ' 18A8L8L $ #ZZ / / 9M 9 9 %n*>nNNOI H 5 5
!1II.....v $& !'Hb!9!9!?!?!A!A : : Hg7""5<(2;;}e44 7"" EE!1'(2C!DEEE&+
#(- % vx 9 9 9 9!--.FGG
\0L
Kg>>> jj++//
yII \7K@@%,( **Xr227799& & & &>
GP
P
r+ c t t t t j | }t | | d S r5 )r6 r r r SCHEMA_PATHSr )r r r s r) _apply_subparsersr 5 s:
,~c.> ? ?FF
G
GFV$$$$$r+ r )maxsizec H t j dt j z } | dd | dg dd | d
d d
| d }t
|d t | | S )Nz CLI for %s.) descriptionz --log-configzlogging config filenamer z--console-log-level)ERRORWARNINGINFODEBUGz%Level of logging input to the consoler7 r z
--remote-addrc ( t | r| nd S r5 )r )ips r) z#create_cli_parser..F s 0 4 4>$ r+ z2Client's IP address for adding it to the whitelist)r r r root)r ArgumentParserConfigNAMErV r r _apply_completions_parser)rX r s r) create_cli_parserr : s
$ 1L
M
M
MF
-FGGG
555
4
>
>
A
&&,@&AAJj&))) j))) Mr+ c ddl m} | dd }| d|d | d
d S ) Nr )SUPPORTED_SHELLS completionsz&Generate shell auto-completion scriptsr shellz!Shell to generate completions forr T)completions_command)!defence360agent.utils.completionsr r rV rW )r r completions_parsers r) r r O s| BBBBBB#..
5 / ##
0 $
###=====r+ )%r r ry functoolsr r itertoolsr typingr r r r r
r defence360agent.applicationr defence360agent.contracts.configr
r defence360agent.rpc_tools.utilsr defence360agent.simple_rpcr defence360agent.utils.clir r r[ r] r r r r r r r r rN r+ r) r s
( ( ( ( ( ( ( ( @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ + + + + + + ; ; ; ; ; ; : : : : : : 0 0 0 0 0 0 8 8 8 8 8 8M# M# M# M# M# M# M# M#`D D D(J
J
J
J
J
J
J
J
Z S
S
S
l% % %
1 ( > > > > >r+