U g@sddlmZddlZddlmZddlmZddlmZddl m Z m Z ddl m Z d d lmZd d lmZmZd d lmZed d dGdddZGddde eZe eddZdS)) annotationsN)ABC) dataclass)Any)RequestHandlerregister_preference)UnsupportedRequest)NoneType) classproperty join_nonempty) std_headersT)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 ValueErrorrrselfrA/tmp/pip-unpacked-wheel-q5ljy6pj/yt_dlp/networking/impersonate.py __post_init__"s  zImpersonateTarget.__post_init__targetcCst|tsdS|jdks.|jdks.|j|jko|jdksN|jdksN|j|jko|jdksn|jdksn|j|jko|jdkp|jdkp|j|jkS)NF) isinstancerrrrrrrrrr __contains__(s  zImpersonateTarget.__contains__cCs(t|j|jdt|j|jdS)N:)r rrrrrstriprrrr__str__2szImpersonateTarget.__str__strcCs.td|}|s td|d|f|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 ddZddZZS)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 stjf|||_dSN)super__init__r3)rr3kwargs __class__rrr6Ssz"ImpersonateRequestHandler.__init__rcCs@t|ttfst|dks |js$dS||s^s z+ImpersonateRequestHandler._check_extensionscst|||jdSr4)r5 _validater=r3rrequestr8rrrAcs z#ImpersonateRequestHandler._validatezImpersonateTarget | NonecCsP|dkr dS|jD]8}||kr|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|jSr4)tupler2keys)r)rrrr;rsz+ImpersonateRequestHandler.supported_targetscCst|tst||dk Sr4)rrr:rFrrrrr<vsz-ImpersonateRequestHandler.is_supported_targetcCs||jdp|jS)z(Get the requested target for the requestr3)rFr@r?r3rBrrr_get_request_targetzsz-ImpersonateRequestHandler._get_request_targetcCsH||j}||dk rDtD] \}}|||kr"||q"|Sr4)Z_merge_headersheadersrJritemsr?pop)rrCrKkvrrr_get_impersonate_headers~s   z2ImpersonateRequestHandler._get_impersonate_headers)r+r,r-r.r2r/r6r=r>rArFr r;r<rJrP __classcell__rrr8rr1=s     r1cCs|jds|jrdSdS)Nr3ir)r@r?r3)ZrhrCrrrimpersonate_preferencesrR) __future__rr&abcrZ dataclassesrtypingrcommonrr exceptionsr Z compat.typesr utilsr r Zutils.networkingrrr1rRrrrrs        -L