U ܟj@s<ddlmZddlZddlZddlZddlZddlZddlZddlm Z m Z ddl m Z ddl mZddlZddlmZmZddlmZe rdd l mZd d d d dZGdddeZdd dddZGdddeZdd dddZdZddddZdd ddd Zddd!d"Zddd d#d$d%Zddd&d'Z dS)() annotationsN) TYPE_CHECKINGList)Path)ArgumentParser)CLIErrorSilentCLIError) BaseModel)_SubParsersActionz!_SubParsersAction[ArgumentParser]None) subparserreturncCs8|d}|jttdd|d}|jttdddS)NmigrateT)funcZ args_modelZallow_unknown_argsgrit) add_parser set_defaultsr MigrateArgsrGritArgs)r subr=/tmp/pip-unpacked-wheel-2qj3ofc_/openai/cli/_tools/migrate.pyregisters  rc@seZdZUgZded<dS)r List[str] unknown_argsN__name__ __module__ __qualname__r__annotations__rrrrrs r)argsrcCs@t}zt|f|jWntjk r:tdYnXdSNinstall subprocess check_callrCalledProcessErrorr r!Z grit_pathrrrr#s rc@seZdZUgZded<dS)rrrNrrrrrr.s rcCsDt}zt|ddf|jWntjk r>tdYnXdS)NapplyZopenair#r(rrrr3s rz custodian-devr)rcCs(tjd}|dk rt|StdS)NXDG_CACHE_HOMEz.cache)osenvirongetrhome)Zxdgrrr _cache_dirCs r/str)messagercCs(tjdsdStjd|ddS)NDEBUGz [DEBUG]:  )r+r,r-sysstdoutwrite)r1rrr_debugKs r7c Cstjdkrtdtdtjdkr(dnd}td}|d}|d }|d }|d }|rzt|d tj|Std ||jddd|r| t }td|d|d|}d|d}tj d|dt ^} | j|dd} | jdkrtd|t|d } | D]} | | q0W5QRXW5QRX|d} | jdddt|d,}tjdkr|j| ddn || W5QRXt| |t| t|t|d tj|S)!z|D]0}|r&|||jq|rt||qdSr")iterdiris_filerenamenameis_dirrP)r`rUitemrrrrPs  rPcCs&t}ddddd}|||S)Nx86_64aarch64)rgamd64armv7lZarm64)rDmachinelowerr-) architectureZarch_maprrrrIs rI)! __future__rr+r4rQrMrDr%typingrrpathlibrargparserrJ_errorsrr Z_modelsr r rrrrrZKEYGEN_ACCOUNTr/r7r$rPrIrrrrs2       ?