hj*ddlmZddlmZddlmZdZdZdZGddZ Gd d e Z Gd d e Z gd Z y))Path) HTTPError) urlretrievecddl}d}|j}|j||dddd}t|}|r||d}|j||d}t ||y) Nrzhttps://docs.google.com/uctdownload)idconfirmexportT)paramsstream)r r )requestsSessiongetget_confirm_tokensave_response_content)r destinationrURLsessionresponsetokenr s G/DATA/.local/lib/python3.12/site-packages/trame_common/assets/remote.pydownload_file_from_google_driversx &C G{{   H h 'E u-;;s6$;?(K0cr|jjD]\}}|jds|cSy)Ndownload_warning)cookiesitems startswith)rkeyvalues rrrs6&&,,. U >>, -L/ rcd}t|jd5}|j|D]}|s|j| dddy#1swYyxYw)Niwb)ropen iter_contentwrite)rr CHUNK_SIZEfchunks rrr&sKJ k    %**:6E7 & % %sAAAc>eZdZdZddZedZdZedZy)AbstractRemoteFilez AbstractRemoteFile provide infrastructure for RemoteFile where only the method fetch() needs to be defined for a concreate implementation. Nctjj|_|Qt|j}||_|j r!|j r|j |_t||_|jjs|j|z |_|jj }|jddy)NT)parentsexist_ok) rcwdresolve_baseexistsis_fileparent _file_path is_absolutemkdir)self local_path local_base parent_dirs r__init__zAbstractRemoteFile.__init__5sXXZ'')  !j)113J#DJ  "z'9'9';'.. z***,"jj:5DO__++ 5rc6|jjS)z?Return true if the file is available locally on the File System)r5r2r8s rlocalzAbstractRemoteFile.localGs%%''rcy)zCPerform the action needed to fetch the content and store it locallyNr>s rfetchzAbstractRemoteFile.fetchLsrcd|js|jt|jS)z!Return the actual local file path)r?rBstrr5r>s rpathzAbstractRemoteFile.pathOs#zz JJL4??##r)NN) __name__ __module__ __qualname____doc__r<propertyr?rBrErArrr+r+/s: 6$((R$$rr+c0eZdZdZ dfd ZdZxZS)GoogleDriveFilezX Helper file to manage caching and retrieving of file available on Google Drive c4t|||||_y)a+ Provide the information regarding where the file should be located and where to fetch it if missing. :param local_path: relative or absolute path :param google_id: Resource ID from google :param local_base: Absolute path when local_path is relative N)superr<_gid)r8r9 google_idr: __class__s rr<zGoogleDriveFile.__init__]s Z0 rc  td|jd|jt|j|jy#t$r8}tt d|jd|j Yd}~yd}~wwxYwNzDownloading: - z - to zFailed to download z. )printrOr5rr RuntimeErrorreasonr8es rrBzGoogleDriveFile.fetchnsq O %dii[8IJ K +DIIt G O ,!4TYYKr!((LM N N OsAA B .BB NNNrFrGrHrIr<rB __classcell__rQs@rrLrLXs  "OrrLc0eZdZdZ dfd ZdZxZS)HttpFilezX Helper file to manage caching and retrieving of file available on HTTP servers c4t|||||_y)a: Provide the information regarding where the file should be located and where to fetch it if missing. :param local_path: relative or absolute path :param remote_url: http(s):// url to fetch the file from :param local_base: Absolute path when local_path is relative N)rNr<_url)r8r9 remote_urlr:rQs rr<zHttpFile.__init__{s Z0 rc * td|jd|jt|jt |jy#t $r8}tt d|jd|jYd}~yd}~wwxYwrS)rTr`r5rrDrrUrVrWs rrBzHttpFile.fetchsv O %dii[8IJ K  3t#7 8 O ,!4TYYKr!((LM N N OsAA B.B  BrYrZr\s@rr^r^vs  "Orr^)r+rLr^N) pathlibr urllib.errorrurllib.requestrrrrr+rLr^__all__rArrrgsM"&10&$&$RO(O<O!O< r