File: //usr/local/lib/python3.10/dist-packages/openai/cli/_tools/__pycache__/migrate.cpython-310.pyc
o
���g� � @ s d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dlm Z m
Z
d dl m Z d dl
mZ d dlZddlmZmZ ddlmZ e rLd d l
mZ d)dd�ZG dd� de�Zd*dd�ZG dd� de�Zd+dd�ZdZd,dd �Z d-d d �Z d,d!d"�Z d.d%d&�Z d/d'd(�Z dS )0� )� annotationsN)�
TYPE_CHECKING�List)�Path)�ArgumentParser� )�CLIError�SilentCLIError)� BaseModel)�_SubParsersAction� subparser�!_SubParsersAction[ArgumentParser]�return�Nonec C s8 | � d�}|jttdd� | � d�}|jttdd� d S )N�migrateT)�func�
args_model�allow_unknown_args�grit)�
add_parser� set_defaultsr � MigrateArgsr �GritArgs)r �sub� r �D/usr/local/lib/python3.10/dist-packages/openai/cli/_tools/migrate.py�register s
r c @ � e Zd ZU g Zded< dS )r � List[str]� unknown_argsN��__name__�
__module__� __qualname__r �__annotations__r r r r r �
r �argsc C s8 t � }z t�|g| j�� W d S tjy t� d �w �N��install�
subprocess�
check_callr �CalledProcessErrorr �r& � grit_pathr r r r # s �r c @ r )r r r Nr r r r r r . r% r c C s< t � }zt�|ddg| j�� W d S tjy t� d �w )N�apply�openair( r- r r r r 3 s �r z
custodian-devr c C s( t j�d�} | d urt| �S t�� d S )N�XDG_CACHE_HOMEz.cache)�os�environ�getr �home)�xdgr r r �
_cache_dirC s r7 �message�strc C s( t j�d�sd S tj�d| � d�� d S )N�DEBUGz [DEBUG]: �
)r2 r3 r4 �sys�stdout�write)r8 r r r �_debugK s r? c C s t jdkr td��td� t jdkrdnd} t� d }|d }|d }|d
}|d }|�� r=t|� d �� t j�� |S td
|� �� |jddd� |�� rS|� � t
� }td|� �� d|� d| � �}d|� d�}t j� d|� d�� t �
� �<} | j|dd�}
|
jdkr�td|� ���t|d��} |
�� D ]} | � | � q�W d � n1 s�w Y W d � n1 s�w Y |d }
|
jddd� t�|d ��}t jd kr�|j|
d d � n|�|
� W d � n1 s�w Y t|
|� t�|
� t�|� t�|d!� t j�� |S )"z<Installs the Grit CLI and returns the location of the binary�win32z1Windows is not supported yet in the migration CLIz Using Grit installer from GitHub�darwinz apple-darwinzunknown-linux-gnuz
openai-pythonz.install�binr zgrit.tmpz already existszUsing Grit CLI path: T)�parents�exist_okzUsing architecture zgrit-�-z;https://github.com/getgrit/gritql/releases/latest/download/z.tar.gzzDownloading Grit CLI from r; )�follow_redirects�� z!Failed to download Grit CLI from �wbNzcli-binzr:gz)� � �data)�filteri� )r< �platformr r? r7 �existsr= �flush�mkdir�unlink� _get_archr> �httpx�Clientr4 � status_code�open�
iter_bytes�tarfile� version_info�
extractall�_move_files_recursively�shutil�rmtreer2 �remove�chmod)rM �dir_name� install_dir�
target_dir� target_path� temp_file�arch� file_name� download_url�client�download_response�file�chunk� unpacked_dir�archiver r r r) R s^
� �� �
� �
r) �
source_dirrb c C s> | � � D ]}|�� r|�||j � q|�� r t||� qd S r' )�iterdir�is_file�rename�name�is_dirr[ )rn rb �itemr r r r[ � s
��r[ c C s& t �� �� } ddddd�}|�| | �S )N�x86_64�aarch64)ru �amd64�armv7l�arm64)rM �machine�lowerr4 )� architecture�arch_mapr r r rR � s � rR )r r
r r )r& r r r )r& r r r )r r )r8 r9 r r )rn r rb r r r )r r9 )!�
__future__r r2 r< r\ rX rM r* �typingr r �pathlibr �argparser rS �_errorsr r �_modelsr
r r r r r r �KEYGEN_ACCOUNTr7 r? r) r[ rR r r r r �<module> s4
?