o
    DDi_                     @   sD   d dl Z d dlZd dlmZ G dd dejZG dd dejZdS )    Nc                   @   s>   e Zd ZeeeddejdejdejfddZ	dS )SimplifyUnionsTransformerUnionoriginal_nodeupdated_nodereturnc                 C   s*   t |jdkrt|jd jtjjS |S )N   r   )lenslicecstensure_typeIndexvalue)selfr   r    r   R/var/www/Datamplify/venv/lib/python3.10/site-packages/libcst/codegen/transforms.py_leave_union   s   z&SimplifyUnionsTransformer._leave_unionN)
__name__
__module____qualname__mleave	SubscriptNamer
   BaseExpressionr   r   r   r   r   r      s    r   c                   @   s6   e Zd Zee dejdejdejfddZdS )!DoubleQuoteForwardRefsTransformerr   r   r   c                 C   s^   d|j v r-d|jdd  d}z|jt|kr |j|dW S W |S  ty,   Y |S w |S )N'"r   )r   )quoter   evaluated_valueastliteral_evalwith_changesSyntaxError)r   r   r   	new_valuer   r   r   leave_SimpleString!   s   
z4DoubleQuoteForwardRefsTransformer.leave_SimpleStringN)	r   r   r   r   call_if_inside
Annotationr
   SimpleStringr%   r   r   r   r   r       s    r   )	r    libcstr
   libcst.matchersmatchersr   MatcherDecoratableTransformerr   r   r   r   r   r   <module>   s
   