o d۷i@sddlZddlmZddlmZmZddlZddlmZddl m Z ddl m Z ddl mZddlmZmZdd lmZeeZGd d d eZGd d d eZdedefddZdededefddZ d)dededededdf ddZeZe d*deedBej!ddfdefddZ"e ej!ddddfdedBdeedBej!ddfdefd d!Z#e ej!dd"dej!dd#dddfdedBdedBdeedBej!ddfd$eeej$d%d&dfdef d'd(Z%dS)+N)Path) AnnotatedAny) BaseModel) APIClient)get_app_config)Identity)get_rich_toolkithandle_http_errors)"validate_environment_variable_namec@s&eZdZUeed<dZedBed<dS)EnvironmentVariablenameNvalue)__name__ __module__ __qualname__str__annotations__rrrT/home/ubuntu/vllm_env/lib/python3.10/site-packages/fastapi_cloud_cli/commands/env.pyr s r c@seZdZUeeed<dS)EnvironmentVariableResponsedataN)rrrlistr rrrrrrs rapp_idreturncCsRt}|d|d}|t|WdS1s"wYdS)N/apps//environment-variables/)rgetraise_for_statusrmodel_validatejson)rclientresponserrr_get_environment_variabless  $r#r cCsVt}|d|d|}Wdn1swY|jdkr%dS|dS)NrriFT)rdelete status_coder)rr r!r"rrr_delete_environment_variable"s r&Fr is_secretcCsRt}|jd|d|||dd}|WddS1s"wYdS)Nrr)r rr')r )rpostr)rr rr'r!r"rrr_set_environment_variable.s   "r)pathz:A path to the folder containing the app you want to deploy)helpc Cs>t}tdd}|s|jdddtd|pt}t|}|s3|d|dtd|j d dd !}t | t |j }Wd n1sOwYWd n1s^wY|j st|d  Wd d S|d ||j D] }|d|jdqWd d S1swYd S)z5 List the environment variables for the app. Tminimalenvironment_variablesenv_varrrrr<sD          "rz.The name of the environment variable to deletec Cst}tdd}|s|jdddtd|pt}t|}|s3|d|dtd|s|j d dd !}t | t |j }Wd n1sQwYWd n1s`wY|j sv|d  Wd d S|jd dd|j Dd}|sJnt|s|d|dtd||j ddd "}t |t|j |}Wd n1swYWd n1swY|s|dtd|d|dWd d S1swYd S)z6 Delete an environment variable from the app. Tr,r.r/r0r2r3r4r5r6Nr8z*Select the environment variable to delete:cSsg|] }|j|jdqS))r r)r ).0rErrr s zdelete..)optionsz$The environment variable name [bold]z[/] is invalid.zDeleting environment variablezEnvironment variable not found.Environment variable [bold]z [/] deleted.)rr r9r:r;r<rr=rr>r r#rraskr r?r&) r r*r@rApath_to_deployrCr>rDdeletedrrrr$ost              "r$z+The name of the environment variable to setz,The value of the environment variable to setsecretz--secretz'Mark the environment variable as secretc Cst}tdd}|s|jdddtd|pt}t|}|s3|d|dtd|sB|r=| d }n| d }|sS|rN|j d dd }n| d }|j ddd0}|dusaJ|dusgJt |t |j |||Wdn1s~wYWdn1swY|r|d|dn|d|dWddSWddS1swYdS)z2 Set an environment variable for the app. Tr,r.r/r0r2r3r4z$Enter the name of the secret to set:z2Enter the name of the environment variable to set:zEnter the secret value:)passwordz,Enter the value of the environment variable:zSetting environment variabler6Nz"Secret environment variable [bold]z[/] set.rI)rr r9r:r;r<rr=rinputr>r r)r) r rr*rMr@rArKrCr>rrrsetsT           +"rP)F)N)&loggingpathlibrtypingrrr;pydanticrfastapi_cloud_cli.utils.apirfastapi_cloud_cli.utils.appsrfastapi_cloud_cli.utils.authrfastapi_cloud_cli.utils.clir r fastapi_cloud_cli.utils.envr getLoggerrloggerr rrr#boolr&r)Typerenv_appcommandArgumentrr$OptionrPrrrrs           2  Q