o +ke&@sddlZddlZddlZddlZddlmZddlmZeZddlm Z ddZ Gd d d Z Gd d d e Z Gd dde Z GdddZdS)N)get_terminal_size) create_logger)ellipsis_truncatecCs$tddd}|dkr||S|S)Nrfallbackrr)rljust)messageterminal_spacer9usr/lib/python3.10/site-packages/borg/helpers/progress.pyjustify_to_terminal_size s rc@sPeZdZdZdZdZdZeddZdddZ d d Z dd d d Z ddZ dS)ProgressIndicatorBasezborg.output.progressNFrcCs|jd7_|jS)zYUnique number, can be used by receiving applications to distinguish different operations.r)operation_id_counter)clsrrr operation_idsz"ProgressIndicatorBase.operation_idcCsd|_t|j|_||_||_|jjsitj t j d|_|j tj td}z|j}|jr3dnd}|j|_Wn tyEd}Ynw|j|||j_|j|j|jjtjkre|j tjd|j_|jtj k|_dS)N)streamborg  F)handlerlogging getLoggerLOGGERloggerridmsgidhandlers StreamHandlersysstderrsetLevelINFO formatterjsonAttributeError setFormatter terminator addHandlerlevelNOTSETWARN propagategetEffectiveLevelemit)selfrrr&r*rrr__init__"s,      zProgressIndicatorBase.__init__cCs*|jdur|j|j|jdSdSN)rr removeHandlercloser2rrr__del__Is zProgressIndicatorBase.__del__finishedc KsP|jsJ|js dS|t|j|j|j|tdtt |t j dddS)N)Z operationrtyper:timeT)fileflush) r'r1updatedictrr JSON_TYPEr<printdumpsr"r#)r2r:kwargsrrr output_jsonNs z!ProgressIndicatorBase.output_jsoncCs$|jr |jdddS|ddS)NTr9)r'rEoutputr7rrrfinish[szProgressIndicatorBase.finishr4) __name__ __module__ __qualname__rrAr'r classmethodrr3r8rErHrrrrrs  ' rc@seZdZdZddZdS)ProgressIndicatorMessageZprogress_messagecCs*|jr |j|ddS|jt|dS)N)r )r'rErinfor)r2msgrrrrGeszProgressIndicatorMessage.outputN)rIrJrKrArGrrrrrMbs rMcs@eZdZdZdfdd Zdd d Zdd d ZdddZZS)ProgressIndicatorPercentZprogress_percentr%3.0f%%Ncs0d|_||_||_||_||_tj|ddS)a Percentage-based progress indicator :param total: total amount of items :param step: step size in percent :param start: at which percent value to start :param msg: output message, must contain one %f placeholder for the percentage r)rN)countertotal trigger_atsteprOsuperr3)r2rTrVstartrOr __class__rrr3os  z!ProgressIndicatorPercent.__init__rcCsN|dur||_|jd|j}|j|7_||jkr%|j|j7_|SdS)Nd)rSrTrUrV)r2currentincreasepctrrrprogresss z!ProgressIndicatorPercent.progresscCs|||}|durX|durP|js@ddlm}tddd}|dkr@|||jt|g|dddg}t|d||d<|j|jt|g|d |d S||j|SdS) z Show and output the progress message :param current: set the current percentage [None] :param increase: increase the current percentage [None] :param info: array of strings to be formatted with msg [None] Nr)swidthrr rrrFF)justifyrN) r_r'platformr`rrOtuplerrG)r2r\r]rNr^r`r spacerrrshows  * zProgressIndicatorPercent.showTcCs<|jr|j||j|j|ddS|rt|}|j|dS)N)r r\rTrN)r'rErSrTrrrN)r2r rarNrrrrGs zProgressIndicatorPercent.output)rrQrrRN)Nr)NrN)TN) rIrJrKrAr3r_rerG __classcell__rrrYrrPls   rPc@s6eZdZd ddZddZddZd d Zd d ZdS)ProgressIndicatorEndless NcCs*d|_d|_||_|durtj}||_dS)z Progress indicator (long row of dots) :param step: every Nth call, call the func :param file: output file, default: sys.stderr rN)rS triggeredrVr"r#r=)r2rVr=rrrr3s  z!ProgressIndicatorEndless.__init__cCs4|jd7_|j|jdk}|r|jd7_|S)Nrr)rSrVrir2Ztriggerrrrr_s z!ProgressIndicatorEndless.progresscCs|}|r ||jSdSr4)r_rGrirjrrrres zProgressIndicatorEndless.showcCstdd|jdddS)N.rFT)endr=r>rBr=)r2rirrrrGszProgressIndicatorEndless.outputcCst|jddS)N)r=rmr7rrrrHszProgressIndicatorEndless.finish)rhN)rIrJrKr3r_rerGrHrrrrrgs   rg)rr'r"r<shutilrrrZ parseformatrrrrMrPrgrrrrs   M >