ij7(ddlmZddlZddlmZddlZddlZddlZ ddl Z ddl Z ddl m Z ddl mZddl mZddl m Z ddlmZGd d Zd+dZ d,d-d#Zd.d$Z d/d0d)Z d1d2d*ZdS)3) annotationsN)GoogleDriveFileToDownload) _get_session)_sanitize_filename)download) DownloadErrorc&eZdZdZ ddd Zdd ZdS)_GoogleDriveFilez"application/vnd.google-apps.folderNidstrnametypechildrenlist[_GoogleDriveFile] | NonereturnNonecF||_||_||_||ng|_dSN)r rrr)selfr rrrs l/home/spazio/transcriptions/margarita_drive_19vsq/venv/lib/python3.11/site-packages/gdown/download_folder.py__init__z_GoogleDriveFile.__init__s-  $,$8b boolc"|j|jkSr)r TYPE_FOLDER)rs r is_folderz_GoogleDriveFile.is_folder"syD,,,rr) r r rr rr rrrr)rr)__name__ __module__ __qualname__rrrrrr r sP6K37 A A A A A------rr gdrive_file previous_pathr rlist[tuple[str | None, str]]cg}|jD]}t|j|_|rp|dt j||jft|t j||jD]}|||js4||jt j||jf|S)zEConverts a Google Drive folder structure into a local directory list.filenameN) rrrrappendospjoin_get_directory_structurer )r"r#directory_structurefileis rr+r+&s $VV& ::: >>   V  & &ch}di.P.P'Q R R R-dCH]DI4V4VWW . .#**1---- . V  & &-1S1S'T U U U rFTurl str | Noner outputquietrproxyspeed float | None use_cookiesverify bool | str user_agent skip_downloadresume+list[str] | list[GoogleDriveFileToDownload]c |du|duz std||} n|Jt|} |d}t|||\} } |stdtjt | | ||}t|j|_|s6td tjtd tjt|d }|std tj| tj tj z}|tj rtj||j}n|}| s(tj|stj|g}|D]\}}tj||}|+| s(tj|stj|G| r&|t'|||otj|dr|}n!tj|tj z}t-d|z||||||| }|||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)r3r6r9zRetrieving folder contents)r-sess folder_idr2r7r&z$Retrieving folder contents completedzBuilding directory structure)r#z&Building directory structure completed)r path local_pathrzhttps://drive.google.com/uc?id=)r/r1r2r3r4r6r7r;zDownload completed) ValueError_extract_folder_idrprintsysstderr%_download_and_parse_google_drive_linkrrr+osgetcwdr)sependswithr*existsmakedirsr(rsplitextdirnamer)r/r r1r2r3r4r6r7r9r:r;r@r?_r"r,root_dirfilesrBrCdownload_outputs rdownload_folderrV7sJ $J3$; 'A?@@@ ~ &s++ Q KJWWWGD! = *<<<<7  K*;3CDDDK ? 43:FFFF ,3:>>>>2;bQQQ I 6SZHHHH ~sw& sw8FK$455 H!5!5 H E'%%DXh-- :  (J)?)? ( J'''   % LL)RdzRRR    |J''* D","%+j"9"9CG"C!5:&'   J LL $ $ $ $ 5 "4444 Lrctj|jdddS)N/)urllibparseurlparserBrstripsplit)r/s rrErEs< <  % % * 1 1# 6 6 < > >D zTZ.6 3i 3 3 3   *#K+-HC((""yy$$$$$  XJ      &&q))GT22I OOWi1KL M M M XD      &&q))GT22I OOWi1KL M M M xL     *0033Od33J OO_j:J:VW X X X     ""rct|||\}}t||tj}|D]\}}} | tjkr>|std|||jt||| T|std||t ||||} |j| |S)N)r?r@r7)r rrzProcessing filezRetrieving folderr>)rr rrFrr(rI) r?r@r2r7ryrr"child_idr child_typechilds rrIrIs4 8 YvK#    )K -5++(*j )5 5 5 %  ' ' ##      #    6     ##E**** r)r"r r#r rr$) NNNFNNTTNFF)r/r0r r0r1r0r2rr3r0r4r5r6rr7r8r9r0r:rr;rrr<)r/r rr )T)r?r_r@r r7r8rr`)FT) r?r_r@r r2rr7r8rr ) __future__rrJos.pathrBr)rrrG urllib.parserZrlrequestsrrrr exceptionsr r r+rVrErrIr!rrrs""""""   //////""""""((((((%%%%%%--------&$!OOOOOdFFFF?#?#?#?#?#J .......r