jrdZddlmZddlZddlmZddlmZddlm Z m Z GddZ dddZ dS)zBUpload utilities for Ultralytics, mirroring downloads.py patterns.) annotationsN)Path)sleep)LOGGERTQDMc,eZdZdZdZddZdZdZdS) _ProgressReaderz>File wrapper that reports read progress for upload monitoring.ct|d|_||_tj||_dS)Nrb)openfilepbarospathgetsize_size)self file_pathrs ^/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/ultralytics/utils/uploads.py__init__z_ProgressReader.__init__s3D))  W__Y// c|j|}|r.|jr'|jt ||S)z"Read data and update progress bar.)r readrupdatelen)rsizedatas rrz_ProgressReader.readsGy~~d##  (DI ( I  SYY ' ' ' rc|jS)z+Return file size for Content-Length header.)rrs r__len__z_ProgressReader.__len__s zrc8|jdS)zClose the file.N)r closer s rr#z_ProgressReader.close s rN)r)__name__ __module__ __qualname____doc__rrr!r#rrr r s[HH000 rr XFr str | Pathurlstrheaders dict | Noneretryinttimeoutprogressboolreturnc ddl}t|}|std||j}d|j}ddi} |r| |d} t|dzD]L} d} d} |rt||dd d } t|| } | || | | }| | d} | r| d} tjd |jd | r| | r| d Sd S#|jj$r}|j |jjnd}d|cxkrdkrmnnj|dvrftj|d|dt+|jddYd}~| r| | r| dSdSd|} Yd}~n)d}~wt,$r}t/|} Yd}~nd}~wwxYw| r| | r| n2#| r| | r| wwxYw| |kr=d| dzz}tj|d| d| dzd|d|d t1|Ntj|d|dzd| dS) aUpload a file to a URL with retry logic and optional progress bar. Args: file (str | Path): Path to the file to upload. url (str): The URL endpoint to upload the file to (e.g., signed GCS URL). headers (dict, optional): Additional headers to include in the request. retry (int, optional): Number of retry attempts on failure (default: 2 for 3 total attempts). timeout (int, optional): Request timeout in seconds. progress (bool, optional): Whether to display a progress bar during upload. Returns: (bool): True if upload succeeded, False otherwise. Examples: >>> from ultralytics.utils.uploads import safe_upload >>> success = safe_upload("model.pt", "https://storage.googleapis.com/...", progress=True) rNzFile not found: z Uploading z Content-Typezapplication/octet-streamBTi)totaldescunit unit_scale unit_divisor)rr.r2z Uploaded u ✅ii>z failed:  reasonFzHTTP r)z failed (z ), retrying /z in zs...z failed after z attempts: )requestsrexistsFileNotFoundErrorstatst_sizenamerrangerr putraise_for_statusr#rinfo exceptions HTTPErrorresponse status_codewarninggetattr Exceptionr-r)r r,r.r0r2r3rD file_sizer:upload_headers last_errorattemptrreaderrestatus wait_times r safe_uploadr^%s2OOO ::D ;;==; 94 9 9::: #I # # #D%&@AN'g&&&J####  f)$ST`deee$T400F Sv~w WWA   LLNNNF   K3DI333 4 4 4       ", * * */0z/EQZ++1Ff""""s"""""vZ'?'?$]]]]'!*hXZ:[:[]]^^^uuu         *))JJJJJJ   QJJJJJJ           U??gk*I NdmmZmmWq[mmSXmm^gmmm n n n )    NdLL%!)LL LLMMM 5sJBE%%H34AH  I#H I# H3H.)I#.H33I##/J)Nr)r*F)r r+r,r-r.r/r0r1r2r1r3r4r5r4) r' __future__rrpathlibrtimerultralytics.utilsrrr r^r(rrrcsHH"""""" ********6  NNNNNNNr