0IjM+4ddlZddlZddlZddlZddlmZddlZddlZddl Z ddl m Z ddl m Z ddl Z ddlmZddlmZddlmZddlmZd ZGd d eZd Z dd ZdZej4ddZ ddede e ee edffdZy)N)List)Union) _get_session)download)FolderContentsMaximumLimitError)is_google_drive_url2ceZdZdZddZdZy)_GoogleDriveFilez"application/vnd.google-apps.folderNcP||_||_||_|||_yg|_yN)idnametypechildren)selfrrrrs X/root/repos/pocs/video-evals/.venv/lib/python3.12/site-packages/gdown/download_folder.py__init__z_GoogleDriveFile.__init__s)  $,$8 b c4|j|jk(Sr)r TYPE_FOLDER)rs r is_folderz_GoogleDriveFile.is_foldersyyD,,,,rr)__name__ __module__ __qualname__rrrrrr r s6KA -rr cftj|d}d}|jdD]l}|j}d|vst j dj |} ttj|ddjd}n| td tj5tjd t |j#d j%d }dddt'j(}|dgn|d} d} |j*j,dj/| } t1| dk\r| j3| dd} n1tdj5|j*j,dt7|j/dd| t6j8} | Dcgc].}|d|dj#dj%d |df0}}| |fS#t$r tdwxYw#1swY,xYwcc}w)zBExtracts information about the current page file and its children.z html.parser)featuresNscript _DRIVE_ivdz'((?:[^'\\]|\\.)*)'rz*Couldn't find the folder encoded JS stringzCannot retrieve the folder information from the link. You may need to change the permission to 'Anyone with the link', or have had many accesses. Check FAQ in https://github.com/wkentaro/gdown?tab=readme-ov-file#faq.ignore)categoryzutf-8unicode_escaperz - z-file/folder name cannot be extracted from: {}/rrrraw_unicode_escape)bs4 BeautifulSoupselectdecode_contentsrecompilefinditernext itertoolsislicegroup StopIteration RuntimeErrorwarningscatch_warningsfilterwarningsDeprecationWarningencodedecodejsonloadstitlecontentssplitlenjoinformatr r)urlcontent folder_soup encoded_datar inner_html regex_iterdecoded folder_arrfolder_contentssepsplittedr gdrive_fileeid_name_type_iters r_parse_google_drive_filerT#sA##GmDKL$$X. ++- : %$:;DDZPJ Q#I$4$4ZD$IJPPQRS    U    "H3EF%%g.556FGHG$J&qM1bz!}O C  ))!,2237H 8}xx" & ; B B!!**1-   # 99S>"    ) )K!  1qt{{/077@!A$G ) ))U! Q"#OPP QHH0s(/H ; &"K" ->!+(*j )55 5%  ' ' ## "D((  #  C9HD%   U $ $##E*C!+D!!5!56:JJ 2((3::s:C,338H3I=  .g66  ##rcg}|jD]}|jjtjd|_|j rr|j dtj||jft|tj||jD]}|j ||jr|j |jtj||jf|S)zEConverts a Google Drive folder structure into a local directory list.rgN) rrreplaceosprOrrcrD_get_directory_structurer)rQ previous_pathdirectory_structurefileis rrrrrs$$VII%%cggs3 >>   & &chh}dii.P'Q R-dCHH]DII4VW .#**1- .  & &-1S'T UV rGoogleDriveFileToDownloadrpath local_path skip_downloadreturnc Z|du|duz s td|dj|}| d} t||| } |stdtj t | |||| \} }| std tj y|s6td tj td tj t|d }|stdtj |%tjtjz}|jtjr!tj||j}n|}| s*tj|stj |g}|D]\}}tj||}|-| s*tj|stj |L| r|j#t%|||l| rQtj&j)|r2|std|tj |j#|t+d|z||||||| }||stdtj y|j#| |stdtj |S)aDownloads entire folder from URL. Parameters ---------- url: str URL of the Google Drive folder. Must be of the format 'https://drive.google.com/drive/folders/{url}'. id: str Google Drive's folder ID. output: str, optional String containing the path of the output folder. Defaults to current working directory. quiet: bool, optional Suppress terminal output. proxy: str, optional Proxy. speed: float, optional Download byte size per second (e.g., 256KB/s = 256 * 1024). use_cookies: bool, optional Flag to use cookies. Default is True. verify: bool or string Either a bool, in which case it controls whether the server's TLS certificate is verified, or a string, in which case it must be a path to a CA bundle to use. Default is True. user_agent: str, optional User-agent to use in the HTTP request. skip_download: bool, optional If True, return the list of files to download without downloading them. Defaults to False. resume: bool Resume interrupted transfers. Completed output files will be skipped. Partial tempfiles will be reused, if the transfer is incomplete. Default is False. Returns ------- files: List[str] or List[GoogleDriveFileToDownload] or None If dry_run is False, list of local file paths downloaded or None if failed. If dry_run is True, list of GoogleDriveFileToDownload that contains id, path, and local_path. Example ------- gdown.download_folder( "https://drive.google.com/drive/folders/" + "1ZXEhzbLRLU1giKKRJkjm8N04cO_JoYE2", ) Nz$Either url or id has to be specifiedz+https://drive.google.com/drive/folders/{id})rzyMozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/98.0.4758.102 Safari/537.36)proxy use_cookies user_agentzRetrieving folder contents)ru)rZr[rWz"Failed to retrieve folder contentsz$Retrieving folder contents completedzBuilding directory structure)rsz&Building directory structure completedrxz!Skipping already downloaded file zhttps://drive.google.com/uc?id=)rFoutputrZr~speedrrWresumezDownload ended unsuccessfullyzDownload completed) ValueErrorrErrbsysstderrrdrrosgetcwdrqrOendswithrDrexistsmakedirsrcrwryisfiler)rFrrrZr~rrr[rWrr{rrY is_successrQrtroot_dirfilesryrzs rdownload_folderrsK~ $J3$; '?@@ ~;BBbBIQ e TD  *<C  ! J   2D  43::F ,3::>2;bQ  6SZZH ~sww& sww88FK$4$45 H!5 H E'$%DXXh- : J)? J'   LL)RdzR "''..4;J<H ZZ Z(!5:!' J!9 K LL $I$%J  "4 Lr)FFT) NNNFNNTFTNFF) collectionsr3r>ros.pathryrqr/rr8typingrrr+rr exceptionsr parse_urlr reobjectr rTrdrr namedtuplerwboolstrrrrrrs   "7* -v -:*@   Q$h 3K22!=       NN 49d45t ;<Nr