IR-edZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z dZn #e$rdZYnwxYwddlmZddlmZddlmZgd Zd ZGd d Zd,d ZdZd-dZdZdZdZdddZdZ dZ!dZ"GddZ#GddZ$GddZ%Gdd Z&d.d#Z'Gd$d%Z(Gd&d'Z)Gd(d)Z*Gd*d+Z+dS)/z) Utilities for console input and output. NTF)conf) classproperty) isiterable) isatty color_print human_timehuman_file_size ProgressBarSpinnerprint_code_lineProgressBarOrSpinner terminal_sizezutf-8cjeZdZdZedZedZedZedZ dS)_IPythonz5Singleton class given access to IPython streams, etc.c6 ddlm}n#t$rYnwxYw|S)Nr) get_ipython)IPythonr ImportError)clsrs 5lib/python3.11/site-packages/astropy/utils/console.pyrz_IPython.get_ipython4sB  + + + + + + +    D s  c\t|dsd|_ |n#t$rYdSwxYw ddlm}nY#t $rL ddlm}nA#t $r4ddlm }|ddkrYYdS ddl m}n#t $rYYYdSwxYwYnwxYwYnwxYw||_|jS)N _OutStreamr) OutStream version_info) hasattrrr NameErroripykernel.iostreamrrIPython.zmq.iostreamrrIPython.kernel.zmq.iostream)rrrs rrz_IPython.OutStream<sEsL)) '!CN !!!!   tt  $8888888 $ $ $ $>>>>>>>" $ $ $444444#A!++#ttt$IIIIIII&$$$#tttt$JI $ $'CN~sr. <<A BABB5B:BB B B BBBBBBBc|t|ds& ddlm}||_n#t$r d|_YnwxYw|jS)N_ipyior)io)r IPython.utilsr%r$r)rr%s ripyioz_IPython.ipyioYshsH%%  ,,,,,,   " " "!  "zs  44c,t|j|SN)getattrr')rstreams r get_streamz_IPython.get_streamdssy&)))N) __name__ __module__ __qualname____doc__rrrr' classmethodr,r-rrr1s??]]8]**[***r-rc>|rd}nd}tt|}|S)a_ This utility function contains the logic to determine what streams to use by default for standard out/err. Typically this will just return `sys.stdout`, but it contains additional logic for use in IPython on Windows to determine the correct stream to use (usually ``IPython.util.io.stdout`` but only if sys.stdout is a TTY). stderrstdout)r*sys)r5r+ sys_streams r _get_stdoutr9is-f%%J r-ctjjdkstjjdkrdSt |dr|Stjt|tjsdSt|dddkrdSt |d r. d d l m }n#t$rYdSwxYwt|j|SdS) z Returns `True` if ``file`` is a tty. Most built-in Python file-like objects have an `isatty` member, but some user-defined types may not, so this assumes those are not ttys. MainProcess MainThreadFrNnamer6Tr+r)Console)multiprocessingcurrent_processr= threadingcurrent_threadrrrr isinstancer*pyreadline.consoler>rr+)filePyreadlineConsoles rrr{s ')).-??  # % % *l : :utX{{}}!*T8;M*N*N!u tVT""h..ttX :   G G G G G G G   55 $+'8999 5s/B66 CCcp|t} tjddddd}tj|t j|}tjd|\}}}}|dkr|dz}|dkr|dz}|dks|dkrtd||fS#t$r ttj d ttj d fcYS#t$r)tj}tj}|d }|d }||fcYcYSwxYwwxYw) z Returns a tuple (height, width) containing the height and width of the terminal. This function will look for the width in height in multiple areas before falling back on the width and height in astropy's configuration. NHHHHr  rzunable to get terminal sizeLINESCOLUMNSP)r9structpackfcntlioctltermios TIOCGWINSZunpack Exceptionintosenvironget TypeErrorr max_lines max_width)rEsxlineswidthxpixelsypixelss rrrs^ |}} K1a + + Kg0! 4 4+1=+C+C(w 2:: QJE 2:: QJE A::!9:: :u~       w//00#bjnnY6O6O2P2PQ Q Q Q NENE}}%<       s1BB D5$AC>;D5>.D1,D50D11D5ciddddddddd d d d d ddddddddddddddddddd d!d"}tjd#krtj|S||d}d%|d&|d'S)(aeReturns a string wrapped in ANSI color codes for coloring the text in a terminal. :: colored_text = color_text('Here is a message', 'blue') This won't actually effect the text until it is printed to the terminal. Parameters ---------- text : str The string to return, bounded by the color codes. color : str An ANSI terminal color name. Must be one of: black, red, green, brown, blue, magenta, cyan, lightgrey, default, darkgrey, lightred, lightgreen, yellow, lightblue, lightmagenta, lightcyan, white, or '' (the empty string). blackz0;30redz0;31greenz0;32brownz0;33bluez0;34magentaz0;35cyanz0;36 lightgreyz0;37defaultz0;39darkgreyz1;30lightredz1;31 lightgreenz1;32yellowz1;33 lightbluez1;34 lightmagentaz1;35 lightcyanz1;36whitez1;37win32Nz[mz)r7platformrrr[)textcolor color_mapping color_codes r _color_textr~s( v      6   V 6 F F f & V  V!" #M( |w8#5#= ""5&11J -: - - - - --r-ctj} ||S#t$r t}YnwxYw||S#t $r|dcYSwxYw)zDecode the supplied byte string using the preferred encoding for the locale (`locale.getpreferredencoding`) or, if the default encoding is invalid, fall back first on utf-8, then on latin-1 if the message cannot be decoded with utf-8. latin-1)localegetpreferredencodingdecode LookupError_DEFAULT_ENCODINGUnicodeDecodeError)r_encs r_decode_preferred_encodingrs  % ' 'C# $88C==  $ $ $#CCC $xx}} ###xx """""#s"+?A?AA98A9c |||S#t$rYnwxYwtj} tj|}n)#t $rtjt }YnwxYw||}|j} |||S#t$r)tjd}||}|j}YnwxYw|||S)zWrite the supplied string with the given write function like ``write(s)``, but use a writer for the locale's preferred encoding in case of a UnicodeEncodeError. Failing that attempt to write with 'utf-8' or 'latin-1'. r)UnicodeEncodeErrorrrcodecs getwriterrrwrite)r_rfileobjrWriterfs r_write_with_fallbackr s   a        % ' 'C5!#&& 555!"3445 wA GE a !),, F7OO  E!HHH Ls/  A#A.-A. B0CC )endc|dt}|j}t|rtjr}t dt|dD]Q}||}|dzt|krd}n ||dz}|rt||}t|||}R||dSt dt|dD]}||}||||dS)a Prints colors and styles to the terminal uses ANSI escape sequences. :: color_print('This is the color ', 'default', 'GREEN', 'green') Parameters ---------- positional args : str The positional arguments come in pairs (*msg*, *color*), where *msg* is the string to display and *color* is the color to display it in. *color* is an ANSI terminal color name. Must be one of: black, red, green, brown, blue, magenta, cyan, lightgrey, default, darkgrey, lightred, lightgreen, yellow, lightblue, lightmagenta, lightcyan, white, or '' (the empty string). file : writable file-like, optional Where to write to. Defaults to `sys.stdout`. If file is not a tty (as determined by calling its `isatty` member, if one exists), no coloring will be included. end : str, optional The ending of the message. Defaults to ``\n``. The end will be printed after resetting any color or font state. rErrN) r[r9rrr use_colorrangelenr~r)rargskwargsrErimsgr{s rrr0s< ::fkmm , ,D JE d||q#d))Q'' ; ;Aq'C1uD !!QU  .!#u-- )eT::EE c q#d))Q''  Aq'C E#JJJJ c r-c.tjdd|S)z2 Remove ANSI color codes from the string. z\[([0-9]+)(;[0-9]+)*mr)resub)r_s rstrip_ansi_codesrjs 6.A 6 66r-cgd}t|}|dkrd|ddStt|dz D]D}||\}}||dz\}}||kr#d||z|||z|z|cSEdS) a Returns a human-friendly time string that is always exactly 6 characters long. Depending on the number of seconds given, can be one of:: 1w 3d 2d 4h 1h 5m 1m 4s 15s Will be in color if console coloring is turned on. Parameters ---------- seconds : int The number of seconds to represent Returns ------- time : str A human-friendly representation of the given number of seconds that is always exactly 6 characters. ))yi)wi: )diQ)hi)rx<)r_rrz 2dr_rz{:2d}{}{:2d}{}z ~inf)rXrrformat)secondsunitsrunit1limit1unit2limit2s rr r qs4   E'llG||"W""""" 3u::> " "a va!e v f  #**6!57V+;*F     8r-c:t|dr&ddlm}|||jj}d}|dkrd}nHt tjtj |tj dz }|dkrd}n||}t tj d|}||z }t|}|dkr|d | d }n!|d d kr |d d }n |d d }|d |S)a Returns a human-friendly string representing a file size that is 2-4 characters long. For example, depending on the number of bytes given, can be one of:: 256b 64k 1.1G Parameters ---------- size : int The size of the file (in bytes) Returns ------- size : str A human-friendly representation of the size of the file unitr)rz kMGTPEZYi? N.rz>3s) rastropyrQuantitybytevaluerXmathfloorlogpowstrindex)sizeusuffixes num_scalesuffixr str_values rr r s1,tV. '&&&&&zz$''-H qyy  48D>>DHTNN#BCCDD 1}})$DHT9--..I 9 EE I }}4  4 445 1  bqbM bqbM  % %V % %%r-ceZdZdZdZdZdS)_mapfuncz: A function wrapper to support ProgressBar.map(). c||_dSr)_func)selffuncs r__init__z_mapfunc.__init__s  r-c:|\}}|||fSr)r)ri_argrargs r__call__z_mapfunc.__call__s 3$**S//!!r-Nr.r/r0r1rrr3r-rrrs<"""""r-rceZdZdZddZddZdZdZdZd Z dd Z dd Z dd Z dd Z e ddZe ddZdS)r aR A class to display a progress bar in the terminal. It is designed to be used either with the ``with`` statement:: with ProgressBar(len(items)) as bar: for item in enumerate(items): bar.update() or as a generator:: for item in ProgressBar(items): item.process() FNc0|t}|s#t|s|j|_d|_nd|_t |r)t ||_t||_ nY t||_ t t|j |_n#t$rtdwxYw||_ tj|_t!|j |_||_d|_|skt(o|j |_||jr+t/jt.j|jd|_|ddS)a Parameters ---------- total_or_items : int or sequence If an int, the number of increments in the process being tracked. If a sequence, the items to iterate over. ipython_widget : bool, optional If `True`, the progress bar will display as an IPython notebook widget. file : writable file-like, optional The file to write the progress bar to. Defaults to `sys.stdout`. If ``file`` is not a tty (as determined by calling its `isatty` member, if any, or special case hacks to detect the IPython console), the progress bar will be completely silent. NTFz&First argument must be int or sequencer)r9r_silent_updateupdate_silentriter_itemsr_totalrXrr\_filetime _start_timer _human_total_ipython_widget _signal_set_CAN_RESIZE_TERMINAL_should_handle_resize_handle_resizesignalSIGWINCH)rtotal_or_itemsipython_widgetrEs rrzProgressBar.__init__s}& <==D !fTll !-DKDLL DL n % % 7~..DKn--DKK 7!.11 #5#5#566  J J J HIII J  9;;+DK88-  ()=)U$*BSBSBUBUD &    ! ! !) ( fot/BCCC#'  As 6B11C cNt|jd}|dz |_dS)Nr%)rr _bar_length)rsignumframeterminal_widths rrzProgressBar._handle_resize's)&tz2215)B.r-c|Sr)r3rs r __enter__zProgressBar.__enter__+ r-c|js|||j|jd|j|jr-tjtjtj dSdSdS)Nr) rrrrrflushrrrSIG_DFLrexc_type exc_value tracebacks r__exit__zProgressBar.__exit__.s| ? DK((( J  T " " " J      ? fov~>>>>>  ? ?  ? ?r-c|Sr)r3rs r__iter__zProgressBar.__iter__7rr-c t|j}||S#t$r|dddwxYwr))nextrr StopIterationr)rrvs r__next__zProgressBar.__next__:s` dk""B KKMMMI     MM$d + + +  s ,#Ac| |jdz}||_|jr||dS||dS)zN Update progress bar via the console or notebook accordingly. Nr)_current_valuer_update_ipython_widget_update_consolerrs rrzProgressBar.updateDs_ ='!+E#   (  ' ' . . . . .   ' ' ' ' 'r-cd|jdkrd}n$t|t|jz }|j}|j}|dkrt |j}n$t t|j|z}|dt d|zd|d||jkr,t d d |d|d |j|z dz z|d ||jkrtj|jz }d }n1|dkrd}d}n&tj|jz d|z z|z }d}|dt|dd|j d|d|dd||||t||j dS)zq Update the progress bar to the given value (out of the total given to the constructor). rg?rz |=rjrrEr>rh-|z Nz ETA rz>4s/ (>6.2%)) rfloatrrrXrrrrr rr r)rrfracrErbar_filltprefixs rrzProgressBar._update_consoleSs ;!  DD<<% "4"44Dz  !884+,,HH5!122T9::H e C(NF2>>>> d& & & W4R 8 8 8 8 E#)H4q89 : : : c DK   d..AFF aZZAFF)++ 00S4Z@DHAF F/%((FFFt/@FFFGGG  4    !!! f = E*Q--  r-ct|dsddlm}|ddkr ddlm}||_n8tddl m}| |_ddl m }||jd|j_ ||jz }|dz|j_ d|d d |j_d S) z Update the progress bar to the given value (out of a total given to the constructor). This method is for use in the IPython notebook 2+. _widgetrrr)widgets)displaydrrrN)rrr IPython.htmlrFloatProgressWidgetrrr ipywidgets FloatProgressIPython.displayrrr description)rrrrrrs rrz"ProgressBar._update_ipython_widget{stY'' # , , , , , ,A""000000&::<< $$&&&......&4466 / / / / / / GDL ! ! !!"DL t{"!CZ #5#5#5#5#5    r-cdSr)r3rs rrzProgressBar._silent_updates r-rc |r+t|}tt|}||||||||}|r(t t |\} }t|}|S)aMap function over items while displaying a progress bar with percentage complete. The map operation may run in arbitrary order on the items, but the results are returned in sequential order. :: def work(i): print(i) ProgressBar.map(work, range(50)) Parameters ---------- function : function Function to call for each step items : sequence Sequence where each element is a tuple of arguments to pass to *function*. multiprocess : bool, int, optional If `True`, use the `multiprocessing` module to distribute each task to a different processor core. If a number greater than 1, then use that number of cores. ipython_widget : bool, optional If `True`, the progress bar will display as an IPython notebook widget. file : writable file-like, optional The file to write the progress bar to. Defaults to `sys.stdout`. If ``file`` is not a tty (as determined by calling its `isatty` member, if any), the scrollbar will be completely silent. step : int, optional Update the progress bar at least every *step* steps (default: 100). If ``multiprocess`` is `True`, this will affect the size of the chunks of ``items`` that are submitted as separate tasks to the process pool. A large step size may make the job complete faster if ``items`` is very long. multiprocessing_start_method : str, optional Useful primarily for testing; if in doubt leave it as the default. When using multiprocessing, certain anomalies occur when starting processes with the "spawn" method (the only option on Windows); other anomalies occur with the "fork" method (the default on Linux). ) multiprocessrEsteprmultiprocessing_start_method)rlist enumerate map_unorderedzipsorted) rfunctionitemsr#rEr$rr%results_s rmapzProgressBar.mapsz  +))H5))**E##  %))E$    $fWoo.JAw7mmGr-c ddlm}m} g} |t}|t |||5} | jr|} nOt ttt || j z d} t| |} |r|dkrRt|D]A\}}| ||| zdkr| |Bntj|}t!|}|d d|durt|ndi|5t| fd |DD]A\}}| || |B dddn #1swxYwYdddn #1swxYwY| S) aMap function over items, reporting the progress. Does a `map` operation while displaying a progress bar with percentage complete. The map operation may run on arbitrary order on the items, and the results may be returned in arbitrary order. :: def work(i): print(i) ProgressBar.map(work, range(50)) Parameters ---------- function : function Function to call for each step items : sequence Sequence where each element is a tuple of arguments to pass to *function*. multiprocess : bool, int, optional If `True`, use the `multiprocessing` module to distribute each task to a different processor core. If a number greater than 1, then use that number of cores. ipython_widget : bool, optional If `True`, the progress bar will display as an IPython notebook widget. file : writable file-like, optional The file to write the progress bar to. Defaults to `sys.stdout`. If ``file`` is not a tty (as determined by calling its `isatty` member, if any), the scrollbar will be completely silent. step : int, optional Update the progress bar at least every *step* steps (default: 100). If ``multiprocess`` is `True`, this will affect the size of the chunks of ``items`` that are submitted as separate tasks to the process pool. A large step size may make the job complete faster if ``items`` is very long. multiprocessing_start_method : str, optional Useful primarily for testing; if in doubt leave it as the default. When using multiprocessing, certain anomalies occur when starting processes with the "spawn" method (the only option on Windows); other anomalies occur with the "fork" method (the default on Linux). r)ProcessPoolExecutor as_completedN)rrEr) mp_context max_workersTc3DK|]}|VdSr))submit).0itemr+ps r z,ProgressBar.map_unordered..Ls1$P$P$QXXh%=%=$P$P$P$P$P$Pr-r3)concurrent.futuresr1r2r9rrmaxrXrrminr'appendrr? get_contextdictresult)rr+r,r#rEr$rr%r1r2r-bar chunksize default_steprr8ctxrrr9s ` @rr(zProgressBar.map_unordereds@ IHHHHHHH <==D SUN F F F 3#" 4 "3uSZZ'8'83?'J#K#KQOO  d33  3>222I !++ 1 & &12NOO---((-9-E-EL)))4  3  )$ $P$P$P$P$P%$P$P$PPP!!331 1 qxxzz2222 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 36s7C5F9-A(F" F9"F& &F9)F& *F99F=F=)FN)NNr))FNrFN)r.r/r0r1rrrrrrrrrrr2r/r(r3r-rr r sB  4444l////??? ( ( ( (&&&&P6666@       %)NNN[N`   %)aaa[aaar-r cNeZdZdZdZdZddZdZd Zd Z d Z d Z dd Z dZ dS)r z A class to display a spinner in the terminal. It is designed to be used with the ``with`` statement:: with Spinner("Reticulating splines", "green") as s: for item in enumerate(items): s.update() u ◓◑◒◐z-/|\rnNrcF|t}||_||_||_||_|t jr|j}n|j}||_ t| |_ |j r| |_ dS||_ dS)aw Parameters ---------- msg : str The message to print color : str, optional An ANSI terminal color name. Must be one of: black, red, green, brown, blue, magenta, cyan, lightgrey, default, darkgrey, lightred, lightgreen, yellow, lightblue, lightmagenta, lightcyan, white. file : writable file-like, optional The file to write the spinner to. Defaults to `sys.stdout`. If ``file`` is not a tty (as determined by calling its `isatty` member, if any, or special case hacks to detect the IPython console), the spinner will be completely silent. step : int, optional Only update the spinner every *step* steps chars : str, optional The character sequence to use for the spinner N)r9_msg_colorr_steprunicode_output_default_unicode_chars_default_ascii_chars_charsrr_silent_iterator_iter _iterator)rrr{rEr$charss rrzSpinner.__init__bs4 <==D    =" 231 !$<<' < *..00DJJJ))DJJJr-c#K|j}d}|j}|j}|j}d} |dt |j|j|d|d |rt||||}n|||n*#t$r|j }|||d}YnwxYw|dVt|j D]}dV|dzt|z}) NrT rr rFr) rNrrrrrHrIr UnicodeErrorrMrrJr)rrRrrErr try_fallbackrs rrQzSpinner._iterators4 z    - E$KKK  4;Tr B B B B E#JJJ %(0uudKKEEE%,''' % % %1eEl###$  % EGGG EEE4:&&  QY#e**,E+ -s+B$B+*B+c|Sr)r3rs rrzSpinner.__enter__rr-c|j}|j}|j}|js(|dt |j|j|d|t dd|nt dd||dS) NrTrr z [Done]rhrEz [Failed]rg)rrrrrrHrI)rrrrrErrs rrzSpinner.__exit__sz  | C E$KKK  4;Tr B B B B    7 6 6 6 6 6  U 6 6 6 6 r-c|Sr)r3rs rrzSpinner.__iter__rr-c.t|jdSr))rrPrs rrzSpinner.__next__s TZr-c$t|dS)zUpdate the spin wheel in the terminal. Parameters ---------- value : int, optional Ignored (present just for compatibility with `ProgressBar.update`). N)rrs rrzSpinner.updates T r-c#Kt|j|j|jd|j dV)Nrr )rrHrIrrrs rrOzSpinner._silent_iteratorsJDIt{DDDD   EEE r-)rnNrNr))r.r/r0r1rLrMrrQrrrrrrOr3r-rr r Ts,"-*-*-*-*^--->       r-r c,eZdZdZddZdZdZdZdS) ra A class that displays either a `ProgressBar` or `Spinner` depending on whether the total size of the operation is known or not. It is designed to be used with the ``with`` statement:: if file.has_length(): length = file.get_length() else: length = None bytes_read = 0 with ProgressBarOrSpinner(length) as bar: while file.read(blocksize): bytes_read += blocksize bar.update(bytes_read) rnNc|t}|t|s d|_t||||_dSd|_t |||t |||_dS)a Parameters ---------- total : int or None If an int, the number of increments in the process being tracked and a `ProgressBar` is displayed. If `None`, a `Spinner` is displayed. msg : str The message to display above the `ProgressBar` or alongside the `Spinner`. color : str, optional The color of ``msg``, if any. Must be an ANSI terminal color name. Must be one of: black, red, green, brown, blue, magenta, cyan, lightgrey, default, darkgrey, lightred, lightgreen, yellow, lightblue, lightmagenta, lightcyan, white. file : writable file-like, optional The file to write the to. Defaults to `sys.stdout`. If ``file`` is not a tty (as determined by calling its `isatty` member, if any), only ``msg`` will be displayed: the `ProgressBar` or `Spinner` will be silent. NT)r{rEFrY)r9r _is_spinnerr _objrr )rtotalrr{rEs rrzProgressBarOrSpinner.__init__sx4 <==D =t =#D 5t<<rso   !LLLMMMNNN!!! !%%%%%%   5*5*5*5*5*5*5*5*p$'''T$ $ $ $ N-.-.-.`###"   F 77777t777...b/&/&/&d " " " " " " " "ppppppppf CCCCCCCCLC C C C C C C C LJ+J+J+J+h*0))))))))))s;AA