o d۷i` @sddlZddlZddlZddlZddlZddlZddlmZddlm Z m Z ddl m Z ddl mZmZddlZddlZddlZddlmZddlmZmZmZmZmZddlmZdd lmZdd lm Z dd l!m"Z"dd l#m$Z$m%Z%m&Z&m'Z'm(Z(dd l)m*Z*m+Z+m,Z,ddl-m.Z.ddl/m0Z0m1Z1e2e3Z4de5dBde5dBfddZ6ee5dBee6fZ7de5ddfddZ8de de5fddZ9de de:fddZ;de de de fddZe=fd!d"Z?Gd#d$d$eZ@d%e5d&e5dBde@fd'd(ZAd)e5d*e5d&e5dBde@fd+d,ZBGd-d.d.eZCd%e5deCfd/d0ZDGd1d2d2eZEde5d3e ddfd4d5ZFd6e5de@dBfd7d8ZGd)e5de>e@fd9d:ZHgd;ZIgd<ZJd=ed>e de*fd?d@ZKd=edAe%d%e5dBeCddf dCdDZLd=ed%e5dBeCddfdEdFZMGdGdHdHeZNdIeNd=eddfdJdKZOd=eddfdLdMZP  N d[dee dBejQdOdPfdQee:ej dRdSdPfdTee5dBej dUdVdWdXfdefdYdZZRdS)\N)cycle)Path PurePosixPath)dedent) AnnotatedAny)Client)AfterValidator BaseModelEmailStr TypeAdapterValidationError)Text) RichToolkit)Option)login)SUCCESSFUL_STATUSES APIClientDeploymentStatusStreamLogErrorTooManyRetriesError) AppConfigget_app_configwrite_app_config)Identity)get_rich_toolkithandle_http_errorsvreturncCsx|durdS|}|sdS|drtdt|}|r#tdd|jvr,td|}td|s:td|S)N~zcannot start with '~'z%must be a relative path, not absolutez..z!cannot contain '..' path segmentsz[A-Za-z0-9._/ -]+zGcontains invalid characters (allowed: letters, numbers, space, / . _ -)) strip startswith ValueErrorr is_absolutepartsas_posixre fullmatch)rpath normalizedr*W/home/ubuntu/vllm_env/lib/python3.10/site-packages/fastapi_cloud_cli/commands/deploy.pyvalidate_app_directory$s$   r, deployment_idc Cstd|z*t}|d|d}|tdWdWdS1s)wYWdStyI}z td|WYd}~dSd}~ww)Nz$Cancelling upload for deployment: %s /deployments/z/upload-cancelledz2Upload cancellation notification sent successfullyz5Failed to notify server about upload cancellation: %s)loggerdebugrpostraise_for_status Exception)r-clientresponseer*r*r+_cancel_uploadEs  &r7r(cCs|jSN)namer(r*r*r+ _get_app_nameRsr;csNgd}tfdd|DrdSjdkrdSjdks#jdr%dSdS) N)z.venv __pycache__z .mypy_cachez .pytest_cachez.gitz .gitignore.fastapicloudignorec3s|]}|jvVqdSr8)r$).0partr:r*r+ bsz(_should_exclude_entry..Tz.pycz.envz.env.F)anysuffixr9r!)r(parts_to_excluder*r:r+_should_exclude_entryWs  rDtar_pathcCstd|tj|tdgdd}td|d}t|d(}|D]}|r(q!||}td||j ||d |d 7}q!Wdn1sIwYtd ||S) Nz&Starting archive creation for path: %sr=F)should_exclude_entryadditional_ignore_paths ignore_hiddenzArchive will be created at: %srzw:zstzAdding %s to archive)arcnamez*Archive created successfully with %s files) r/r0rignorewalkrDfastaropenis_dir relative_toappend)r(rEfiles file_counttarfilenamerIr*r*r+archivens*      rVc@s&eZdZUeed<eed<eed<dS)Teamidslugr9N__name__ __module__ __qualname__str__annotations__r*r*r*r+rWs  rWcCsRt}|d}||d}Wdn1swYdd|DS)Nz/teams/datacSg|]}t|qSr*)rWmodel_validater>teamr*r*r+ z_get_teams..rgetr2json)r4r5r`r*r*r+ _get_teamss  rjc@s*eZdZUeed<eed<edBed<dS) AppResponserXrYN directoryrZr*r*r*r+rks rkapp_idrlcCsXt}|jd|d|id}|t|WdS1s%wYdS)N/apps/rlri)rpatchr2rkrbri)rmrlr4r5r*r*r+ _update_apps $rqteam_idapp_namecCsVt}|jd|||dd}|t|WdS1s$wYdS)Nrn)r9rrrlro)rr1r2rkrbri)rrrsrlr4r5r*r*r+ _create_apps  $rtc@s>eZdZUeed<eed<eed<eed<eed<eed<dS)CreateDeploymentResponserXrmrYstatus dashboard_urlurlN)r[r\r]r^r_rr*r*r*r+rus  rucCsRt}|d|d}|t|WdS1s"wYdS)Nrnr.)rr1r2rurbri)rmr4r5r*r*r+_create_deployments  $ryc@s&eZdZUeed<eeefed<dS)RequestUploadResponserxfieldsN)r[r\r]r^r_dictr*r*r*r+rzs rz archive_pathc CsNtd|td||jt}tn}td|d|d}|t | }td|j tdt |d}|j|j |j d |id }Wdn1sYwY|td td |d|d }|tdWdn1swYWddSWddS1swYdS)Nz-Starting deployment upload for deployment: %sz Archive path: %s, size: %s byteszRequesting upload URL from APIr.z/uploadzReceived upload URL: %szStarting file upload to S3rbfile)r`rRz"File upload completed successfullyz%Notifying API that upload is completez/upload-completez%Upload notification sent successfully)r/r0statst_sizerrr1r2rzrbrirxrNr{) r-r}fastapi_clientr4r5 upload_data archive_fileupload_responsenotify_responser*r*r+_upload_deployments@       Prapp_slugcCslt&}|d|}|jdkr WddS||}Wdn1s,wYt|S)Nrni)rrh status_coder2rirkrb)rr4r5r`r*r*r+_get_apps   rcCsZt}|jdd|id}||d}Wdn1s!wYdd|DS)Nrnrr)paramsr`cSrar*)rkrbr>appr*r*r+re rfz_get_apps..rg)rrr4r5r`r*r*r+ _get_appss r) u+🚀 Preparing for liftoff! Almost there...uA👹 Sneaking past the dependency gremlins... Don't wake them up!u>🤏 Squishing code into a tiny digital sandwich. Nom nom nom.u;🐱 Removing cat videos from our servers to free up space.uM🐢 Uploading at blazing speeds of 1 byte per hour. Patience, young padawan.uN🔌 Connecting to server... Please stand by while we argue with the firewall.uQ💥 Oops! We've angered the Python God. Sacrificing a rubber duck to appease it.u3🧙 Sprinkling magic deployment dust. Abracadabra!uB👀 Hoping that @tiangolo doesn't find out about this deployment.uF🍪 Cookie monster detected on server. Deploying anti-cookie shields.)uS😅 Well, that's embarrassing. We're still waiting for the deployment to finish...u9🤔 Maybe we should have brought snacks for this wait...u🥱 Yawn... Still waiting...uK🤯 Time is relative... Especially when you're waiting for a deployment...toolkitpath_to_deployc Cs|jd|ddd||d!}t|dd t}Wdn1s)wYWdn1s8wY||jdd d d |Dd }||jd ddd}|d}|s|d#}t|dd t|j}Wdn1s~wYWdn1swY||s|dt d|jddd |Dd}|r|j n|j dt |d}||r|jnd} |j dd| pddttd} | r| nd} ||jdd d||d!|jd"|d#|d"|d$| pd%d"||jd&d'td(d)d*td+d,d*gd } || d,kr4|d-t d.|r| |jkr}|jd/d00}t|t|j| d1} |d2| pWd%d3Wdn 1sgwYWdn 1swwYn@|} n=|jd4d0.}t|t|j|| d1} Wdn 1swY|d5| j Wdn 1swYt| j|jd6}t|||S)7NzSetting up and deploying [blue]z[/blue]r(tagzFetching teams...z-Error fetching teams. Please try again later.)default_messagez&Select the team you want to deploy to:rdcSg|] }t|j|dqSr9value)rr9rcr*r*r+re2z"_configure_app..roptionsz Do you want to create a new app?rT)rdefaultzFetching apps...z,Error fetching apps. Please try again later.z=No apps found in this team. You can create a new app instead.rJz%Select the app you want to deploy to:cSrr)rrYrr*r*r+reQr)rzWhat's your app name?)titlerz>Path to the directory containing your app (e.g. src, backend):dirz:[italic]Leave empty if it's the current directory[/italic])rrr placeholder validatorzDeployment configuration:summaryz Team: [bold]z[/bold]zApp name: [bold]zDirectory: [bold].zDoes everything look right?confirmzYes, start the deployment!deployrzNo, let me start overcancelzDeployment cancelled.rzUpdating app directory...)r)rlzApp directory updated to ''zCreating app...z$App created successfully! App slug: )rmrr)print print_lineprogressrrjaskrrrXtyperExitrYinputr;rlr AppDirectoryr9rrqlogrtrr)rrrteamsrdcreate_new_app selected_appappsrsinitial_directorydirectory_inputrlchoicer app_configr*r*r+_configure_app"s                 rr4 deploymentc Cs|jddddk}z |||j}Wn$tttfy6d|jd<d|jd|jd |_YWddSw|t vrHd |j d|j d |_n#d |jd<d |_t |}| d |d|jd|jd tdWddS1svwYdS)NzVerifying deployment...Tu✅)r inline_logs done_emojiu⚠️rz@Could not confirm deployment status. Check the dashboard: [link=][/link]u3Ready the chicken! 🐔 Your app is ready at [link=u❌zDeployment failedu😔 Oh no! Deployment failed: z. Check out the logs at [link=rJ)rpoll_deployment_statusrX TimeoutErrorrrmetadatarwcurrent_messagerrxrto_human_readablerrr)rr4rmrr final_status human_statusr*r*r+_verify_deploymentsD    "rc Cstt}|jddd|d}t}t}t}|jt|dddd}d } zc| |j D]Z} t|}| j d krJ| t | j| j d krVd} d |_n7| j d krs| d| d|jd|jdtd|dkr{tt}t|dkrt||_t}q2Wn*tttfy} z|td| d|jd|jdtddd} ~ wwWdn1swY| r|t||||WddSWddS1swYdS)Nu+Checking the status of your deployment 👀cloudrgTu🚀)r lines_to_showrFmessagecompletezBuild complete!failedru>😔 Oh no! Something went wrong. Check out the logs at [link=rrrJz8 [error]Build log streaming failed: z\[/] Unable to stream build logs. Check the dashboard for status: [link=z[/link] )rWAITING_MESSAGESrrtime monotonicrrnextstream_build_logsrXtyperr from_ansirrstriprrwrrLONG_WAIT_MESSAGESrrr set_errorrr r) rrmrmessages time_elapsed started_atlast_message_changed_atr4rbuild_completerr6r*r*r+_wait_for_deployments|        01"rc@seZdZUeed<dZedBed<dZedBed<dZedBed<dZ edBed<dZ edBed<dZ edBed<dZ edBed <dS) SignupToWaitingListemailNr9 organizationrole team_sizelocationuse_case secret_code) r[r\r]r r_r9r^rrrrrrr*r*r*r+rs rresultc Cs|dB}t)}t||jd|d}|Wdn1s&wYWdn1s5wY|dWddS1sJwYdS)NzSending your request...z/users/waiting-listrou9Let's go! Thanks for your interest in FastAPI Cloud! 🚀)rrrr1 model_dumpr2r)rrrr4r5r*r*r+_send_waitlist_forms    "rcCsddlm}|jddd||jddttd}|td |i}|j d ddr||d |j d }|j d ddd|j dddd|j dddd|j dddd|j dddd|j dddd|j d d!d"d| }z td |i|}Wnt y|d#YdSw||j d$d%dr|t||tttj gd&tjtjd'd(WddS1swYdSdS))Nr)FormzkWe're currently in private beta. If you want to be notified when we launch, please fill out the form below.waitlistrzEnter your email:T)requiredrrz?Do you want to get access faster by giving us more information?z Waitlist form)styler9NamezJohn Doe)labelrr Organizationz Acme Inc.rdrWzTeam ArRole DeveloperrLocationz San Franciscorz%How do you plan to use FastAPI Cloud?zI'm building a web apprz Secret code123456z.[error]Invalid form data. Please try again.[/]zDo you agree to - Terms of Service: [link=https://fastapicloud.com/legal/terms]https://fastapicloud.com/legal/terms[/link] - Privacy Policy: [link=https://fastapicloud.com/legal/privacy-policy]https://fastapicloud.com/legal/privacy-policy[/link] terms)rNz-gu!raycast://confetti?emojis=🐔⚡F)stdoutstderrcheck)rich_toolkit.formrrrrr r rrbrr add_inputrunr r contextlibsuppressr3 subprocessDEVNULL)rrrrformr*r*r+_waitlist_form!s~   "rFz:A path to the folder containing the app you want to deploy)help skip_waitz --no-waitz"Skip waiting for deployment statusprovided_app_idz--app-idzApplication ID to deploy toFASTAPI_CLOUD_APP_ID)renvvarc Cs tdtd|||t}t}|shtd|jddd||jr8|r8|j dddn|j d dd||j d d t d d dt dddgd}|d kr_t n t |td|jddd||pxt}td|t|}|r|r|j|kr| d|d|jd||j dddtdd|r|}n|r|j}ntdt||d}||j}|r| d|dn| d||jd d!d"3} t| td#|t|} Wdn1swY| std$| d%Wdn 1swY| s5||s0|j d&ddtdt} td't| d(} t|pLt| |jd)d*d+S} t| >td,| jt| j} z| d-| j| d.t | j| | d/Wnt!yt"| jwWdn 1swYWdn 1swYWdn 1swY||std0t#|| j| d1ntd2| d3| j$d4| j$d5WddSWddS1swYdS)6uB Deploy a [bold]FastAPI[/bold] app to FastAPI Cloud. 🚀 zDeploy command startedz*Deploy path: %s, skip_wait: %s, app_id: %sz3User not logged in, prompting for login or waitlistzWelcome to FastAPI Cloud!FastAPIrz.Your session has expired. Please log in again.infoz4You need to be logged in to deploy to FastAPI Cloud.zWhat would you like to do?authzLogin to my existing accountrrzJoin the waiting listrrrJzStarting deploymentzDeploying from path: %sz[error]Error: Provided app ID (z#) does not match the local config (z).[/]zRun [bold]fastapi cloud unlink[/] to remove the local config, or remove --app-id / unset FASTAPI_CLOUD_APP_ID to use the configured app.tipNz(No app config found, configuring new app)rzDeploying to app [blue]z [/blue]...zDeploying app...zChecking app...T) transientzChecking app with ID: %szApp not found in APIz>App not found. Make sure you're logged in the correct account.zeIf you deleted this app, you can run [bold]fastapi cloud unlink[/] to unlink the local configuration.zCreating archive for deploymentz archive.tarzCreating deploymentu📦)rrzCreating deployment for app: %sz2Deployment created successfully! Deployment slug: zUploading deployment...z!Deployment uploaded successfully!z"Waiting for deployment to complete)rz%Skipping deployment wait as requestedz-Check the status of your deployment at [link=rr)%r/r0rr is_logged_in print_titlertoken is_expiredrrrrrrrrcwdrrmrrrrrtempfileTemporaryDirectoryrVrXryrrYrKeyboardInterruptr7rrw)r(rridentityrrrr target_app_idrrtemp_dirr}rr*r*r+rrs                             $r)NFN)Srloggingr&rrr itertoolsrpathlibrrtextwraprtypingrrrMrKrhttpxrpydanticr r r r r rich.textr rich_toolkitrrich_toolkit.menur fastapi_cloud_cli.commands.loginrfastapi_cloud_cli.utils.apirrrrrfastapi_cloud_cli.utils.appsrrrfastapi_cloud_cli.utils.authrfastapi_cloud_cli.utils.clirr getLoggerr[r/r^r,rr7r;boolrDrVrWlistrjrkrqrtruryrzrrrrrrrrrrrArgumentrr*r*r*r+s              )  z $ H  W