&SrSSKJr SSKJr SSKJr SSKrSSKJr SSKJ r SSK J r SS K J r SSKJr SS KJr SS KJr S rS rSr"SS\5rSrg)z@Utility methods to aid in interacting with a GCS results bucket.)absolute_import)division)unicode_literalsN) exceptions)transfer)util)apis)log) resourceszgs://iic:\rSrSrSrSrSrSrS SjrSr S r g) ResultsBucketOps'z@A utility class to encapsulate operations on the results bucket.cXlX`l[R"SS5UlX0lU(aUR U5 OURXE5n[RRUSS9nURUl SRX R S9UlSRX R S9Ul[ R""SRUR55 g ) aConstruct a ResultsBucketOps object to be used with a single matrix run. Args: project: string containing the Google Cloud Platform project id. bucket_name: string with the user-supplied name of a GCS bucket, or None. unique_obj_name: a unique-per-matrix GCS object name which will prefix all raw test result files within the supplied bucket_name. tr_client: ToolResults API client library generated by Apitools. tr_messages: ToolResults API messages library generated by Apitools. storage_client: Cloud Storage API client library generated by Apitools. Attributes: gcs_results_root: string containing the root path for the test results in 'gs://{bucket}/{timestamp-suffix}' format. storagev1zstorage.buckets) collectionz>https://console.developers.google.com/storage/browser/{b}/{t}/)btz gs://{b}/{t}/zRaw results root path is: [{0}]N)_project_storage_client core_apisGetMessagesModule_storage_messages_gcs_object_nameEnsureBucketExists_GetDefaultBucketr REGISTRYParsebucket_results_bucketformat_gcs_results_urlgcs_results_rootr info)selfproject bucket_nameunique_obj_name tr_client tr_messagesstorage_client bucket_refs :lib/googlecloudsdk/api_lib/firebase/test/results_bucket.py__init__ResultsBucketOps.__init__*s"M)&88DID+ k***9Bk##)) 1*3J%,,D I +!6!6 7 -$f{6K6KfL HH . 5 5d6K6K LMcURURS9nURRU5nUR$![ R akn[R"U5upgU[:XaSRXgUR5nOSRXg5n[R"U5eSnAff=f)zEFetch the project's default GCS bucket name for storing tool results.) projectIdztPermission denied while fetching the default results bucket (Error {0}: {1}). Is billing enabled for project: [{2}]?zSHttp error while trying to fetch the default results bucket: ResponseError {0}: {1}N) ,ToolresultsProjectsInitializeSettingsRequestrprojectsInitializeSettings defaultBucketapitools_exceptions HttpErrorrGetErrorCodeAndMessageHTTP_FORBIDDENr!r HttpException) r%r)r*requestresponseerrorcodeerr_msgmsgs r-r"ResultsBucketOps._GetDefaultBucketRsFF--G!G *##66w?h  # ##  ( ( *11%8md  Jtdmm4 (t%   $ $S )) *s&ACA&B<<Cc 6URRUS9nURRR U5 g![ R aPn[R"U5upEU[:wa$[R"SRXUS95eSnAOSnAff=f[RRSR[ XR"S95 URR$nUR&R(nUR*R(nURR%URR-US9UUUR"S9n URRR/U 5 g![ R a{n[R"U5upEU[0:XaS RXR"S 9n O$S RU[R2"U5S 9n [R"U 5eSnAff=f) aCreate a GCS bucket if it doesn't already exist. Args: bucket_name: the name of the GCS bucket to create if it doesn't exist. Raises: BadFileException if the bucket name is malformed, the user does not have access rights to the bucket, or the bucket can't be created. )rNzCould not access bucket [{b}]. Response error {c}: {e}. Please supply a valid bucket name or use the default bucket provided by Firebase Test Lab.)rcez2Creating results bucket [{g}{b}] in project [{p}].)grp)name)r predefinedAclpredefinedDefaultObjectAclr&zUPermission denied while creating bucket [{b}]. Is billing enabled for project: [{p}]?)rrGz!Failed to create bucket [{b}] {e})rrE)rStorageBucketsGetRequestrbucketsGetr7r8rr9HTTP_NOT_FOUNDrBadFileExceptionr!r statusPrint GCS_PREFIXrStorageBucketsInsertRequestPredefinedAclValueValuesEnumprojectPrivate)PredefinedDefaultObjectAclValueValuesEnumBucketInsertr:GetError) r%r'get_reqerrr?r@ bucket_reqaclobjacl insert_reqrAs r-r#ResultsBucketOps.EnsureBucketExistses$$==>G 7 ""&&w/  ( (711#6md  )) -VkWV 5 7 7 7JJIfz[MMfJL''CCJ  1 1 @ @C  A A P PF''CC%%,,+,>#) DJ - ""))*5  ( ( -11#6md  8 6 3 c(:;  ' ' ,, -s1%AB%AB  B%#%F HA6HHNc [RRSRUS95 UR [ 5(a[ U5upEURRUUURSRURU=(d [RRU5S9S9nURRR!U5 g[RR#U5nURR+US9n[,R.R1XS9n URR7URSRURU=(d [RRU5S9US 9n URRR9U U S 9n U R:U:wa:[&R("S RU R:X[=U 555eg![R$a& [&R("SRU55ef=f![2R4a" [,R.R1US S9n GNDf=f![2R>aGn [&R("S RXR@[BRD"U 5S95eSn A ff=f)aLUpload a file to the GCS results bucket using the storage API. Args: path: str, the absolute or relative path of the file to upload. File may be in located in GCS or the local filesystem. mimetype: str, the MIME type (aka Content-Type) that should be applied to files being copied from a non-GCS source to GCS. MIME types for GCS->GCS file uploads are not modified. destination_object: str, the destination object path in GCS to upload to, if it's different than the base name of the path argument. Raises: BadFileException if the file upload is not successful. z'Uploading [{f}] to Firebase Test Lab...)fz {obj}/{name})objrH) sourceBucket sourceObjectdestinationBucketdestinationObjectz![{0}] not found or not accessible)size) mime_typezapplication/octet-stream)rrHobject)uploadzoCloud storage upload failure: Insert response.size={0} bytes but [{1}] contains {2} bytes. Insert response: {3}z$Could not copy [{f}] to [{gcs}] {e}.)rbgcsrEN)#r rPrQr! startswithrR_SplitBucketAndObjectrStorageObjectsCopyRequestr rospathbasenamerobjectsCopygetsizer>rrOObjectrUploadFromFiler7InvalidUserInputErrorStorageObjectsInsertRequestrXrhreprr8r#rrY) r%rqmimetypedestination_object file_bucketfile_objcopy_req file_sizesrc_objrkr_r=r[s r-UploadFileToGcs ResultsBucketOps.UploadFileToGcssJJ>EEEMN*L  $ $ 5d ; ))CC$!"22,33))'E277+;+;H+E4G DH $$))(3 ;ggood+)((//Y/? :??++D+E& ++GG''&&))'A277+;+;D+A'C H ''//66z>D7F ==I %++DvhmmTd8nEG G &%xx ;++,O-3VD\; ; ; #88 :??++8,:& :"  ( (L  ' ' 0 6D33t}}S7I6 J LLLsVB0I> H?I>I6CI>:II>2I;7I>:I;;I>>KAKKct[RRSRUR55 g)Nz7Raw results will be stored in your GCS bucket at [{0}] )r rPrQr!r")r%s r-LogGcsResultsUrl!ResultsBucketOps.LogGcsResultsUrls'JJOfT2235r0)rr"rr rrr#)NN) __name__ __module__ __qualname____firstlineno____doc__r.rrrr__static_attributes__r0r-r r 's#H&NP*&2-h:Lx5r0r cU[[5SRS5RSS5n[U5S:wa%[R "SR U55eU$)zHSplit a GCS path into bucket & object tokens, or raise BadFileException.N/z.[{0}] is not a valid Google Cloud Storage path)lenrRstripsplitrrOr!)gcs_pathtokenss r-rnrns_ C O$ % + +C 0 6 6sA >&[A  % %8??I KK -r0)r __future__rrrrpapitools.base.pyrr7r$googlecloudsdk.api_lib.firebase.testrgooglecloudsdk.api_lib.utilr rgooglecloudsdk.calliopegooglecloudsdk.corer r rRr:rNrjr rnrr0r-rsQ G&' >%59.#) p5vp5fr0