gB&dZddlZddlZddlZddlZddlZddlZddlZddlm Z m Z m Z m Z m Z mZddlmZddlmZej&eZGddZded ed ed efd Zd!d eded efdZej6dfdede ed dfdZedZde defd efdZded e fdZd"de ed efdZ de eefd dfdZ!de d e fdZ"dZ#d e ee ffd Z$y)#z/Various helper functions implemented by pytube.N)AnyCallableDictListOptionalTypeVar)request)RegexMatchErrorc^eZdZdZdZdZdefdZdZdefdZ de fdZ de fd Z d Zd Zy ) DeferredGeneratorLista A wrapper class for deferring list generation. Pytube has some continuation generators that create web calls, which means that any time a full list is requested, all of those web calls must be made at once, which could lead to slowdowns. This will allow individual elements to be queried, so that slowdowns only happen as necessary. For example, you can iterate over elements in the list without accessing them all simultaneously. This should allow for speed improvements for playlist and channel interactions. c ||_g|_y)aConstruct a :class:`DeferredGeneratorList `. :param generator generator: The deferrable generator to create a wrapper for. :param func func: (Optional) A function to call on the generator items to produce the list. N)gen _elements)self generators W/var/www/it7/html/youtubeDownloader/venv/lib/python3.12/site-packages/pytube/helpers.py__init__zDeferredGeneratorList.__init__sct||k(S)z.We want to mimic list behavior for comparison.)list)rothers r__eq__zDeferredGeneratorList.__eq__'sDzU""rreturnct|ttfs td|}t|trt||dzd}t |j |j krT t|j}|j j|t |j |j krT|j |S#t$rtwxYw)z)Only generate items as they're asked for.z"Key must be either a slice or int.) isinstanceintslice TypeErrorlenrstopnextrappend StopIteration IndexError)rkey key_slice next_items r __getitem__z!DeferredGeneratorList.__getitem__+s#U|,@A A c3 c37A.I$..!INN2 1 N %%i0$..!INN2~~c"" ! !   !s 'C Cc#NKd} ||}||dz }#t$rYywxYww)z/Custom iterator for dynamically generated list.rrN)r%)r iter_index curr_items r__iter__zDeferredGeneratorList.__iter__BsE   ,  a   s% % "%"%cv ||j}|xjdz c_|S#t$rtwxYw)zFetch next element in iterator.r)r+r%r$)r curr_elements r__next__zDeferredGeneratorList.__next__NsB 0L 1   s(8cL|jt|jS)z#Return length of list of all items.) generate_allr rrs r__len__zDeferredGeneratorList.__len__W 4>>""rcL|jt|jS)z#String representation of all items.)r2strrr3s r__repr__zDeferredGeneratorList.__repr__\r5rcF|j|jdddS)N)r2rr3s r __reversed__z"DeferredGeneratorList.__reversed__as! ~~dd##rc t|j}|jj|2#t$rYywxYw)zGenerate all items.N)r"rrr#r$)rr(s rr2z"DeferredGeneratorList.generate_allesD 1 N %%i0 !  s4 AAN)__name__ __module__ __qualname____doc__rrrr)r-r0rr4r7r8r;r2rrr r sP  ####.  ### ### $1rr patternstringgrouprctj|}|j|}|s td|tj d||j |S)a?Shortcut method to search a string for a given pattern. :param str pattern: A regular expression pattern. :param str string: A target string to search. :param int group: Index of group to return. :rtype: str or tuple :returns: Substring pattern matches. regex_search)callerrBzmatched regex search: %s)recompilesearchr loggerdebugrD)rBrCrDregexresultss rrFrFpsN JJw Ell6"G ^WEE LL+W5 == rs max_lengthctddDcgc] }t|}}gd}dj||z}tj|tj }|j d|}|d|jdddScc}w)atSanitize a string making it safe to use as a filename. This function was based off the limitations outlined here: https://en.wikipedia.org/wiki/Filename. :param str s: A string to make safe for use as a file name. :param int max_length: The maximum filename character length. :rtype: str :returns: A sanitized string. r)"z\#z\$z\%'z\*z\,z\.z\/z\:rSz\;z\z\?z\\z\^z\|z\~z\\\\|N )rangechrjoinrHrIUNICODEsubrsplit)rOrPintfs_characters charactersrBrMfilenames r safe_filenamerbs(-Q|4!s1v4O4J,hh34G JJw +EyyQH KZ ' 'Q / 2255sB level log_filenamecfd}d}tj||}tjd}|j|tj}|j ||j ||8tj|}|j ||j |yy)z~Create a configured instance of logger. :param int level: Describe the severity level of the logs to handle. z6[%(asctime)s] %(levelname)s in %(module)s: %(message)sz%H:%M:%S)datefmtpytubeN)logging Formatter getLoggersetLevel StreamHandler setFormatter addHandler FileHandler)rcrdfmtdate_fmt formatterrKstream_handler file_handlers r setup_loggerrus CCH!!#x8I  x (F OOE**,N * n%**<8 !!),,' r GenericTypefunc.c6tj|S)z1 mypy compatible annotation wrapper for lru_cache) functools lru_cache)rws rcacher{s 9    &&rreasoncfd}|S)z This is a decorator which can be used to mark functions as deprecated. It will result in a warning being emitted when the function is used. cNdtjfd}|S)Nz.Call to deprecated function {name} ({reason}).ctjdttjj j tdtjdt|i|S)Nalways)namer|)category stackleveldefault)warnings simplefilterDeprecationWarningwarnformatr=)argskwargsfunc1messager|s r new_func1z0deprecated..decorator..new_func1s]  ! !(,> ? MMENN6B+   ! !)-? @$)&) )r)rywraps)rrrr|s` @r decoratorzdeprecated..decorators+B   *  *rrA)r|rs` r deprecatedrs r output_pathc|rRtjj|sGtjjtj|}ntj}tj |d|S)a4 Function for determining target directory of a download. Returns an absolute path (if relative one given) or the current path (if none given). Makes directory if it does not exist. :type output_path: str :rtype: str :returns: An absolute directory path as a string. T)exist_ok)ospathisabsrZgetcwdmakedirs)rs rtarget_directoryrsPww}}[)'',,ryy{K@Kiik KK d+ r proxy_handlerctj|}tj|}tj|yN)r ProxyHandler build_openerinstall_opener)r proxy_supportopeners r install_proxyrs0((7M  ! !- 0F 6"r duped_listcRi}g}|D]}||vrd||<|j||S)zRemove duplicate items from a list, while maintaining list order. :param List duped_list List to remove duplicates from :return List result De-duplicated list T)r#)rseenresultitems r uniqueifyrsCD F 4< T  d  Mrc0gd}|D] }t|y)zRegenerate the video mock json files for all current test videos. This should automatically output to the test/mocks directory. ) 2lAe1cqCOXo 5YceQ8YqYMc irauhITDrsE m8uHb5jIGN8 QRS8MkLhQmM WXxV9g7lsFEN)create_mock_html_json) test_vid_idsvid_ids rgenerate_all_html_json_mocksrs" L&f%&rcddlm}d|z}tjj tjj tjj ttjj}tjj |dd}tjj ||}|d|}|j|j|j|j|jd}tjd|t!j"|d 5}|j%t'j(|j+d d d d |S#1swY|SxYw) zGenerate a json.gz file with sample html responses. :param str vid_id YouTube video id :return dict data Dict used to generate the json.gz file r)YouTubezyt-video-%s-html.json.gztestsmocksz https://www.youtube.com/watch?v=)urljs embed_html watch_htmlvid_infozOutputing json.gz file to wbzutf-8N)rgrrrabspathrZdirname__file__pardir watch_urlrrrrrKinfogzipopenwritejsondumpsencode) rr gzip_filenamepytube_dir_pathpytube_mocks_path gzip_filepathyt html_datafs rrr,s .7Mggoo  GGOOH % GGNN O  _gwGGGLL!2MBM 3F8< =B||eemmmmKK I KK,]O<= =$ '71  9%,,W567 7 s 64E44E>)r)%r@ryrrrhrrHrtypingrrrrrrurllibr pytube.exceptionsr rjr=rKr r7rrFrbERRORrurvr{rrrrrrrArrrsH5  ??-   8 $\1\1~ # s 3 3 0)3S)3c)3C)3X&mm4((8C=(TX(0m$ 'k)*'{' sx4(3-3(#c3h#D# $4&&"#T#s(^#r