o -wiA;@sdZddlZddlmZmZmZmZerddlmZddlm Z ddl Z ddl m Z ddl mZmZdd lmZGd d d e ZGd d d e ZGddde ZdS)zPublic API: registries search.N) TYPE_CHECKINGAnyDictOptional)Client)gql) Paginator)_gql_artifact_fragment_gql_registry_fragment)ensure_registry_prefix_on_namesc seZdZdZedeZ  ddddedee ee fd ee ffd d Z d d Z ddZd dee ee fddfddZd dee ee fddfddZeddZeddZeddZddZZS)! Registriesz!Iterator that returns Registries.a query Registries($organization: String!, $filters: JSONString, $cursor: String, $perPage: Int) { organization(name: $organization) { orgEntity { name projects(filters: $filters, after: $cursor, first: $perPage) { pageInfo { endCursor hasNextPage } edges { node { ...RegistryFragment } } } } } } Ndclientr organizationfilterper_pagecs@||_||_t|p i|_|t|jd}t|||dS)N)rfilters)rrr rjsondumpssuper__init__)selfrrrr variables __class__k/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/wandb/apis/public/registries/registries_search.pyr0s zRegistries.__init__cCs t|jSN)boolobjectsrrrr__bool__As zRegistries.__bool__cCF|jd7_t|j|jkr|stt|j|jks|j|jSNr indexlenr _load_page StopIterationr!rrr__next__D  zRegistries.__next__return CollectionscCst|j|j|j|dS)N)registry_filtercollection_filter)r-rrrrrrrr collectionsLs zRegistries.collectionsVersionscCst|j|j|jd|dSN)r.r/artifact_filter)r2rrrr0rrrversionsTszRegistries.versionscC$|jrt|jddddSdS)Nr orgEntityprojectsedges last_responser'r!rrrlength]s zRegistries.lengthcC$|jr|jdddddSdS)Nrr7r8pageInfo hasNextPageTr;r!rrrmorefzRegistries.morecC$|jr|jdddddSdS)Nrr7r8r> endCursorr@r!rrrcursororBzRegistries.cursorcsfjsgSjdrjddstdjdddlmfddjddd d DS) Nrr7Organization ';' not found. Please verify the organization name is correctr)Registryc s:g|]}jjjddd|dd|dqS)rr7namenode)rrr;.0rrHrrr s z.Registries.convert_objects..r8r9)r; ValueErrorr%wandb.apis.public.registries.registryrHr!rrNrconvert_objectsxs    zRegistries.convert_objects)Nrr)__name__ __module__ __qualname____doc__rr QUERYstrrrrintrr"r*r1r5propertyr<rArErR __classcell__rrrrr s:     r c seZdZdZedZ   ddddedeeee fd eeee fd ee f fd d Z d dZ ddZ ddeeee fddfddZeddZeddZeddZddZZS) r-z;Iterator that returns Artifact collections in the Registry.a query Collections( $organization: String!, $registryFilter: JSONString, $collectionFilter: JSONString, $collectionTypes: [ArtifactCollectionType!], $cursor: String, $perPage: Int ) { organization(name: $organization) { orgEntity { name artifactCollections( projectFilters: $registryFilter, filters: $collectionFilter, collectionTypes: $collectionTypes, after: $cursor, first: $perPage ) { totalCount pageInfo { endCursor hasNextPage } edges { cursor node { id name description createdAt tags { edges { node { name } } } project { name entity { name } } defaultArtifactType { name } aliases { edges { node { alias } } } } } } } } } Nrrrrr.r/rcsh||_||_||_|p i|_|jrt|jnd|jr!t|jnd|jdg|d}t|||dS)N PORTFOLIO)registryFiltercollectionFilterrcollectionTypesperPage)rrr.r/rrrr)rrrr.r/rrrrrrs  zCollections.__init__cCt|dkp t|jdkSNrr'r r!rrrr"zCollections.__bool__cCr#r$r%r!rrrr*r+zCollections.__next__rr,r2cCst|j|j|j|j|dSr3)r2rrr.r/r0rrrr5szCollections.versionscCs |jr|jddddSdS)Nrr7artifactCollections totalCountr@r!rrrr<s zCollections.lengthcCr=)Nrr7rer>r?Tr@r!rrrrA zCollections.morecCrC)Nrr7rer>rDr@r!rrrrErgzCollections.cursorcsfddlmjs gSjdrjdds tdjdfddjddd d DS) Nr)ArtifactCollectionrr7rFrGc s\g|]*}j|dddd|ddd|dd|dddj|dddqS)rJprojectentityrIdefaultArtifactTypeF) is_sequence)rrrKrhrrrrO-s  z/Collections.convert_objects..rer9)wandb.apis.publicrhr;rPrr!rrmrrR s"     zCollections.convert_objects)NNrr)rSrTrUrVrrWrXrrrrYrr"r*r5rZr<rArErRr[rrrrr-s:D    r-cseZdZdZ    ddddedeeeefdeeeefd eeeefd ef fd d Z d dZ ddZ e ddZ e ddZe ddZddZZS)r2z8Iterator that returns Artifact versions in the Registry.Nrrrrr.r/r4rcs||_||_||_||_|pi|_tdtdd|_|jr$t |jnd|jr.t |jnd|jr8t |jnd|jd}t |||dS)Na[ query Versions( $organization: String!, $registryFilter: JSONString, $collectionFilter: JSONString, $artifactFilter: JSONString, $cursor: String, $perPage: Int ) { organization(name: $organization) { orgEntity { name artifactMemberships( projectFilters: $registryFilter, collectionFilters: $collectionFilter, filters: $artifactFilter, after: $cursor, first: $perPage ) { pageInfo { endCursor hasNextPage } edges { node { artifactCollection { project { name entity { name } } name } versionIndex artifact { ...ArtifactFragment } aliases { alias } } } } } } } F)include_aliases)r]r^artifactFilterr) rrr.r/r4rr rWrrrr)rrrr.r/r4rrrrrrAs   06 zVersions.__init__cCr#r$r%r!rrrr*r+zVersions.__next__cCrarbrcr!rrrr"rdzVersions.__bool__cCr6)Nrr7artifactMembershipsr9r:r!rrrr<szVersions.lengthcCr=)Nrr7rqr>r?Tr@r!rrrrArgz Versions.morecCrC)Nrr7rqr>rDr@r!rrrrErgzVersions.cursorcs\jsgSjdrjddstdjdfddjddddD}|S) Nrr7rFrGc 3s|]>}tj|ddddd|dddd|ddddt|dd|ddjdd |dd DVqd S) rJartifactCollectionrirjrIz:v versionIndexartifactcSsg|]}|dqS)aliasr)rLrurrrrOsz6Versions.convert_objects...aliasesN)wandbArtifact _from_attrsrXr)rLar!rr s    z+Versions.convert_objects..rqr9)r;rPr)r artifactsrr!rrRs"    zVersions.convert_objects)NNNr)rSrTrUrVrXrrrrYrr*r"rZr<rArErRr[rrrrr2>s8Q   r2)rVrtypingrrrr wandb_gqlrrrwwandb.apis.paginatorr&wandb.sdk.artifacts._graphql_fragmentsr r _utilsr r r-r2rrrrs    ~,