jKu ddlmZmZmZmZmZmZmZddlm Z ddl m Z ddl m Z ddlTddlmZe jdZedd ZeeeefZeeeefefZd Zd e d ed edefdZ ddedeeefdedededef dZerddlmZGddZGddeZGddeZ d S) )IO TYPE_CHECKINGAnyDictSetTypeVarUnion) BaseModel)VERSION)*)modelsz2.Modelr )boundNmodelargskwargsreturncDtr |j|i|S|j|i|SN) PYDANTIC_V2model_dump_jsonjson)rrrs RC:\Projects\DSCons\.venv\Lib\site-packages\qdrant_client\http\api\snapshots_api.pyto_jsonrs9+$u$d5f555uz4*6***Tobjincludeby_alias skip_defaults exclude_unset exclude_nonec t|dst|dr$t||||t|p||S|S)Nrr)rexcluderr r!)hasattrrbool)rrr#rrr r!s rjsonable_encoderr&sesF ws,=>>  }= >>%      Jr) ApiClientceZdZddZ d defdZ d dededefd Z d dedefd Z d d edefd Z d deded edefd Z d ded edefdZ d efdZ deded efdZded efdZdZdedefdZdefdZ d!dededejfdZ d"dededededeef dZ d!dedededejfdZ d"dedededededeef dZdedefdZdS)# _SnapshotsApi api_client Union[ApiClient, AsyncApiClient]c||_dSr)r*)selfr*s r__init__z_SnapshotsApi.__init__4s $rNwaitci}|$t||d<i}|jtjdd|r|nd|S): Create new snapshot of the whole storage Nr/POST /snapshots)type_methodurlheadersparamsstrlowerr*requestmInlineResponse20015)r-r/ query_paramsr7s r_build_for_create_full_snapshotz-_SnapshotsApi._build_for_create_full_snapshot7sh  #&t99??#4#4L &&'&0GGD '   rcollection_nameshard_idct|t|d}i}|$t||d<i}|jtjdd|r|nd||S)A Create new snapshot of a shard for a collection rArBNr/r2:/collections/{collection_name}/shards/{shard_id}/snapshotsr4r5r6r7 path_paramsr8r9)r-rArBr/rHr?r7s r _build_for_create_shard_snapshotz._SnapshotsApi._build_for_create_shard_snapshotKs #?33H     #&t99??#4#4L &&'L&0GGD# '   rcdt|i}i}|$t||d<i}|jtjdd|r|nd||S)6 Create new snapshot for a collection rANr/r2(/collections/{collection_name}/snapshotsrGr9)r-rAr/rHr?r7s r_build_for_create_snapshotz(_SnapshotsApi._build_for_create_snapshotgs s?33   #&t99??#4#4L &&':&0GGD# '   r snapshot_namecdt|i}i}|$t||d<i}|jtjdd|r|nd||S)6 Delete snapshot of the whole storage rNNr/DELETE/snapshots/{snapshot_name}rGr:r;r*r<r=InlineResponse20013)r-rNr/rHr?r7s r_build_for_delete_full_snapshotz-_SnapshotsApi._build_for_delete_full_snapshots S//   #&t99??#4#4L &&',&0GGD# '   rc t|t|t|d}i}|$t||d<i}|jtjdd|r|nd||S)= Delete snapshot of a shard for a collection rArBrNNr/rQJ/collections/{collection_name}/shards/{shard_id}/snapshots/{snapshot_name}rGrS)r-rArBrNr/rHr?r7s r _build_for_delete_shard_snapshotz._SnapshotsApi._build_for_delete_shard_snapshots #?33H  //    #&t99??#4#4L &&'\&0GGD# '   rct|t|d}i}|$t||d<i}|jtjdd|r|nd||S)2 Delete snapshot for a collection rArNNr/rQ8/collections/{collection_name}/snapshots/{snapshot_name}rGrS)r-rArNr/rHr?r7s r_build_for_delete_snapshotz(_SnapshotsApi._build_for_delete_snapshots #?33 //    #&t99??#4#4L &&'J&0GGD# '   rcxdt|i}i}|jtdd|r|nd|S)L Download specified snapshot of the whole storage as a file rNGETrRNr4r5r6r7rHr:r*r<file)r-rNrHr7s r_build_for_get_full_snapshotz*_SnapshotsApi._build_for_get_full_snapshotsV S// &&,&0GGD# '   rct|t|t|d}i}|jtdd|r|nd|S)T Download specified snapshot of a shard from a collection as a file rXrbrYNrcrd)r-rArBrNrHr7s r_build_for_get_shard_snapshotz+_SnapshotsApi._build_for_get_shard_snapshotsm #?33H  //  &&\&0GGD# '   rct|t|d}i}|jtdd|r|nd|S)I Download specified snapshot from a collection as a file r]rbr^Nrcrd)r-rArNrHr7s r_build_for_get_snapshotz%_SnapshotsApi._build_for_get_snapshotsd #?33 //  &&J&0GGD# '   rc^i}|jtjdd|r|ndS)< Get list of snapshots of the whole storage rbr3N)r4r5r6r7)r*r<r=InlineResponse20014)r-r7s r_build_for_list_full_snapshotsz,_SnapshotsApi._build_for_list_full_snapshotss@ &&'&0GGD '   rct|t|d}i}|jtjdd|r|nd|S)C Get list of snapshots for a shard of a collection rErbrFNrcr:r*r<r=ror-rArBrHr7s r_build_for_list_shard_snapshotsz-_SnapshotsApi._build_for_list_shard_snapshots%se #?33H   &&'L&0GGD# '   rcdt|i}i}|jtjdd|r|nd|S)8 Get list of snapshots for a collection rArbrLNrcrs)r-rArHr7s r_build_for_list_snapshotsz'_SnapshotsApi._build_for_list_snapshots;sY s?33 &&':&0GGD# '   rsnapshot_recoverc dt|i}i}|$t||d<i}t|}d|vrd|d<|jt jdd|r|nd|||S) Recover local collection data from a snapshot. This will overwrite any data, stored on this node, for the collection. If collection does not exist - it will be created. rANr/ Content-Typeapplication/jsonPUTz0/collections/{collection_name}/snapshots/recoverr4r5r6r7rHr8contentr:r;r&r*r<r=rT)r-rAr/ryrHr?r7bodys r _build_for_recover_from_snapshotz._SnapshotsApi._build_for_recover_from_snapshotOs s?33   #&t99??#4#4L  011  ( (&8GN #&&'B&0GGD#'   rprioritychecksumsnapshotc >dt|i}i}|$t||d<|t||d<|t||d<i}i} i} ||| d<|jtjdd|r|nd||| | S) Recover local collection data from an uploaded snapshot. This will overwrite any data, stored on this node, for the collection. If collection does not exist - it will be created. rANr/rrrr2z//collections/{collection_name}/snapshots/uploadr4r5r6r7rHr8datafilesrS) r-rAr/rrrrHr?r7rrs r)_build_for_recover_from_uploaded_snapshotz7_SnapshotsApi._build_for_recover_from_uploaded_snapshotns s?33   #&t99??#4#4L  '*8}}L $  '*8}}L $$&!   (E* &&'A&0GGD#'   rshard_snapshot_recoverc "t|t|d}i}|$t||d<i}t|}d|vrd|d<|jt jdd|r|nd|||S) Recover shard of a local collection data from a snapshot. This will overwrite any data, stored in this shard, for the collection. rENr/r|r}r~zB/collections/{collection_name}/shards/{shard_id}/snapshots/recoverrr) r-rArBr/rrHr?r7rs r&_build_for_recover_shard_from_snapshotz4_SnapshotsApi._build_for_recover_shard_from_snapshots #?33H     #&t99??#4#4L  677  ( (&8GN #&&'T&0GGD#'   rc Zt|t|d}i}|$t||d<|t||d<|t||d<i} i} i} ||| d<|jtjdd| r| nd||| | S) Recover shard of a local collection from an uploaded snapshot. This will overwrite any data, stored on this node, for the collection shard. rENr/rrrr2zA/collections/{collection_name}/shards/{shard_id}/snapshots/uploadrrS) r-rArBr/rrrrHr?r7rrs r/_build_for_recover_shard_from_uploaded_snapshotz=_SnapshotsApi._build_for_recover_shard_from_uploaded_snapshots #?33H     #&t99??#4#4L  '*8}}L $  '*8}}L $$&!   (E* &&'S&0GGD#'   rct|t|d}i}|jtdd|r|nd|S)H Stream the current state of a shard as a snapshot file rErbz9/collections/{collection_name}/shards/{shard_id}/snapshotNrcrdrts r _build_for_stream_shard_snapshotz._SnapshotsApi._build_for_stream_shard_snapshotsb #?33H   &&K&0GGD# '   r)r*r+rNNNNNN)__name__ __module__ __qualname__r.r%r@r:intrIrMrUrZr_rfrirlrprurxr=SnapshotRecoverrSnapshotPriorityrrrShardSnapshotRecoverrrrrrr)r)3s%%%%       0          >       :       >            D          8     (        0      ,          ,     ..2     +     D%) & & & & # &  & S' & & & & X9=       !" 6     J%) ( ( ( (  ( # (  ( S'( ( ( ( T        rr)cpeZdZ ddedejfdZ ddedededejfdZ ddededejfdZ dd ededej fd Z ddeded ededej f d Z dded ededej fd Zd edefd Zdeded edefdZded edefdZdejfdZdededejfdZdedejfdZ ddededejdej fdZ ddededededeedej f dZ ddedededejdej f dZ ddedededededeedej fdZdededefdZ dS) AsyncSnapshotsApiNr/rc>K||d{VS)r1r/Nr@r-r/s rcreate_full_snapshotz&AsyncSnapshotsApi.create_full_snapshotsF99:         rrArBcBK||||d{VS)rDrArBr/NrIr-rArBr/s rcreate_shard_snapshotz'AsyncSnapshotsApi.create_shard_snapshotsL::+;         rc@K|||d{VS)rKrAr/NrMr-rAr/s rcreate_snapshotz!AsyncSnapshotsApi.create_snapshotsI44+5         rrNc@K|||d{VS)rPrNr/NrUr-rNr/s rdelete_full_snapshotz&AsyncSnapshotsApi.delete_full_snapshot sI99':         rcDK|||||d{VS)rWrArBrNr/NrZr-rArBrNr/s rdelete_shard_snapshotz'AsyncSnapshotsApi.delete_shard_snapshot-sO::+' ;         rcBK||||d{VS)r\rArNr/Nr_r-rArNr/s rdelete_snapshotz!AsyncSnapshotsApi.delete_snapshot>sL44+'5         rc>K||d{VS)rarNNrfr-rNs rget_full_snapshotz#AsyncSnapshotsApi.get_full_snapshotMsF66'7         rcBK||||d{VS)rhrXNrir-rArBrNs rget_shard_snapshotz$AsyncSnapshotsApi.get_shard_snapshotXsL77+'8         rc@K|||d{VS)rkr]Nrlr-rArNs r get_snapshotzAsyncSnapshotsApi.get_snapshotgsI11+'2         rc:K|d{VS)rnNrpr-s rlist_full_snapshotsz%AsyncSnapshotsApi.list_full_snapshotsts, 88:::::::::rc@K|||d{VS)rrrENrur-rArBs rlist_shard_snapshotsz&AsyncSnapshotsApi.list_shard_snapshots|sI99+:         rc>K||d{VS)rwrANrxr-rAs rlist_snapshotsz AsyncSnapshotsApi.list_snapshotssF33+4         rrycBK||||d{VS)r{rAr/ryNrr-rAr/rys rrecover_from_snapshotz'AsyncSnapshotsApi.recover_from_snapshotsL::+-;         rrrrcFK||||||d{VS)rrAr/rrrNrr-rAr/rrrs rrecover_from_uploaded_snapshotz0AsyncSnapshotsApi.recover_from_uploaded_snapshotsSCC+ D         rrcDK|||||d{VS)rrArBr/rNrr-rArBr/rs rrecover_shard_from_snapshotz-AsyncSnapshotsApi.recover_shard_from_snapshotsP@@+#9 A         rcHK|||||||d{VS)rrArBr/rrrNrr-rArBr/rrrs r$recover_shard_from_uploaded_snapshotz6AsyncSnapshotsApi.recover_shard_from_uploaded_snapshotsVII+ J         rc@K|||d{VS)rrENrrs rstream_shard_snapshotz'AsyncSnapshotsApi.stream_shard_snapshotsI::+;         rrrr!rrrr%r=r>rr:rrrrTrrrrerrrrorrrrrrrrrrrrrrrrrrsY                         $                          $              *                                                 ; ;;;;                      .2       +       $%)     #    S'       .9=       !" 6       *%)       #    S'      *              rrcpeZdZ ddedejfdZ ddedededejfdZ ddededejfdZ dd ededej fd Z ddeded ededej f d Z dded ededej fd Zd edefd Zdeded edefdZded edefdZdejfdZdededejfdZdedejfdZ ddededejdej fdZ ddededededeedej f dZ ddedededejdej f dZ ddedededededeedej fdZdededefdZ dS) SyncSnapshotsApiNr/rc.||S)r1rrrs rrz%SyncSnapshotsApi.create_full_snapshots$334   rrArBc2||||S)rDrrrs rrz&SyncSnapshotsApi.create_shard_snapshots*44+5   rc0|||S)rKrrrs rrz SyncSnapshotsApi.create_snapshots'..+/   rrNc0|||S)rPrrrs rrz%SyncSnapshotsApi.delete_full_snapshots'33'4   rc4|||||S)rWrrrs rrz&SyncSnapshotsApi.delete_shard_snapshots-44+' 5   rc2||||S)r\rrrs rrz SyncSnapshotsApi.delete_snapshot0s*..+'/   rc.||S)rarrrs rrz"SyncSnapshotsApi.get_full_snapshot?s$00'1   rc2||||S)rhrXrrs rrz#SyncSnapshotsApi.get_shard_snapshotJs*11+'2   rc0|||S)rkr]rrs rrzSyncSnapshotsApi.get_snapshotYs'+++',   rc*|S)rnrrs rrz$SyncSnapshotsApi.list_full_snapshotsfs 22444rc0|||S)rrrErrs rrz%SyncSnapshotsApi.list_shard_snapshotsns'33+4   rc.||S)rwrrrs rrzSyncSnapshotsApi.list_snapshots{s$--+.   rryc2||||S)r{rrrs rrz&SyncSnapshotsApi.recover_from_snapshots*44+-5   rrrrc6||||||S)rrrrs rrz/SyncSnapshotsApi.recover_from_uploaded_snapshots0==+ >   rrc4|||||S)rrrrs rrz,SyncSnapshotsApi.recover_shard_from_snapshots-::+#9 ;   rc8|||||||S)rrrrs rrz5SyncSnapshotsApi.recover_shard_from_uploaded_snapshots4CC+ D   rc0|||S)rrErrs rrz&SyncSnapshotsApi.stream_shard_snapshots'44+5   rrrrrrrrrrsY                         $                          $              *                                                 5 5555                      .2       +       $%)     #    S'       .9=       !" 6       *%)       #    S'      *              rr)NNTNTT)!typingrrrrrrr pydanticr pydantic.mainpydantic.versionr PYDANTIC_VERSIONqdrant_client.http.modelsr r= startswithrrrr: SetIntStr DictIntStrAnyrerr%r&qdrant_client.http.api_clientr'r)rrrrrrsADDDDDDDDDDDDDDDDDD######888888''''111111))$// {+++ c3h U38_c)*  +9+S+C+C++++04   9m+ ,     ,8777777B B B B B B B B Jo o o o o o o o do o o o o }o o o o o r