Format: application/x-python-code
1 o
2
3 ��h) � @ sL d Z ddlZddlmZ G dd� de�ZG dd� de�ZG dd � d e�ZdS )
4 zIO capturing utilities.� N)�StringIOc @ sh e Zd Zddd�Zdd� Zdd� Zdd d
5 �Zd d � Zd
6 d� Zdd� Z dd� Z
7 dd� Z dd� Z dd� Z
8 dS )�
9 RichOutputNFc C s( |pi | _ |pi | _|p
10 i | _|| _d S �N��data�metadata� transient�update)�selfr r r r � r �Kc:\Users\shubh\Videos\agent\nova\Lib\site-packages\IPython/utils/capture.py�__init__ s
11
12
13
14 zRichOutput.__init__c C s( ddl m} || j| j| j| jd� d S )Nr )�publish_display_datar )�IPython.displayr r r r r )r
15 r r r r �display s
16
17 �zRichOutput.displayc C s4 || j vrd S | j | }|| jv r|| j| fS |S r �r r )r
18 �mimer r r r � _repr_mime_ s
19
20
21 zRichOutput._repr_mime_c C s | j | jfS r r )r
22 �include�excluder r r �_repr_mimebundle_% s z RichOutput._repr_mimebundle_c C �
23 | � d�S )Nz text/html�r �r
24 r r r � _repr_html_( �
25 zRichOutput._repr_html_c C r )Nz
26 text/latexr r r r r � _repr_latex_+ r zRichOutput._repr_latex_c C r )Nzapplication/jsonr r r r r � _repr_json_. r zRichOutput._repr_json_c C r )Nzapplication/javascriptr r r r r �_repr_javascript_1 r z RichOutput._repr_javascript_c C r )Nz image/pngr r r r r �
27 _repr_png_4 r zRichOutput._repr_png_c C r )Nz
28 image/jpegr r r r r � _repr_jpeg_7 r zRichOutput._repr_jpeg_c C r )Nz
29 image/svg+xmlr r r r r �
30 _repr_svg_: r zRichOutput._repr_svg_)NNNF)NN)�__name__�
31 __module__� __qualname__r
32 r r r r r r r r r r! r r r r r s
33
34 r c @ sR e Zd ZdZddd�Zdd� Zedd� �Zed d
35 � �Zed d � �Z d
36 d� Z
37 e
38 Z dS )�
39 CapturedIOa� Simple object for containing captured stdout/err and rich display StringIO objects
40
41 Each instance `c` has three attributes:
42
43 - ``c.stdout`` : standard output as a string
44 - ``c.stderr`` : standard error as a string
45 - ``c.outputs``: a list of rich display outputs
46
47 Additionally, there's a ``c.show()`` method which will print all of the
48 above in the same order, and can be invoked simply via ``c()``.
49 Nc C s" || _ || _|d u r g }|| _d S r )�_stdout�_stderr�_outputs)r
50 �stdout�stderr�outputsr r r r
51 K s
52
53 zCapturedIO.__init__c C s | j S r )r) r r r r �__str__R s zCapturedIO.__str__c C � | j sdS | j �� S )zCaptured standard output� )r& �getvaluer r r r r) U �
54 zCapturedIO.stdoutc C r- )zCaptured standard errorr. )r' r/ r r r r r* \ r0 zCapturedIO.stderrc C s dd� | j D �S )a A list of the captured rich display outputs, if any.
55
56 If you have a CapturedIO object ``c``, these can be displayed in IPython
57 using::
58
59 from IPython.display import display
60 for o in c.outputs:
61 display(o)
62 c S s g | ] }t d i |���qS )r )r )�.0�kargsr r r �
63 <listcomp>n s z&CapturedIO.outputs.<locals>.<listcomp>)r( r r r r r+ c s zCapturedIO.outputsc C sR t j�| j� t j�| j� t j�� t j�� | jD ] }tdi |���� qdS )z0write my output to sys.stdout/err as appropriateNr )�sysr) �writer* �flushr( r r )r
64 r2 r r r �showp s
65
66
67 �zCapturedIO.showr ) r" r# r$ �__doc__r
68 r, �propertyr) r* r+ r7 �__call__r r r r r% > s
69
70
71
72 r% c @ s6 e Zd ZdZdZdZdZd
73 dd�Zdd� Zdd� Z d S ) �capture_outputz(context manager for capturing stdout/errTc C s || _ || _|| _d | _d S r )r) r* r �shell)r
74 r) r* r r r r r
75 � s
76 zcapture_output.__init__c C s� ddl m} ddlm} ddlm} tj| _tj | _
77 | j r,|� | _ | j d u r,d | _
78 d| _ d } }}| jr;t� }t_| j rDt� }t_ | j rb| j j| _
79 |� | j _| j jj}tj| _|| j |d�t_t|||�S )Nr )� get_ipython)�CapturingDisplayPublisher)�CapturingDisplayHookF)r< r+ )ZIPython.core.getipythonr= ZIPython.core.displaypubr> ZIPython.core.displayhookr? r4 r) �
80 sys_stdoutr* �
81 sys_stderrr r< �save_display_pubr � display_pubr+ � displayhook�save_display_hookr% )r
82 r= r> r? r) r* r+ r r r � __enter__� s0
83
84
85
86 � zcapture_output.__enter__c C s: | j t_| jt_| jr| jr| j| j_| j t_
87 d S d S d S r ) r@ r4 r) rA r* r r< rB rC rE rD )r
88 �exc_type� exc_value� tracebackr r r �__exit__� s
89 �zcapture_output.__exit__N)TTT)
90 r" r# r$ r8 r) r* r r
91 rF rJ r r r r r; | s
92 r; )r8 r4 �ior �objectr r% r; r r r r �<module> s .>