a Oc<@sddlZddlZddlZddlZddlmZddlmZmZddlm Z ddl m Z ddl m Z ddZd d ZGd d d ZGd ddZddZGdddZGdddZGdddZGdddZGdddZGdddZdS)N)config)resourceutil)Client)dump) WhoisFormatcsfdd}t|S)zDecorate a command handlercs|i|}|durd}|S)Nr)akrfuncrI/var/www/html/gsl.tools/lib/python3.9/site-packages/peeringdb/commands.py_wrappedsz_handler.._wrapped) staticmethod)r rrr r_handlers rcCs^|}|D]L\}}|j||jd}t|dd}|r<||t|dd}|r |j|dq dS)zAdd commands to a parser)help add_argumentsNhandle)handler)add_subparsers add_parser__doc__getattr set_defaults)parsercommandsZsubpscmdclsZsubpZadd_argsrrrradd_subcommandss   rc@s"eZdZdZdddZddZdS) CommandGroupzA group of nested subcommands.NcCs||_|r||_dS)zc Arguments: - commands: dict of command names to handler classes N)rr)selfrrrrr__init__*szCommandGroup.__init__cs,t|jfdd}j|ddS)Ncs dS)N) print_usage_rrr_usage6sz*CommandGroup.add_arguments.._usage)Z usage_handler)rritemsr)r!rr(rr'rr3s zCommandGroup.add_arguments)N)__name__ __module__ __qualname__rr"rrrrrr 's r c@s(eZdZdZeddZeddZdS)GetzGet a resourcecCsN|jdddd|jdddtdd |jd d d d d|jdddddddS)Npoids+z Object IDsnargsrz--depthz-Drz*How many levels of nested objects to fetch)defaulttyper--output-formatz-OyamlzOutput data formatr2rz--remotez-R store_trueFz/Fall back to API request if object is not foundactionr2r add_argumentintr'rrrr@s$zGet.add_argumentsc Kst|}|D]}t|\}} t|} t} z|| | } WnX| | | y|rr| | | |d} n"t d|d| t j dYdSYn0t| |t jq dS)Nrz Not found: -filer#)rr split_refr get_resource peeringdb get_backendgetZobject_missing_errorZ get_concretefetchprintsysstderrrstdout) rr. output_formatdepthremoter&clientpoidtagpkresBobjrrrrUs z Get.handleNr*r+r,rrrrrrrrrr-=s  r-cCsB|dkr|tj|dS|dkr,|tj|dS|t||dSdS)Nas)ZasnZixnets)Z ix_id__in)id)rZNetworkrA)rOkeyZgetfuncrrr _lookup_taghs rXc@s(eZdZdZeddZeddZdS)Whoisz|Simulate a whois lookup; supports as : query by AS; ixnets : query networks on an IX cCs|jdddddS)Nr.r/z.Object IDs, or { as | ixnets }...r0r;r'rrrrxszWhois.add_argumentsc st|fdd}t}|D]h}t|\}}zt|||}Wn4tjjyttd|d|t j dYdS0| ||dqdS)Ncs&j|fddi|}|s"tjj|S)NrK)Z fetch_allrBsyncNotFoundException)rQkwargsobjsrMrrrDszWhois.handle..getzNot found: resource for =r>r#r) rrrr@rXrBr\r]rFrGrHZdisplay) rr.r&rDfmtrNrOrWdatarr`rr~s  z Whois.handleNrTrrrrrYqs  rYc@s(eZdZdZeddZeddZdS) DumpConfigzOutput current configcCs|jdddddS)Nr4r5output data formatr6rZr'rrrrszDumpConfig.add_argumentscKst|}||tjdSN)mungeZ get_codecrrGrI)rrJr&codecrrrrs zDumpConfig.handleNrTrrrrrds  rdc@s(eZdZdZeddZeddZdS) PromptConfigz2Prompt for configuration values and save to a filecCs(|jdddd|jddddd d dS) Nr4r5rer6z --defaultsz-nr7Fz*do not prompt, just write defaults to filer8rZr'rrrrszPromptConfig.add_argumentscKs>|r|}|}ntjtj|d}td|}tj|||ddS)N)defaultszOutput directory)rh)cfgZ prompt_configZ CLIENT_SCHEMArpromptZ write_config)rrjZ config_dirrJr&Z newconfigZoutdirrrrrs  zPromptConfig.handleNrTrrrrris  ric@seZdZdZeddZdS) ListCodecszList available codecscKstdtjdS)N )rFjoinrgrhZ list_codecsr%rrrrszListCodecs.handleNr*r+r,rrrrrrrrmsrmc@s(eZdZdZeddZeddZdS)Syncz%Synchronize local tables to PeeringDBcCs`|jddddd|jddddd|jd d d d d |jdd d dd |jdddtdddS)Nz-vz --verbosecountzBe more verbose)r9rz-qz--quietz Be more quietz --dry-runr7FzDo not actually perform updatesr8z--initz)Only initialize the database; do not syncz--sincestorez6Only fetch updates since when (<0 for since last sync))r9r2r3rr:r'rrrrs* zSync.add_argumentsc Kst}d|pd|pd}|dkr0ttj|dkrDttjt|fi|}|r\dS|dkrvtd|dd|dkrd}| ||dS)Nr#rz Syncing tor\url) rZ all_resourcesrBZ _config_logsloggingDEBUGWARNINGrrFZ update_all) rverbosequietinitZsincer^rsZloglvlrMrrrrs  z Sync.handleNrTrrrrrqs  rqc@seZdZdZeddZdS) DropTableszDrop all database tablescKst|t}|dSrf)rrBrCZ delete_all)rr&rRrrrrszDropTables.handleNrprrrrr}sr})rvrGrgrBrrkrrZpeeringdb.clientrZpeeringdb.output._yamlrZpeeringdb.whoisrrrr r-rXrYrdrirmrqr}rrrrs&      + "7