a &jf@slddlmZddlZddlZddlZddlmZmZmZddlm Z ddl m Z m Z mZmZmZmZmZmZmZddlmZmZmZmZmZmZmZddlmZmZm Z m!Z!m"Z"dd l#m$Z$m%Z%m&Z&d d l'm(Z(d d l)m*Z*d d l+m,Z,dZ-edddZ.GdddZ/eGdddZ0eGddde0Z1eGddde0Z2eGddde0Z3eGddde0Z4dS)) annotationsN)asdict dataclassfield)fields) AnyCallableDictListMappingOptionalSetTypeVarUnion)BROWSER_HEADERSDEFAULT_ALLOW_HTTPDEFAULT_CLIENT_IDENTIFIER DEFAULT_DEBUG DEFAULT_HTTP2DEFAULT_PROTOCOL_RACINGDEFAULT_TIMEOUT) CookiesTypesIdentifierTypes MethodTypes SessionIdURLTypes) to_base64to_bytesto_json) StreamEncoder) TLSLibrary) StatusCodes) TLSClient TLSResponse TLSConfigCustomTLSClientConfigTLSRequestCookiesConfigT _BaseConfig)boundc@sJeZdZUdZdZded<dZded<dZded<dZded<dZ ded <dZ ded <dZ ded <d d ddZ e ddZe ddddddZe dddddZe dd ddZe dddd d!Ze d"d#Ze dd d$d%d&Ze d'dd(d)d*Ze d'd+d,d-Ze d.d/Ze d0dd1d2d3Ze d0dd1d4d5ZdS)6r$a TLSClient The `TLSClient` class provides a high-level interface for performing secure TLS-based HTTP operations. It encapsulates interactions with a custom TLS library, offering functionality for managing sessions, cookies, and HTTP requests. This class is designed to be extensible and integrates seamlessly with the `TLSResponse` and `TLSConfig` classes for handling responses and configuring requests. Attributes: _library (Optional): Reference to the loaded TLS library. _getCookiesFromSession (Optional): Function for retrieving cookies from a session. _addCookiesToSession (Optional): Function for adding cookies to a session. _destroySession (Optional): Function for destroying a specific session. _destroyAll (Optional): Function for destroying all active sessions. _request (Optional): Function for performing a TLS-based HTTP request. _freeMemory (Optional): Function for freeing allocated memory for responses. Methods: setup(cls): Loads and sets up the TLS library and initializes function bindings. get_cookies(cls, session_id: TLSSessionId, url: str) -> TLSResponse: Retrieves cookies from a session for the given URL. add_cookies(cls, session_id: TLSSessionId, payload: dict): Adds cookies to a specific session. destroy_all(cls) -> bool: Destroys all active TLS sessions. Returns `True` if successful. destroy_session(cls, session_id: TLSSessionId) -> bool: Destroys a specific TLS session. Returns `True` if successful. request(cls, payload): Performs a TLS-based HTTP request with the provided payload. free_memory(cls, response_id: str) -> None: Frees the memory allocated for a specific response. response(cls, raw: bytes) -> TLSResponse: Processes a raw byte response and returns a `TLSResponse` object. _make_request(cls, fn: callable, payload: dict): Helper method to handle request processing and response generation. Example: Initialize the client and perform operations: >>> from tls_requests.tls import TLSClient >>> client = TLSClient.initialize() >>> session_id = "my-session-id" >>> url = "https://example.com" >>> response = client.get_cookies(session_id, url) >>> print(response) Nz Optional[Any]_libraryzOptional[Callable]_getCookiesFromSession_addCookiesToSession_destroySession _destroyAll_request _freeMemoryNonereturncCs|jdur|dSN)r, initializeselfr:=E:\soccer\ml_env\Lib\site-packages\tls_requests/models/tls.py__init__es zTLSClient.__init__cCsvt|_dD]L}d|}t||t|j|dt||d}|rt|rtjg|_tj|_ q|jj |_ tj|j _ |S)N)ZgetCookiesFromSessionZaddCookiesToSessionZdestroySessionZ freeMemoryrequestz_%s) r"loadr,setattrgetattrcallablectypesc_char_pargtypesrestypeZ destroyAllr0)clsnamefn_namefnr:r:r;r7is       zTLSClient.initializerstr 'TLSResponse') session_idurlr5cCs*|jdur|||j||d}|S)N) sessionIdrM)r-r7_send)rFrLrMresponser:r:r; get_cookies~s zTLSClient.get_cookiesdict)rLpayloadcCs(|jdur|||d<||j|S)NrN)r.r7rO)rFrLrSr:r:r; add_cookiess zTLSClient.add_cookiesboolcCs.|jdur|t|}|jr*dSdS)NTF)r0r7r% from_bytessuccess)rFrPr:r:r; destroy_alls  zTLSClient.destroy_all)rLr5cCs.|jdur|||jd|i}|jp,dS)NrNF)r/r7rOrW)rFrLrPr:r:r;destroy_sessions zTLSClient.destroy_sessioncCs |jdur|||j|Sr6)r1r7rOrFrSr:r:r;r=s zTLSClient.request) response_idr5cCs$|jdur||t|dSr6)r2r7r)rFr[r:r:r; free_memorys zTLSClient.free_memorybytesrawr5cCs t|}|jr||j|Sr6)r%rVidr\)rFr_rPr:r:r;rPs  zTLSClient.response)r_cs\t|>}ddd|2IdH}tjfit|WdS1sN0YdS)Ncsg|z3dHW}|q6Sr6r:).0chunkr:r:r; raz'TLSClient.aresponse..)r!rVjoinr% from_kwargsr)rFr_streamcontentr:r:r; aresponses zTLSClient.aresponsecs&|jdur|||j|IdHSr6)r1r7_areadrZr:r:r;arequests zTLSClient.arequestr)rIrScCs||t|Sr6)rPrrFrIrSr:r:r;rOszTLSClient._sendcs||t|IdHSr6)rirrlr:r:r;rjszTLSClient._aread)__name__ __module__ __qualname____doc__r,__annotations__r-r.r/r0r1r2r< classmethodr7rQrTrXrYr=r\rPrirkrOrjr:r:r:r;r$%sB 7           r$c@seZdZUdZeedddZded<edddd Z ed d d d ddZ ddddZ ddddZ edddddZ dS)r*z!Base configuration for TLSSessionF)default_factoryinitreprrR _extra_configzSet[str]r4cCsddt|DS)NcSsh|]}|jds|jqS)_)rG startswith)rbZ model_fieldr:r:r; raz/_BaseConfig.model_fields_set..) get_fields)rFr:r:r;model_fields_setsz_BaseConfig.model_fields_setztype[T]rr))rFkwargsr5c sPfdd|D}fdd|D}fi|}||_|S)Ncs$i|]\}}|vr||qSr: to_camel_caserbkvrFr{r:r; raz+_BaseConfig.from_kwargs..cs$i|]\}}|vr||qSr:r}rrr:r;rra)r{itemsrv)rFr|Z known_kwargs extra_kwargsinstancer:rr;rfs z_BaseConfig.from_kwargscCs0t|}t|dr||jdd|DS)NrvcSs(i|] \}}|ds|dur||qS)rwN)rxrr:r:r;rraz'_BaseConfig.to_dict..)rhasattrupdatervr)r9datar:r:r;to_dicts  z_BaseConfig.to_dictcCs|Sr6)rr8r:r:r; to_payloadsz_BaseConfig.to_payloadrJ)rGr5cCsdddt|dDS)zConvert a string to camelCase.css&|]\}}|dkr|n|VqdS)rN) capitalize)rbiwordr:r:r; raz,_BaseConfig.to_camel_case..rw)re enumeratesplit)rFrGr:r:r;r~sz_BaseConfig.to_camel_caseN)rmrnrorprrRrvrqrrr{rfrrr~r:r:r:r;r*s c@seZdZUdZdZded<dZded<dZded<dZded <dZ ded <e e d Z d ed <e e d Z d ed<dZded<dZded<edddddZeddddZddZdS)r%aTLS Response Attributes: id (Optional[str]): A unique identifier for the response. Defaults to `None`. sessionId (Optional[str]): The session ID associated with the response. Defaults to `None`. status (Optional[int]): The HTTP status code of the response. Defaults to `0`. target (Optional[str]): The target URL or endpoint of the response. Defaults to `None`. body (Optional[str]): The body content of the response. Defaults to `None`. headers (Optional[dict]): A dictionary containing the headers of the response. Defaults to an empty dictionary. cookies (Optional[dict]): A dictionary containing the cookies of the response. Defaults to an empty dictionary. success (Optional[bool]): Indicates if the response was successful. Defaults to `False`. usedProtocol (Optional[str]): The protocol used in the response. Defaults to `"HTTP/1.1"`. Methods: from_bytes(cls, raw: bytes) -> TLSResponse: Parses a raw byte stream and constructs a `TLSResponse` object. reason_phrase -> str: A property that provides the reason phrase associated with the HTTP status code. If the status code is `0`, it returns `"Bad Request"`. N Optional[str]r`rNrintstatustargetbodyrszDict[str, Any]headerscookiesFrUrWzHTTP/1.1rJ usedProtocolr]rKr^cCsLt|.}d|}|jfit|WdS1s>0YdS)Nra)r!rVrerfr)rFr_rgrSr:r:r;rV s  zTLSResponse.from_bytesr4cCs t|jSr6)r#Z get_reasonrr8r:r:r;reasonszTLSResponse.reasoncCs d|jS)Nz)rr8r:r:r;__repr__szTLSResponse.__repr__)rmrnrorpr`rqrNrrrrrRrrrWrrrrVpropertyrrr:r:r:r;r%s        r%c@s"eZdZUdZded<ded<dS)r(a Request Cookies Configuration Represents a single request cookie with a _name and value. Attributes: name (str): The _name of the cookie. value (str): The value of the cookie. Example: Create a `TLSRequestCookiesConfig` object: >>> from tls_requests.tls import TLSRequestCookiesConfig >>> kwargs = { ... "_name": "foo2", ... "value": "bar2", ... } >>> obj = TLSRequestCookiesConfig(**kwargs) rJrGvalueN)rmrnrorprqr:r:r:r;r(s r(c@seZdZUdZdZded<dZded<dZded<dZded <dZ d ed <dZ ded <dZ ded <dZ ded<dZ ded<dZded<dZded<dZded<dZded<dS)r'a Custom TLS Client Configuration The `CustomTLSClientConfig` class defines advanced configuration options for customizing TLS client behavior. It includes support for ALPN, ALPS protocols, certificate compression, HTTP/2 settings, JA3 fingerprints, and other TLS-related settings. Attributes: alpnProtocols (list[str], optional): ALPN protocols. Defaults to `None`. alpsProtocols (list[str], optional): ALPS protocols. Defaults to `None`. certCompressionAlgo (str, optional): Certificate compression algorithm. Defaults to `None`. connectionFlow (int, optional): Connection flow. Defaults to `None`. h2Settings (list[str], optional): HTTP/2 settings. Defaults to `None`. h2SettingsOrder (list[str], optional): Order of HTTP/2 settings. Defaults to `None`. headerPriority (list[str], optional): Priority of headers. Defaults to `None`. ja3String (str, optional): JA3 string. Defaults to `None`. keyShareCurves (list[str], optional): Key share curves. Defaults to `None`. priorityFrames (list[str], optional): Priority of frames. Defaults to `None`. pseudoHeaderOrder (list[str], optional): Order of pseudo headers. Defaults to `None`. supportedSignatureAlgorithms (list[str], optional): Supported signature algorithms. Defaults to `None`. supportedVersions (list[str], optional): Supported versions. Defaults to `None`. Example: Create a `CustomTLSClientConfig` instance with specific settings: >>> from tls_requests.tls import CustomTLSClientConfig >>> kwargs = { ... "alpnProtocols": ["h2", "http/1.1"], ... "alpsProtocols": ["h2"], ... "certCompressionAlgo": "brotli", ... "connectionFlow": 15663105, ... "h2Settings": { ... "HEADER_TABLE_SIZE": 65536, ... "MAX_CONCURRENT_STREAMS": 1000, ... "INITIAL_WINDOW_SIZE": 6291456, ... "MAX_HEADER_LIST_SIZE": 262144 ... }, ... "h2SettingsOrder": [ ... "HEADER_TABLE_SIZE", ... "MAX_CONCURRENT_STREAMS", ... "INITIAL_WINDOW_SIZE", ... "MAX_HEADER_LIST_SIZE" ... ], ... "headerPriority": None, ... "ja3String": "771,2570-4865-4866-4867-49195-49199-49196-49200-52393-52392-49171-49172-156-157-47-53,2570-0-23-65281-10-11-35-16-5-13-18-51-45-43-27-17513-2570-21,2570-29-23-24,0", ... "keyShareCurves": ["GREASE", "X25519"], ... "priorityFrames": [], ... "pseudoHeaderOrder": [ ... ":method", ... ":authority", ... ":scheme", ... ":path" ... ], ... "supportedSignatureAlgorithms": [ ... "ECDSAWithP256AndSHA256", ... "PSSWithSHA256", ... "PKCS1WithSHA256", ... "ECDSAWithP384AndSHA384", ... "PSSWithSHA384", ... "PKCS1WithSHA384", ... "PSSWithSHA512", ... "PKCS1WithSHA512" ... ], ... "supportedVersions": ["GREASE", "1.3", "1.2"] ... } >>> obj = CustomTLSClientConfig.from_kwargs(**kwargs) NzOptional[List[str]] alpnProtocols alpsProtocolsrcertCompressionAlgo Optional[int]connectionFlowzOptional[Dict[str, int]] h2Settingsh2SettingsOrderheaderPriority ja3StringkeyShareCurvespriorityFramespseudoHeaderOrdersupportedSignatureAlgorithmssupportedVersions)rmrnrorprrqrrrrrrrrrrrrr:r:r:r;r'3s E            r'cseZdZUdZdZded<eedZded<dZ d ed <dZ ded <dZ ded <ee dZ d ed<eedZded<dZded<dZded<dZded<dZded<dZded<dZded<ee dZded<dZded<dZded<ed d!dZded"<dZd#ed$<d%Zd&ed'<eZd(ed)<eZded*<eZ ded+<dZ!ded,<e"Z#ded-<dZ$ded.<dZ%ded/<d0d1fd2d3 Z&dCdd4d5dd6d7d8dd8d9d8d(d8d8d8d#d:d;d<fd=d> Z'e(dddddddde)e*ddee"edfdd4d5dd6d7ddd?d@dd(dddd#d:d;d<fdAdB Z+Z,S)Dr&a4 TLS Configuration The `TLSConfig` class provides a structured and flexible way to configure TLS-specific settings for HTTP requests. It supports features like custom headers, cookie handling, proxy configuration, and advanced TLS options. Methods: to_dict(self) -> dict Converts the TLS configuration object into a dictionary. copy_with(self, **kwargs) -> "TLSConfig" Creates a new `TLSConfig` object with updated properties. from_kwargs(cls, **kwargs) -> "TLSConfig" Creates a `TLSConfig` instance from keyword arguments. Example: Initialize a `TLSConfig` object using predefined or custom settings: >>> from tls_requests.tls import TLSConfig >>> kwargs = { ... "catchPanics": false, ... "certificatePinningHosts": {}, ... "customTlsClient": {}, ... "followRedirects": false, ... "forceHttp1": false, ... "headerOrder": [ ... "accept", ... "user-agent", ... "accept-encoding", ... "accept-language" ... ], ... "headers": { ... "accept": "text/html,application/xhtml+xml,application/xml;q=0.9,image/avif,image/webp,image/apng,*/*;q=0.8,application/signed-exchange;v=b3;q=0.9", ... "user-agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/105.0.0.0 Safari/537.36", ... "accept-encoding": "gzip, deflate, br", ... "accept-language": "de-DE,de;q=0.9,en-US;q=0.8,en;q=0.7" ... }, ... "insecureSkipVerify": false, ... "isByteRequest": false, ... "isRotatingProxy": false, ... "proxyUrl": "", ... "requestBody": "", ... "requestCookies": [ ... { ... "_name": "foo", ... "value": "bar", ... }, ... { ... "_name": "bar", ... "value": "foo", ... }, ... ], ... "requestMethod": "GET", ... "requestUrl": "https://microsoft.com", ... "sessionId": "2my-session-id", ... "timeoutSeconds": 30, ... "tlsClientIdentifier": "chrome_120", ... "withDebug": false, ... "withDefaultCookieJar": false, ... "withRandomTLSExtensionOrder": false, ... "withoutCookieJar": false ... } ... >>> obj = TLSConfig.from_kwargs(**kwargs) FrU catchPanicsrzMapping[str, str]certificatePinningHostsNzOptional[CustomTLSClientConfig]customTlsClientfollowRedirects forceHttp1z List[str] headerOrderrinsecureSkipVerify isByteRequestTisByteResponseisRotatingProxyrrJproxyUrlz,Union[str, bytes, bytearray, Optional[None]] requestBodyzList[TLSRequestCookiesConfig]requestCookieszOptional[MethodTypes] requestMethodr requestUrlcCs ttSr6)rJuuiduuid4r:r:r:r;razTLSConfig.rNrstreamIDrtimeoutSecondszOptional[IdentifierTypes]tlsClientIdentifier withAllowHTTP withDebugwithDefaultCookieJarwithProtocolRacingwithRandomTLSExtensionOrderwithoutCookieJarrRr4cs|jr d|_d|_|jr.rv)rrrrrvrrf)r9rLrrrrMrrrrrrrrrrrr|mappingZfiltered_mappingZcurrent_kwargsrrrr:r; copy_withs4     zTLSConfig.copy_withzOptional[Union[bool, str]]zUnion[float, int]c s~| dur|d| | dur(|d| |dur<|d||durP|d||durd|d|| pt|dptt}t|}|dur|n|d}i}|stD]D\}}||vr|}|dkr2t d |}|r| d }|d d }t d d|||d <d|vr|d}| d||d<n|dkrt d|}|r| d }|d d }t dd||}t dd|||d <nX|dkrt d|pt d|}|r| d }|d d }t dd|||d <qq||rt |n||p g|||||t| t| ttfr6t| nt| || |||d}|D]\}}|||qTtjfi|S)z6Creates a `TLSConfig` instance from keyword arguments.Nrrrrrrchromez chrome_(\d+)r z user-agentrz Chrome/\d+zChrome/z sec-ch-uaZ133Zfirefoxz firefox_(\d+)zrv:\d+zrv:z Firefox/\d+zFirefox/Zsafarizsafari_ios_(\d+)z safari_(\d+)z Version/\d+zVersion/)rNrrrrrrrrrrrrrrr) setdefaultgetrrJlowerrrcopyresearchgroupsubreplacerRrUrrrrrrf)rFrLrrrrMrrrrrrrrrrrr| identifierZidentifier_strZresolved_headersZinjected_headersZbrowserZbrowser_headersmatchversionZuavaldefaultskeyrrr:r;rf3s~                  zTLSConfig.from_kwargs)NNNNNNNNNNNNNNNN)-rmrnrorprrqrrRrrrrlistrrrrrrrrrrrrNrrrrrrrrrrrrrrrrrrrrf __classcell__r:r:rr;r&s~ A                     67r&)5 __future__rrBrr dataclassesrrrrrztypingrrr r r r r rrsettingsrrrrrrrtypesrrrrrutilsrrrencodersr! librariesr" status_codesr#__all__r)r$r*r%r(r'r&r:r:r:r;s4  ,$     #!/U