
    o i6                     v   d Z ddlZddlZddlmZ ddlmZ ddlmZ ddl	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mZmZmZmZmZmZ d	dlmZ d	dlmZmZ e
rd	dl m!Z!  ej"        e#          Z$ ej%        d          Z&dZ'dddZ(e G d d                      Z)e	 	 	 	 d8dee)         dee*         de*de*dee*         dee*         dee+e*e*f                  dee,e*                  de-e,e+         e,e+         ee*         f         fd            Z. G d de          Z/ G d  d!e          Z0	 	 	 d9d"d#d$e+dee*         dee+e*e*f                  dee*         ddfd%Z1d&e+fd'Z2d(e+fd)Z3d*e+fd+Z4d"d#d,e*ddfd-Z5d"d#d.e+d/e6d,e*ddf
d0Z7d.e+d1e)d/e6de,e*         fd2Z8d3e,e+         d4e*de0fd5Z9d"d#d6e,e*         d/e6de,e+         fd7Z:dS ):z.Git LFS related type definitions and utilities    N)	dataclass)ceil)getsize)TYPE_CHECKINGBinaryIOIterableOptional	TypedDict)unquote)	constants   )build_hf_headersfix_hf_endpoint_in_urlget_sessionhf_raise_for_statushttp_backoffloggingvalidate_hf_hub_args)SliceFileObj)sha256sha_fileobj)CommitOperationAddz^[0-9a-f]{40}$zlfs-multipart-uploadzapplication/vnd.git-lfs+json)AcceptzContent-Typec                       e Zd ZU dZeed<   eed<   eed<   edefd            Z	edefd            Z
ed	efd
            ZdS )
UploadInfoad  
    Dataclass holding required information to determine whether a blob
    should be uploaded to the hub using the LFS protocol or the regular protocol

    Args:
        sha256 (`bytes`):
            SHA256 hash of the blob
        size (`int`):
            Size in bytes of the blob
        sample (`bytes`):
            First 512 bytes of the blob
    r   sizesamplepathc                     t          |          }t          j        |d          5 }|                    d          d d         }t	          |          }d d d            n# 1 swxY w Y    | |||          S )Nrb   r   r   r   )r   ioopenpeekr   )clsr   r   filer   shas         S/var/www/html/auto_sub_bot/venv/lib/python3.11/site-packages/huggingface_hub/lfs.py	from_pathzUploadInfo.from_pathJ   s    t}}WT4   	$DYYs^^DSD)Fd##C	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ sS8888s   -AA"%A"datac                     t          |                                          } | t          |          |d d         |          S )Nr!   )r   r   r   )r   digestlen)r&   r+   r(   s      r)   
from_byteszUploadInfo.from_bytesR   s>    Tll!!##sD		$tt*SAAAA    fileobjc                    |                     d          }|                    dt          j                   t	          |          }|                                }|                    dt          j                    | |||          S )Nr!   r   r"   )readseekr#   SEEK_SETr   tell)r&   r1   r   r(   r   s        r)   from_fileobjzUploadInfo.from_fileobjW   sp    c""Q$$$'""||~~Q$$$sS8888r0   N)__name__
__module____qualname____doc__bytes__annotations__intclassmethodstrr*   r/   r   r7    r0   r)   r   r   7   s           MMM
IIIMMM9S 9 9 9 [9 Be B B B [B 98 9 9 9 [9 9 9r0   r   upload_infostoken	repo_typerepo_idrevisionendpointheaders	transfersreturnc                    ||nt           j        }d}|t           j        v rt           j        |         }| d| | d}	d||nddgd | D             d	d
}
|dt          |          i|
d<   i t          t          |          |pi }t                                          |	||
          }t          |           |	                                }|
                    dd          }t          |t                    st          d          |
                    d          }t          |t                    r|nd}d |D             d |D             |fS )a  
    Requests the LFS batch endpoint to retrieve upload instructions

    Learn more: https://github.com/git-lfs/git-lfs/blob/main/docs/api/batch.md

    Args:
        upload_infos (`Iterable` of `UploadInfo`):
            `UploadInfo` for the files that are being uploaded, typically obtained
            from `CommitOperationAdd.upload_info`
        repo_type (`str`):
            Type of the repo to upload to: `"model"`, `"dataset"` or `"space"`.
        repo_id (`str`):
            A namespace (user or an organization) and a repo name separated
            by a `/`.
        revision (`str`, *optional*):
            The git revision to upload to.
        headers (`dict`, *optional*):
            Additional headers to include in the request
        transfers (`list`, *optional*):
            List of transfer methods to use. Defaults to ["basic", "multipart"].

    Returns:
        `LfsBatchInfo`: 3-tuple:
            - First element is the list of upload instructions from the server
            - Second element is a list of errors, if any
            - Third element is the chosen transfer adapter if provided by the server (e.g. "basic", "multipart", "xet")

    Raises:
        [`ValueError`](https://docs.python.org/3/library/exceptions.html#ValueError)
            If an argument is invalid or the server response is malformed.
        [`HfHubHTTPError`]
            If the server returned an error.
    N /z.git/info/lfs/objects/batchuploadbasic	multipartc                 P    g | ]#}|j                                         |j        d $S )oidr   )r   hexr   ).0rN   s     r)   
<listcomp>z'post_lfs_batch_info.<locals>.<listcomp>   sG     
 
 

  }((** 
 
 
r0   r   )	operationrI   objects	hash_algonameref)rC   rH   jsonrX   zMalformed response from servertransferc                 4    g | ]}d |vt          |          S error)_validate_batch_actionsrU   objs     r)   rV   z'post_lfs_batch_info.<locals>.<listcomp>   s+    OOO#G3<N<N	 	%	%<N<N<Nr0   c                 4    g | ]}d |v t          |          S r`   )_validate_batch_errorrc   s     r)   rV   z'post_lfs_batch_info.<locals>.<listcomp>   s&    III'S..	s	#	#...r0   )r   ENDPOINTREPO_TYPES_URL_PREFIXESr   LFS_HEADERSr   r   postr   r]   get
isinstancelist
ValueErrorr@   )rB   rC   rD   rE   rF   rG   rH   rI   
url_prefix	batch_urlpayloadresp
batch_inforX   chosen_transfers                  r)   post_lfs_batch_inforu   a   s   X $/xxY5GHJI5556yA
NNjN'NNNI"+"7YYg{=S
 

 '
 
 
  G  '("3"34


'
'
' =bG
 ==iwGGDJnnY--Ggt$$ ;9::: nnZ00O)3OS)I)ISootO 	POOOOIIwIII r0   c                   $    e Zd ZU eed<   eed<   dS )PayloadPartT
partNumberetagN)r8   r9   r:   r>   r=   r@   rA   r0   r)   rw   rw      s"         OOO
IIIIIr0   rw   c                   4    e Zd ZU dZeed<   ee         ed<   dS )CompletionPayloadTz?Payload that will be sent to the Hub when uploading multi-part.rS   partsN)r8   r9   r:   r;   r@   r=   rm   rw   rA   r0   r)   r{   r{      s2         II	HHHr0   r{   rW   r   lfs_batch_actionc                    t          |           |                    d          }|%t                              d| j         d           dS |d         d         }t          |           |d                             d          }|t          |           |                    di           }|                    d          }	t          |d	         |
          }
|	M	 t          |	          }	n(# t          t          f$ r t          d|	 d          w xY wt          | ||	|
           nt          | |
           |t          |           t          |d	         |          }t                                          |t          ||          | j        j                                        | j        j        d          }t'          |           t                              | j         d           dS )a9  
    Handles uploading a given object to the Hub with the LFS protocol.

    Can be a No-op if the content of the file is already present on the hub large file storage.

    Args:
        operation (`CommitOperationAdd`):
            The add operation triggering this upload.
        lfs_batch_action (`dict`):
            Upload instructions from the LFS batch endpoint for this object. See [`~utils.lfs.post_lfs_batch_info`] for
            more details.
        headers (`dict`, *optional*):
            Headers to include in the request, including authentication and user agent headers.

    Raises:
        [`ValueError`](https://docs.python.org/3/library/exceptions.html#ValueError)
            If `lfs_batch_action` is improperly formatted
        [`HfHubHTTPError`]
            If the upload resulted in an error
    actionsNzContent of file z. is already present upstream - skipping uploadrN   verifyheader
chunk_sizehref)rG   zTMalformed response from LFS batch endpoint: `chunk_size` should be an integer. Got 'z'.)rW   r   r   
upload_url)rW   r   )rC   rH   rR   r\   z: Upload successful)rb   rk   loggerdebugpath_in_repo_validate_lfs_actionr   r>   rn   	TypeError_upload_multi_part_upload_single_partr   rj   r   upload_infor   rT   r   r   )rW   r}   rC   rH   rG   r   upload_actionverify_actionr   r   r   
verify_urlverify_resps                r)   
lfs_uploadr      s   8 ,---""9--Gn	(>nnnooo %Y/9M'''$Y/33H==M ]+++ x,,FL))J'f(=QQQJ	ZJJI& 	 	 	ugquuu  	 	Yv*akllllliJGGGG  ]++++M&,A8LL
!mm(($5'BBB".599;;YEZE_`` ) 
 

 	K(((
LLI*???@@@@@s   C* *%D
lfs_actionc                     t          |                     d          t                    r=|                     d          7t          |                     d          t                    st	          d          | S ).validates response from the LFS batch endpointr   r   Nz"lfs_action is improperly formatted)rl   rk   r@   dictrn   )r   s    r)   r   r   
  se     	:>>&))3//?^^H%%-JNN8<T<TVZ1[1[-=>>>r0   lfs_batch_actionsc                    t          |                     d          t                    r(t          |                     d          t                    st	          d          |                     di                               d          }|                     di                               d          }|t          |           |t          |           | S )r   rS   r   z)lfs_batch_actions is improperly formattedr   rN   r   )rl   rk   r@   r>   rn   r   )r   r   r   s      r)   rb   rb     s    (,,U33S99 FjIZI^I^_eIfIfhk>l>l FDEEE%)))R88<<XFFM%)))R88<<XFFM ]+++ ]+++r0   lfs_batch_errorc                    t          |                     d          t                    r(t          |                     d          t                    st	          d          |                     d          }t          |t
                    rPt          |                    d          t                    r(t          |                    d          t                    st	          d          | S )r   rS   r   z'lfs_batch_error is improperly formattedra   messagecode)rl   rk   r@   r>   rn   r   )r   
error_infos     r)   rf   rf   "  s    **511377 DJGZGZ[aGbGbdg<h<h DBCCC $$W--J:t$$Dz~~i00#66D z~~f--s33D
 BCCCr0   r   c                     |                      d          5 }t          d||          }t          |           ddd           dS # 1 swxY w Y   dS )aK  
    Uploads `fileobj` as a single PUT HTTP request (basic LFS transfer protocol)

    Args:
        upload_url (`str`):
            The URL to PUT the file to.
        fileobj:
            The file-like object holding the data to upload.

    Raises:
        [`HfHubHTTPError`]
            If the upload resulted in an error.
    T	with_tqdmPUTr+   N)as_filer   r   )rW   r   r1   responses       r)   r   r   0  s     
		T		*	* &gz@@@H%%%& & & & & & & & & & & & & & & & & &s   "AA
A
r   r   c                    t          || j        |          }t          | ||          }t                                          |t          || j        j                                                  t                    }t          |           dS )z@
    Uploads file using HF multipart LFS transfer protocol.
    )r   r   r   )rW   sorted_parts_urlsr   )r]   rH   N)
_get_sorted_parts_urlsr   _upload_parts_iterativelyr   rj   _get_completion_payloadr   rT   ri   r   )rW   r   r   r   r   response_headerscompletion_ress          r)   r   r   D  s    
 /f)J_lvwww 1/@Z  
 !]]''$%5y7L7S7W7W7Y7YZZ (  N
 '''''r0   r   c                     d t          d |                                 D             d           D             }t          |          }|t          |j        |z            k    rt          d          |S )Nc                     g | ]\  }}|S rA   rA   )rU   _r   s      r)   rV   z*_get_sorted_parts_urls.<locals>.<listcomp>Z  s,     
 
 
Az 	
 
 
r0   c                     g | ]>\  }}|                                 r%t          |          d k    ,t          |d          |f?S )r   
   )isdigitr.   r>   )rU   part_numr   s      r)   rV   z*_get_sorted_parts_urls.<locals>.<listcomp>]  s\       (Hj##%% +.h--!*;*; Xr""J/*;*;*;r0   c                     | d         S )Nr   rA   )ts    r)   <lambda>z(_get_sorted_parts_urls.<locals>.<lambda>b  s
    !A$ r0   )keyz0Invalid server response to upload large LFS file)sorteditemsr.   r   r   rn   )r   r   r   sorted_part_upload_urls	num_partss        r)   r   r   Y  s    
 
# ,2LLNN  
 
 
 

 
 
 +,,ID)J67777KLLL""r0   r   rS   c                     g }t          |           D ]U\  }}|                    d          }||dk    rt          d| d|dz              |                    |dz   |d           V||dS )Nry   rL   zInvalid etag (`z`) returned for part r   )rx   ry   )rS   r|   )	enumeraterk   rn   append)r   rS   r|   part_numberr   ry   s         r)   r   r   k  s     "E()9:: 	
 	
Vzz&!!<42::[t[[+XY/[[\\\)Ao 	
 	
 	
 	
 '''r0   r   c                 b   g }|                      d          5 }t          |          D ]m\  }}t          |||z  |          5 }t          d||          }t	          |           |                    |j                   d d d            n# 1 swxY w Y   n	 d d d            n# 1 swxY w Y   |S )NTr   )	seek_from
read_limitr   r   )r   r   r   r   r   r   rH   )	rW   r   r   rH   r1   part_idxpart_upload_urlfileobj_slicepart_upload_ress	            r)   r   r   z  sK    G			T		*	* 
8g)23D)E)E 		8 		8%Ho$x/%   8 ".uoM"Z"Z"Z#O44467778 8 8 8 8 8 8 8 8 8 8 8 8 8 8		8
8 
8 
8 
8 
8 
8 
8 
8 
8 
8 
8 
8 
8 
8 
8 Ns5   *B$<B?B$BB$BB$$B(+B()NNNN)NNN);r;   r#   redataclassesr   mathr   os.pathr   typingr   r   r   r	   r
   urllib.parser   huggingface_hubr   utilsr   r   r   r   r   r   r   
utils._lfsr   	utils.shar   r   _commit_apir   
get_loggerr8   r   compile	OID_REGEXLFS_MULTIPART_UPLOAD_COMMANDri   r   r@   r   rm   tupleru   rw   r{   r   r   rb   rf   r   r>   r   r   r   r   rA   r0   r)   <module>r      s   5 4 				 				 ! ! ! ! ! !             I I I I I I I I I I I I I I             % % % % % %                  % $ $ $ $ $ * * * * * * * *  0//////		H	%	%BJ())	5  -2  &9 &9 &9 &9 &9 &9 &9 &9R  #"(,%)S S:&SC=S S 	S
 smS smS d38n%S S	"S 4:tDz8C=01S S S Sl    9   
         (,"CA CA#CACA C=CA d38n%	CA
 smCA 
CA CA CA CALT    t    4    &#7 &S &T & & & &(("6 ( (RU (cf (ko ( ( ( (*#4 #j #c #VZ[^V_ # # # #$(d4j (s (GY ( ( ( (#8<S	OR	$Z     r0   