6dZddlmZddlmZddlmZddlZddlZddlZddlZddl m Z ddl m Z ddl m Z dd l m Z dd l mZdd lmZddlm Zdd lmZdd lmZddlmZddlZddlZdZdZdZdZGdde j<ZGddeZGddeZ GddeZ!GddeZ"dZ#dZ$dZ% d$d Z&d%d!Z'Gd"d#e(Z)y)&z.RetryIfs4 ++55 5 2    * *c 13rc.tjdy)NzRetrying request...)rdebug) unused_resultrZs r StatusUpdatez!_RawRequest..StatusUpdatesII#$rrd) max_retrialsexponential_sleep_multiplier jitter_msstatus_update_funci)should_retry_ifsleep_msr6)tbN)rRetryerRetryOnException!_ExecuteRequestAndRaiseExceptionsRetryException last_resultrreraise)argskwargsr[r_retryerr's rrFrFs3% MM#$%  ''   # #)  $     }}Qq)!,q1A!1DE  s;BABBcddlm}|j}|jdr$|j dt j |j|||d}|j|S)aPExecutes an HTTP request using requests. Args: url: str, the url to download. headers: obj, the headers to include in the request. timeout: int, the timeout length for the request. Returns: A response object from the request. Raises: requests.exceptions.HTTPError in the case of a client or server error. r)rGzfile://T)r7r8stream) googlecloudsdk.corerG GetSessionr<mountrLocalFileAdaptergetraise_for_status)rKr7r8 core_requestsrequests_sessionrIs rrkrksk<"--/^^I9&8&I&I&KL  ! ! 7GD":(  /rc|xstj}tjj |st j |tjj|tjj|}tjj |rtj| t||}t j|5}d}t|j}|jtD],} |j!| |t| z }|||z . ddd|d|S#1swYxYw#t"j$j&t(f$r} t+| d} ~ wwxYw)aDownload the given tar file. Args: url: str, The URL to download. download_dir: str, The path to put the temporary download file into. progress_callback: f(float), A function to call with the fraction of completeness. command_path: the command path to include in the User-Agent header if the URL is HTTP Returns: str, The path of the downloaded tar file. Raises: URLFetchError: If there is a problem fetching the given URL. r) chunk_sizeNr6)r DefaultProgressBarCallbackospathexists file_utilsMakeDirjoinbasenameremoverTBinaryFileWriterlencontent iter_contentWRITE_BUFFER_SIZEwriterGrrHOSErrorr-) rK download_dirprogress_callbackrRdownload_file_pathrIfp total_written total_sizechunkr's r DownloadTarrs9$(P:+P+P  %|$ww||L"''2B2B32GHWW^^&'II ! 3 -H  $ $%7 8Bmx''(j((4E(F% U# -*45G 9a  9 8    ' ' 1  s1;!EAE :E EE#F ; FF c~|xstj}tjj |st j |tj|5}|j}t|}g}t|dD]\}}|j|jr|jdzn |j|j||tjj!||j} tjj#| rYtj$| tj&s5tj(| t*j,t*j.z|||z |ddddtj0|S#1swY xYw)aPExtracts the given archive. Args: downloaded_archive: str, The path to the archive downloaded previously. extract_dir: str, The path to extract the tar into. progress_callback: f(float), A function to call with the fraction of completeness. Returns: [str], The files that were extracted from the tar file. )namer6)start/N)r r~rrrrrtarfileopen getmembersr enumerateappendisdirrextractrisfileaccessW_OKchmodstatS_IWUSRS_IREADr) downloaded_archive extract_dirrtarmembers total_filesr nummember full_paths r ExtractTarrs/(P:+P+P  ${#||+,nnGg,K E 2 V ll 6;;$FKKH kk&+&'',,{FKK8i  "299Y+H DLL56k)*3a- ))  ,#-,s D5F33F<c>eZdZdZdZdZdZdZd dZd dZ d d Z y) r=zBA class to install Cloud SDK components of different source types.z .downloadz!https://storage.cloud.google.com/zhttps://storage.googleapis.com/c||_||_tjj |jt j |_y)zInitializes an installer for components of different source types. Args: sdk_root: str, The path to the root directory of all Cloud SDK files. state_directory: str, The path to the directory where the local state is stored. N)_ComponentInstaller__sdk_root$_ComponentInstaller__state_directoryrrrr=DOWNLOAD_DIR_NAME'_ComponentInstaller__download_directory)r&sdk_rootstate_directorys rr"zComponentInstaller.__init__%s;DO,D "   2 D D!FDrNc|j}|sy|jdk(r|j|||Stdj |j)aDownloads the given component for whatever source type it has. Args: component: schemas.Component, The component from the snapshot to install. progress_callback: f(float), A function to call with the fraction of completeness. command_path: the command path to include in the User-Agent header if the URL is HTTP Returns: Optional[str], The path of the downloaded archive, or None if the component has no actual sources. Raises: UnsupportedSourceError: If the component data source is of an unknown type. URLFetchError: If the URL associated with the component data source cannot be fetched. NrrrRz4tar is the only supported source format [{datatype}])datatype)datatype _DownloadTarr3r#)r& componentrrRrs rDownloadzComponentInstaller.Download2sj( >>D   yyE    '8#%% !>EEYY F  !!rc:|gSt||j|S)apExtracts the archive previously downloaded from self.Download(). Args: downloaded_archive: Optional[str], The path to the archive downloaded previously. progress_callback: f(float), A function to call with the fraction of completeness. Returns: list of str, The files that were installed or [] if nothing was installed. )r)rr)r&rrs rExtractzComponentInstaller.ExtractUs(! i DOO+ --rc6|jj}|sytjd|s$t dj |j  t||j||S#ttf$r}t|j |d}~wwxYw)aDownload implementation for a component with source in a .tar.gz. Downloads the .tar for the component and returns its path. Args: component: schemas.Component, The component to install. progress_callback: f(float), A function to call with the fraction of completeness. command_path: the command path to include in the User-Agent header if the URL is HTTP Returns: Optional[str], The path of the downloaded archive, or None if the component has no actual sources. Raises: ValueError: If the source URL for the tar file is relative, but there is no location information associated with the snapshot we are installing from. URLFetchError: If there is a problem fetching the component's URL. Nz^\w+://zhCannot install component [{0}] from a relative path because the base URL of the snapshot is not defined.r) rsourceresearch ValueErrorr#idrrr-r/r)r&rrrRrKr's rrzComponentInstaller._DownloadTaris. ..  C   99Z % Ny||, ..:  t((r@r"rrrrrrr=r=s8J! ;5. F!!F-(7; )&:rr=rr)*r __future__rrrrrrrrtrrrr r googlecloudsdk.core.consoler r;rNgooglecloudsdk.core.utilr rr rrGr$rDrCrErrrr-r/r3rTrFrkrrobjectr=rrrrs C&' *2#*)2J82* / !J   ;5;E O%OU @6F :46:&&R!Hx:x:r