
    <i?l                     t   S SK JrJrJrJrJrJrJr  S SKJ	r	  S SK
J	r	  S SKJr  S SK7  S SKJr  \R"                  " S5      r\" SSS	9r\\\\4      r\\\\4   \4   rS
rS\	S\S\S\4S jr      SS\S\\\4   S\S\S\S\4S jjr\(       a  S SKJr   " S S5      r " S S\5      r " S S\5      r g
)    )IOTYPE_CHECKINGAnyDictSetTypeVarUnion)	BaseModel)VERSION)*)modelsz2.Modelr
   )boundNmodelargskwargsreturnc                 `    [         (       a  U R                  " U0 UD6$ U R                  " U0 UD6$ N)PYDANTIC_V2model_dump_jsonjson)r   r   r   s      ^/home/james-whalen/.local/lib/python3.13/site-packages/qdrant_client/http/api/snapshots_api.pyto_jsonr      s1    {$$d5f55zz4*6**    objincludeby_aliasskip_defaultsexclude_unsetexclude_nonec           
          [        U S5      (       d  [        U S5      (       a   [        U UUU[        U=(       d    U5      US9$ U $ )Nr   r   )r   excluder   r    r!   )hasattrr   bool)r   r   r#   r   r   r    r!   s          r   jsonable_encoderr&      sN     sFws,=>>}=>%
 	
 Jr   )	ApiClientc                      \ rS rSrSS jr SS\4S jjr SS\S\S\4S jjr	 SS\S\4S	 jjr
 SS
\S\4S jjr SS\S\S
\S\4S jjr SS\S
\S\4S jjrS
\4S jrS\S\S
\4S jrS\S
\4S jrS rS\S\4S jrS\4S jr  S S\S\S\R*                  4S jjr    S!S\S\S\S\S\\   4
S jjr  S S\S\S\S\R6                  4S jjr    S!S\S\S\S\S\S\\   4S jjrSrg)"_SnapshotsApi3   c                     Xl         g r   
api_client)selfr-   s     r   __init___SnapshotsApi.__init__4   s    $r   Nwaitc                     0 nUb  [        U5      R                  5       US'   0 nU R                  R                  [        R
                  SSU(       a  UUS9$ SUS9$ )*
Create new snapshot of the whole storage
Nr1   POST
/snapshots)type_methodurlheadersparamsstrlowerr-   requestmInlineResponse20011)r.   r1   query_paramsr9   s       r   _build_for_create_full_snapshot-_SnapshotsApi._build_for_create_full_snapshot7   ss     #&t9??#4L &&''&G ' 
 	
 -1 ' 
 	
r   collection_nameshard_idc           	          [        U5      [        U5      S.n0 nUb  [        U5      R                  5       US'   0 nU R                  R                  [        R
                  SSU(       a  UOSUUS9$ )1
Create new snapshot of a shard for a collection
rD   rE   Nr1   r4   :/collections/{collection_name}/shards/{shard_id}/snapshotsr6   r7   r8   r9   path_paramsr:   r;   )r.   rD   rE   r1   rK   rA   r9   s          r    _build_for_create_shard_snapshot._SnapshotsApi._build_for_create_shard_snapshotK   sy      #?3H

 #&t9??#4L &&''L&GD# ' 
 	
r   c           	          S[        U5      0n0 nUb  [        U5      R                  5       US'   0 nU R                  R                  [        R
                  SSU(       a  UOSUUS9$ )&
Create new snapshot for a collection
rD   Nr1   r4   (/collections/{collection_name}/snapshotsrJ   r;   )r.   rD   r1   rK   rA   r9   s         r   _build_for_create_snapshot(_SnapshotsApi._build_for_create_snapshotg   st     s?3
 #&t9??#4L &&'':&GD# ' 
 	
r   snapshot_namec           	          S[        U5      0n0 nUb  [        U5      R                  5       US'   0 nU R                  R                  [        R
                  SSU(       a  UOSUUS9$ )&
Delete snapshot of the whole storage
rS   Nr1   DELETE/snapshots/{snapshot_name}rJ   r<   r=   r-   r>   r?   InlineResponse2009)r.   rS   r1   rK   rA   r9   s         r   _build_for_delete_full_snapshot-_SnapshotsApi._build_for_delete_full_snapshot   st     S/
 #&t9??#4L &&&&,&GD# ' 
 	
r   c           	          [        U5      [        U5      [        U5      S.n0 nUb  [        U5      R                  5       US'   0 nU R                  R                  [        R
                  SSU(       a  UOSUUS9$ )-
Delete snapshot of a shard for a collection
rD   rE   rS   Nr1   rV   J/collections/{collection_name}/shards/{shard_id}/snapshots/{snapshot_name}rJ   rX   )r.   rD   rE   rS   r1   rK   rA   r9   s           r    _build_for_delete_shard_snapshot._SnapshotsApi._build_for_delete_shard_snapshot   s      #?3H /
 #&t9??#4L &&&&\&GD# ' 
 	
r   c           	          [        U5      [        U5      S.n0 nUb  [        U5      R                  5       US'   0 nU R                  R                  [        R
                  SSU(       a  UOSUUS9$ )"
Delete snapshot for a collection
rD   rS   Nr1   rV   8/collections/{collection_name}/snapshots/{snapshot_name}rJ   rX   )r.   rD   rS   r1   rK   rA   r9   s          r   _build_for_delete_snapshot(_SnapshotsApi._build_for_delete_snapshot   sz      #?3 /

 #&t9??#4L &&&&J&GD# ' 
 	
r   c                 z    S[        U5      0n0 nU R                  R                  [        SSU(       a  UUS9$ SUS9$ )<
Download specified snapshot of the whole storage as a file
rS   GETrW   Nr6   r7   r8   r9   rK   r<   r-   r>   file)r.   rS   rK   r9   s       r   _build_for_get_full_snapshot*_SnapshotsApi._build_for_get_full_snapshot   s`     S/
 &&,&G# ' 
 	
 -1# ' 
 	
r   c                     [        U5      [        U5      [        U5      S.n0 nU R                  R                  [        SSU(       a  UUS9$ SUS9$ )D
Download specified snapshot of a shard from a collection as a file
r^   rj   r_   Nrk   rl   )r.   rD   rE   rS   rK   r9   s         r   _build_for_get_shard_snapshot+_SnapshotsApi._build_for_get_shard_snapshot   sm      #?3H /
 &&\&G# ' 
 	
 -1# ' 
 	
r   c                     [        U5      [        U5      S.n0 nU R                  R                  [        SSU(       a  UUS9$ SUS9$ )9
Download specified snapshot from a collection as a file
rd   rj   re   Nrk   rl   )r.   rD   rS   rK   r9   s        r   _build_for_get_snapshot%_SnapshotsApi._build_for_get_snapshot  sf      #?3 /

 &&J&G# ' 
 	
 -1# ' 
 	
r   c                 p    0 nU R                   R                  [        R                  SSU(       a  US9$ SS9$ ),
Get list of snapshots of the whole storage
rj   r5   N)r6   r7   r8   r9   )r-   r>   r?   InlineResponse20010)r.   r9   s     r   _build_for_list_full_snapshots,_SnapshotsApi._build_for_list_full_snapshots  sM     &&''&G	 ' 
 	
 -1	 ' 
 	
r   c                     [        U5      [        U5      S.n0 nU R                  R                  [        R                  SSU(       a  UUS9$ SUS9$ )3
Get list of snapshots for a shard of a collection
rH   rj   rI   Nrk   r<   r-   r>   r?   rz   )r.   rD   rE   rK   r9   s        r   _build_for_list_shard_snapshots-_SnapshotsApi._build_for_list_shard_snapshots%  sk      #?3H

 &&''L&G# ' 
 	
 -1# ' 
 	
r   c                     S[        U5      0n0 nU R                  R                  [        R                  SSU(       a  UUS9$ SUS9$ )(
Get list of snapshots for a collection
rD   rj   rP   Nrk   r   )r.   rD   rK   r9   s       r   _build_for_list_snapshots'_SnapshotsApi._build_for_list_snapshots;  sf     s?3
 &&'':&G# ' 
 	
 -1# ' 
 	
r   snapshot_recoverc           
          S[        U5      0n0 nUb  [        U5      R                  5       US'   0 n[        U5      nSU;  a  SUS'   U R                  R	                  [
        R                  SSU(       a  UOSUUUS9$ )	
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.
rD   Nr1   Content-Typeapplication/jsonPUTz0/collections/{collection_name}/snapshots/recoverr6   r7   r8   r9   rK   r:   contentr<   r=   r&   r-   r>   r?   rY   )r.   rD   r1   r   rK   rA   r9   bodys           r    _build_for_recover_from_snapshot._SnapshotsApi._build_for_recover_from_snapshotO  s     s?3
 #&t9??#4L  01(&8GN#&&&&B&GD# ' 
 	
r   prioritychecksumsnapshotc                 *   S[        U5      0n0 nUb  [        U5      R                  5       US'   Ub  [        U5      US'   Ub  [        U5      US'   0 n0 n	0 n
Ub  XYS'   U R                  R                  [        R
                  SSU(       a  UOSUUU
U	S	9$ )

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.
rD   Nr1   r   r   r   r4   z//collections/{collection_name}/snapshots/uploadr6   r7   r8   r9   rK   r:   datafilesrX   )r.   rD   r1   r   r   r   rK   rA   r9   r   r   s              r   )_build_for_recover_from_uploaded_snapshot7_SnapshotsApi._build_for_recover_from_uploaded_snapshotn  s     s?3
 #&t9??#4L '*8}L$'*8}L$$&! (*&&&&A&GD# ' 	
 		
r   shard_snapshot_recoverc           
         [        U5      [        U5      S.n0 nUb  [        U5      R                  5       US'   0 n[        U5      nSU;  a  SUS'   U R                  R	                  [
        R                  SSU(       a  UOSUUUS9$ )	
Recover shard of a local collection data from a snapshot. This will overwrite any data, stored in this shard, for the collection.
rH   Nr1   r   r   r   zB/collections/{collection_name}/shards/{shard_id}/snapshots/recoverr   r   )	r.   rD   rE   r1   r   rK   rA   r9   r   s	            r   &_build_for_recover_shard_from_snapshot4_SnapshotsApi._build_for_recover_shard_from_snapshot  s      #?3H

 #&t9??#4L  67(&8GN#&&&&T&GD# ' 
 	
r   c                 >   [        U5      [        U5      S.n0 nUb  [        U5      R                  5       US'   Ub  [        U5      US'   Ub  [        U5      US'   0 n	0 n
0 nUb  XjS'   U R                  R                  [        R
                  SSU	(       a  U	OSUUUU
S	9$ )

Recover shard of a local collection from an uploaded snapshot. This will overwrite any data, stored on this node, for the collection shard.
rH   Nr1   r   r   r   r4   zA/collections/{collection_name}/shards/{shard_id}/snapshots/uploadr   rX   )r.   rD   rE   r1   r   r   r   rK   rA   r9   r   r   s               r   /_build_for_recover_shard_from_uploaded_snapshot=_SnapshotsApi._build_for_recover_shard_from_uploaded_snapshot  s      #?3H

 #&t9??#4L '*8}L$'*8}L$$&! (*&&&&S&GD# ' 	
 		
r   r,   )r-   z Union[ApiClient, AsyncApiClient]r   NNNNNN)__name__
__module____qualname____firstlineno__r/   r%   rB   r<   intrL   rQ   rZ   r`   rf   rn   rr   rv   r{   r   r   r?   SnapshotRecoverr   SnapshotPriorityr   r   r   ShardSnapshotRecoverr   r   __static_attributes__ r   r   r)   r)   3   sD   %
 

0 	

 
 	
> 

 
: 

 
> 

 
 	

 
D 	

 
 	
8

(

 
 	
0

 
,


 
,

. .2	

 
 ++	
D %) &
&
 &
 #	&

 &
 S'&
X 9=

 
 	

 !" 6 6
J %) (
(
 (
 	(

 #(
 (
 S'(
 (
r   r)   c                       \ rS rSr SS\S\R                  4S jjr SS\S\	S\S\R                  4S jjr
 SS\S\S\R                  4S	 jjr SS
\S\S\R                  4S jjr SS\S\	S
\S\S\R                  4
S jjr SS\S
\S\S\R                  4S jjrS
\S\4S jrS\S\	S
\S\4S jrS\S
\S\4S jrS\R(                  4S jrS\S\	S\R(                  4S jrS\S\R(                  4S jr  SS\S\S\R0                  S\R                  4S jjr    S S\S\S\S\S\\   S\R                  4S jjr  SS\S\	S\S\R<                  S\R                  4
S jjr    S S\S\	S\S\S\S\\   S\R                  4S jjr Sr!g)!AsyncSnapshotsApii  Nr1   r   c                 <   #    U R                  US9I Sh  vN $  N7f)r3   r1   NrB   r.   r1   s     r   create_full_snapshot&AsyncSnapshotsApi.create_full_snapshot  s,      99 : 
 
 	
 
   rD   rE   c                 @   #    U R                  UUUS9I Sh  vN $  N7f)rG   rD   rE   r1   NrL   r.   rD   rE   r1   s       r   create_shard_snapshot'AsyncSnapshotsApi.create_shard_snapshot  s2      ::+ ; 
 
 	
 
   c                 >   #    U R                  UUS9I Sh  vN $  N7f)rO   rD   r1   NrQ   r.   rD   r1   s      r   create_snapshot!AsyncSnapshotsApi.create_snapshot  s/      44+ 5 
 
 	
 
   rS   c                 >   #    U R                  UUS9I Sh  vN $  N7f)rU   rS   r1   NrZ   r.   rS   r1   s      r   delete_full_snapshot&AsyncSnapshotsApi.delete_full_snapshot
  s/      99' : 
 
 	
 
r   c                 B   #    U R                  UUUUS9I Sh  vN $  N7f)r]   rD   rE   rS   r1   Nr`   r.   rD   rE   rS   r1   s        r   delete_shard_snapshot'AsyncSnapshotsApi.delete_shard_snapshot  s5      ::+'	 ; 
 
 	
 
   c                 @   #    U R                  UUUS9I Sh  vN $  N7f)rc   rD   rS   r1   Nrf   r.   rD   rS   r1   s       r   delete_snapshot!AsyncSnapshotsApi.delete_snapshot(  s2      44+' 5 
 
 	
 
r   c                 <   #    U R                  US9I Sh  vN $  N7f)ri   rS   Nrn   r.   rS   s     r   get_full_snapshot#AsyncSnapshotsApi.get_full_snapshot7  s,      66' 7 
 
 	
 
r   c                 @   #    U R                  UUUS9I Sh  vN $  N7f)rq   r^   Nrr   r.   rD   rE   rS   s       r   get_shard_snapshot$AsyncSnapshotsApi.get_shard_snapshotB  s2      77+' 8 
 
 	
 
r   c                 >   #    U R                  UUS9I Sh  vN $  N7f)ru   rd   Nrv   r.   rD   rS   s      r   get_snapshotAsyncSnapshotsApi.get_snapshotQ  s/      11+' 2 
 
 	
 
r   c                 >   #    U R                  5       I Sh  vN $  N7f)ry   Nr{   r.   s    r   list_full_snapshots%AsyncSnapshotsApi.list_full_snapshots^  s      88::::r   c                 >   #    U R                  UUS9I Sh  vN $  N7f)r~   rH   Nr   r.   rD   rE   s      r   list_shard_snapshots&AsyncSnapshotsApi.list_shard_snapshotsf  s/      99+ : 
 
 	
 
r   c                 <   #    U R                  US9I Sh  vN $  N7f)r   rD   Nr   r.   rD   s     r   list_snapshots AsyncSnapshotsApi.list_snapshotss  s,      33+ 4 
 
 	
 
r   r   c                 @   #    U R                  UUUS9I Sh  vN $  N7f)r   rD   r1   r   Nr   r.   rD   r1   r   s       r   recover_from_snapshot'AsyncSnapshotsApi.recover_from_snapshot~  s2      ::+- ; 
 
 	
 
r   r   r   r   c                 D   #    U R                  UUUUUS9I Sh  vN $  N7f)r   rD   r1   r   r   r   Nr   r.   rD   r1   r   r   r   s         r   recover_from_uploaded_snapshot0AsyncSnapshotsApi.recover_from_uploaded_snapshot  s9      CC+ D 
 
 	
 
s     r   c                 B   #    U R                  UUUUS9I Sh  vN $  N7f)r   rD   rE   r1   r   Nr   r.   rD   rE   r1   r   s        r   recover_shard_from_snapshot-AsyncSnapshotsApi.recover_shard_from_snapshot  s6      @@+#9	 A 
 
 	
 
r   c           	      F   #    U R                  UUUUUUS9I Sh  vN $  N7f)r   rD   rE   r1   r   r   r   Nr   r.   rD   rE   r1   r   r   r   s          r   $recover_shard_from_uploaded_snapshot6AsyncSnapshotsApi.recover_shard_from_uploaded_snapshot  s<      II+ J 
 
 	
 
s   !!r   r   r   r   "r   r   r   r   r%   r?   r@   r   r<   r   r   r   rY   r   r   r   rm   r   r   r   rz   r   r   r   r   r   r   r   r   r  r   r  r  r   r   r   r   r   r     s'    	
	
 
			
 	

 
 	

 
		
$ 

 
 
			
  

 
 
			
$ 

 
 	

 
 
		
* 	

 
 	

 
		
	
	
 
	


 
 	

 



 
 
	
;	
		;

 
 
			
	
	
 
			
 .2	

 
 ++	

 
		
$ %) 

 
 #	

 
 S'
 
		
. 9=

 
 	

 !" 6 6
 
		
* %) 

 
 	

 #
 
 S'
 
		
 
r   r   c                       \ rS rSr SS\S\R                  4S jjr SS\S\	S\S\R                  4S jjr
 SS\S\S\R                  4S	 jjr SS
\S\S\R                  4S jjr SS\S\	S
\S\S\R                  4
S jjr SS\S
\S\S\R                  4S jjrS
\S\4S jrS\S\	S
\S\4S jrS\S
\S\4S jrS\R(                  4S jrS\S\	S\R(                  4S jrS\S\R(                  4S jr  SS\S\S\R0                  S\R                  4S jjr    S S\S\S\S\S\\   S\R                  4S jjr  SS\S\	S\S\R<                  S\R                  4
S jjr    S S\S\	S\S\S\S\\   S\R                  4S jjr Sr!g)!SyncSnapshotsApii  Nr1   r   c                      U R                  US9$ )r3   r   r   r   s     r   r   %SyncSnapshotsApi.create_full_snapshot  s     33 4 
 	
r   rD   rE   c                 $    U R                  UUUS9$ )rG   r   r   r   s       r   r   &SyncSnapshotsApi.create_shard_snapshot  s%     44+ 5 
 	
r   c                 "    U R                  UUS9$ )rO   r   r   r   s      r   r    SyncSnapshotsApi.create_snapshot  s"     ..+ / 
 	
r   rS   c                 "    U R                  UUS9$ )rU   r   r   r   s      r   r   %SyncSnapshotsApi.delete_full_snapshot  s"     33' 4 
 	
r   c                 &    U R                  UUUUS9$ )r]   r   r   r   s        r   r   &SyncSnapshotsApi.delete_shard_snapshot  s(     44+'	 5 
 	
r   c                 $    U R                  UUUS9$ )rc   r   r   r   s       r   r    SyncSnapshotsApi.delete_snapshot  s%     ..+' / 
 	
r   c                      U R                  US9$ )ri   r   r   r   s     r   r   "SyncSnapshotsApi.get_full_snapshot  s     00' 1 
 	
r   c                 $    U R                  UUUS9$ )rq   r^   r   r   s       r   r   #SyncSnapshotsApi.get_shard_snapshot'  s%     11+' 2 
 	
r   c                 "    U R                  UUS9$ )ru   rd   r   r   s      r   r   SyncSnapshotsApi.get_snapshot6  s"     +++' , 
 	
r   c                 "    U R                  5       $ )ry   r   r   s    r   r   $SyncSnapshotsApi.list_full_snapshotsC  s     2244r   c                 "    U R                  UUS9$ )r~   rH   r   r   s      r   r   %SyncSnapshotsApi.list_shard_snapshotsK  s"     33+ 4 
 	
r   c                      U R                  US9$ )r   r   r   r   s     r   r   SyncSnapshotsApi.list_snapshotsX  s     --+ . 
 	
r   r   c                 $    U R                  UUUS9$ )r   r   r   r   s       r   r   &SyncSnapshotsApi.recover_from_snapshotc  s%     44+- 5 
 	
r   r   r   r   c                 (    U R                  UUUUUS9$ )r   r  r  r  s         r   r  /SyncSnapshotsApi.recover_from_uploaded_snapshotr  s+     ==+ > 
 	
r   r   c                 &    U R                  UUUUS9$ )r   r  r	  r
  s        r   r  ,SyncSnapshotsApi.recover_shard_from_snapshot  s(     ::+#9	 ; 
 	
r   c           	      *    U R                  UUUUUUS9$ )r   r  r  r  s          r   r  5SyncSnapshotsApi.recover_shard_from_uploaded_snapshot  s/     CC+ D 
 	
r   r   r   r   r   r  r   r   r   r  r    s'    	
	
 
			
 	

 
 	

 
		
$ 

 
 
			
  

 
 
			
$ 

 
 	

 
 
		
* 	

 
 	

 
		
	
	
 
	


 
 	

 



 
 
	
5	
		5

 
 
			
	
	
 
			
 .2	

 
 ++	

 
		
$ %) 

 
 #	

 
 S'
 
		
. 9=

 
 	

 !" 6 6
 
		
* %) 

 
 	

 #
 
 S'
 
		
 
r   r  )NNTNTT)!typingr   r   r   r   r   r   r	   pydanticr
   pydantic.mainpydantic.versionr   PYDANTIC_VERSIONqdrant_client.http.modelsr   r?   
startswithr   r   r   r<   	SetIntStrDictIntStrAnyrm   r   r%   r&   qdrant_client.http.api_clientr'   r)   r   r  r   r   r   <module>r@     s   D D D  # 8 ' 1))$/{+c3h 	U38_c)*+9 +S +C +C + 04	9m+, 	
   , 7l
 l
^b
 b
Jb
} b
r   