o
�J�h� � @ sp d dl mZ d dlZd dlmZmZ G dd� d�Zedkr6ed�Ze ej
dd gd
d gd d
ddgg d�� dS dS )� )� annotationsN)�Mapping�Sequencec @ sX e Zd ZU dZe�eej�Zded< ded< e ddd��Z
ddd
d �Z dddd�Z d S )� CodeTemplatez-(^[^\n\S]*)?\$([^\d\W]\w*|\{,?[^\d\W]\w*\,?})�str�pattern�filename�returnc C s: t | ��}t|�� | �W d � S 1 sw Y d S �N)�openr �read)r �f� r �TC:\pinokio\api\whisper-webui.git\app\env\lib\site-packages\torchgen\code_template.py� from_file s
$�zCodeTemplate.from_file� �Nonec C s || _ || _d S r
)r r )�selfr r r r r �__init__ s
zCodeTemplate.__init__N�env�Mapping[str, object] | None�kwargs�objectc sF � d u ri � d� �fdd� �dd d
��d��fd
d� }| j �|| j�S )N�vr r r c s$ � d usJ �| �v r�| S � | S r
r )r )r r r r �lookup' s z'CodeTemplate.substitute.<locals>.lookup�indent�Sequence[object]c s d� � fdd�|D ���� S )Nr c s* g | ]}t |��� D ]}� | d �q
qS )�
)r �
splitlines)�.0�e�l�r r r �
<listcomp>- s * zACodeTemplate.substitute.<locals>.indent_lines.<locals>.<listcomp>)�join�rstrip)r r r r" r � indent_lines+ s
��z-CodeTemplate.substitute.<locals>.indent_lines�match�
re.Match[str]c s� | � d�}| � d�}d}d}|d dkr6|dd� }|d dkr(d}|dd � }|d dkr6d}|d d� }�|�}|d urKt|t�sF|g}� ||�S t|t�rhd�d d
� |D ��}t|�dkrb|S || | S t|�S ) N� � r r �{������,z, c S s g | ]}t |��qS r )r )r �xr r r r# C s z<CodeTemplate.substitute.<locals>.replace.<locals>.<listcomp>)�group�
isinstance�listr$ �lenr )r' r �keyZ comma_beforeZ comma_afterr �middle)r& r r r �replace0 s.
z(CodeTemplate.substitute.<locals>.replace)r r r r )r r r r r r )r' r( r r )� substitution�subr )r r r r5 r )r r& r r r �
substitute! s
zCodeTemplate.substitute)r r r r )r )r r r r r r r
)r r r r r r )
�__name__�
__module__� __qualname__Zsubstitution_str�re�compile� MULTILINEr6 �__annotations__� staticmethodr r r8 r r r r r s
�r �__main__z� int foo($args) {
$bar
$bar
$a+$b
}
int commatest(int a${,stuff})
int notest(int a${,empty,})
�hi� �what� � � z things...�others)�args�bar�a�b�stuff�empty) �
__future__r r< �typingr r r r9 �c�printr8 r r r r �<module> s&
<� ���