yi(bddlmZddlZddlZddlmZddlZddlZddl Z ddl Z ddl Z ddl mZddl mZddl m Z ddlmZGdd Z dd Zej(d d Z d dd ZddZ d ddZ d ddZy)) annotationsN) _get_session)_sanitize_filename)download) DownloadErrorc6eZdZdZ d ddZddZy)_GoogleDriveFilez"application/vnd.google-apps.folderNcP||_||_||_|||_yg|_yN)idnametypechildren)selfr rrrs I/home/ubuntu/.local/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_folder"syyD,,,,rr ) r strrrrrrzlist[_GoogleDriveFile] | NonereturnNone)rbool)__name__ __module__ __qualname__rrrrrr r sL6K37 A  A A A 0 A  A-rr cg}|jD]}t|j|_|jrr|j dt j ||jft|t j ||jD]}|j ||jr|j |jt j ||jf|S)zEConverts a Google Drive folder structure into a local directory list.filenameN) rrrrappendospjoin_get_directory_structurer ) gdrive_file previous_pathdirectory_structurefileis rr&r&&s $$V& : >>   & &chh}dii.P'Q R-dCHH]DII4VW .#**1- .  & &-1S'T UV rGoogleDriveFileToDownloadr path local_pathc |du|duz s td||} n|Jt|} |d}t|||\} } |stdtj t | | ||}t|j|_|s6td tj td tj t|d }|std tj |%tjtjz}|jtjr!tj||j}n|}| s*tj |stj"|g}|D]\}}tj||}|-| s*tj |stj"|K| r|j%t'|||ktj(|dr|}n&tj*|tjz}t-d|z||||||| }|j%||stdtj |S)aDownloads entire folder from URL. Parameters ---------- url: URL of the Google Drive folder. Must be of the format 'https://drive.google.com/drive/folders/{url}'. id: Google Drive's folder ID. output: String containing the path of the output folder. Defaults to current working directory. quiet: Suppress terminal output. proxy: Proxy. speed: Download byte size per second (e.g., 256KB/s = 256 * 1024). use_cookies: Flag to use cookies. Default is True. verify: 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: User-agent to use in the HTTP request. skip_download: If True, return the list of files to download without downloading them. Defaults to False. resume: Resume interrupted transfers. Completed output files will be skipped. Partial tempfiles will be reused, if the transfer is incomplete. Default is False. Returns ------- files: If skip_download is False, list of local file paths downloaded. If skip_download is True, list of GoogleDriveFileToDownload that contains id, path, and local_path. Raises ------ ValueError If neither url nor id is specified, or both are specified. DownloadError If a file in the folder fails to download. Example ------- gdown.download_folder( "https://drive.google.com/drive/folders/" + "1ZXEhzbLRLU1giKKRJkjm8N04cO_JoYE2", ) Nz$Either url or id has to be specifiedzyMozilla/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)r*sess folder_idquietverifyr!z$Retrieving folder contents completedzBuilding directory structure)r(z&Building directory structure completedr-rzhttps://drive.google.com/uc?id=)urloutputr7r1speedr2r8resumezDownload completed) ValueError_extract_folder_idrprintsysstderr%_download_and_parse_google_drive_linkrrr&osgetcwdr$sependswithr%existsmakedirsr#r,splitextdirnamer)r:r r;r7r1r<r2r8r3 skip_downloadr=r6r5_r'r)root_dirfilesr.r/download_outputs rdownload_folderrQ<sJ $J3$; '?@@ ~ &s+ Q KJWGD!  *<7  K*;3C3CDK  43::F ,3::>2;bQ  6SZZH ~sww& sww88FK$4$45 H!5 H E'%DXXh- : J)? J'   LL)RdzR ||J'*","%++j"9CGG"C!5:&' J LL $=%>  "4 Lrctjj|jj dj ddS)N/)urllibparseurlparser.rstripsplit)r:s rr?r?s8 <<  % * * 1 1# 6 < D zzTZZ..6=i[I3 3  **##K+-HC("yy$$$ XXJ   &&q)GT2I OOWi1KL M XXD   &&q)GT2I OOWi1KL M xxL  *003Od3J OO_j:J:V:VW X E"H  ""rc|t|||\}}t||tj}|D]\}}} | tjk7r7|s td|||jj t||| Q|s td||t ||||} |jj | |S)N)r5r6r8)r rrzProcessing filezRetrieving folderr4)r{r rr@rr#rC) r5r6r7r8rrrr'child_idrz child_typechilds rrCrCs 8 YvK#    ) )K -5+(*j )55 5%  ' ' ##   #  6   ##E*;+< r)r'r r(rrzlist[tuple[str | None, str]]) NNNFNNTTNFF)r: str | Noner rr;rr7rr1rr<z float | Noner2rr8 bool | strr3rrLrr=rrz+list[str] | list[GoogleDriveFileToDownload])r:rrr)T)r5requests.Sessionr6rr8rrz&tuple[str, list[tuple[str, str, str]]])FT) r5rr6rr7rr8rrr ) __future__r collectionsrDos.pathr.r$rkrA urllib.parserUrerequestsrrr exceptionsrr r& namedtupler,rQr?r{rCrrrrs"   "(%--&!25!"3K22!= !O OO O  O  O  OO OOO O1OdF?# ?#?# ?#, ?#J . .. .  .  .r