a d- @sddlmZddlZddlZddlmZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlZddlZddlmZdZedZddZd d Zdd d ZdddZdS))print_functionN) parse_urli~csfdd}d|S)Nc3s*dD]}|r|n|Vq dS)NT) splitlinesstrip)lineprefixtextW/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/gdown/download.pyprefixed_linesszindent..prefixed_lines)join)r r rr r r indentsrcCsd}|D]}td|}|r@d|d}|dd}qtd|}|rl|d}|dd}qtd|}|r|d}|d d }|d d}qtd |}|r |d}t|q |std |S)Nrz"href="(\/uc\?export=download[^"]+)zhttps://docs.google.comrz&&z!id="download-form" action="(.+?)"z"downloadUrl":"([^"]+)z\u003d=z\u0026z'

(.*)

zCannot retrieve the public link of the file. You may need to change the permission to 'Anyone with the link', or have had many accesses.)rresearchgroupsreplace RuntimeError)contentsurlrmerrorr r r get_url_from_gdrive_confirmation s6             rFc Cst}|jddi|dur<||d|_td|tjdt t d}t |r|rt |}t |}Wdn1s~0Y|D]\}}||j|<q|r||fS|SdS)Nz User-Agentz/Mozilla/5.0 (Macintosh; Intel Mac OS X 10_12_6))httphttpsz Using proxy:filez.cache/gdown/cookies.json)requestssessionheadersupdateproxiesprintsysstderrosprhomeexistsopenjsonloadcookies) proxy use_cookiesreturn_cookies_filesess cookies_filefr0kvr r r _get_session@s    (  r9Tc # Cs|du|duAstd|dur,dj|d}|} t||dd\} } t|| d\}}|rp|rpdj|d}|} d}z| j|d|d}WnJtjjy}z.td | j t j d t|t j d WYd}~dSd}~00|| kr|j d krd j|d}qp|j d drtd|j}|rJ|ddrJdj|| dur>dn| d}qpnr|r|ddrdj|| durxdn| d}qpn8|r|ddrdj|| durdn| d}qpnFd|j vr|j ddr| dvrdj|| durdn| d}qp|r|tt| s,tt| t| d4}dd| jD}tj||d d!Wdn1sr0Yd|j vrq.|r.|sq.zt|j}Wqpty*}zftd"d# t!"t#|}t$|d$}td#|d#t j d td%t j d td&| d#t j d WYd}~dSd}~00qp|rv|rvt%j&j'(|j d}td'|}|d}|)tj*d(}n t+|}|dur|}t,|t%j-}|r|tj*rt|st|t ||}|rg}t.t|pd)D]0}|t+|r|/t t||q| r|rt0|d*krtd+t j d td#|D]}td$|t j d qVtd#td,t j d dS|d}n"d-} t1j2t1j3t+|t|d.}t|d/}nd}|}|dur|4dkrd0d1|4i}| j||d|d2}|std3t j d | r.td4|t j d | |krZtd5| t j d td6|t j d ntd7|t j d td8|r~t5|n|t j d z$z|j d9}|durt6|}|st7j7|d:dd;}t88}|j9t:d<D]`} |;| |s||!|"q|sH|?|rb|?t@A||Wn@tBy}z&t|t j d WYd}~W| ?dSd}~00W| ?n | ?0|S)>aDownload file from URL. Parameters ---------- url: str URL. Google Drive URL is also supported. output: str Output filename. Default is basename of URL. quiet: bool Suppress terminal output. Default is False. proxy: str Proxy. speed: float Download byte size per second (e.g., 256KB/s = 256 * 1024). use_cookies: bool 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. id: str Google Drive's file ID. fuzzy: bool Fuzzy extraction of Google Drive's file Id. Default is False. resume: bool Resume the download from existing tmp file if possible. Default is False. format: str, optional Format of Google Docs, Spreadsheets and Slides. Default is: - Google Docs: 'docx' - Google Spreadsheet: 'xlsx' - Google Slides: 'pptx' Returns ------- output: str Output filename. Nz$Either url or id has to be specifiedz#https://drive.google.com/uc?id={id})idT)r1r2r3)warning)streamverifyz"An error has occurred using proxy:r iz%https://drive.google.com/open?id={id}z Content-Typez text/htmlz(.+)rz - Google Docsz>https://docs.google.com/document/d/{id}/export?format={format}Zdocx)r:formatz - Google SheetszBhttps://docs.google.com/spreadsheets/d/{id}/export?format={format}Zxlsxz - Google SlideszBhttps://docs.google.com/presentation/d/{id}/export?format={format}pptxzContent-Disposition>Nr?wcSs"g|]\}}|ds||fqS)Zdownload_warning_) startswith).0r7r8r r r s zdownload..)rz'Access denied with the following error:  z:You may still be able to access the file from the browser:z zfilename\*=UTF-8''(.*)_.rz-There are multiple temporary files to resume:z4Please remove them except one to resume downloading.F)suffixr dirabRangez bytes={}-)r$r<r=zDownloading...zResume:zFrom (uriginal):zFrom (redirected):zFrom:zTo:zContent-LengthB)totalunitZ unit_scale) chunk_sizeg?)C ValueErrorr>r9rgetr" exceptions ProxyErrorr'r&r(r) status_coder$rArrr rendswithr*r,dirnameosmakedirsr-r0itemsr.dumprrrtextwrapwrapstrrsixmoves urllib_parseunquotersepbasename isinstance string_typeslistdirappendlentempfilemktemptemplatetellabspathinttqdmtime iter_content CHUNK_SIZEwriter%nsleepcloseshutilmoveIOError)#routputquietr1speedr2r=r:Zfuzzyresumer>Z url_originr4r5Zgdrive_file_idZis_gdrive_download_linkreserr6r0rcontent_dispositionZfilename_from_urlZoutput_is_pathZexisting_tmp_filesr!Ztmp_filer$rNZpbarZt_startchunkZelapsed_time_expectedZ elapsed_timer r r downloadYsb3    0                       r)F) NNFNNTTNFFN) __future__rr.rXos.pathpathr*rrxr(rjr\rqr"r_rprrs expanduserr+rrr9rr r r r s<