o xi/@sUdZddlmZddlZddlZddlmZmZmZddl m Z m Z ddl m Z mZmZmZmZmZddlmZddlmZmZmZdd lmZmZdd lmZdd lmZdd l m!Z!d dl"m#Z#m$Z$e ryddl%m&Z&ddl m'Z'ddl(m)Z)edddZ*edZ+edZ,edZ-dZ.de/d<dZ0de/d<dZ1de/d<e2d Z3d!e/d"<e2d#Z4d!e/d$<e2d%Z5d!e/d&<eGd'd(d(Z6dgd,d-Z7dgd.d/Z8dhd3d4Z9did6d7Z:e;d8ZZ>de/d?< d@dAdBdCZ?dDe/dE< dkdGdHZ@e dldLdMZAeABeCdeABeDdmdOdPZEeABeFdndQdPZEdodUdVZGeee+e-fe,fZH dpdYdZZIdpd[d\ZJdqd_d`ZKdrdadbZLeGdcddddZMeGdedfdfeMZNdS)sz=Internal validation utilities that are specific to artifacts.) annotationsN) dataclassfieldreplace)singledispatchwraps) TYPE_CHECKINGAnyCallableLiteralOptionalTypeVar)r) Concatenate ParamSpecSelf) always_list unique_list) from_json)nameof)json_friendly_val)ArtifactFinalizedErrorArtifactNotLoggedError)Iterable)Final)Artifact ArtifactTr)boundSelfTRPzwandb-registry-z Final[str]REGISTRY_PREFIXdz Final[int]MAX_ARTIFACT_METADATA_KEYS NAME_MAXLEN/zFinal[frozenset[str]]INVALID_ARTIFACT_NAME_CHARSz/\#?%: INVALID_URL_CHARSz/:ARTIFACT_SEP_CHARSc@s~eZdZUdZded<ded<ded<ded<ded<ed d d Zd ed <ed edZded<e dddZ e dddZ dS)LinkArtifactFieldszLKeep this list updated with fields where linked and source artifacts differ.str entity_name project_namenameversion list[str]aliasesFT)initdefaultz Literal[True]_is_link)r2default_factorylist[Artifact]_linked_artifactsreturnboolcC|jSN)r4selfr>S/home/ubuntu/.local/lib/python3.10/site-packages/wandb/sdk/artifacts/_validators.pyis_link8zLinkArtifactFields.is_linkcCr:r;)r7r<r>r>r?linked_artifacts<rAz#LinkArtifactFields.linked_artifactsNr8r9)r8r6) __name__ __module__ __qualname____doc____annotations__rr4listr7propertyr@rBr>r>r>r?r**s  r*r.r+r8cCs^t|tkr|dtd}tdtd|t|r-tddttd||S)z~Validate the artifact name, returning it if successful. Raises: ValueError: If the artifact name is invalid. Nz ...zArtifact name is longer than z characters: zAArtifact names must not contain any of the following characters: , z. Got: )lenr% ValueErrorr' intersectionjoinsorted)r. trunc_namer>r>r?validate_artifact_nameAs   rRcCs|std|t}stdt|tkr9||kr,d|dtttd}t|d|dtd}t|tt|}rZ|pE|}dt t t |}td|d ||S) zValidate a project name according to W&B rules. Return the original name if successful. Args: name: The project name string. Raises: ValueError: If the name is invalid (too long or contains invalid characters). zProject name cannot be emptyzRegistry name cannot be emptyzInvalid registry name z , must be z characters or lesszInvalid project name rKzInvalid project/registry name z, cannot contain characters: ) rM removeprefixr!rLr%setr(rNrOrPmaprepr)r. registry_namemsg invalid_chars error_nameinvalid_chars_reprr>r>r?validate_project_nameVs"  r\r1Iterable[str] | strr0cCs@t|}tdd|Drdtttt}td||S)zValidate the artifact aliases and return them as a list. Raises: ValueError: If any of the aliases contain invalid characters. cs|]}t|VqdSr;r)rN.0r.r>r>r? ~z#validate_aliases..rKz:Aliases must not contain any of the following characters: )ranyrOrPrUrVr)rM)r1 aliases_listrYr>r>r?validate_aliaseswsrftypescCsbt|}tdd|Drdtttt}td|tdd|Dr/tdtd|S)z;Validate the artifact type names and return them as a list.csr^r;r_r`r>r>r?rbrcz*validate_artifact_types..rKzAArtifact types must not contain any of the following characters: css|] }t|tkVqdSr;)rLr%r`r>r>r?rbz-Artifact types must be less than or equal to z characters) rrdrOrPrUrVr)rMr%)rg types_listrYr>r>r?validate_artifact_typess rjz^[-\w]+( +[-\w]+)*$zre.Pattern[str] TAG_REGEXtagscCs*tt|}tdd|Drtd|S)zValidate artifact tag names and return them as a deduped list. In the case of duplicates, keep the first tag and maintain the order of appearance. Raises: ValueError: If any of the tags contain invalid characters. css|] }t| VqdSr;)rkmatch)ratagr>r>r?rbrhz validate_tags..zqInvalid tag(s). Tags must only contain alphanumeric characters separated by hyphens, underscores, and/or spaces.)rrrdrM)rl tags_listr>r>r? validate_tagss rpzwandb-RESERVED_ARTIFACT_TYPE_PREFIXzrun-zsource-)job run_tablecodezFinal[dict[str, str]]%RESERVED_ARTIFACT_NAME_PREFIX_BY_TYPEtypcCs:t|}dur||s|trtd|d|S)z5Validate the artifact type and return it as a string.NzArtifact type z; is reserved for internal use. Please use a different type.)rvget startswithrqrM)rwr. bad_prefixr>r>r?validate_artifact_types r{metadatadict[str, Any] | str | Nonedict[str, Any]cCstdt|d)z7Validate the artifact metadata and return it as a dict.z Cannot parse z as artifact metadata) TypeErrortyper|r>r>r?validate_metadatar str | NonecCs|rtt|SiSr;)rrrr>r>r?_rrcCs6ttjt|dd}t|tkrtdtd|S)NF) allow_nanz!Artifact must not have more than z metadata keys.)rjsondumpsrrLr#rMrr>r>r?rs   ttl_duration_secondsint int | NonecCs|dkr|SdS)zValidate the `ttlDurationSeconds` value from a GraphQL response. A non-positive value indicates that TTL is DISABLED (-2), which we convert to `None`. rNr>)rr>r>r?validate_ttl_duration_secondssrmethodMethodT[ArtifactT, P, R]c$ttd fd d }|S) zEnsure an artifact method runs only if the artifact has been logged. If the method is called on an artifact that's not logged, `ArtifactNotLoggedError` is raised. r=rargsP.argskwargsP.kwargsr8rcs*|r t|d|g|Ri|SN)fullnameobj)is_draftrr=rrrmethod_fullnamer>r?wrappers zensure_logged..wrapperNr=rrrrrr8rrrrrr>rr? ensure_loggedrcr) zEnsure an `Artifact` method runs only if the artifact is not finalized. If the method is called on an artifact that's not logged, `ArtifactFinalizedError` is raised. r=rrrrrr8rcs(|jr t|d|g|Ri|Sr)_finalrrrr>r?r s z%ensure_not_finalized..wrapperNrrrr>rr?ensure_not_finalizedrrprojectr9cCs |tSr;)ryr!rr>r>r?is_artifact_registry_projects rcCs&t|std|dt|tS)NzProject z- is not a registry project. Must start with: )rrMr!rSrr>r>r?remove_registry_prefixs  rc@sheZdZUded< dZded< dZded< edd d Zdd d ZddddddZ dddZ dS) ArtifactPathr+r.Nz Optional[str]rprefixpathr8rc Cs|d\}}}d\}}t|d}dkr|d}nt|dkr'|\}}nt|dkr3|\}}}ntd|||||||d S) zInstantiate by parsing a string artifact path. Raises: ValueError: If the string is not a valid artifact path. :)NNr&rrzInvalid artifact path: )rrr.) partitionrLsplitrM) clsrcollection_pathsepaliasrrpartsr.r>r>r?from_str*s      zArtifactPath.from_strcCs$|j|j|jf}ddd|DS)z>Returns the slash-separated string representation of the path.r&css|]}|r|VqdSr;r>)rapartr>r>r?rbCsz&ArtifactPath.to_str..)rrr.rO)r= ordered_partsr>r>r?to_str@szArtifactPath.to_strrrrcCst||jp||jp |dS)zJReturns a copy of this path with missing values set to the given defaults.r)rrr)r=rrr>r>r? with_defaultsEs zArtifactPath.with_defaultsr9cCst|j}o t|S)z8Returns True if this path appears to be a registry path.)r9rr)r=pr>r>r?is_registry_pathRszArtifactPath.is_registry_path)rr+r8r)r8r+)rrrrr8rrC) rDrErFrHrr classmethodrrrrr>r>r>r?r!s      rc@s*eZdZUdZded<ded<ded<dS)FullArtifactPathz2Same as ArtifactPath, but with all parts required.r+r.rrN)rDrErFrGrHr>r>r>r?rWs  r)r.r+r8r+)r1r]r8r0)rgr]r8r0)rlr]r8r0)rwr+r.r+r8r+)r|r}r8r~)r|rr8r~)r|r~r8r~)rrr8r)rrr8r)rr+r8r9)rr+r8r+)OrG __future__rrre dataclassesrrr functoolsrrtypingrr r r r r pydantic.dataclassespydantic_dataclasstyping_extensionsrrrwandb._iterutilsrrwandb._pydanticrwandb._strutilsr wandb.utilr exceptionsrrcollections.abcrrwandb.sdk.artifacts.artifactrrrrr r!rHr#r% frozensetr'r(r)r*rRr\rfrjcompilerkrprqrvr{rregisterrr+rdictrMethodTrrrrrrr>r>r>r?s~                !             5