a a&@sddlZddlZddlmZddlmZddlmZddlmZddl Z ddl Z ddl Z GdddeZ Gdd d ej ZGd d d eZGd d d eZGdddeZdS)N)colour) SambaOption)get_samba_logger)LdbErrorc@seZdZejZdS)OptionN)__name__ __module__ __qualname__optparseZ SUPPRESS_HELPr r 7/usr/lib/python3/dist-packages/samba/netcmd/__init__.pyrsrc@seZdZdddZddZdS)PlainHelpFormattercsF|j|jd|j|d}fdd|D}d|d}|S)N  csg|]}tj|dqS))Zinitial_indentZsubsequent_indent)textwrapZfill).0pZ desc_widthindentr r *s z9PlainHelpFormatter.format_description..)widthZcurrent_indentsplitjoin)self descriptionZ paragraphsZwrapped_paragraphsresultr rr format_description&s    z%PlainHelpFormatter.format_descriptioncCs|rd|dSdSdS)Nrrr )repilogr r r format_epilog3s z PlainHelpFormatter.format_epilogN)r)rrr rrr r r r r %s r c@seZdZdZddZeeZddZeeZddZ ee Z dZ gZ gZ iZd ZdZdZdZejejfd d Zdd d ZddZdddZddZddZddZdddZddZdS) CommandzA samba-tool command.cCs|jddS)Nrr)__doc__ splitlinesrstriprr r r _get_short_description=szCommand._get_short_descriptioncCs0|jd}|ddtd|ddS)Nrr)r!rrdedentr)rlinesr r r _get_full_descriptionBs zCommand._get_full_descriptioncCs"|jj}|dr|ddS|S)NZcmd_) __class__r startswith)rnamer r r _get_nameHs  zCommand._get_nameNFcCs||_||_dSN)outferrf)rr0r1r r r __init__]szCommand.__init__cCs||\}}|dSr/)_create_parser print_usage)rprogparser_r r r usageasz Command.usagec Cs2t|tr(|j\}}}|j}|j}d}nt\}}}|}d}d}t|trl|j\}} |j d|| fnt|t r|j d|d}n~t|t r|j d||fn^t |tur|j d||fd}n8|dur|j d |n|j d t|||fd}|std kr.tj||j d dS) zdisplay a command errorFzuncaught exceptionTzERROR(ldb): %s - %s zERROR(assert): %s zERROR(runtime): %s - %s zERROR(exception): %s - %s Nz ERROR: %s zERROR(%s): %s - %s )file) isinstance CommandErrorexception_infoinner_exceptionmessagesysexc_inforargsr1writeAssertionError RuntimeErrortype ExceptionstrsambaZget_debug_level tracebackprint_tb) reetypeZevalueZ etracebackr>r?Zforce_tracebackZ ldb_ecodeZldb_emsgr r r show_command_errores4       zCommand.show_command_errorcCsjtj|j|jt||d}||ji}t|j D](}|j|}||||<| ||q8||fS)N)r8rZ formatterr5r) r Z OptionParsersynopsisfull_descriptionr Z add_options takes_optionssortedtakes_optiongroupskeysZadd_option_group)rr5rr6 optiongroupsr-Z optiongroupr r r r3s   zCommand._create_parsercCs|j|ddS)Nr)r0rC)rtextr r r r?szCommand.messagec Gs\||d\}}|t|\}}|dd}t|j}|jD]"}|jD]}|jdurJ||j=qJq@||d} d} d} t |j D]N\} } | ddkr| ddkr| d7} | ddks| ddkrd} q| d7} qt || ks| st || kr| dSt||_ ||_||_z|j|i|WStyV}z||WYd}~dSd}~00dS) Nrr&F?*+T)r3 parse_argslistdict__dict__Z option_groupsZ option_listdestupdate enumerate takes_argslenr4raw_argvraw_args raw_kwargsrunrGrN)rargvr6rUoptsrBkwargsZ option_groupZoptionZmin_argsZmax_argsZundetermined_max_argsiargrLr r r _runs:          z Command._runcCst|jdS)z=Run the command. This should be overridden by all subclasses.N)NotImplementedErrorrgr$r r r rgsz Command.runrcKs tf|p |j|j||d|S)zGet a logger object.)r-streamverbosequiet)rr-r1)rr-rprqrjr r r get_loggers  zCommand.get_loggercCsj|}|dkrtnL|dkr,tn:|dkr^t|jdrT|jrTtqftntddS)zHeuristics to work out whether the user wants colour output, from a --color=yes|no|auto option. This alters the ANSI 16 bit colour "constants" in the colour module to be either real colours or empty strings. ZnoZyesautoisattyz;Unknown --color option: %s please choose from yes, no, autoN)lowerrZswitch_colour_offZswitch_colour_onhasattrr0rtr<)rZ requestedr r r apply_colour_choices    zCommand.apply_colour_choice)N)NN)rFF)rrr r!r%propertyshort_descriptionr)rPr.r-rOrbrQrShiddenrdrerfr@stdoutstderrr2r8rNr3r?rmrgrrrwr r r r r :s0 ! ( r c@s"eZdZdZdZiZdddZdS) SuperCommandz&A samba-tool command with subcommands.z%prog NcGs||jvr*|j|jd||fg|RS|dvrFtd||fg}|dkrt|dkr|j|d}t|tr|jd||dfdg|ddRS|dur|jd||dfdg|ddRSd}d}t|j}| t d d |D}|D].}|j|} | j s|d | || j f7}q|d |7}|j ||d \} } t|} |rh| d|| | \} }| dS)Nz%s %s)--helphelpNz%s: no such subcommand: %s rrr&r~z Available subcommands: cSsg|] }t|qSr )rc)rcr r r rz%SuperCommand._run..z %*s - %s zQFor more help on a specific subcommand, please type: %s (-h|--help) )rrW) subcommandsrmprintrcgetr;r}r\rTsortmaxrzryr3insertr[Z print_help)rZmynameZ subcommandrBsubrZsubcmdsZ max_lengthZcmd_namecmdr6rUZ args_listrir r r rmsN           zSuperCommand._run)N)rrr r!rOrrmr r r r r}sr}c@s"eZdZdZdddZddZdS)r<z1An exception class for samba-tool Command errors.NcCs||_||_t|_dSr/)r?r>r@rAr=)rr?r>r r r r2szCommandError.__init__cCs d|jS)NzCommandError(%s))r?r$r r r __repr__szCommandError.__repr__)N)rrr r!r2rr r r r r<s r<)r rIrZ samba.getoptrZ samba.loggerrZldbrr@rJrrZIndentedHelpFormatterr objectr r}rGr<r r r r s    )1