o -wi+!@sdZddlZddlZddlmZddlZddlmZddlm Z ddl Z ddl m Z ddl m Z ddlmZdd lmZdd lmZdd lmZdd lmZdd lmZdZGdddedZGddde ZdS)zPublic API: files.N)Optional)gql) RetryError)util)Attrs)normalize_exceptions)SizedPaginator)utils)Api)RETRY_TIMEDELTA)retryafragment RunFilesFragment on Run { files(names: $fileNames, after: $fileCursor, first: $fileLimit) { edges { node { id name url(upload: $upload) directUrl sizeBytes mimetype updatedAt md5 } cursor } pageInfo { endCursor hasNextPage } } }csleZdZdZedeZdfdd Ze dd Z e d d Z e d d Z ddZ ddZddZZS)Filesz)An iterable collection of `File` objects.a query RunFiles($project: String!, $entity: String!, $name: String!, $fileCursor: String, $fileLimit: Int = 50, $fileNames: [String] = [], $upload: Boolean = false) {{ project(name: $project, entityName: $entity) {{ internalId run(name: $name) {{ fileCount ...RunFilesFragment }} }} }} {} N2Fcs4||_|j|j|j|p g|d}t|||dS)N)projectentityname fileNamesupload)runrridsuper__init__)selfclientrnamesper_pager variables __class__T/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/wandb/apis/public/files.pyr?szFiles.__init__cCs |js||jdddS)Nrr fileCount) last_response _load_pagerrrr _lengthJsz Files._lengthcCs$|jr|jdddddSdS)NrrfilespageInfo hasNextPageTr"r$rrr moreQs z Files.morecCs(|jr|jddddddSdS)Nrrr&edgescursorr)r$rrr r-Zsz Files.cursorcCs|j|j|jddS)N) fileLimit fileCursor)rupdaterr-r$rrr update_variablesazFiles.update_variablescs$fddjddddDS)Ncs g|] }tj|djqS)node)Filerr).0rr$rr esz)Files.convert_objects..rrr&r+r)r$rr$r convert_objectsds zFiles.convert_objectscCsdd|jjt|S)Nz/)formatjoinrpathlenr$rrr __repr__jr2zFiles.__repr__)NrF)__name__ __module__ __qualname____doc__rr: FILE_FRAGMENTQUERYrpropertyr%r*r-r1r8r> __classcell__rrrr r ,s     r r4c seZdZdZdfdd ZeddZedefdd Ze e j e e jeejfd  dd edededeedejf ddZe ddZddZe defddZZS)r4aFile is a class associated with a file saved by wandb. Attributes: name (string): filename url (string): path to file direct_url (string): path to file in the bucket md5 (string): md5 of file mimetype (string): mimetype of file updated_at (string): timestamp of last update size (int): size of file in bytes path_uri (str): path to file in the bucket, currently only available for files stored in S3 Ncs,||_||_||_d|_tt|dSN)r_attrsr+server_supports_delete_file_with_project_idrrdict)rrattrsrrrr r|s z File.__init__cCs|jd}|dur t|SdS)N sizeBytesr)rHint)r size_bytesrrr sizes z File.sizereturncCsVd}z t|jd}W|StytdY|Sty*tdY|Sw)zI Returns the uri path to the file in the storage bucket. directUrlz1path_uri is only available for files stored in S3z!Unable to find direct_url of file)r parse_s3_url_to_s3_urirH ValueErrorwandbtermwarn LookupError)rpath_urirrr rXs  z File.path_uri)retry_timedeltacheck_retry_fnretryable_exceptions.Frootreplaceexist_okapicCs^|durt}tj||j}tj|r"|s"|rt|Stdt ||j |j t|S)aDownloads a file previously saved by a run from the wandb server. Args: replace (boolean): If `True`, download will overwrite a local file if it exists. Defaults to `False`. root (str): Local directory to save the file. Defaults to ".". exist_ok (boolean): If `True`, will not raise ValueError if file already exists and will not re-download unless replace=True. Defaults to `False`. api (Api, optional): If given, the `Api` instance used to download the file. Raises: `ValueError` if file already exists, replace=False and exist_ok=False. NzgFile already exists, pass replace=True to overwrite or exist_ok=True to leave it as is and don't error.) rUr osr<r;rexistsopenrTrdownload_file_from_urlurlapi_key)rr]r^r_r`r<rrr downloadsz File.downloadcCsXd}d}d|jgi}|r|jj|d<d}d}d||}t|}|jj||ddS)NrQr& projectIdz, $projectId: IntzprojectId: $projectIdz mutation deleteFiles($files: [ID!]!{}) {{ deleteFiles(input: {{ files: $files {} }}) {{ success }} }} )variable_values)r*_server_accepts_project_id_for_delete_filer_project_internal_idr:rrexecute)rproject_id_mutation_fragmentproject_id_variable_fragmentrimutation_stringmutationrrr deletes     z File.deletecCs d|j|jtj|jtjdS)Nz)units)r:rmimetyper to_human_sizerO POW_2_BYTESr$rrr r>s z File.__repr__cCsNd}|jdur$t|}|j|}ddd|didigDv|_|jS)zReturns True if the server supports deleting files with a projectId. This check is done by utilizing GraphQL introspection in the available fields on the DeleteFiles API. a query ProbeDeleteFilesProjectIdInput { DeleteFilesProjectIdInputType: __type(name:"DeleteFilesInput") { inputFields{ name } } } NrhcSsg|]}|dqS)rr)r5xrrr r7szCFile._server_accepts_project_id_for_delete_file..DeleteFilesProjectIdInputType inputFields)rIrrrlget)r query_stringqueryresrrr rjs     z/File._server_accepts_project_id_for_delete_filerG)r\FFN)r?r@rArBrrErOstrrXrr retriabler r no_retry_authrrequestsRequestExceptionboolrr io TextIOWrapperrgrqr>rjrFrrrr r4nsB   # )rBrratypingrr wandb_gqlrwandb_gql.clientrrUrwandb.apis.attrsrwandb.apis.normalizerwandb.apis.paginatorrwandb.apis.publicr wandb.apis.public.apir wandb.apis.public.constr wandb.sdk.libr rCr r4rrrr s&           B