FidZddlmZddlZddlmZddlmZddlm Z m Z GddZ d d d Z y) zBUpload utilities for Ultralytics, mirroring downloads.py patterns.) annotationsN)Path)sleep)LOGGERTQDMc*eZdZdZdZddZdZdZy)_ProgressReaderz>File wrapper that reports read progress for upload monitoring.c|t|d|_||_tjj ||_y)Nrb)openfilepbarospathgetsize_size)self file_pathrs [/var/www/pod-logistic/pod-ai/venv/lib/python3.12/site-packages/ultralytics/utils/uploads.py__init__z_ProgressReader.__init__s+D)  WW__Y/ c|jj|}|r0|jr$|jjt ||S)z"Read data and update progress bar.)r readrupdatelen)rsizedatas rrz_ProgressReader.reads8yy~~d# DII II  SY ' rc|jS)z+Return file size for Content-Length header.)rrs r__len__z_ProgressReader.__len__s zzrc8|jjy)zClose the file.N)r closers rr"z_ProgressReader.close s rN))__name__ __module__ __qualname____doc__rrr r"rrr r sH0 rr c ddl}t|}|jstd||j j }d|j }ddi} |r| j|d} t|dzD]} d} d} |rt||dd d } t|| } |j|| | | }|j| jd} | r| jd} tjd |j d | r| j| r| jy y tj(|d|dzd| y#|j j"$r}|j$|j$j&nd}d|cxkrdkrennb|dvr^tj(|d|dt+|j$ddYd}~| r| j| r| jyyd|} Yd}~n$d}~wt,$r}t/|} Yd}~nd}~wwxYw| r| j| r;| jn*#| r| j| r| jwwxYw| |ksd| dzz}tj(|d| d| dzd|d|d t1|L) 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)rheaderstimeoutz Uploaded u ✅ii>z failed:  reasonFzHTTP z failed (z ), retrying /z in zs...z failed after z attempts: )requestsrexistsFileNotFoundErrorstatst_sizenamerrangerr putraise_for_statusr"rinfo exceptions HTTPErrorresponse status_codewarninggetattr Exceptionstrr)r urlr1retryr2progressr: file_sizer-upload_headers last_errorattemptrreaderrestatus wait_times r safe_uploadrX%s2 :D ;;="24& 9:: ##I  { #D%&@ANg&J## )$ST`de$T40F Sv~w WA   LLNF  KK)DII;d3 4  ;#J NNdV>%!)K |LM )"",, */0zz/EQZZ++1Ff"s"vZ'?$y'!**hXZ:[9\]^    !)J QJ      U?gk*I NNdV9ZL Wq[MQRSXRYY]^g]hhlm n ) sJB EH56A(HI HI H5 H0+I0H55I'J)Nr8iXF)r z str | PathrLrKr1z dict | NonerMintr2rYrNboolreturnrZ) r' __future__rrpathlibrtimerultralytics.utilsrrr rXr(rrr`s~H" *6  N N NN  N  N  N Nr