
    Ng*l                        d dl mZmZmZmZmZmZmZ d dlm	Z	 d dl
m	Z	 d dlmZ d dlT d dlmZ  ej        d          Z edd	          Zeeeef                  Zeeeef         ef         Zd
Zde	dededefdZ	 	 	 	 	 	 ddedeeef         dedededefdZerd dlmZ  G d d          Z G d de          Z G d de          Z d
S )     )IOTYPE_CHECKINGAnyDictSetTypeVarUnion)	BaseModel)VERSION)*)modelsz2.Modelr
   )boundNmodelargskwargsreturnc                 D    t           r | j        |i |S  | j        |i |S N)PYDANTIC_V2model_dump_jsonjson)r   r   r   s      `/var/www/html/ai-engine/env/lib/python3.11/site-packages/qdrant_client/http/api/snapshots_api.pyto_jsonr      s9     +$u$d5f555uz4*6***    Tobjincludeby_aliasskip_defaultsexclude_unsetexclude_nonec           	          t          | d          st          | d          r$t          | |||t          |p|          |          S | S )Nr   r   )r   excluder   r    r!   )hasattrr   bool)r   r   r#   r   r   r    r!   s          r   jsonable_encoderr&      se     sF 
ws,=>> 
}=>>%
 
 
 	
 Jr   )	ApiClientc                      e Zd ZddZ	 ddefdZ	 ddededefd	Z	 ddedefd
Z		 ddedefdZ
	 ddedededefdZ	 ddededefdZdefdZdededefdZdedefdZd ZdedefdZdefdZ	 	 d dededej        fdZ	 	 	 	 d!dededededee         f
dZ	 	 d dedededej        fdZ	 	 	 	 d!dedededededee         fdZdS )"_SnapshotsApi
api_client Union[ApiClient, AsyncApiClient]c                     || _         d S r   )r*   )selfr*   s     r   __init__z_SnapshotsApi.__init__4   s    $r   Nwaitc                     i }|$t          |                                          |d<   i }| j                            t          j        dd|r|nd|          S ):
        Create new snapshot of the whole storage
        Nr/   POST
/snapshots)type_methodurlheadersparamsstrlowerr*   requestmInlineResponse20011)r-   r/   query_paramsr7   s       r   _build_for_create_full_snapshotz-_SnapshotsApi._build_for_create_full_snapshot7   sh     #&t99??#4#4L &&'&0GGD ' 
 
 	
r   collection_nameshard_idc                     t          |          t          |          d}i }|$t          |                                          |d<   i }| j                            t          j        dd|r|nd||          S )A
        Create new snapshot of a shard for a collection
        rA   rB   Nr/   r2   :/collections/{collection_name}/shards/{shard_id}/snapshotsr4   r5   r6   r7   path_paramsr8   r9   )r-   rA   rB   r/   rH   r?   r7   s          r    _build_for_create_shard_snapshotz._SnapshotsApi._build_for_create_shard_snapshotK   s      #?33H
 

 #&t99??#4#4L &&'L&0GGD# ' 
 
 	
r   c                     dt          |          i}i }|$t          |                                          |d<   i }| j                            t          j        dd|r|nd||          S )6
        Create new snapshot for a collection
        rA   Nr/   r2   (/collections/{collection_name}/snapshotsrG   r9   )r-   rA   r/   rH   r?   r7   s         r   _build_for_create_snapshotz(_SnapshotsApi._build_for_create_snapshotg   s     s?33
 #&t99??#4#4L &&':&0GGD# ' 
 
 	
r   snapshot_namec                     dt          |          i}i }|$t          |                                          |d<   i }| j                            t          j        dd|r|nd||          S )6
        Delete snapshot of the whole storage
        rN   Nr/   DELETE/snapshots/{snapshot_name}rG   r:   r;   r*   r<   r=   InlineResponse200)r-   rN   r/   rH   r?   r7   s         r   _build_for_delete_full_snapshotz-_SnapshotsApi._build_for_delete_full_snapshot   s     S//
 #&t99??#4#4L &&%,&0GGD# ' 
 
 	
r   c                    t          |          t          |          t          |          d}i }|$t          |                                          |d<   i }| j                            t          j        dd|r|nd||          S )=
        Delete snapshot of a shard for a collection
        rA   rB   rN   Nr/   rQ   J/collections/{collection_name}/shards/{shard_id}/snapshots/{snapshot_name}rG   rS   )r-   rA   rB   rN   r/   rH   r?   r7   s           r    _build_for_delete_shard_snapshotz._SnapshotsApi._build_for_delete_shard_snapshot   s      #?33H //
 
 #&t99??#4#4L &&%\&0GGD# ' 
 
 	
r   c                     t          |          t          |          d}i }|$t          |                                          |d<   i }| j                            t          j        dd|r|nd||          S )2
        Delete snapshot for a collection
        rA   rN   Nr/   rQ   8/collections/{collection_name}/snapshots/{snapshot_name}rG   rS   )r-   rA   rN   r/   rH   r?   r7   s          r   _build_for_delete_snapshotz(_SnapshotsApi._build_for_delete_snapshot   s      #?33 //
 

 #&t99??#4#4L &&%J&0GGD# ' 
 
 	
r   c                 x    dt          |          i}i }| j                            t          dd|r|nd|          S )L
        Download specified snapshot of the whole storage as a file
        rN   GETrR   Nr4   r5   r6   r7   rH   r:   r*   r<   file)r-   rN   rH   r7   s       r   _build_for_get_full_snapshotz*_SnapshotsApi._build_for_get_full_snapshot   sV     S//
 &&,&0GGD# ' 
 
 	
r   c                     t          |          t          |          t          |          d}i }| j                            t          dd|r|nd|          S )T
        Download specified snapshot of a shard from a collection as a file
        rX   rb   rY   Nrc   rd   )r-   rA   rB   rN   rH   r7   s         r   _build_for_get_shard_snapshotz+_SnapshotsApi._build_for_get_shard_snapshot   sm      #?33H //
 
 &&\&0GGD# ' 
 
 	
r   c                     t          |          t          |          d}i }| j                            t          dd|r|nd|          S )I
        Download specified snapshot from a collection as a file
        r]   rb   r^   Nrc   rd   )r-   rA   rN   rH   r7   s        r   _build_for_get_snapshotz%_SnapshotsApi._build_for_get_snapshot  sd      #?33 //
 

 &&J&0GGD# ' 
 
 	
r   c                 ^    i }| j                             t          j        dd|r|nd          S )<
        Get list of snapshots of the whole storage
        rb   r3   N)r4   r5   r6   r7   )r*   r<   r=   InlineResponse20010)r-   r7   s     r   _build_for_list_full_snapshotsz,_SnapshotsApi._build_for_list_full_snapshots  s@     &&'&0GGD	 ' 
 
 	
r   c                     t          |          t          |          d}i }| j                            t          j        dd|r|nd|          S )C
        Get list of snapshots for a shard of a collection
        rE   rb   rF   Nrc   r:   r*   r<   r=   ro   )r-   rA   rB   rH   r7   s        r   _build_for_list_shard_snapshotsz-_SnapshotsApi._build_for_list_shard_snapshots%  se      #?33H
 

 &&'L&0GGD# ' 
 
 	
r   c                     dt          |          i}i }| j                            t          j        dd|r|nd|          S )8
        Get list of snapshots for a collection
        rA   rb   rL   Nrc   rs   )r-   rA   rH   r7   s       r   _build_for_list_snapshotsz'_SnapshotsApi._build_for_list_snapshots;  sY     s?33
 &&':&0GGD# ' 
 
 	
r   snapshot_recoverc           	         dt          |          i}i }|$t          |                                          |d<   i }t          |          }d|vrd|d<   | j                            t
          j        dd|r|nd|||          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.
        rA   Nr/   Content-Typeapplication/jsonPUTz0/collections/{collection_name}/snapshots/recoverr4   r5   r6   r7   rH   r8   contentr:   r;   r&   r*   r<   r=   rT   )r-   rA   r/   rx   rH   r?   r7   bodys           r    _build_for_recover_from_snapshotz._SnapshotsApi._build_for_recover_from_snapshotO  s     s?33
 #&t99??#4#4L  011((&8GN#&&%B&0GGD# ' 
 
 	
r   prioritychecksumsnapshotc           
      >   dt          |          i}i }|$t          |                                          |d<   |t          |          |d<   |t          |          |d<   i }i }	i }
|||	d<   | j                            t          j        dd|r|nd|||
|		          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.
        rA   Nr/   r   r   r   r2   z//collections/{collection_name}/snapshots/uploadr4   r5   r6   r7   rH   r8   datafilesrS   )r-   rA   r/   r   r   r   rH   r?   r7   r   r   s              r   )_build_for_recover_from_uploaded_snapshotz7_SnapshotsApi._build_for_recover_from_uploaded_snapshotn  s     s?33
 #&t99??#4#4L '*8}}L$'*8}}L$$&! (E*&&%A&0GGD# ' 	
 	
 		
r   shard_snapshot_recoverc           	      "   t          |          t          |          d}i }|$t          |                                          |d<   i }t          |          }d|vrd|d<   | j                            t
          j        dd|r|nd|||          S )	
        Recover shard of a local collection data from a snapshot. This will overwrite any data, stored in this shard, for the collection.
        rE   Nr/   r{   r|   r}   zB/collections/{collection_name}/shards/{shard_id}/snapshots/recoverr~   r   )	r-   rA   rB   r/   r   rH   r?   r7   r   s	            r   &_build_for_recover_shard_from_snapshotz4_SnapshotsApi._build_for_recover_shard_from_snapshot  s      #?33H
 

 #&t99??#4#4L  677((&8GN#&&%T&0GGD# ' 
 
 	
r   c           
      Z   t          |          t          |          d}i }|$t          |                                          |d<   |t          |          |d<   |t          |          |d<   i }	i }
i }|||
d<   | j                            t          j        dd|	r|	nd||||
	          S )

        Recover shard of a local collection from an uploaded snapshot. This will overwrite any data, stored on this node, for the collection shard.
        rE   Nr/   r   r   r   r2   zA/collections/{collection_name}/shards/{shard_id}/snapshots/uploadr   rS   )r-   rA   rB   r/   r   r   r   rH   r?   r7   r   r   s               r   /_build_for_recover_shard_from_uploaded_snapshotz=_SnapshotsApi._build_for_recover_shard_from_uploaded_snapshot  s      #?33H
 

 #&t99??#4#4L '*8}}L$'*8}}L$$&! (E*&&%S&0GGD# ' 	
 	
 		
r   )r*   r+   r   NNNNNN)__name__
__module____qualname__r.   r%   r@   r:   intrI   rM   rU   rZ   r_   rf   ri   rl   rp   rt   rw   r=   SnapshotRecoverr   SnapshotPriorityr   r   r   ShardSnapshotRecoverr   r    r   r   r)   r)   3   sw       % % % %
 
 

 
 
 
0 	
 

 
 	
 
 
 
> 
 

 
 
 
 
: 
 

 
 
 
 
> 
 

 
 	

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

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

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

 #(
 (
 S'(
 (
 (
 (
 (
 (
r   r)   c                   \   e Zd Z	 ddedej        fdZ	 ddedededej        fdZ		 ddededej        fdZ
	 dd	ededej        fd
Z	 ddeded	ededej        f
dZ	 dded	ededej        fdZd	edefdZdeded	edefdZded	edefdZdej        fdZdededej        fdZdedej        fdZ	 	 ddededej        dej        fdZ	 	 	 	 ddededededee         dej        fdZ	 	 ddedededej        dej        f
dZ	 	 	 	 ddedededededee         dej        fdZdS )AsyncSnapshotsApiNr/   r   c                 >   K   |                      |           d{V S )r1   r/   Nr@   r-   r/   s     r   create_full_snapshotz&AsyncSnapshotsApi.create_full_snapshot  sF       99 : 
 
 
 
 
 
 
 
 	
r   rA   rB   c                 B   K   |                      |||           d{V S )rD   rA   rB   r/   NrI   r-   rA   rB   r/   s       r   create_shard_snapshotz'AsyncSnapshotsApi.create_shard_snapshot  sL       ::+ ; 
 
 
 
 
 
 
 
 	
r   c                 @   K   |                      ||           d{V S )rK   rA   r/   NrM   r-   rA   r/   s      r   create_snapshotz!AsyncSnapshotsApi.create_snapshot  sI       44+ 5 
 
 
 
 
 
 
 
 	
r   rN   c                 @   K   |                      ||           d{V S )rP   rN   r/   NrU   r-   rN   r/   s      r   delete_full_snapshotz&AsyncSnapshotsApi.delete_full_snapshot
  sI       99' : 
 
 
 
 
 
 
 
 	
r   c                 D   K   |                      ||||           d{V S )rW   rA   rB   rN   r/   NrZ   r-   rA   rB   rN   r/   s        r   delete_shard_snapshotz'AsyncSnapshotsApi.delete_shard_snapshot  sO       ::+'	 ; 
 
 
 
 
 
 
 
 	
r   c                 B   K   |                      |||           d{V S )r\   rA   rN   r/   Nr_   r-   rA   rN   r/   s       r   delete_snapshotz!AsyncSnapshotsApi.delete_snapshot(  sL       44+' 5 
 
 
 
 
 
 
 
 	
r   c                 >   K   |                      |           d{V S )ra   rN   Nrf   r-   rN   s     r   get_full_snapshotz#AsyncSnapshotsApi.get_full_snapshot7  sF       66' 7 
 
 
 
 
 
 
 
 	
r   c                 B   K   |                      |||           d{V S )rh   rX   Nri   r-   rA   rB   rN   s       r   get_shard_snapshotz$AsyncSnapshotsApi.get_shard_snapshotB  sL       77+' 8 
 
 
 
 
 
 
 
 	
r   c                 @   K   |                      ||           d{V S )rk   r]   Nrl   r-   rA   rN   s      r   get_snapshotzAsyncSnapshotsApi.get_snapshotQ  sI       11+' 2 
 
 
 
 
 
 
 
 	
r   c                 :   K   |                                   d{V S )rn   Nrp   r-   s    r   list_full_snapshotsz%AsyncSnapshotsApi.list_full_snapshots^  s,       88:::::::::r   c                 @   K   |                      ||           d{V S )rr   rE   Nrt   r-   rA   rB   s      r   list_shard_snapshotsz&AsyncSnapshotsApi.list_shard_snapshotsf  sI       99+ : 
 
 
 
 
 
 
 
 	
r   c                 >   K   |                      |           d{V S )rv   rA   Nrw   r-   rA   s     r   list_snapshotsz AsyncSnapshotsApi.list_snapshotss  sF       33+ 4 
 
 
 
 
 
 
 
 	
r   rx   c                 B   K   |                      |||           d{V S )rz   rA   r/   rx   Nr   r-   rA   r/   rx   s       r   recover_from_snapshotz'AsyncSnapshotsApi.recover_from_snapshot~  sL       ::+- ; 
 
 
 
 
 
 
 
 	
r   r   r   r   c                 F   K   |                      |||||           d{V S )r   rA   r/   r   r   r   Nr   r-   rA   r/   r   r   r   s         r   recover_from_uploaded_snapshotz0AsyncSnapshotsApi.recover_from_uploaded_snapshot  sS       CC+ D 
 
 
 
 
 
 
 
 	
r   r   c                 D   K   |                      ||||           d{V S )r   rA   rB   r/   r   Nr   r-   rA   rB   r/   r   s        r   recover_shard_from_snapshotz-AsyncSnapshotsApi.recover_shard_from_snapshot  sP       @@+#9	 A 
 
 
 
 
 
 
 
 	
r   c                 H   K   |                      ||||||           d{V S )r   rA   rB   r/   r   r   r   Nr   r-   rA   rB   r/   r   r   r   s          r   $recover_shard_from_uploaded_snapshotz6AsyncSnapshotsApi.recover_shard_from_uploaded_snapshot  sV       II+ J 
 
 
 
 
 
 
 
 	
r   r   r   r    r   r   r   r%   r=   r>   r   r:   r   r   r   rT   r   r   r   re   r   r   r   ro   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r     s)        	
 	
	
 
		
 	
 	
 	
 	
 

 
 	

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

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

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

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

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

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

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

 #
 
 S'
 
	
 
 
 
 
 
r   r   c                   \   e Zd Z	 ddedej        fdZ	 ddedededej        fdZ		 ddededej        fdZ
	 dd	ededej        fd
Z	 ddeded	ededej        f
dZ	 dded	ededej        fdZd	edefdZdeded	edefdZded	edefdZdej        fdZdededej        fdZdedej        fdZ	 	 ddededej        dej        fdZ	 	 	 	 ddededededee         dej        fdZ	 	 ddedededej        dej        f
dZ	 	 	 	 ddedededededee         dej        fdZdS )SyncSnapshotsApiNr/   r   c                 .    |                      |          S )r1   r   r   r   s     r   r   z%SyncSnapshotsApi.create_full_snapshot  s$     33 4 
 
 	
r   rA   rB   c                 2    |                      |||          S )rD   r   r   r   s       r   r   z&SyncSnapshotsApi.create_shard_snapshot  s*     44+ 5 
 
 	
r   c                 0    |                      ||          S )rK   r   r   r   s      r   r   z SyncSnapshotsApi.create_snapshot  s'     ..+ / 
 
 	
r   rN   c                 0    |                      ||          S )rP   r   r   r   s      r   r   z%SyncSnapshotsApi.delete_full_snapshot  s'     33' 4 
 
 	
r   c                 4    |                      ||||          S )rW   r   r   r   s        r   r   z&SyncSnapshotsApi.delete_shard_snapshot  s-     44+'	 5 
 
 	
r   c                 2    |                      |||          S )r\   r   r   r   s       r   r   z SyncSnapshotsApi.delete_snapshot  s*     ..+' / 
 
 	
r   c                 .    |                      |          S )ra   r   r   r   s     r   r   z"SyncSnapshotsApi.get_full_snapshot  s$     00' 1 
 
 	
r   c                 2    |                      |||          S )rh   rX   r   r   s       r   r   z#SyncSnapshotsApi.get_shard_snapshot'  s*     11+' 2 
 
 	
r   c                 0    |                      ||          S )rk   r]   r   r   s      r   r   zSyncSnapshotsApi.get_snapshot6  s'     +++' , 
 
 	
r   c                 *    |                                  S )rn   r   r   s    r   r   z$SyncSnapshotsApi.list_full_snapshotsC  s     22444r   c                 0    |                      ||          S )rr   rE   r   r   s      r   r   z%SyncSnapshotsApi.list_shard_snapshotsK  s'     33+ 4 
 
 	
r   c                 .    |                      |          S )rv   r   r   r   s     r   r   zSyncSnapshotsApi.list_snapshotsX  s$     --+ . 
 
 	
r   rx   c                 2    |                      |||          S )rz   r   r   r   s       r   r   z&SyncSnapshotsApi.recover_from_snapshotc  s*     44+- 5 
 
 	
r   r   r   r   c                 6    |                      |||||          S )r   r   r   r   s         r   r   z/SyncSnapshotsApi.recover_from_uploaded_snapshotr  s0     ==+ > 
 
 	
r   r   c                 4    |                      ||||          S )r   r   r   r   s        r   r   z,SyncSnapshotsApi.recover_shard_from_snapshot  s-     ::+#9	 ; 
 
 	
r   c                 8    |                      ||||||          S )r   r   r   r   s          r   r   z5SyncSnapshotsApi.recover_shard_from_uploaded_snapshot  s4     CC+ D 
 
 	
r   r   r   r   r   r   r   r   r   r     s)        	
 	
	
 
		
 	
 	
 	
 	
 

 
 	

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

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

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

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

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

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

 !" 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DictIntStrAnyre   r   r%   r&   qdrant_client.http.api_clientr'   r)   r   r   r   r   r   <module>r     sA   D D D D D D D D D D D D D D D D D D       # # # # # # 8 8 8 8 8 8 ' ' ' ' 1 1 1 1 1 1))$//{+++c3h 	U38_c)*+9 +S +C +C + + + + 04 	9m+, 	
      ,  8777777l
 l
 l
 l
 l
 l
 l
 l
^b
 b
 b
 b
 b
 b
 b
 b
Jb
 b
 b
 b
 b
} b
 b
 b
 b
 b
r   