$TXg]JdZddlZddlZddlZddlZddlZddlmZddlm Z m Z m Z m Z m Z ddlmZddlmZddlmZddlmZmZdd lmZmZdd lmZmZdd lmZdd lmZm Z m!Z!m"Z"m#Z#dd l$m%Z%m&Z&m'Z'm(Z(ddl)m*Z*m+Z+m,Z,m-Z-ddl.m/Z/m.Z.ddl0m1Z1ddl2m3Z3ddl4m5Z5ddl6m7Z7ddl8m9Z9m:Z:ddl;mZ>e#e?Z@eAejdxrejjZDGdde!ZEe5ejjdr eEdZHne!dZHdDdZIGd d!ZJGd"d#eJZKeKZLGd$d%eZMedEd&ZNedFd'ZOedGd(ZPeeMjeMjfd)ZRed*ZSed+ZTed,ZUedGd-ZVe dd.ddfd/ZWdd0d1ZXe7d2d3d45Gd6d7ZYe7d2d3d85Gd9d:ZZGd;de Z\e Z d?Z]Gd@dAeJZ^dBZ_e?dCk(re_yy)HzCommon I/O utilities.N) defaultdict)ExecutorFutureThreadPoolExecutor_base as_completed) _WorkItem)contextmanager)Enum)EPIPE ESHUTDOWN)partialwraps)BytesIOStringIO)cycle)CRITICALWARN Formatter StreamHandler getLogger)dirnameisdirisfilejoin)EventLockRLockThread)sleeptime) memoizemethod) NullHandler)boolify) deprecated)encode_environmenton_win)NULL)expandisattyc.eZdZdZdfd ZfdZxZS)DeltaSecondsFormatteraw Logging formatter with additional attributes for run time logging. Attributes: `delta_secs`: Elapsed seconds since last log/format call (or creation of logger). `relative_created_secs`: Like `relativeCreated`, time relative to the initialization of the `logging` module but conveniently scaled to seconds as a `float` value. cFt|_t| ||y)N)fmtdatefmt)r! prev_timesuper__init__)selfr0r1 __class__s /lib/python3.12/site-packages/conda/common/io.pyr4zDeltaSecondsFormatter.__init__0s S'2ct}|j}t|j||_||z |_|jdz |_t ||S)Ni)r!r2max delta_secsrelativeCreatedrelative_created_secsr3format)r5recordnowr2r6s r7r>zDeltaSecondsFormatter.format4sSfNN T^^S1)O'-'='='D$w~f%%r8NN)__name__ __module__ __qualname____doc__r4r> __classcell__r6s@r7r.r.$s 3&&r8r.CONDA_TIMED_LOGGINGzm%(relative_created_secs) 7.2f %(delta_secs) 7.2f %(levelname)s %(name)s:%(funcName)s(%(lineno)d): %(message)sz<%(levelname)s %(name)s:%(funcName)s(%(lineno)d): %(message)sc8djfd|DS)Nc3LK|]}ddzzdzt|zyw)  z- N)str).0xindents r7 zdashlist..Is)I14#,&-A6s!$)r)iterablerQs `r7dashlistrTHs 77II IIr8ceZdZdZdZy)ContextDecoratorzyBase class for a context manager class (implementing __enter__() and __exit__()) that also makes it a decorator. c2tfd}|S)NcD5|i|cdddS#1swYyxYwN)argskwdsfr5s r7 decoratedz,ContextDecorator.__call__..decoratedTs$'$'s)r)r5r]r^s`` r7__call__zContextDecorator.__call__Ss! q (  (r8N)rBrCrDrEr_rZr8r7rVrVLs  r8rVceZdZdZdZy)SwallowBrokenPipecyrYrZr5s r7 __enter__zSwallowBrokenPipe.__enter__`s r8cz|r9t|tr(t|ddr|jtt fvryyyyy)NerrnoT) isinstanceEnvironmentErrorgetattrrfr r r5exc_typeexc_valexc_tbs r7__exit__zSwallowBrokenPipe.__exit__csD 7$45$/ %!33406 r8N)rBrCrDrdrnrZr8r7rara\s  r8raceZdZdZdZdZy) CaptureTargetzConstants used for contextmanager captured. Used similarly like the constants PIPE, STDOUT for stdlib's subprocess.Popen. N)rBrCrDrESTRINGSTDOUTrZr8r7rprpps F Fr8rpc#pK|i}t|}i}|jD]?\}}tjj |t ||<|tj|<A |r||r|dd|jD]2\}}|t urtj|= |tj|<4|r||r |dyy#|jD]2\}}|t urtj|= |tj|<4|r||r |dwwxYww)NTF)r(itemsosenvirongetr*)var_mapcallbackstack_callback new_var_map saved_varsnamevalues r7env_varsrzs$W-KJ"((* e::>>$5 4  4+"  J  4  %++-KD%}JJt$#( 4 .  J  5 ! &++-KD%}JJt$#( 4 .  J  5 ! s A$D6'C>AD6AD33D6c#bK||i}t|||5}|dddy#1swYyxYww)N)r{r|r)rrr{r|dess r7env_varrs. u A !h~ F" G F Fs/# /,/c#VKt|5}|dddy#1swYyxYww)N)r{r)r{rs r7env_unmodifiedrs 8 $ % $ $s ) )&)c#Kd}Gdd}tjtj}}|tjk(r8t }|j |_t|||_|t_n|}| |t_|tjk(r8t }|j |_t|||_|t_n0|tjk(r|xt_}n|}| |t_|}tjd ||tjk(r|j|_n||_|tjk(r|j|_n"|tjk(rd|_n||_||ct_t_tjdy#|tjk(r|j|_n||_|tjk(r|j|_n"|tjk(rd|_n||_||ct_t_tjdwxYww)aCapture outputs of sys.stdout and sys.stderr. If stdout is STRING, capture sys.stdout as a string, if stdout is None, do not capture sys.stdout, leaving it untouched, otherwise redirect sys.stdout to the file-like object given by stdout. Behave correspondingly for stderr with the exception that if stderr is STDOUT, redirect sys.stderr to stdout target and set stderr attribute of yielded object to None. .. code-block:: pycon >>> from conda.common.io import captured >>> with captured() as c: ... print("hello world!") ... >>> c.stdout 'hello world!\n' Args: stdout: capture target for sys.stdout, one of STRING, None, or file-like object stderr: capture target for sys.stderr, one of STRING, STDOUT, None, or file-like object Yields: CapturedText: has attributes stdout, stderr which are either strings, None or the corresponding file-like function argument. cxt|drd|jvrt}nt|trt}nt }t||s_t|dr#|j d}|j|yt|dr#|jd}|j|yy|j|y)Nmodebdecodezutf-8encode) hasattrrbytesrgrrNr old_writer)r5to_writewanteddecodedrs r7 write_wrapperzcaptured..write_wrappers 4 SDII%5F g &FF(F+x*"//'2w'8,OOG,q!- NN8 $r8c eZdZy)captured..CapturedTextNrBrCrDrZr8r7 CapturedTextrs r8rNzovertaking stderr and stdoutzstderr and stdout yielding back) sysstdoutstderrrprsrwriterrrtlogdebuggetvalue) rrrr saved_stdout saved_stderroutfileerrfilecs r7capturedrs:%(  "%SZZ,L %%%*#MM w7     CJ %%%*#MM w7  ='' '&& W   CJAII,-5 ])) )'')AHAH ])) )'')AH }++ +AHAH!-| CJ 34 ])) )'')AHAH ])) )'')AH }++ +AHAH!-| CJ 34s DI:"G&B(I:B)I77I:c#Ktj}|t_ d|t_y#|t_wxYwwrY)rargv) args_list saved_argss r7rrs,JCH :s?/ ? <?c#Ktj dtjy#tjwxYwwrY)logging _acquireLock _releaseLockrZr8r7 _logger_lockr s2  sA 1A AA c#XKt|}|j|j|j}}}t }t 5|j ||jtdzd\|_|_ddd dt 5|j|||c|_|_||_dddy#1swYIxYw#1swYyxYw#t 5|j|||c|_|_||_dddw#1swYwxYwxYww)Nr')TF) rleveldisabled propagater$r addHandlersetLevelr removeHandler) logger_namelogr_lvl_dsbld_prpgt null_handlers r7disable_loggerrs [ !D::t}}dnn&&D=L   % hl#(3% t~ $ ^   | ,(,f %DJ #DN^ ^\^   | ,(,f %DJ #DN^^sfAD*9C?D*C D*(C> D*C D*CD* D'*(D D'D$ D''D*c#Kt|}|j|j|j|jf\}}}}t t j}d|_|j||jtt5|j|gddc|_|_|_|j||j|ddd dt5|||c|_|_|_||_dddy#1swY?xYw#1swYyxYw#t5|||c|_|_|_||_dddw#1swYwxYwxYww)NrF)rhandlersrrrrrrrr setFormatter _FORMATTERrr)rrr_hndlrsrrrhandlers r7stderr_log_levelr's [ !D     %!GT66 CJJ'GGL U $  e795%4 t}dn   e  $ ^7>f 4DM4:t}#DN^ ^\^7>f 4DM4:t}#DN^^sgBE2A DE2$D1( E22D% E2D"E2%D.*E21 E/<E# E/#E, (E//E2Fct|}td|jDd}ttj }d|_|j||j|xst|xsdD]}|j|t5|r|j||j|||jkr|j|||_dddy#1swYyxYw)a[Attach a new `stderr` handler to the given logger and configure both. This function creates a new StreamHandler that writes to `stderr` and attaches it to the logger given by `logger_name` (which maybe `None`, in which case the root logger is used). If the logger already has a handler by the name of `stderr`, it is removed first. The given `level` is set **for the handler**, not for the logger; however, this function also sets the level of the given logger to the minimum of its current effective level and the new handler level, ensuring that the handler will receive the required log records, while minimizing the number of unnecessary log events. It also sets the loggers `propagate` property according to the `propagate` argument. The `formatter` argument can be used to set the formatter of the handler. c3@K|]}|jdk(s|yw)rN)r)rOrs r7rRz(attach_stderr_handler..YsJ W1I sNrrZ)rnextrrrrrrrr addFilterrrrgetEffectiveLevelr) rrr formatterfiltersrold_stderr_handlernew_stderr_handlerfilter_s r7attach_stderr_handlerrAs, [ !DJ JD 'szz2&&##I$;<=b=$$W-!     1 2 *+ 4))+ + MM% " s AC22C;)default_returncDtr ||i|SGddtfd}tjtj|tj | ||i|}tj d|S#t$r|cYSwxYw#tf$r|cYSwxYw)z[Enforce a maximum time for a callable to complete. Not yet implemented on Windows. c eZdZy)!timeout..TimeoutExceptionNrrZr8r7TimeoutExceptionr{s r8rcrYrZ)signumframers r7 interruptztimeout..interrupt~s "$ $r8r)r)KeyboardInterrupt ExceptionsignalSIGALRMalarm) timeout_secsfuncrr[kwargsrretrs @r7timeoutrns "(( (  y  %  fnni0 \" "''C LLOJ!! "! ! ""!"34 "! ! "s#A>B> B  B  BBz25.3z25.9z*Use `conda.reporters.get_spinner` instead.)addendumcPeZdZdZedZd dZdZdZdZ e dZ dZ y ) Spinnera Args: message (str): A message to prefix the spinner with. The string ': ' is automatically appended. enabled (bool): If False, usage is a no-op. json (bool): If True, will not output non-json to stdout. z/-\|c.||_||_||_t|_t |j |_tt|jdz|_ tj|_|xr | xrt|_||_y)N)targetr')messageenabledjsonr _stop_runningr_start_spinning_spinner_threadlenr spinner_cycle_indicator_lengthrrfhIS_INTERACTIVE show_spin fail_message)r5rrrrs r7r4zSpinner.__init__sw   "W%T-A-AB!$T$*<*<%=!>!B** @X@.(r8c|jr|jjy|js6|jj d|jj yy)Nz...working... )rrstartrrrflushrcs r7rz Spinner.startsE >>  & & ( GGMM* + GGMMOr8c|jr<|jj|jj d|_yyNF)rrsetrrrcs r7stopz Spinner.stops9 >>    " " $  % % '"DN r8c |jjs|jjt |j dz|jj td|jjd|jz|jjsyy#t$r4}|jttfvr|jnYd}~yd}~wwxYw)NrMg?)ris_setrrrrrr rOSErrorrfr r r)r5es r7rzSpinner._start_spinnings ((//1 d4#5#56<= d  dT%;%;;< ((//1  ww5),,  sB2B77 C4*C//C4c|jsJtjj|jdtjj |j y)N: )rrrrrrrrcs r7rdzSpinner.__enter__s?yy JJ   ~R0 1 JJ    r8cJ|j|js{t5|s|r*tjj |j ntjj dtjjdddyy#1swYyxYw)Nzdone )rrswallow_broken_piperrrrrrjs r7rnzSpinner.__exit__sh yy$wJJ$$T%6%67JJ$$X.   " %$$$s A,BB"N)TFzfailed ) rBrCrDrErrr4rrrrrdrnrZr8r7rrs? 'NM )#  #r8rz/Use `conda.reporters.get_progress_bar` instead.cXeZdZedZ d dZdZdZdZe dZ e dZ y) ProgressBarcPt|dst|_|jS)N_lock)rrr)clss r7get_lockzProgressBar.get_locks!sG$CIyyr8Nc d||_||_||_|ry|r_tr/d} |j ||ddt j |||_yd|_t j j|dyy#t$r+}|jttfvrd|_nYd}~yd}~wwxYw)a Args: description (str): The name of the progress bar, shown on left side of output. enabled (bool): If False, usage is a no-op. json (bool): If true, outputs json progress to stdout rather than a progress bar. Currently, the json format assumes this is only used for "fetch", which maintains backward compatibility with conda 4.3 and earlier behavior. z!{desc}{bar} | {percentage:3.0f}% Tr')desc bar_formatasciitotalfilepositionleaveFNz ...working...) descriptionrrr_tqdmrrpbarrrfr r r)r5rrrrrr rs r7r4zProgressBar.__init__s'    @  $ (#-" ZZ!)#!+!DI!    K=!?@)ww5)"44', %s+A;; B/!B**B/c |jr|jrK|j5tjj d|j d|dddddytr3|jj||jjz y|dk(r tjj dyyy#1swYyxYw#t$r+}|jttfvrd|_nYd}~yd}~wwxYw)N {"fetch":"z)","finished":false,"maxval":1,"progress":r]z} r' done F)rrr rrrrrrupdatenrrfr r )r5fractionrs r7 update_tozProgressBar.update_tos ||99 (()$*:*:);;demnodppvw)$II$$X %;<]JJ$$Y/#( ww5),,$  s@(C2CC%8C$CC CC D!DDc&|jdy)Nr')rrcs r7finishzProgressBar.finish!s qr8cz|jr/|js"tr|jj yyyy)z-Force refresh i.e. once 100% has been reachedN)rrrrrefreshrcs r7r zProgressBar.refresh$s* << n II   /= 3B'>#B$!B'#B$$B''B0) )rBrCrDr4r5rFrGs@r7r>r>^s &!r8r>cntddd}ttjj d|S)N~z.condazinstrumentation-record.csv!CONDA_INSTRUMENTATION_RECORD_FILE)rr+rwrxry)default_record_files r7get_instrumentation_record_filerMs3sH.JK  : !3!3!8!8!:;J HH%"":."":.    \*  , Y^ !F* $(" : + $**Z4k RS!  s ACC(__main__)r")NNNrArY)`rErrrwrr collectionsrconcurrent.futuresrrrrrconcurrent.futures.threadr contextlibr enumr rfr r functoolsrriorr itertoolsrrrrrros.pathrrrr threadingrrrrr!r auxlib.decoratorsr# auxlib.logzr$auxlib.type_coercionr% deprecationsr&compatr(r) constantsr*pathr+rBrrrr,rr.rxryrrTrVrarrprrrrsrrrrrrrrrr)r>rMrOrrZr8r7rs~  #XX/%"$ GG0000-%*%.X.F3::3D3D3F&I&2 2::>>/ 01& GJ FJ J   ("()D""6  !((1E1E[5[5|$$"$$4   *#Z7;"<   9 D#D#  D#N   > _%_%  _%F"H"<%&8%P =3$=3@T8 z r8