#dZddlmZddlmZddlmZddlmZddlZddlZddlm Z ddl m Z dd l m Z dd l mZdd lmZdd lmZdd lmZddlmZGddeZdZdZGddZy)z=Helper functions for working with Apigee archive deployments.)absolute_import)division)print_function)unicode_literalsN)apigee)errors)log)requests)resource_projector)archive)files)urllibceZdZdZdZgdZejjdddZ dZ dZ d Z d Z d Zd Zd ZdZy)LocalDirectoryArchivezManages a local zip archive.zapigee_archive_deployment.zip)z.graphqlz.jarz.javaz.jsz.jscz.jsonz.oasz .propertiesz.pyz.securityPolicyz.wsdlz.xmlz.xsdz.xslz.yamlz.ymlsrcmainrc|j||r4tjj|st j |||nt j|_t j|_ yN) _CheckIfPathExistsospathisdirrSourcePathIsNotDirectoryErrorr GetCWD_src_dirTemporaryDirectory_tmp_dir)selfsrc_dirs 1lib/googlecloudsdk/command_lib/apigee/archives.py__init__zLocalDirectoryArchive.__init__<sWG$rww}}W-  0 0 99&2G DM,,.DMc|rDtjj|s$tjdj |yy)z'Checks that the given file path exists.z-Path to archive deployment does not exist: {}N)rrexistsr MissingFileErrorformat)rrs r rz(LocalDirectoryArchive._CheckIfPathExistsDs> BGGNN4(  " " 9 @ @ F HH)tr"c|j|jsytjj |\}}tjj |j |}tjj|jdrytjj|rytjj|r|j|jvryy)zEFilter all files in the archive directory to only allow Apigee files.F.T) startswith _ARCHIVE_ROOTrrsplitextjoinrbasenamerisfilelower_APIGEE_ARCHIVE_FILE_EXTENSIONS)r file_name_ext full_paths r _ZipFileFilterz$LocalDirectoryArchive._ZipFileFilterJs    2 2 3  WW  i (FAs T]]I6I  ww "--c2  ww}}Y  y! t;;;  r"ctjj|jj|j}t j ||j|j|S)z1Creates a zip archive of the specified directory.) rrr,r_ARCHIVE_FILE_NAMEr MakeZipFromDirrr5)rdst_files r ZipzLocalDirectoryArchive.ZipasGww||DMM..0G0GHH 8T]]D4G4GH Or"cz|j|tj|stj|y)zCChecks that the zip file path exists and the file is a zip archvie.N)rzipfile is_zipfilerBundleFileNotValidError)rzip_paths r ValidateZipFilePathz)LocalDirectoryArchive.ValidateZipFilePathgs4H%   h '  * *8 44 (r"c6|jjS)z&Deletes the local temporary directory.)rClosers r rBzLocalDirectoryArchive.Closems ==    r"c|SrrCs r __enter__zLocalDirectoryArchive.__enter__qs Kr"c^ |jy#tjdYyxYw)Nz1Temporary directory was not successfully deleted.T)rBr warning)rexc_typevaltbs r __exit__zLocalDirectoryArchive.__exit__ts' jjl kkEF s,N)__name__ __module__ __qualname____doc__r7r0rrr,r*r!rr5r:r@rBrFrLrEr"r rr#sV$7%!$'',,ufh7-/H . 5 !r"rctjj|}|jj d}|dS)aHelper function to extract the upload file id from the signed URL. Archive deployments must be uploaded to a provided signed URL in the form of: https://storage.googleapis.com//.zip? This function extracts the file id from the URL (e.g., .zip). Args: upload_url: A string of the signed URL. Returns: A string of the file id. /)rparseurlparsersplit) upload_urlurl split_paths r GetUploadFileIdrZ|s4  j)#xx~~c"* Br"ctj}ddd}tj|5}|j |||}ddd|S#1swYSxYw)a-Uploads the specified zip file with a PUT request to the provided URL. Args: upload_url: A string of the URL to send the PUT request to. Required to be a signed URL from GCS. zip_file: A string of the file path to the zip file to upload. Returns: A requests.Response object. zapplication/zipz 0,1073741824)z content-typezx-goog-content-length-range)dataheadersN)r GetSessionr BinaryFileReaderput)rWzip_filesessr]r\responses r UploadArchiverds\    $(%3 ' h'4xx wx?H( /( /s AAc"eZdZdZdZdZdZy) ListArchivesz5Adds additional helpful fields to a list of archives.c||_tj||_d|_d|_d|_d|_d|_d|_ y)NDeployedz In ProgressFailedz Not FoundUnknownMissing) _orgr LROPoller _lro_helper_deployed_status_inprogress_status_failed_status_not_found_status_unknown_status_missing_status)rorgs r r!zListArchives.__init__sLDI'',D&D+D"D(D$D$Dr"clt|dd}d}t|D]\}}tj|}|r|j|d<nd|vrt j jd|did} t j j|j|d }|j|} | d |d<| d |jk(r |||<|cSd | vr| d |d <n|j|d<|||<|S#tj$r|j|d<Y-d}|j|d<YAxYw) zGiven a list of archives, extends them with a status and error field where needed. Args: archives: A list of archives to extend with a status and potential error. Returns: A list of archives with their associated status. c |dS)N createdAtrE)ks r z/ListArchives.ExtendedArchives..s+r"T)keyreverseFoperationStatus operationnameuuid)organizationsId operationsIdstatuserror)sorted enumerater MakeSerializablersrOperationsClient SplitNameDescriberl_StatusFromOperationrorEntityNotFoundErrorrrrt) rarchivesextended_archivescascade_unknownidxaserilized_archiveroprs r ExtendedArchiveszListArchives.ExtendedArchivess~.>O-.Q,==a@ /3/C/C+, ! &&00&!K.2344:< F&&//!%"1  ",,R0&171A - . H !6!6 6%6 c "$ $& )/ g &04/C/C+,07/8 )) H151G1G - . F /151E1E - .s6AC; C;;"D3D3ci} |jj|}|r|j|d<|S|j|d< |S#tj $r|j |d<|dd|d<Y|SwxYw)zGathers given an LRO, determines the associated archive status. Args: op: An Apigee LRO Returns: A dict in the format of {"status": "{status}", "error": "{error if present on LRO}"}. rrmessage)rnIsDonerorpr RequestErrorrq)rrris_dones r rz!ListArchives._StatusFromOperationsF/  ''+g  00x M 22x M   /,,fX7 I.fWo M /s,AA-A54A5N)rMrNrOrPr!rrrEr"r rfrfs=%0dr"rf)rP __future__rrrrrr<googlecloudsdk.api_libr!googlecloudsdk.command_lib.apigeergooglecloudsdk.corer r googlecloudsdk.core.resourcer googlecloudsdk.core.utilr r six.movesrobjectrrZrdrfrEr"r rsZD&%' )4#(;,*VFVr$,YYr"