'j*ddlmZddlmZddlmZdZdZdZGddZ Gd d e Z Gd d e Z gd Z dS))Path) HTTPError) urlretrievecddl}d}|}|||dddd}t|}|r||d}|||d}t ||dS) Nrzhttps://docs.google.com/uctdownload)idconfirmexportT)paramsstream)r r )requestsSessiongetget_confirm_tokensave_response_content)r destinationrURLsessionresponsetokenr s T/DATA/AppData/hermes/venv/lib/python3.11/site-packages/trame_common/assets/remote.pydownload_file_from_google_driversOOO &C  G{{    H h ' 'E @u--;;s6$;??(K00000cv|jD]\}}|dr|cSdS)Ndownload_warning)cookiesitems startswith)rkeyvalues rrrsN&,,.. U >>, - - LLL  4rcd}t|d5}||D]}|r|| ddddS#1swxYwYdS)Niwb)ropen iter_contentwrite)rr CHUNK_SIZEfchunks rrr&sJ k     % %**:66  E  s0A##A'*A'cLeZdZdZddZedZdZedZdS)AbstractRemoteFilez AbstractRemoteFile provide infrastructure for RemoteFile where only the method fetch() needs to be defined for a concreate implementation. Nctj|_|\t|}||_|r |r |j|_t||_|js|j|z |_|jj}| dddS)NT)parentsexist_ok) rcwdresolve_baseexistsis_fileparent _file_path is_absolutemkdir)self local_path local_base parent_dirs r__init__zAbstractRemoteFile.__init__5sXZZ''))  !j))1133J#DJ  "" /z'9'9';'; /'. z****,, 6"j:5DO_+ 55555rc4|jS)z?Return true if the file is available locally on the File System)r5r2r8s rlocalzAbstractRemoteFile.localGs%%'''rcdS)zCPerform the action needed to fetch the content and store it locallyNr>s rfetchzAbstractRemoteFile.fetchLsrc`|js|t|jS)z!Return the actual local file path)r?rBstrr5r>s rpathzAbstractRemoteFile.pathOs+z  JJLLL4?###r)NN) __name__ __module__ __qualname____doc__r<propertyr?rBrErArrr+r+/sz 6666$((X(RRR$$X$$$rr+c0eZdZdZ dfd ZdZxZS)GoogleDriveFilezX Helper file to manage caching and retrieving of file available on Google Drive NcZt||||_dS)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* Z000 rc  td|jd|jt|j|jdS#t$r7}tt d|jd|jYd}~dSd}~wwxYwNzDownloading: - z - to zFailed to download z. )printrOr5rr RuntimeErrorreasonr8es rrBzGoogleDriveFile.fetchns O JdiJJJJ K K K +DIt G G G G G O O O ,LTYLL!(LLMM N N N N N N N N N Os9= A>,A99A>NNNrFrGrHrIr<rB __classcell__rQs@rrLrLXsh  "OOOOOOOrrLc0eZdZdZ dfd ZdZxZS)HttpFilezX Helper file to manage caching and retrieving of file available on HTTP servers NcZt||||_dS)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* Z000 rc  td|jd|jt|jt |jdS#t $r7}tt d|jd|jYd}~dSd}~wwxYwrS)rTr`r5rrDrrUrVrWs rrBzHttpFile.fetchs O JdiJJJJ K K K  3t#7#7 8 8 8 8 8 O O O ,LTYLL!(LLMM N N N N N N N N N OsAA B ,BB rYrZr\s@rr^r^vsh  "OOOOOOOrr^)r+rLr^N) pathlibr urllib.errorrurllib.requestrrrrr+rLr^__all__rArrrgs """"""&&&&&&1110&$&$&$&$&$&$&$&$ROOOOO(OOO<OOOOO!OOO<   r