a h#@sddlZddlZddlZddlZddlmZddlmZmZm Z m Z m Z m Z m Z mZmZmZmZmZmZmZmZmZmZddlmZmZGdddeZGd d d eZGd d d eZdS) N) InfoExtractor)ExtractorError ISO639Utils determine_ext filter_dict float_or_none int_or_none join_nonemptymake_archive_id parse_codecs parse_iso8601parse_qs smuggle_urlunified_timestamp unsmuggle_url url_or_noneurljoinvariadic)require traverse_objc@speZdZdgZdZiZddZdddZdd Zd d Z e d d Z ddZ dddZ dddZe ddZdS) ZDFBaseIEZDE)zdfz api-tokencCs|js"|j|jj|jdiit|jdthfddttkrv|j| dddd|jj g|j|jR|jdd |jd S) Ndefaultexpiresrrz0https://zdf-prod-futura.zdf.de/mediathekV2/tokenzDownloading API tokenzFailed to download API tokentype token) _token_cacheupdatecacheload_TOKEN_CACHE_PARAMSrr inttime_download_jsonstore)selfr*I/mnt/pikpak/tmp/myenv/lib/python3.9/site-packages/yt_dlp/extractor/zdf.py_get_api_token"s" zZDFBaseIE._get_api_tokenNcCs(|j||d|d|td|idS)N Downloading Failed to download Api-Auth)headers)r'r)r)urlvideo_iditem api_tokenr*r*r+ _call_api1s zZDFBaseIE._call_apicCsB|rt|tsdStd|}|r>t|dt|dSdS)Nz(?P\d+):(?P\d+)widthheight) isinstancestrrematchr%group)r) aspect_ratioZmobjr*r*r+_parse_aspect_ratio6s zZDFBaseIE._parse_aspect_ratiocCs(t|dddthfdthfdfp&dS)NcSs|dS)N anchorOffsetr*_vr*r*r+=z-ZDFBaseIE._extract_chapters..r?Z anchorLabel) start_timetitle)rr r9)r)datar*r*r+_extract_chapters<s   zZDFBaseIE._extract_chapterscCsbt}i}|D]N}t|d}|r||vr.q|||dpDd}||gd|iq|S)Nurilanguagedeur1)setrgetadd setdefaultappend)src seen_urls subtitlescaptionZ subtitle_urllangr*r*r+_extract_subtitlesBs   zZDFBaseIE._extract_subtitlescCst||ddS)Nz {playerId}Zandroid_native_6)rreplace)r)Z api_base_urltemplater*r*r+_expand_ptmd_templateQszZDFBaseIE._expand_ptmd_templatecCsd}d}gg}}t} t|D]} t| i\} } | ddk} || |d|} | dpl|jd| ddd}|sz| sz|}|st| dd d td d hf}|t| d dthf7}t| ddddthfD]}t|ddthfD]}t|ddddfD]}|d}|| vrq| |t |}|dkrB|j ||dddd}n|dvrt |d}|rr|rrt ||nd}d|it|d||td |d!t |jd"|d#ddd$g}n|jd%|d&|d'g}|d(}|D]}t|d)p|d)pd*}|d+d,k}|i|t|d-| o6d.t| oD|| oLd/|d0d1d2| rdd3nd4||r~|d0d5ks|s|d6krd7n6|d8kr|d9krd:n|d8krd;n|d9krdS)?Nvod_media_typeZDGSz PTMD databasenamez/vod/ptmd/[^/?#]+/(\w+)z content IDr attributesdurationvaluei)Zscalecaptions.Z priorityListZ formitaetenZ qualitiesaudioZtrackscSs t|dS)NrIrr@r*r*r+rCqrDz)ZDFBaseIE._extract_ptmd..rIZm3u8mp4ZhlsF)Zm3u8_idfatal)rbwebmZhighestVerticalResolutionr1Z mimeCodechttprz_(\d+)k_tbr)r7r6 format_idrfz Skipping unsupported extension "")r2classrJZvcodecnonergZdgszGerman Sign Language format_notez, delimZAudiodeskriptionadirKmain r)rgrl preferencerJZlanguage_preference)idr]formatsrS)rLrrrMr5 _search_regexrr dictrNrZ_extract_m3u8_formatsr roundr r Zreport_warningr short2longlowerrPrV)r) ptmd_urlsr2r4r=Z content_idr]rwZ src_captionsrRptmd_urlZ smuggled_dataZis_dgsZptmdr[streamZqualityvariantZ format_urlextZfmtsr7r6Zf_classfZf_langZ is_audio_onlyr*r*r+ _extract_ptmdTs             zZDFBaseIE._extract_ptmdc Cs^|s|sJd|jd|d|d|||r\d+|auto)[Xx](?P\d+|auto)r6r7)r6r7)r9rr:searchr ).0rgr1r*r*r+ sz1ZDFBaseIE._extract_thumbnails..cSs t|dS)Nrrar@r*r*r+rCrDz/ZDFBaseIE._extract_thumbnails..)rryitems)sourcer*r*r+_extract_thumbnailsszZDFBaseIE._extract_thumbnails)N)NN)NN)__name__ __module__ __qualname__Z_GEO_COUNTRIESr$r r,r5r>rH staticmethodrVrYrrrr*r*r*r+rs   S rcsfeZdZddgZdZdddddd d dd d d dgd ddddddd d dd d d dgd ddiddddddd d dd d d dgd ddiddddddddddddd d!d"d#d$d%gd&d'd(ddddddddddd d!d"d#d$d%gd&ddid)d*ddddddddddd d!d"d#d$d%gd&ddid)d+d,d-dd.d/d0d1d2d3d4d5d6d7d8d9d:d;gd<d'd=d>d?dd@dAdBdCdDdEdFgdG d'dHdIdJddKdLdMdNdOdPdQdRdSd dTdUdVgd&d'dWdXddYdZd[d\d]d^d_d`dadbdcdddegd&ddfdgdhdidjdkdldmdndd dSd dodpdqgd&ddriddsddtduddtdvddtdwddtdxddtdyddtdzddtd{ddtd|ddtd}ddtgZd~ZfddZddZddZ Z S)ZDFIEzFhttps?://(?:www\.)?zdf\.de/(?:video|play)/(?:[^/?#]+/)*(?P[^/?#]+)z=https?://(?:www\.)?zdf\.de/(?:[^/?#]+/)*(?P[^/?#]+)\.htmlrzrhttps://www.zdf.de/video/dokus/sylt---deutschlands-edles-nordlicht-movie-100/sylt-deutschlands-edles-nordlicht-100%sylt-deutschlands-edles-nordlicht-100rb#Sylt - Deutschlands edles Nordlicht$md5:35407b810c2e1e33efbe15ef6e4c06c3P@https://www.zdf.de/filme/filme-sonstige/der-hauptmann-112.htmlz8https://www.zdf.de/wissen/nano/nano-21-mai-2019-102.htmlzYhttps://www.zdf.de/service-und-hilfe/die-neue-zdf-mediathek/zdfmediathek-trailer-100.htmlz\https://www.zdf.de/filme/taunuskrimi/die-lebenden-und-die-toten-1---ein-taunuskrimi-100.htmlzqhttps://www.zdf.de/dokumentation/planet-e/planet-e-uebersichtsseite-weitere-dokumentationen-von-planet-e-100.htmlzVhttps://www.zdf.de/arte/todliche-flucht/page-video-artede-toedliche-flucht-16-100.htmlzShttps://www.zdf.de/dokumentation/terra-x/unser-gruener-planet-wuesten-doku-100.htmla$ query VideoByCanonical($canonical: String!) { videoByCanonical(canonical: $canonical) { canonical title leadParagraph editorialDate teaser { description image { list } } episodeInfo { episodeNumber seasonNumber } smartCollection { canonical title } currentMedia { nodes { ptmdTemplate ... on VodMedia { duration aspectRatio streamAnchorTags { nodes { anchorOffset anchorLabel } } vodMediaType label } ... on LiveMedia { start stop encryption liveMediaType label } id } } } } cs0tj|i|}|d}t||g|d<|S)Nrvr)superrpopr )r)argskwargsZ ptmd_dataZold_archive_id __class__r*r+rs zZDFIE._extract_ptmdc Cs|jd||ddd}|d}t|dthttdhf}g}t|dtjhd fD]F\}}t|d thf}|spqR|||t| d t| d d qRd|it|ddt hfddt hfdt htfdd|j hfd| |||d|iS)Nz4https://zdf-prod-futura.zdf.de/mediathekV2/document/zDownloading fallback metadataz$Failed to download fallback metadata)ZnoteZerrnotedocument)streamApiUrlAndroid)ZstreamsrrzPTMD URLZ teaserBild.r1r6r7)r1rvr6r7 thumbnailsZtitelZ beschreibung))rdate)meta editorialDater_)rFrrrSrv)r'rranyrryrrPr rMr9rrVrr,) r)Z document_idvideorr~rZ thumbnail_keyrZ thumbnail_urlr*r*r+_extract_fallbacksH       zZDFIE._extract_fallbackc Cs8||}|j|dd|jd|idddd}|s>||Sd}g}t|dd d d fD]N}|d |d }|d}rt|d|i}|||sZ| |d}qZit|dt hfdt t hfdt hfddd|j hfddthfddthfddt hfddt hfdd ddd |jhfd |||||d|iS)Nzvideo metadataZVideoByCanonical canonical) operationNamer variables)rrGZvideoByCanonicalZ currentMedianodescSs|dS)N ptmdTemplater*r@r*r*r+rC3rDz%ZDFIE._real_extract..zhttps://api.zdf.derZ vodMediaTyperZZ aspectRatiorFZ leadParagraph)teaserrrrimagelist episodeInfo episodeNumber seasonNumberZsmartCollectionrZstreamAnchorTags) rFrrrrrrrrrv) _match_idr_GRAPHQL_QUERYrrrYrMrrPr>r9rrrr rHrr,) r)r1r2 video_datar=r}noder~rZr*r*r+ _real_extract&sL          zZDFIE._real_extract) rrr _VALID_URLIE_NAME_TESTSrrrr __classcell__r*r*rr+rs  |2 %rcseZdZdZdZddddddd d dddddd egd d d dddddddddgd ddidddddddd dd d!d"dd#d d$d%d&d"dd'd(d)d*d+d,dd-d d.d/d0d,dd1d2d3gZd4Ze fd5d6Z d>d8d9Z d:d;Z d[^/?#]+)z zdf:channelz1https://www.zdf.de/sport/das-aktuelle-sportstudiozdas-aktuelle-sportstudio-220zdas aktuelle sportstudioz$md5:e46c785324238a03edcf8b301c5fd5dc)rvrFr)r1rZplaylist_mincountz5https://www.zdf.de/sport/das-aktuelle-sportstudio-220zFhttps://www.zdf.de/dokus/sylt---deutschlands-edles-nordlicht-movie-100rrbrrrrrrrrrrT)Zadd_ier1rrz9https://www.zdf.de/gesellschaft/aktenzeichen-xy-ungeloestzaktenzeichen-xy-ungeloest-110uAktenzeichen XY... Ungelöstz$md5:b79ac0d64b979e53cbe510c0ca9cb7berz&https://www.zdf.de/serien/taunuskrimi/ztaunuskrimi-100Z Taunuskrimiz$md5:ee7204e9c625c3b611d1274f9d0e3070z0https://www.zdf.de/serien/taunuskrimi/?staffel=1ztaunuskrimi-100-s1zTaunuskrimi - Season 1r)r1rplaylist_countz-https://www.zdf.de/magazine/heute-journal-104rrz$md5:6edad39189abf8431795d3d6d7f986b3iz:https://www.zdf.de/magazine/heute-journal-104?staffel=2024zheute-journal-104-s2024zheute journal - Season 2024z>Video count changes daily, needs support for playlist_maxcount)r1rrskipcst|rdSt|S)NF)rsuitabler)clsr1rr*r+rszZDFChannelIE.suitableNc CsT|j|d|d|ddtt|||j|dtdddd id d d d S)Nzseason z page z JSONZseasonByCanonical)Z seasonIndexrZepisodesPageSizeZ episodesAfterpersistedQueryrZ@9412a0f4ac55dc37d46975d461ec64bfd14380d815df843a1492348f77b5c99aversionZ sha256Hashrr extensionsrrGsmartCollectionByCanonical)rrrr _PAGE_SIZE)r) playlist_id canonical_id season_idxr page_numbercursorr*r*r+ _fetch_pages*zZDFChannelIE._fetch_pagec cst|D]\}}|dur"||kr"qd}tdD]}|||||||} t| d} t| dddddfD]^} |j| dtfit| d thfd d thfd tthfd t hfddt hfddt hfdVqht| dddt hfpi} | dr| dsq| d}q0qdS)Nr)seasonsr..ZepisodesrcSs t|dS)N sharingUrlrar@r*r*r+rCrDz'ZDFChannelIE._entries..rrrrFrrrrr)rvrFrrrrrpZpageInfoZ hasNextPageZ endCursor) enumerate itertoolscountrr url_resultrr9rrr ryrM) r)rrseason_numbersZrequested_season_numberrrrrpagerrZ page_infor*r*r+_entriess8        zZDFChannelIE._entriesc Csd||}|||}|j|j|jddd}tt|ddthf}t||oRd|}|j |ddt |d d t d d d didddd}t|dt hfpi}t|ddddthf}| |o|t|do|ds||dt|dS|dur||vrtd|d|||||||tt|dthf|oLd|ddt|dthfS)Nz channel idrv)r<Zstaffelrpszsmart collection dataZGetSmartCollectionByCanonicald)rZ videoPageSizerrZ@cb49420e133bd668ad895a8cea0e65cba6aa11ac1cacb02341ff5cf32a17cd02rrrrGrrr.numberrrzSeason z% was not found in the collection datarFz - rmZinfoText)rZ_request_webpagerxrr1rrr r rrrryZ _yes_playlistrrMrrrZplaylist_resultrr9) r)r1rZurlhrrZcollection_datarrr*r*r+rsR    zZDFChannelIE._real_extract)N)rrrrrrZie_keyrr classmethodrrrrrr*r*rr+rMs       O r)rrr:r&commonrutilsrrrrr r r r r rrrrrrrrZutils.traversalrrrrrr*r*r*r+s L%