B
    [;                 @   s   d Z ddlmZmZ ddlmZ ddlmZ ddl	m
Z
 dZdZdZdddZdd Zdd Zdd Zdg d
d
fddZdd Zdd	dddd
d
d
dddg fddZdS )z+Tools for setting up interactive sessions.     )print_functiondivision)LooseVersion)import_module)init_printingzfrom __future__ import division
from sympy import *
x, y, z, t = symbols('x y z t')
k, m, n = symbols('k m n', integer=True)
f, g, h = symbols('f g h', cls=Function)
init_printing()
ziThese commands were executed:
%(source)s
Documentation can be found at http://docs.sympy.org/%(version)s
zCouldn't locate IPython. Having IPython installed is greatly recommended.
See http://ipython.scipy.org for more details. If you use Debian/Ubuntu,
just install the 'ipython' package and start isympy again.
TFNc             C   sD  ddl m} ddlm} ddlm} ddl m} ddl}ddl}|rHdS d|j	dd	  }	| rdd
}
nd}
d| g}|
d}|dk	r| dkr|d |r|d |
||	|d|f}d| }|dkrt}d}x8|ddd D ]"}|s|d7 }q|d| d 7 }qW |}d|kr"d}nd| }|dt||d  7 }|S )z,Create a banner for an interactive session. r   )__version__)GROUND_TYPES)ARCH)SYMPY_DEBUGN z%d.%d.%d   IPythonPythonzground types: %sZSYMPY_USE_CACHEZnoz
cache: offzdebugging: onz, z,%s console for SymPy %s (Python %s-%s) (%s)

z>>> Zdevz%s/)sourceversion)Zsympyr   Zsympy.polys.domainsr   Zsympy.utilities.miscr	   r
   sysosversion_infogetenvlowerappendjoinpreexec_sourcesplitverbose_message)ipythonquietr   Zsympy_versionr   r	   r
   r   r   Zpython_versionZ
shell_nameinfocacheargsmessageZ_sourcelineZdoc_version r$   8lib/python3.7/site-packages/sympy/interactive/session.py_make_message    sD    





r&   c             C   s   ddl m}m}m}m}m} ddlm} dd }g }||| j}	xX|	D ]P\}
}}}}|
|kr||r|	|df|df||f|dfg qH|
|
|f qHW ||S )	a  
    Wrap integer literals with Integer.

    This is based on the decistmt example from
    http://docs.python.org/library/tokenize.html.

    Only integer literals are converted.  Float literals are left alone.
    Examples
    ========

    >>> from __future__ import division
    >>> from sympy.interactive.session import int_to_Integer
    >>> from sympy import Integer
    >>> s = '1.2 + 1/2 - 0x12 + a1'
    >>> int_to_Integer(s)
    '1.2 +Integer (1 )/Integer (2 )-Integer (0x12 )+a1 '
    >>> s = 'print (1/2)'
    >>> int_to_Integer(s)
    'print (Integer (1 )/Integer (2 ))'
    >>> exec(s)
    0.5
    >>> exec(int_to_Integer(s))
    1/2
    r   )generate_tokens
untokenizeNUMBERNAMEOP)StringIOc             S   s(   d| ks d|   ks d|   kr$dS dS )z]
        Returns true if string value num (with token NUMBER) represents an integer.
        .jeFT)r   )Znumr$   r$   r%   _is_intt   s     zint_to_Integer.<locals>._is_intZInteger())tokenizer'   r(   r)   r*   r+   Zsympy.core.compatibilityr,   readlineextendr   )sr'   r(   r)   r*   r+   r,   r0   resultgtoknumtokval_r$   r$   r%   int_to_IntegerX   s    	r<   c                s&   ddl  | j fdd}|| _dS )zM
    Allow IPython to automatically convert integer literals to Integer.
    r   Nc                s@   y  |  W n tk
r"   Y n
X t| } | f|| d S )N)parseSyntaxErrorr<   )Zcellr!   kwargs)astold_run_cellr$   r%   my_run_cell   s    z7enable_automatic_int_sympification.<locals>.my_run_cell)r@   run_cell)shellrB   r$   )r@   rA   r%   "enable_automatic_int_sympification   s    rE   c                s2   ddl }|d d fdd	}| tf| dS )z>Allow IPython to automatially create symbols (``isympy -a``). r   Nz8name '(?P<symbol>[A-Za-z_][A-Za-z0-9_]*)' is not definedc                s   |t kr|jr|jjs t|}|dk	r| jdd|di dd zBy| jd d }W n ttfk
rt   Y nX | j|dd dS W d| jd|d dd X | j	j
||||d	}| ||| dS )
zJHandle :exc:`NameError` exception and allow injection of missing symbols. Nz!%(symbol)s = Symbol('%(symbol)s')symbolF)Zstore_historyZInr   zdel %s)	tb_offset)	NameErrortb_nextmatchstrrC   groupZuser_nsKeyError
IndexErrorZInteractiveTBZstructured_tracebackZ_showtraceback)selfetypevaluetbrG   rJ   codeZstb)re_nameerrorr$   r%   _handler   s     
z*enable_automatic_symbols.<locals>._handler)N)recompileZset_custom_excrH   )rD   rV   rU   r$   )rT   r%   enable_automatic_symbols   s
    'rX   c             C   s   ddl }t|jdkr|| s`t|jdkr6ddlm} nddlm} | }d|_|| |j	} |rlt
|  |rxt|  | S ddlm} ||S dS )zConstruct new IPython session. r   Nz0.11z1.0)ipappF)make_IPython)r   Vr   ZIPython.terminalrY   ZIPython.frontend.terminalZTerminalIPythonAppZdisplay_bannerZ
initializerD   rX   rE   ZIPython.ShellrZ   )rD   argvauto_symbolsauto_int_to_Integerr   rY   ZapprZ   r$   r$   r%   init_ipython_session   s"    
r_   c                 s&   ddl m  G  fddd } |  S )zConstruct new Python session. r   )InteractiveConsolec                   s   e Zd ZdZ fddZdS )z)init_python_session.<locals>.SymPyConsolez.An interactive console with readline support. c                s     |  ydd l}W n tk
r*   Y ndX dd l}dd l}|d t|dr|jd}y|	| W n t
k
r~   Y nX ||j| d S )Nr   ztab: completeread_history_filez~/.sympy-history)__init__r4   ImportErrorr   atexitparse_and_bindhasattrpath
expanduserra   IOErrorregisterwrite_history_file)rO   r4   r   rd   history)r`   r$   r%   rb     s    


z2init_python_session.<locals>.SymPyConsole.__init__N)__name__
__module____qualname____doc__rb   r$   )r`   r$   r%   SymPyConsole  s   rq   )rS   r`   )rq   r$   )r`   r%   init_python_session  s    rr   c          
      s  ddl }d}| dk	ryddl}W n( tk
rH   | dkr@tdd Y n2X yddlm} |  W n tk
rx   d Y nX t }| dkr|} | dkrt   j}n`t |||d t	|j
dkrd fd
d	 _y jdd W n tk
r   Y nX |s j}|r(| r t	|j
dk r(td|rL| rDt	|j
dk rLtdt} j|d	d t|||| ||	|
d t| ||}|st| |  |d nt| ddl}|dd  dS )u  
    Initialize an embedded IPython or Python session. The IPython session is
    initiated with the --pylab option, without the numpy imports, so that
    matplotlib plotting can be interactive.

    Parameters
    ==========

    pretty_print: boolean
        If True, use pretty_print to stringify;
        if False, use sstrrepr to stringify.
    order: string or None
        There are a few different settings for this parameter:
        lex (default), which is lexographic order;
        grlex, which is graded lexographic order;
        grevlex, which is reversed graded lexographic order;
        old, which is used for compatibility reasons and for long expressions;
        None, which sets it to lex.
    use_unicode: boolean or None
        If True, use unicode characters;
        if False, do not use unicode characters.
    use_latex: boolean or None
        If True, use latex rendering if IPython GUI's;
        if False, do not use latex rendering.
    quiet: boolean
        If True, init_session will not print messages regarding its status;
        if False, init_session will print messages regarding its status.
    auto_symbols: boolean
        If True, IPython will automatically create symbols for you.
        If False, it will not.
        The default is False.
    auto_int_to_Integer: boolean
        If True, IPython will automatically wrap int literals with Integer, so
        that things like 1/2 give Rational(1, 2).
        If False, it will not.
        The default is False.
    ipython: boolean or None
        If True, printing will initialize for an IPython console;
        if False, printing will initialize for a normal console;
        The default is None, which automatically determines whether we are in
        an ipython instance or not.
    str_printer: function, optional, default=None
        A custom string printer function. This should mimic
        sympy.printing.sstrrepr().
    pretty_printer: function, optional, default=None
        A custom pretty printer. This should mimic sympy.printing.pretty().
    latex_printer: function, optional, default=None
        A custom LaTeX printer. This should mimic sympy.printing.latex()
        This should mimic sympy.printing.latex().
    argv: list of arguments for IPython
        See sympy.bin.isympy for options that can be used to initialize IPython.

    See Also
    ========

    sympy.interactive.printing.init_printing: for examples and the rest of the parameters.


    Examples
    ========

    >>> from sympy import init_session, Symbol, sin, sqrt
    >>> sin(x) #doctest: +SKIP
    NameError: name 'x' is not defined
    >>> init_session() #doctest: +SKIP
    >>> sin(x) #doctest: +SKIP
    sin(x)
    >>> sqrt(5) #doctest: +SKIP
      ___
    \/ 5
    >>> init_session(pretty_print=False) #doctest: +SKIP
    >>> sqrt(5) #doctest: +SKIP
    sqrt(5)
    >>> y + x + y**2 + x**2 #doctest: +SKIP
    x**2 + x + y**2 + y
    >>> init_session(order='grlex') #doctest: +SKIP
    >>> y + x + y**2 + x**2 #doctest: +SKIP
    x**2 + y**2 + x + y
    >>> init_session(order='grevlex') #doctest: +SKIP
    >>> y * x**2 + x * y**2 #doctest: +SKIP
    x**2*y + x*y**2
    >>> init_session(order='old') #doctest: +SKIP
    >>> x**2 + y**2 + x + y #doctest: +SKIP
    x + y + x**2 + y**2
    >>> theta = Symbol('theta') #doctest: +SKIP
    >>> theta #doctest: +SKIP
    theta
    >>> init_session(use_unicode=True) #doctest: +SKIP
    >>> theta # doctest: +SKIP
    θ
    r   NFTz'IPython is not available on this system)get_ipython)r\   r]   r^   z0.11execc                s     | dS )NF)rC   )srcrF   )ipr$   r%   <lambda>  s    zinit_session.<locals>.<lambda>)Z
import_allzKautomatic construction of symbols is possible only in IPython 0.11 or abovezQautomatic int to Integer transformation is possible only in IPython 0.11 or above)rF   )pretty_printorderuse_unicode	use_latexrv   str_printerpretty_printerlatex_printerzExiting ...c               S   s   t dS )NzExiting ...
)printr$   r$   r$   r%   rw     s    )rt   )r   r   rc   RuntimeErrorrs   boolrr   Zinteractr_   r[   r   Z	runsourceZenable_pylab	Exceptionmainloopr   r   r&   r   exitrd   rj   )r   rx   ry   rz   r{   r   r]   r^   r|   r}   r~   r\   r   Z
in_ipythonr   rs   r   Z_preexec_sourcer"   rd   r$   )rv   r%   init_session-  s`    _


r   )TFN)rp   Z
__future__r   r   Zdistutils.versionr   r[   Zsympy.externalr   Zsympy.interactive.printingr   r   r   Z
no_ipythonr&   r<   rE   rX   r_   rr   r   r$   r$   r$   r%   <module>   s"   	
84G!!