U tdG@sdZddlmZmZddlZddlZddlmZddlm Z m Z ddl m Z GdddZ Gd d d ZGd d d ed ZGdddeZGdddZGdddeeZGdddeeZddZedkreedS)agCommand-line user interface of igraph The command-line interface launches a Python shell with the igraph module automatically imported into the main namespace. This is mostly a convenience module and it is used only by the C{igraph} command line script which executes a suitable Python shell and automatically imports C{igraph}'s classes and functions in the top-level namespace. Supported Python shells are: - IDLE shell (class L{IDLEShell}) - IPython shell (class L{IPythonShell}) - Classic Python shell (class L{ClassicPythonShell}) The shells are tried in the above mentioned preference order one by one, unless the C{global.shells} configuration key is set which overrides the default order. IDLE shell is only tried in Windows unless explicitly stated by C{global.shells}, since Linux and Mac OS X users are likely to invoke igraph from the command line. )ABCMetaabstractmethodN __version__)set_progress_handlerset_status_handler) Configurationc@seZdZdZdZdZdZdZdZdZ dZ dZ dZ dZ dZdZdZdZdZdZdZZZZZZZZdZZZZdZ Z!Z"Z#d$Z%d$Z&d$Z'e(j)fddZ*e+dd Z,e+d d Z-d d Z.ddZ/dS)TerminalControllera A class that can be used to portably generate formatted output to a terminal. C{TerminalController} defines a set of instance variables whose values are initialized to the control sequence necessary to perform a given action. These can be simply included in normal output to the terminal: >>> term = TerminalController() >>> print('This is '+term.GREEN+'green'+term.NORMAL) This is green Alternatively, the L{render()} method can used, which replaces C{${action}} with the string required to perform C{action}: >>> term = TerminalController() >>> print(term.render('This is ${GREEN}green${NORMAL}')) This is green If the terminal doesn't support a given action, then the value of the corresponding instance variable will be set to ''. As a result, the above code will still work on terminals that do not support color, except that their output will not be colored. Also, this means that you can test whether the terminal supports a given action by simply testing the truth value of the corresponding instance variable: >>> term = TerminalController() >>> if term.CLEAR_SCREEN: ... print 'This terminal supports clearning the screen.' ... Finally, if the width and height of the terminal are known, then they will be stored in the C{COLS} and C{LINES} attributes. @author: Edward Loper z BOL=cr UP=cuu1 DOWN=cud1 LEFT=cub1 RIGHT=cuf1 CLEAR_SCREEN=clear CLEAR_EOL=el CLEAR_BOL=el1 CLEAR_EOS=ed BOLD=bold BLINK=blink DIM=dim REVERSE=rev UNDERLINE=smul NORMAL=sgr0 HIDE_CURSOR=cinvis SHOW_CURSOR=cnormz.BLACK BLUE GREEN CYAN RED MAGENTA YELLOW WHITEz.BLACK RED GREEN YELLOW BLUE MAGENTA CYAN WHITEc Csz ddl}Wntk r"YdSX|s0dSz |Wntk rRYdSX|d|_|d|_|jD](}| d\}}t ||| |pdqr| d}|rt t t|j|jD] \}}t |||||pdq| d} | r.t t t|j|jD]$\}}t |||| |p&dq| d } | r~t t t|j|jD](\}}t |d ||| |pvdqT| d } | rt t t|j|jD](\}}t |d ||| |pdqdS) aR Create a C{TerminalController} and initialize its attributes with appropriate values for the current terminal. C{term_stream} is the stream that will be used for terminal output; if this stream is not a tty, then the terminal is assumed to be a dumb terminal (i.e., have no capabilities). rNcolslines=r ZsetfZsetafZsetbZBG_Zsetab)curses ImportErrorisattyZ setupterm ExceptionZtigetnumCOLSLINES_STRING_CAPABILITIESsplitsetattr _tigetstrziprangelen_COLORS_tparm _ANSICOLORS) selfZ term_streamrZ capabilityZattribcap_nameZset_fgicolorZ set_fg_ansiZset_bgZ set_bg_ansir"I/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/igraph/app/shell.py__init__os>         " zTerminalController.__init__cCs.ddl}||pd}|d}tdd|S)zcRewrites string capabilities to remove "delays" which are not required for modern terminalsrNlatin-1z \$<\d+>[/*]?r )rZtigetstrdecoderesub)rrcapr"r"r#rs zTerminalController._tigetstrcCs(ddl}||d|pd}|dS)Nrr&r%)rZtparmencoder')rparamrr*r"r"r#rszTerminalController._tparmcCstd|j|S)z Replace each $-substitutions in the given template string with the corresponding terminal control string (if it's defined) or '' (if it's not). z r\$\$|\${\w+})r(r) _render_sub)rtemplater"r"r#renderszTerminalController.rendercCs*|}|dkr|St||ddSdS)zHelper function for L{render}z$$N)groupgetattr)rmatchsr"r"r#r-szTerminalController._render_subN)0__name__ __module__ __qualname____doc__BOLUPZDOWNLEFTRIGHTZ CLEAR_SCREEN CLEAR_EOLZ CLEAR_BOLZ CLEAR_EOSZBOLDZBLINKZDIMZREVERSEZNORMAL HIDE_CURSOR SHOW_CURSORZBLACKZBLUEZGREENZCYANZREDZMAGENTAZYELLOWZWHITEZBG_BLACKZBG_BLUEZBG_GREENZBG_CYANZBG_REDZ BG_MAGENTAZ BG_YELLOWZBG_WHITErrrrsysstdoutr$ staticmethodrrr/r-r"r"r"r#r s<( 4  r c@s:eZdZdZdZdZddZd ddZd d Zd d Z dS) ProgressBara: A 2-line progress bar. The progress bar looks roughly like this in the console:: Header 20% [===========----------------------------------] The progress bar is colored, if the terminal supports color output; and adjusts to the width of the terminal. z5%3d%% ${GREEN}[${BOLD}%s%s${NORMAL}${GREEN}]${NORMAL}z${BOLD}${CYAN}%s${NORMAL} cCst||_|jjr|jjr|jjs&td|jjp0d|_||j|_ |j|j d |j|_ d|_ d|_d|_dS)NzKTerminal isn't capable enough -- you should use a simpler progress display.Kr Tr)termr>r;r: ValueErrorrwidthr/BAR progress_barHEADERcenterheadercleared last_percent last_message)rrFr"r"r#r$szProgressBar.__init__Nc Cs|jrtjd|jd|_|dkr.|j}n||_|dkrD|j}n||_t|jd|d}tj|j j |j j |j j |j j |j |j||j|j|d|d|jd|fddS)aUpdates the progress bar. @param percent: the percentage to be shown. If C{None}, the previous value will be used. @param message: the message to be shown above the progress bar. If C{None}, the previous message will be used.  FN gY@r -)rNrArBwriterMrPrOintrHrFr:r;r>r/rKrLrJ)rpercentmessagenr"r"r#updates2 zProgressBar.updatecCs|j|dS)zyUpdates the message of the progress bar. @param message: the message to be shown above the progress bar )rW)rYstrip)rrWr"r"r#update_messageszProgressBar.update_messagecCsT|jsPtj|jj|jj|jj|jj|jj|jjd|_d|_d|_ dS)z9Clears the progress bar (i.e. removes it from the screen)Trr N) rNrArBrTrFr:r>r;rOrPrr"r"r#clears"zProgressBar.clear)NN) r6r7r8r9rIrKr$rYr[r]r"r"r"r#rDs  !rDc@sDeZdZdZddZeddZddZdd Zd d Z d d Z dS)Shellz7Superclass of the embeddable shells supported by igraphcCsdSNr"r\r"r"r#r$"szShell.__init__cCstdSr_)NotImplementedErrorr\r"r"r#__call__%szShell.__call__cCs t|dS)zChecks whether the shell supports progress bars. This is done by checking for the existence of an attribute called C{_progress_handler}._progress_handlerhasattrr\r"r"r#supports_progress_bar)szShell.supports_progress_barcCs t|dS)zChecks whether the shell supports status messages. This is done by checking for the existence of an attribute called C{_status_handler}._status_handlerrcr\r"r"r#supports_status_messages0szShell.supports_status_messagescCs|r|jSdS)z:Returns the progress handler (if exists) or None (if not).N)rerbr\r"r"r#get_progress_handler7szShell.get_progress_handlercCs|r|jSdS)z8Returns the status handler (if exists) or None (if not).N)rgrfr\r"r"r#get_status_handler=szShell.get_status_handlerN) r6r7r8r9r$rrarergrhrir"r"r"r#r^s r^) metaclasscs(eZdZdZfddZddZZS) IDLEShellzmIDLE embedded shell interface. This class allows igraph to be embedded in IDLE (the Tk Python IDE). cstddl}d|j_z tjWntk r>dt_YnX|jjdd}|j || |j |}| szt |j|_||_dS)zConstructor. Imports IDLE's embedded shell. The implementation of this method is ripped from idlelib.PyShell.main() after removing the unnecessary parts.rNTz>>> Idle)Z className)superr$Zidlelib.PyShellZPyShellZuse_subprocessrAps1AttributeErrorZTkZ fixwordbreaksZwithdrawZPyShellFileListZ open_shellr`Zpyshell_shell_root)ridlelibrootflist __class__r"r#r$Ms     zIDLEShell.__init__cCs&|jjd|j|jdS)zStarts the shellfrom igraph import *N)rpZinterpZ execsourcerqmainloopdestroyr\r"r"r#rags zIDLEShell.__call__)r6r7r8r9r$ra __classcell__r"r"rur#rkDs rkc@s8eZdZdZddZddZeddZedd Zd S) ConsoleProgressBarMixinz;Mixin class for console shells that support a progress bar.cCsNztt|j_Wn6tk r.|Yntk rH|YnXdSr_)rDr rvrJrG_disable_handlers TypeErrorr\r"r"r#r$qs  z ConsoleProgressBarMixin.__init__cCsDz |j`Wntk rYnXz |j`Wntk r>YnXdS)zXDisables the status and progress handlers if the terminal is not capable enough.N)rvrbrorfr\r"r"r#r||s  z)ConsoleProgressBarMixin._disable_handlerscCs&|dkr|jn|j||dS)zProgress bar handler, called when C{igraph} reports the progress of an operation @param message: message provided by C{igraph} @param percentage: percentage provided by C{igraph} dN)rJr]rY)clsrW percentager"r"r#rbs z)ConsoleProgressBarMixin._progress_handlercCs|j|dS)zStatus message handler, called when C{igraph} sends a status message to be displayed. @param message: message provided by C{igraph} N)rJr[)rrWr"r"r#rfsz'ConsoleProgressBarMixin._status_handlerN) r6r7r8r9r$r| classmethodrbrfr"r"r"r#r{ns   r{c@s eZdZdZddZddZdS) IPythonShellznIPython embedded shell interface. This class allows igraph to be embedded in IPython's interactive shell.cCst|t|ddl}ddlm}||_zLzddlm}Wn t k r`ddl m}YnX| |_ |j dWnBt k rddl}|j|_ |j jd|j dYnXdS)zTConstructor. Imports IPython's embedded shell with separator lines removed.rNr)TerminalIPythonAppz--noseprwz-nosep)r^r$r{rAIPythonripython_versionZIPython.terminal.ipapprrZIPython.frontend.terminal.ipappinstancerpargvappendZ IPython.ShellstartZIP runsource)rrArrrr"r"r#r$s"     zIPythonShell.__init__cCsPtdtdd|jjjdkrB|j|jjd|jn |j dS)Starts the embedded shell.igraph %s running inside r )endrrwN) printrrprvr6Z initializeshellexrrxr\r"r"r#ras   zIPythonShell.__call__Nr6r7r8r9r$rar"r"r"r#rs rc@s eZdZdZddZddZdS)ClassicPythonShellz_Classic Python shell interface. This class allows igraph to be embedded in Python's shell.cCst|t|d|_dS)z4Constructor. Imports Python's classic shellN)r^r$r{rpr\r"r"r#r$s  zClassicPythonShell.__init__cCsL|jdkr>ddlm}||_tdtdtjd|jd|jdS)rNr)InteractiveConsolerr )rfilerw) rpcoderrrrAstderrrZinteract)rrr"r"r#ras    zClassicPythonShell.__call__Nrr"r"r"r#rsrc Cst}|jr$td|jtjdntdtjdd|krdd|ddD}g}tddt D}|D]6}| |d }|d krtd |tjdqr| |qrn(t t g}d d l}|d kr|d td }|D]>}z|}WqWqtk rd t|krYqXqt|trf|dr^|rHt||r^t||ntdtjdtdtjdd S)zLThe main entry point for igraph when invoked from the command line shellzUsing configuration from %s)rz%No configuration file, using defaultsZshellscSsg|] }|qSr")rZ).0partr"r"r# szmain..,cSs,g|]$\}}t|trt|tr||fqSr") isinstancetype issubclassr^)rkvr"r"r#rs Nz!Warning: unknown shell class `%s'rWindowsZClassicverbosez#No suitable Python shell was found.z.Check configuration variable `general.shells'.)rrfilenamerrArrdictglobalsitemsgetrrrplatformsysteminsertrkrstrrr^rerrhrgrri) configpartsZ shell_classesZavailable_classesrrrrZ shell_classr"r"r#mainsP          r__main__)r9abcrrr(rAZigraphrZigraph._igraphrrZigraph.configurationrr rDr^rkr{rrrr6exitr"r"r"r#s"  )W%*108