Format: application/x-python-code
1 o
2
3 ��h � @ s� d Z ddlmZ ddlZddlmZ ddlmZmZ ddl m
4 Z
5 ddl m Z G dd � d e
6 �ZG d
7 d � d e�ZG d d
8 � d
9 e�ZdS )z"Terminal input and output prompts.� )�TokenN)� DisplayHook)�fragment_list_width�PygmentsTokens)�print_formatted_text)� EditingModec @ sZ e Zd Zdd� Zdd� Zdefdd�Zdd � Zd
10 d � Zdd d
11 �dd�Z dd� Z
12 dd� Z d S )�Promptsc C s
13 || _ d S �N)�shell)�selfr
14 � r �Nc:\Users\shubh\Videos\agent\nova\Lib\site-packages\IPython/terminal/prompts.py�__init__ s
15 zPrompts.__init__c C sr t | jjdd �tjkr7| jjr7t| jjjjj �}|�
16 d�r&|dd� � � }n |�
17 d�r1|dd� }d| d S d
18 S ) NZ editing_modez
19 InputMode.�
20 �
21 zvi-� � �[z] � ) �getattrr
22 �pt_appr ZVIZprompt_includes_vi_mode�str�appZvi_stateZ
23 input_mode�
24 startswith�lower)r �moder r r
25 �vi_mode s �
26
27 zPrompts.vi_mode�returnc C s" | j jd ur| j jjjjpdS dS )Nr )r
28 r Zdefault_bufferZdocumentZcursor_position_row�r r r r
29 � current_line s zPrompts.current_linec C sL t j| �� ft j| jjjd| �� d�ft jdft jt| jj �ft jdfgS )N� ��lineZrel_linezIn [�]: )
30 r �Promptr r
31 �prompt_line_number_format�formatr Z PromptNumr �execution_countr r r r
32 �in_prompt_tokens! s
33 ���zPrompts.in_prompt_tokensc C s t | �� �S r )r r( r r r r
34 �_width/ s zPrompts._widthN)�linenoc C sv |d u r| � � }|d ur|d nd}dt| �� � | jjj||| �� d d� }tj|d|t|� d d fgS )Nr r � r! � z...: ) r) �lenr r
35 r% r& r r r$ )r �widthr* r" �prefixr r r
36 �continuation_prompt_tokens2 s �����z"Prompts.continuation_prompt_tokensc C s | � � }tjd|d d fgS )N�-� z> )r) r r$ �r r. r r r
37 �rewrite_prompt_tokensB s �z Prompts.rewrite_prompt_tokensc C s$ t jdft jt| jj�ft jdfgS )NzOut[r# )r Z OutPromptZ OutPromptNumr r
38 r' r r r r
39 �out_prompt_tokensH s �zPrompts.out_prompt_tokensr ) �__name__�
40 __module__� __qualname__r r �intr r( r) r0 r4 r5 r r r r
41 r
42 s r c @ s. e Zd Zdd� Zd
43 dd�Zdd� Zdd � ZdS ) �ClassicPromptsc C � t jdfgS )Nz>>> �r r$ r r r r
44 r( P � �z ClassicPrompts.in_prompt_tokensNc C r; )Nz... r< r3 r r r
45 r0 U r= z)ClassicPrompts.continuation_prompt_tokensc C � g S r r r r r r
46 r4 Z � z$ClassicPrompts.rewrite_prompt_tokensc C r> r r r r r r
47 r5 ] r? z ClassicPrompts.out_prompt_tokensr )r6 r7 r8 r( r0 r4 r5 r r r r
48 r: O s
49
50 r: c s, e Zd ZdZdd� Zdd � fdd�
51 Z� ZS )
52 �RichPromptDisplayHookz3Subclass of base display hook using coloured promptc C s� t j�| jj� d| _| jrC| jj�� }d� dd� |D ��}|r(|�
53 d�s(d| _| jj r;t t
54 |�| jj jjdd� d S t j�|� d S d S )NTr c s s � | ]\}}|V qd S r r )�.0�_�sr r r
55 � <genexpr>j s � z<RichPromptDisplayHook.write_output_prompt.<locals>.<genexpr>�
56 F)�style�end)�sys�stdout�writer
57 Z separate_outZprompt_end_newlineZ
58 do_full_cacheZpromptsr5 �join�endswithr r r r rF )r �tokensZ
59 prompt_txtr r r
60 �write_output_promptb s
61 ��z)RichPromptDisplayHook.write_output_promptNr c sZ | j d usJ �| j jr$| j j�� D ]\}}||v r#||| d � d S qt� �||� d S r )r
62 Zmime_renderers�items�super�write_format_data)r Z format_dictZmd_dict�mime�handler�� __class__r r
63 rQ v s �z'RichPromptDisplayHook.write_format_datar )r N)r6 r7 r8 �__doc__rN rQ �
64 __classcell__r r rT r
65 r@ ` s r@ )rV Zpygments.tokenr rH ZIPython.core.displayhookr Z prompt_toolkit.formatted_textr r Zprompt_toolkit.shortcutsr Zprompt_toolkit.enumsr �objectr r: r@ r r r r
66 �<module> s B