3607917966

Build Type

push

github-actions

Committed by funilrys

Commit Message

fixup! Introduction of the changeip SPECIAL rule.

Run Details

11094 of 11508 relevant lines covered (96.4%)

11.55 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

25.93

/PyFunceble/query/requests/adapter/https.py

"""
The tool to check the availability or syntax of domain, IP or URL.

::


    ██████╗ ██╗   ██╗███████╗██╗   ██╗███╗   ██╗ ██████╗███████╗██████╗ ██╗     ███████╗
    ██╔══██╗╚██╗ ██╔╝██╔════╝██║   ██║████╗  ██║██╔════╝██╔════╝██╔══██╗██║     ██╔════╝
    ██████╔╝ ╚████╔╝ █████╗  ██║   ██║██╔██╗ ██║██║     █████╗  ██████╔╝██║     █████╗
    ██╔═══╝   ╚██╔╝  ██╔══╝  ██║   ██║██║╚██╗██║██║     ██╔══╝  ██╔══██╗██║     ██╔══╝
    ██║        ██║   ██║     ╚██████╔╝██║ ╚████║╚██████╗███████╗██████╔╝███████╗███████╗
    ╚═╝        ╚═╝   ╚═╝      ╚═════╝ ╚═╝  ╚═══╝ ╚═════╝╚══════╝╚═════╝ ╚══════╝╚══════╝

Provides the our HTTPS adapter.

Author:
    Nissar Chababy, @funilrys, contactTATAfunilrysTODTODcom

Special thanks:
    https://pyfunceble.github.io/#/special-thanks

Contributors:
    https://pyfunceble.github.io/#/contributors

Project link:
    https://github.com/funilrys/PyFunceble

Project documentation:
    https://pyfunceble.readthedocs.io/en/dev/

Project homepage:
    https://pyfunceble.github.io/

License:
::


    Copyright 2017, 2018, 2019, 2020, 2022 Nissar Chababy

    Licensed under the Apache License, Version 2.0 (the "License");
    you may not use this file except in compliance with the License.
    You may obtain a copy of the License at

        http://www.apache.org/licenses/LICENSE-2.0

    Unless required by applicable law or agreed to in writing, software
    distributed under the License is distributed on an "AS IS" BASIS,
    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    See the License for the specific language governing permissions and
    limitations under the License.
"""

import urllib.parse

import requests

import PyFunceble.facility
import PyFunceble.storage
from PyFunceble.query.requests.adapter.base import RequestAdapterBase


class RequestHTTPSAdapter(RequestAdapterBase):
    """
    Provides our HTTP adapter.
    """

    # pylint: disable=arguments-differ
    def send(self, request, **kwargs) -> requests.Response:
        """
        Overwrite the upstream :code:`send` method.

        We basically do the same. We only ensure that we request the IP from the chosen
        DNS record.

        :param request: The :class:`PreparedRequest <PreparedRequest>` being sent.
        :param stream: (optional) Whether to stream the request content.
        :param timeout: (optional) How long to wait for the server to send
            data before giving up, as a float, or
            a :ref:`(connect timeout, read timeout) <timeouts>` tuple.
        :type timeout: float or tuple or urllib3 Timeout object
        :param verify: (optional) Either a boolean, in which case it controls whether
            we verify the server's TLS certificate, or a string, in which case it
            must be a path to a CA bundle to use
        :param cert: (optional) Any user-provided SSL certificate to be trusted.
        :param proxies: (optional) The proxies dictionary to apply to the request.
        :rtype: requests.Response

        .. versionchanged:: 4.1.0b16
            When a proxy is given, it is acceptable to have an unresolvable
            subject. The proxy should handle the situation and give us back a
            proper status or error.
        """

        kwargs["timeout"] = self.timeout

        parsed_url = urllib.parse.urlparse(request.url)
        hostname_ip = self.resolve(parsed_url.hostname)

        kwargs["proxies"] = self.fetch_proxy_from_pattern(parsed_url.hostname)

        PyFunceble.facility.Logger.debug("Parsed URL: %r", parsed_url)
        PyFunceble.facility.Logger.debug("Resolved IP: %r", hostname_ip)
        PyFunceble.facility.Logger.debug("KWARGS: %r", kwargs)
        PyFunceble.facility.Logger.debug(
            "Pool Manager: %r", self.poolmanager.connection_pool_kw
        )

        if hostname_ip or kwargs["proxies"]:
            if hostname_ip:
                request.url = request.url.replace(
                    f"{parsed_url.scheme}://{parsed_url.hostname}",
                    f"{parsed_url.scheme}://{hostname_ip}",
                )

            if parsed_url.scheme == "https":
                self.poolmanager.connection_pool_kw[
                    "server_hostname"
                ] = parsed_url.hostname
                self.poolmanager.connection_pool_kw[
                    "assert_hostname"
                ] = parsed_url.hostname

            # Ensure that the Hosts header is present. Otherwise, connection might
            # not work.
            request.headers["Host"] = parsed_url.hostname
        else:
            self.poolmanager.connection_pool_kw.pop(
                "server_hostname", PyFunceble.storage.NOT_RESOLVED_STD_HOSTNAME
            )
            self.poolmanager.connection_pool_kw.pop(
                "assert_hostname", PyFunceble.storage.NOT_RESOLVED_STD_HOSTNAME
            )

            self.poolmanager.connection_pool_kw.pop(
                "server_hostname", parsed_url.hostname
            )
            self.poolmanager.connection_pool_kw.pop(
                "assert_hostname", parsed_url.hostname
            )

            return self.fake_response()

        response = super().send(request, **kwargs)

        if hostname_ip:
            response.url = response.url.replace(hostname_ip, parsed_url.hostname)

        return response

1	"""
2	The tool to check the availability or syntax of domain, IP or URL.
3
4	::
5
6
7	██████╗ ██╗ ██╗███████╗██╗ ██╗███╗ ██╗ ██████╗███████╗██████╗ ██╗ ███████╗
8	██╔══██╗╚██╗ ██╔╝██╔════╝██║ ██║████╗ ██║██╔════╝██╔════╝██╔══██╗██║ ██╔════╝
9	██████╔╝ ╚████╔╝ █████╗ ██║ ██║██╔██╗ ██║██║ █████╗ ██████╔╝██║ █████╗
10	██╔═══╝ ╚██╔╝ ██╔══╝ ██║ ██║██║╚██╗██║██║ ██╔══╝ ██╔══██╗██║ ██╔══╝
11	██║ ██║ ██║ ╚██████╔╝██║ ╚████║╚██████╗███████╗██████╔╝███████╗███████╗
12	╚═╝ ╚═╝ ╚═╝ ╚═════╝ ╚═╝ ╚═══╝ ╚═════╝╚══════╝╚═════╝ ╚══════╝╚══════╝
13
14	Provides the our HTTPS adapter.
15
16	Author:
17	Nissar Chababy, @funilrys, contactTATAfunilrysTODTODcom
18
19	Special thanks:
20	https://pyfunceble.github.io/#/special-thanks
21
22	Contributors:
23	https://pyfunceble.github.io/#/contributors
24
25	Project link:
26	https://github.com/funilrys/PyFunceble
27
28	Project documentation:
29	https://pyfunceble.readthedocs.io/en/dev/
30
31	Project homepage:
32	https://pyfunceble.github.io/
33
34	License:
35	::
36
37
38	Copyright 2017, 2018, 2019, 2020, 2022 Nissar Chababy
39
40	Licensed under the Apache License, Version 2.0 (the "License");
41	you may not use this file except in compliance with the License.
42	You may obtain a copy of the License at
43
44	http://www.apache.org/licenses/LICENSE-2.0
45
46	Unless required by applicable law or agreed to in writing, software
47	distributed under the License is distributed on an "AS IS" BASIS,
48	WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
49	See the License for the specific language governing permissions and
50	limitations under the License.
51	"""
52
53	import urllib.parse	12✔
54
55	import requests	12✔
56
57	import PyFunceble.facility	12✔
58	import PyFunceble.storage	12✔
59	from PyFunceble.query.requests.adapter.base import RequestAdapterBase	12✔
60
61
62	class RequestHTTPSAdapter(RequestAdapterBase):	12✔
63	"""
64	Provides our HTTP adapter.
65	"""
66
67	# pylint: disable=arguments-differ
68	def send(self, request, **kwargs) -> requests.Response:	12✔
69	"""
70	Overwrite the upstream :code:`send` method.
71
72	We basically do the same. We only ensure that we request the IP from the chosen
73	DNS record.
74
75	:param request: The :class:`PreparedRequest <PreparedRequest>` being sent.
76	:param stream: (optional) Whether to stream the request content.
77	:param timeout: (optional) How long to wait for the server to send
78	data before giving up, as a float, or
79	a :ref:`(connect timeout, read timeout) <timeouts>` tuple.
80	:type timeout: float or tuple or urllib3 Timeout object
81	:param verify: (optional) Either a boolean, in which case it controls whether
82	we verify the server's TLS certificate, or a string, in which case it
83	must be a path to a CA bundle to use
84	:param cert: (optional) Any user-provided SSL certificate to be trusted.
85	:param proxies: (optional) The proxies dictionary to apply to the request.
86	:rtype: requests.Response
87
88	.. versionchanged:: 4.1.0b16
89	When a proxy is given, it is acceptable to have an unresolvable
90	subject. The proxy should handle the situation and give us back a
91	proper status or error.
92	"""
93
94	kwargs["timeout"] = self.timeout	×
95
96	parsed_url = urllib.parse.urlparse(request.url)	×
97	hostname_ip = self.resolve(parsed_url.hostname)	×
98
99	kwargs["proxies"] = self.fetch_proxy_from_pattern(parsed_url.hostname)	×
100
101	PyFunceble.facility.Logger.debug("Parsed URL: %r", parsed_url)
102	PyFunceble.facility.Logger.debug("Resolved IP: %r", hostname_ip)
103	PyFunceble.facility.Logger.debug("KWARGS: %r", kwargs)
104	PyFunceble.facility.Logger.debug(
105	"Pool Manager: %r", self.poolmanager.connection_pool_kw
106	)
107
108	if hostname_ip or kwargs["proxies"]:	×
109	if hostname_ip:	×
110	request.url = request.url.replace(	×
111	f"{parsed_url.scheme}://{parsed_url.hostname}",
112	f"{parsed_url.scheme}://{hostname_ip}",
113	)
114
115	if parsed_url.scheme == "https":	×
116	self.poolmanager.connection_pool_kw[	×
117	"server_hostname"
118	] = parsed_url.hostname
119	self.poolmanager.connection_pool_kw[	×
120	"assert_hostname"
121	] = parsed_url.hostname
122
123	# Ensure that the Hosts header is present. Otherwise, connection might
124	# not work.
125	request.headers["Host"] = parsed_url.hostname	×
126	else:
127	self.poolmanager.connection_pool_kw.pop(	×
128	"server_hostname", PyFunceble.storage.NOT_RESOLVED_STD_HOSTNAME
129	)
130	self.poolmanager.connection_pool_kw.pop(	×
131	"assert_hostname", PyFunceble.storage.NOT_RESOLVED_STD_HOSTNAME
132	)
133
134	self.poolmanager.connection_pool_kw.pop(	×
135	"server_hostname", parsed_url.hostname
136	)
137	self.poolmanager.connection_pool_kw.pop(	×
138	"assert_hostname", parsed_url.hostname
139	)
140
141	return self.fake_response()	×
142
143	response = super().send(request, **kwargs)	×
144
145	if hostname_ip:	×
146	response.url = response.url.replace(hostname_ip, parsed_url.hostname)	×
147
148	return response	×

funilrys / PyFunceble / 3607917966

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous