U e{j @sdZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z z ddl Z ddl Z ddl Z dZWnek rdZYnXddlmZddlmZddlmZd d d d d ddddg ZdZGdddZd6ddZdd Zd7ddZddZddZddZdd d!d Zd"d#Z d$d Z!d%d Z"Gd&d'd'Z#Gd(d d Z$Gd)ddZ%Gd*ddZ&d8d-dZ'Gd.d/d/Z(Gd0d1d1Z)Gd2d3d3Z*Gd4d5d5Z+dS)9z) Utilities for console input and output. NTF)conf) classproperty) isiterableisatty color_print human_timehuman_file_size ProgressBarSpinnerprint_code_lineProgressBarOrSpinner terminal_sizezutf-8c@s@eZdZdZeddZeddZeddZedd Z d S) _IPythonz5Singleton class given access to IPython streams, etc.cCs*zddlm}Wntk r$YnX|S)Nr) get_ipython)IPythonr ImportError)clsrrT/var/www/html/bwave/python/venv/lib/python3.8/site-packages/astropy/utils/console.pyr4s z_IPython.get_ipythoncCst|dsd|_z |Wntk r2YdSXzddlm}Wntk rzddlm}Wn`tk rddlm }|ddkrYYdSzddl m}Wntk rYYYdSXYnXYnX||_|jS)N _OutStreamr) OutStream version_info) hasattrrr NameErrorZipykernel.iostreamrrZIPython.zmq.iostreamrrZIPython.kernel.zmq.iostream)rrrrrrr<s*    z_IPython.OutStreamcCsBt|dsd |||||||Sq>d S)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)r[<)rLrrz Z2drLrz{:2d}{}{:2d}{}z ~inf)rErurvformat)secondsunitsryZunit1Zlimit1Zunit2Zlimit2rrrrus*    cCst|dr&ddlm}|||jj}d}|dkr8d}nttt |t d}|dkrdd}n||}tt d|}||}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.rsz>3s) rastropyrZQuantitybytevaluerEmathfloorlogpowstrindex)sizeusuffixesZ num_scalesuffixrZ str_valuerrrr s&    c@s eZdZdZddZddZdS)_mapfuncz: A function wrapper to support ProgressBar.map(). cCs ||_dSr Z_func)selffuncrrr__init__sz_mapfunc.__init__cCs|\}}|||fSr r)rZi_argryargrrr__call__sz_mapfunc.__call__Nr$r%r&r'rrrrrrrsrc@seZdZdZdddZdddZdd Zd d Zd d ZddZ dddZ d ddZ d!ddZ d"ddZ ed#ddZed$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() FNcCs|dkrt}|s*t|s*|j|_d|_nd|_t|rNt||_t||_ nszProgressBar.__next__cCs8|dkr|jd}||_|jr*||n ||dS)zN Update progress bar via the console or notebook accordingly. Nr)Z_current_valuer_update_ipython_widget_update_consolerrrrrrHs   zProgressBar.updatecCsR|jdkrd}nt|t|j}|j}|j}|dkrBt|j}ntt|j|}|dtd|d|dd||jkrtd d |dd|d |j|d|d ||jkrt|j}d }n0|dkrd}d}nt|jd||}d}|dt |dd|j d|d|dd|||dk rD|t ||j dS)zq Update the progress bar to the given value (out of the total given to the constructor). rg?rz |=rSrtr6rr>rR-|z Nz ETA rz>4s/ (>6.2%)) rfloatrrlrErrrrr rrr)rrfracr6rlZbar_filltprefixrrrrXs8      zProgressBar._update_consolecCst|dsvddlm}|ddkr:ddlm}||_ntddl m}| |_ddl m }||jd|j_ ||j}|d|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)rrrZ IPython.htmlrZFloatProgressWidgetrrr ipywidgetsZ FloatProgressIPython.displayrrr description)rrrrrrrrrrs           z"ProgressBar._update_ipython_widgetcCsdSr rrrrrrszProgressBar._silent_updaterc CsP|rt|}tt|}|j|||||||d}|rLtt|\} }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). ) multiprocessr6steprmultiprocessing_start_method)rlist enumerate map_unorderedzipsorted) rfunctionitemsrr6rrrresults_rrrmaps >  zProgressBar.mapc s<ddlm}m} g} |dkr"t}|t|||d} | jrB|} n&tttt|| j d} t | |} |rt|dkrt |D],\}}| ||| dkr|| |q|nt|}t|d}|fd|dk rt|ndi|Ft | fd d |DD]"\}}| || |qW5QRXW5QRX| 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)rr6r)Z mp_context max_workersTc3s|]}|VqdSr )Zsubmit).0itemrprr Tsz,ProgressBar.map_unordered..)concurrent.futuresrrr,rvrmaxrErrminrappendrr1Z get_contextdictresult)rrrrr6rrrrrrbar chunksizeZ default_stepryrctxrxrorrrrs8@       &zProgressBar.map_unordered)FN)NN)N)N)N)N)FNrFN)FNrFN)r$r%r&r'rrrrrrrrrrr(rrrrrrr s2 6    ) !  Qc@s\eZdZdZdZdZdddZd d Zd d Zd dZ ddZ ddZ dddZ ddZ 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-/|\rVNrcCst|dkrt}||_||_||_||_|dkrBtjr<|j}n|j}||_ t | |_ |j rf| |_ n ||_ 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)r,_msg_colorr_steprunicode_output_default_unicode_chars_default_ascii_chars_charsrr_silent_iterator_iter _iterator)rrzr^r6rcharsrrrrjs  zSpinner.__init__ccs|j}d}|j}|j}|j}d}|dt|j|j|dd|dz&|r\t||||}n |||Wn*tk r|j }|||d}YnX|dVt |j D] }dVq|dt |}q dS) NrT rtrrFr) rrrlrrrrrp UnicodeErrorrrurrv)rrrr6rlrZ try_fallbackryrrrrs,  zSpinner._iteratorcCs|Sr rrrrrrszSpinner.__enter__cCsd|j}|j}|j}|js4|dt|j|j|dd|dkrLtdd|dntdd|d|dS) Nrrtrz [Done]rRr6z [Failed]rQ)rrlrrrrr)rrrrr6rlrrrrrszSpinner.__exit__cCs|Sr rrrrrrszSpinner.__iter__cCst|jdSr )rrrrrrrszSpinner.__next__cCs t|dS)zUpdate the spin wheel in the terminal. Parameters ---------- value : int, optional Ignored (present just for compatibility with `ProgressBar.update`). N)rrrrrrs zSpinner.updateccs,t|j|j|jdd|jdVq dS)Nrtr)rrrrrrrrrrs zSpinner._silent_iterator)rVNrN)N)r$r%r&r'rrrrrrrrrrrrrrr \s  0 c@s2eZdZdZd ddZddZdd Zd d ZdS) r a 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) rVNcCs\|dkrt}|dkst|s6d|_t|||d|_n"d|_t|||dt||d|_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^r6Fr)r,rZ _is_spinnerr _objrr )rtotalrzr^r6rrrrszProgressBarOrSpinner.__init__cCs|Sr rrrrrrszProgressBarOrSpinner.__enter__cCs|j|||Sr )rrrrrrr szProgressBarOrSpinner.__exit__cCs|j|dS)zp Update the progress bar to the given value (out of the total given to the constructor. N)rrrrrrr#szProgressBarOrSpinner.update)rVN)r$r%r&r'rrrrrrrrr s  &Fc CsT|dkrt}tjrd}nd}|j}|dk rb|t|kr@td|d|d}|||d7}|d}|dd|}|dk r||krt |d t||}||} || t|d}|t|8}|}|| 8}t |d |d d t||kr||d|t|t |d |d n|||d|dk rP|d|t dd|d dS)ux Prints a line of source code, highlighting a particular character position in the line. Useful for displaying the context of error messages. If the line is more than ``width`` characters, the line is truncated accordingly and '…' characters are inserted at the front and/or end. It looks like this:: there_is_a_syntax_error_here : ^ Parameters ---------- line : unicode The line of code to display col : int, optional The character in the line to highlight. ``col`` must be less than ``len(line)``. file : writable file-like, optional Where to write to. Defaults to `sys.stdout`. tabwidth : int, optional The number of spaces per tab (``'\t'``) character. Default is 8. All tabs will be converted to spaces to ensure that the caret lines up with the correct column. width : int, optional The width of the display, beyond which the line will be truncated. Defaults to 70 (this matches the default in the standard library's `textwrap` module). Nu…z...z%col must be less the the line length. rrqrrsrWrtrr^rQ) r,rrrlrv ValueErrorcountrstripreplacerr) linecolr6ZtabwidthrOellipsisrlZntabsZnew_coloffsetrrrr +s:&     c@s eZdZdZddZddZdS)Getchz|Get a single character from standard input without screen echo. Returns ------- char : str (one character) c CsTz t|_WnBtk rNz t|_Wn ttfk rHt|_YnXYnXdSr ) _GetchWindowsimplr_GetchMacCarbonAttributeError _GetchUnixrrrrrs  zGetch.__init__cCs|Sr )rrrrrrszGetch.__call__Nrrrrrrs rc@seZdZddZddZdS)rcCsddl}ddl}ddl}dSNr)r+rAtty)rr+rArrrrrsz_GetchUnix.__init__c Csdddl}ddl}ddl}|j}||}z ||j|j d}W5|||j|X|S)Nrr) r+rArstdinfileno tcgetattr tcsetattr TCSADRAINZsetrawread)rr+rArfdZ old_settingschrrrrs  z_GetchUnix.__call__Nr$r%r&rrrrrrrsrc@seZdZddZddZdS)rcCs ddl}dSr)msvcrtrr'rrrrsz_GetchWindows.__init__cCsddl}|Sr)r'Zgetchr(rrrrsz_GetchWindows.__call__Nr&rrrrrsrc@s eZdZdZddZddZdS)rz A function which returns the current ASCII key that is down; if no ASCII key is down, the null string is returned. The page http://www.mactech.com/macintosh-c/chap02-1.html was very helpful in figuring out how to do this. cCsddl}|jdSr)CarbonEvt)rr)rrrrsz_GetchMacCarbon.__init__cCsJddl}|jdddkr dS|jdd\}}}}}t|d@SdS)Nrr rtr)r)r*Z EventAvailZ GetNextEventchr)rr)whatrzwhenwheremodrrrrs  z_GetchMacCarbon.__call__Nrrrrrrsr)F)N)NNr r ),r'rjrarr1rFr{r>r+r3rr@rrArrrrZ decoratorsrmiscr__all__rerr,rrr_rhrprr}rr rr r r r rrrrrrrrsl     8 * (0#;12 y G U