a hk@sddlmZddlZddlmZddlmZddlmZddl m Z m Z m Z ddl mZd d lmZd d lmZmZd d lmZmZed d dGdddZGddde eZe eddZdS)) annotationsN)ABC) dataclass)Any)RequestHandlerregister_preferenceRequest)UnsupportedRequest)NoneType) classproperty join_nonempty) std_headersHTTPHeaderDictT)orderfrozenc@sreZdZUdZdZded<dZded<dZded<dZded<dd Z dd d d Z d dZ e dd ddZ dS)ImpersonateTargeta< A target for browser impersonation. Parameters: @param client: the client to impersonate @param version: the client version to impersonate @param os: the client OS to impersonate @param os_version: the client OS version to impersonate Note: None is used to indicate to match any. Nz str | Noneclientversionos os_versioncCs,|jr|jstd|jr(|js(tddS)Nz$client is required if version is setz#os is required if os_version is set)rr ValueErrorrrselfrR/mnt/pikpak/tmp/myenv/lib/python3.9/site-packages/yt_dlp/networking/impersonate.py __post_init__"s  zImpersonateTarget.__post_init__targetcCst|tsdS|jdus.|jdus.|j|jko|jdusN|jdusN|j|jko|jdusn|jdusn|j|jko|jdup|jdup|j|jkS)NF) isinstancerrrrrrrrrr __contains__(s  zImpersonateTarget.__contains__cCs(t|j|jdt|j|jdS)N:)rrrrrrstriprrrr__str__2szImpersonateTarget.__str__strcCs2td|}|s td|d|fi|S)Nze(?:(?P[^:-]+)(?:-(?P[^:-]+))?)?(?::(?:(?P[^:-]+)(?:-(?P[^:-]+))?)?)?zInvalid impersonate target "")re fullmatchr groupdict)clsrZmobjrrrfrom_str5s zImpersonateTarget.from_str)__name__ __module__ __qualname____doc__r__annotations__rrrrr"r% classmethodr,rrrrrs      rcseZdZUdZiZded<ddddfddZdd d d Zfd d ZfddZ dd ddZ e ddddZ dd ddZ ddZdddddd Zdd!d"d#d$ZZS)%ImpersonateRequestHandlera< Base class for request handlers that support browser impersonation. This provides a method for checking the validity of the impersonate extension, which can be used in _check_extensions. Impersonate targets consist of a client, version, os and os_ver. See the ImpersonateTarget class for more details. The following may be defined: - `_SUPPORTED_IMPERSONATE_TARGET_MAP`: a dict mapping supported targets to custom object. Any Request with an impersonate target not in this list will raise an UnsupportedRequest. Set to None to disable this check. Note: Entries are in order of preference Parameters: @param impersonate: the default impersonate target to use for requests. Set to None to disable impersonation. zdict[ImpersonateTarget, Any]!_SUPPORTED_IMPERSONATE_TARGET_MAPN) impersonaterc stjfi|||_dSN)super__init__r5)rr5kwargs __class__rrr8Ssz"ImpersonateRequestHandler.__init__rcCs@t|ttfsJ|dus |js$dS||sget)r extensionsr:rrr?^s z+ImpersonateRequestHandler._check_extensionscst|||jdSr6)r7 _validater>r5rrequestr:rrrBcs z#ImpersonateRequestHandler._validatezImpersonateTarget | NonecCsP|dur dS|jD]8}||vr|jrB|j|jd|d||SqdS)z'Resolve a target to a supported target.Nz: resolved impersonate target z to )r<verboseZ_loggerstdoutZRH_NAME)rrZsupported_targetrrr_resolve_targetgs z)ImpersonateRequestHandler._resolve_targetztuple[ImpersonateTarget, ...])returncCst|jSr6)tupler4keys)r+rrrr<rsz+ImpersonateRequestHandler.supported_targetscCst|tsJ||duSr6)r rrGr!rrrr=vsz-ImpersonateRequestHandler.is_supported_targetcCs||jdp|jS)z(Get the requested target for the requestr5)rGrAr@r5rCrrr_get_request_targetzsz-ImpersonateRequestHandler._get_request_targetr rNone)rDheadersrHcCsdS)zAdditional operations to prepare headers before building. To be extended by subclasses. @param request: Request object @param headers: Merged headers to prepare Nr)rrDrMrrr_prepare_impersonate_headers~sz6ImpersonateRequestHandler._prepare_impersonate_headerszdict[str, str])rDrHcCsl||j}||durDtD] \}}|||kr"||q"||||jdrd| St |S)z Get headers for external impersonation use. Subclasses may define a _prepare_impersonate_headers method to modify headers after merge but before building. NZkeep_header_casing) Z_merge_headersrMrKritemsr@poprNrAZ sensitivedict)rrDrMkvrrr_get_impersonate_headerss    z2ImpersonateRequestHandler._get_impersonate_headers)r-r.r/r0r4r1r8r>r?rBrGr r<r=rKrNrT __classcell__rrr:rr3=s     r3cCs|jds|jrdSdS)Nr5ir)rAr@r5)ZrhrDrrrimpersonate_preferencesrV) __future__rr(abcrZ dataclassesrtypingrcommonrrr exceptionsr Z compat.typesr utilsr rZutils.networkingrrrr3rVrrrrs       -Z