PyFunceble-dev 4.3.0a13__tar.gz → 4.3.0a14__tar.gz

{pyfunceble_dev-4.3.0a13 → pyfunceble_dev-4.3.0a14}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: PyFunceble-dev
-Version: 4.3.0a13
+Version: 4.3.0a14
 Summary: The tool to check the availability or syntax of domain, IP or URL.
 Home-page: https://github.com/funilrys/PyFunceble
 Author: funilrys
@@ -22,169 +22,169 @@ Classifier: License :: OSI Approved
 Requires-Python: >=3.9, <4
 Description-Content-Type: text/markdown
 License-File: LICENSE
+Requires-Dist: colorama
+Requires-Dist: PyMySQL
+Requires-Dist: python-box[all]~=6.0.0
 Requires-Dist: requests[socks]<3
+Requires-Dist: packaging
+Requires-Dist: inflection
+Requires-Dist: shtab
 Requires-Dist: domain2idna~=1.12.0
 Requires-Dist: setuptools>=65.5.1
+Requires-Dist: dnspython[DOH]~=2.6.0
+Requires-Dist: SQLAlchemy~=2.0
 Requires-Dist: python-dotenv
-Requires-Dist: shtab
 Requires-Dist: PyYAML
-Requires-Dist: packaging
-Requires-Dist: inflection
-Requires-Dist: python-box[all]~=6.0.0
-Requires-Dist: PyMySQL
-Requires-Dist: dnspython[DOH]~=2.6.0
 Requires-Dist: alembic
-Requires-Dist: SQLAlchemy~=2.0
-Requires-Dist: colorama
 Provides-Extra: docs
-Requires-Dist: mkdocs-literate-nav~=0.6; extra == "docs"
-Requires-Dist: mkdocs-gen-files~=0.5; extra == "docs"
-Requires-Dist: mkdocstrings[python]~=0.26; extra == "docs"
-Requires-Dist: zipp>=3.19.1; extra == "docs"
+Requires-Dist: mkdocs-material~=9.5; extra == "docs"
+Requires-Dist: mkdocs-git-revision-date-localized-plugin~=1.2; extra == "docs"
 Requires-Dist: mkdocs-git-authors-plugin~=0.9; extra == "docs"
 Requires-Dist: pymdown-extensions~=10.9; extra == "docs"
-Requires-Dist: mkdocs-git-revision-date-localized-plugin~=1.2; extra == "docs"
-Requires-Dist: mkdocs-macros-plugin~=1.2; extra == "docs"
+Requires-Dist: mkdocs-gen-files~=0.5; extra == "docs"
 Requires-Dist: mkdocs~=1.5; extra == "docs"
+Requires-Dist: mkdocstrings[python]~=0.26; extra == "docs"
+Requires-Dist: mkdocs-literate-nav~=0.6; extra == "docs"
 Requires-Dist: mkdocs-section-index~=0.3; extra == "docs"
-Requires-Dist: mkdocs-material~=9.5; extra == "docs"
+Requires-Dist: zipp>=3.19.1; extra == "docs"
+Requires-Dist: mkdocs-macros-plugin~=1.2; extra == "docs"
 Provides-Extra: dev
 Requires-Dist: isort; extra == "dev"
-Requires-Dist: black; extra == "dev"
 Requires-Dist: flake8; extra == "dev"
 Requires-Dist: pylint; extra == "dev"
+Requires-Dist: black; extra == "dev"
 Provides-Extra: test
 Requires-Dist: coverage; extra == "test"
 Requires-Dist: tox; extra == "test"
 Provides-Extra: psql
+Requires-Dist: colorama; extra == "psql"
+Requires-Dist: PyMySQL; extra == "psql"
+Requires-Dist: python-box[all]~=6.0.0; extra == "psql"
 Requires-Dist: requests[socks]<3; extra == "psql"
+Requires-Dist: packaging; extra == "psql"
+Requires-Dist: inflection; extra == "psql"
+Requires-Dist: shtab; extra == "psql"
 Requires-Dist: domain2idna~=1.12.0; extra == "psql"
 Requires-Dist: setuptools>=65.5.1; extra == "psql"
+Requires-Dist: dnspython[DOH]~=2.6.0; extra == "psql"
+Requires-Dist: SQLAlchemy~=2.0; extra == "psql"
+Requires-Dist: psycopg2; extra == "psql"
 Requires-Dist: python-dotenv; extra == "psql"
-Requires-Dist: shtab; extra == "psql"
 Requires-Dist: PyYAML; extra == "psql"
-Requires-Dist: packaging; extra == "psql"
-Requires-Dist: psycopg2; extra == "psql"
-Requires-Dist: inflection; extra == "psql"
-Requires-Dist: python-box[all]~=6.0.0; extra == "psql"
-Requires-Dist: PyMySQL; extra == "psql"
-Requires-Dist: dnspython[DOH]~=2.6.0; extra == "psql"
 Requires-Dist: alembic; extra == "psql"
-Requires-Dist: SQLAlchemy~=2.0; extra == "psql"
-Requires-Dist: colorama; extra == "psql"
 Provides-Extra: psql-binary
-Requires-Dist: psycopg2-binary; extra == "psql-binary"
+Requires-Dist: colorama; extra == "psql-binary"
+Requires-Dist: PyMySQL; extra == "psql-binary"
+Requires-Dist: python-box[all]~=6.0.0; extra == "psql-binary"
 Requires-Dist: requests[socks]<3; extra == "psql-binary"
+Requires-Dist: packaging; extra == "psql-binary"
+Requires-Dist: inflection; extra == "psql-binary"
+Requires-Dist: shtab; extra == "psql-binary"
+Requires-Dist: psycopg2-binary; extra == "psql-binary"
 Requires-Dist: domain2idna~=1.12.0; extra == "psql-binary"
 Requires-Dist: setuptools>=65.5.1; extra == "psql-binary"
+Requires-Dist: dnspython[DOH]~=2.6.0; extra == "psql-binary"
+Requires-Dist: SQLAlchemy~=2.0; extra == "psql-binary"
 Requires-Dist: python-dotenv; extra == "psql-binary"
-Requires-Dist: shtab; extra == "psql-binary"
 Requires-Dist: PyYAML; extra == "psql-binary"
-Requires-Dist: packaging; extra == "psql-binary"
-Requires-Dist: inflection; extra == "psql-binary"
-Requires-Dist: python-box[all]~=6.0.0; extra == "psql-binary"
-Requires-Dist: PyMySQL; extra == "psql-binary"
-Requires-Dist: dnspython[DOH]~=2.6.0; extra == "psql-binary"
 Requires-Dist: alembic; extra == "psql-binary"
-Requires-Dist: SQLAlchemy~=2.0; extra == "psql-binary"
-Requires-Dist: colorama; extra == "psql-binary"
 Provides-Extra: postgresql
+Requires-Dist: colorama; extra == "postgresql"
+Requires-Dist: PyMySQL; extra == "postgresql"
+Requires-Dist: python-box[all]~=6.0.0; extra == "postgresql"
 Requires-Dist: requests[socks]<3; extra == "postgresql"
+Requires-Dist: packaging; extra == "postgresql"
+Requires-Dist: inflection; extra == "postgresql"
+Requires-Dist: shtab; extra == "postgresql"
 Requires-Dist: domain2idna~=1.12.0; extra == "postgresql"
 Requires-Dist: setuptools>=65.5.1; extra == "postgresql"
+Requires-Dist: dnspython[DOH]~=2.6.0; extra == "postgresql"
+Requires-Dist: SQLAlchemy~=2.0; extra == "postgresql"
+Requires-Dist: psycopg2; extra == "postgresql"
 Requires-Dist: python-dotenv; extra == "postgresql"
-Requires-Dist: shtab; extra == "postgresql"
 Requires-Dist: PyYAML; extra == "postgresql"
-Requires-Dist: packaging; extra == "postgresql"
-Requires-Dist: psycopg2; extra == "postgresql"
-Requires-Dist: inflection; extra == "postgresql"
-Requires-Dist: python-box[all]~=6.0.0; extra == "postgresql"
-Requires-Dist: PyMySQL; extra == "postgresql"
-Requires-Dist: dnspython[DOH]~=2.6.0; extra == "postgresql"
 Requires-Dist: alembic; extra == "postgresql"
-Requires-Dist: SQLAlchemy~=2.0; extra == "postgresql"
-Requires-Dist: colorama; extra == "postgresql"
 Provides-Extra: postgresql-binary
-Requires-Dist: psycopg2-binary; extra == "postgresql-binary"
+Requires-Dist: colorama; extra == "postgresql-binary"
+Requires-Dist: PyMySQL; extra == "postgresql-binary"
+Requires-Dist: python-box[all]~=6.0.0; extra == "postgresql-binary"
 Requires-Dist: requests[socks]<3; extra == "postgresql-binary"
+Requires-Dist: packaging; extra == "postgresql-binary"
+Requires-Dist: inflection; extra == "postgresql-binary"
+Requires-Dist: shtab; extra == "postgresql-binary"
+Requires-Dist: psycopg2-binary; extra == "postgresql-binary"
 Requires-Dist: domain2idna~=1.12.0; extra == "postgresql-binary"
 Requires-Dist: setuptools>=65.5.1; extra == "postgresql-binary"
+Requires-Dist: dnspython[DOH]~=2.6.0; extra == "postgresql-binary"
+Requires-Dist: SQLAlchemy~=2.0; extra == "postgresql-binary"
 Requires-Dist: python-dotenv; extra == "postgresql-binary"
-Requires-Dist: shtab; extra == "postgresql-binary"
 Requires-Dist: PyYAML; extra == "postgresql-binary"
-Requires-Dist: packaging; extra == "postgresql-binary"
-Requires-Dist: inflection; extra == "postgresql-binary"
-Requires-Dist: python-box[all]~=6.0.0; extra == "postgresql-binary"
-Requires-Dist: PyMySQL; extra == "postgresql-binary"
-Requires-Dist: dnspython[DOH]~=2.6.0; extra == "postgresql-binary"
 Requires-Dist: alembic; extra == "postgresql-binary"
-Requires-Dist: SQLAlchemy~=2.0; extra == "postgresql-binary"
-Requires-Dist: colorama; extra == "postgresql-binary"
 Provides-Extra: full
-Requires-Dist: requests[socks]<3; extra == "full"
-Requires-Dist: python-dotenv; extra == "full"
-Requires-Dist: pymdown-extensions~=10.9; extra == "full"
-Requires-Dist: PyYAML; extra == "full"
-Requires-Dist: tox; extra == "full"
-Requires-Dist: mkdocs-git-revision-date-localized-plugin~=1.2; extra == "full"
-Requires-Dist: mkdocs~=1.5; extra == "full"
 Requires-Dist: python-box[all]~=6.0.0; extra == "full"
-Requires-Dist: dnspython[DOH]~=2.6.0; extra == "full"
-Requires-Dist: mkdocs-literate-nav~=0.6; extra == "full"
-Requires-Dist: zipp>=3.19.1; extra == "full"
-Requires-Dist: flake8; extra == "full"
-Requires-Dist: mkdocstrings[python]~=0.26; extra == "full"
-Requires-Dist: setuptools>=65.5.1; extra == "full"
-Requires-Dist: black; extra == "full"
+Requires-Dist: mkdocs-material~=9.5; extra == "full"
 Requires-Dist: packaging; extra == "full"
-Requires-Dist: coverage; extra == "full"
-Requires-Dist: alembic; extra == "full"
-Requires-Dist: SQLAlchemy~=2.0; extra == "full"
+Requires-Dist: inflection; extra == "full"
+Requires-Dist: black; extra == "full"
 Requires-Dist: domain2idna~=1.12.0; extra == "full"
-Requires-Dist: mkdocs-git-authors-plugin~=0.9; extra == "full"
+Requires-Dist: setuptools>=65.5.1; extra == "full"
+Requires-Dist: SQLAlchemy~=2.0; extra == "full"
+Requires-Dist: mkdocstrings[python]~=0.26; extra == "full"
+Requires-Dist: PyYAML; extra == "full"
+Requires-Dist: alembic; extra == "full"
 Requires-Dist: mkdocs-macros-plugin~=1.2; extra == "full"
-Requires-Dist: PyMySQL; extra == "full"
-Requires-Dist: colorama; extra == "full"
-Requires-Dist: mkdocs-gen-files~=0.5; extra == "full"
+Requires-Dist: coverage; extra == "full"
+Requires-Dist: mkdocs-git-authors-plugin~=0.9; extra == "full"
 Requires-Dist: shtab; extra == "full"
-Requires-Dist: inflection; extra == "full"
+Requires-Dist: mkdocs-gen-files~=0.5; extra == "full"
+Requires-Dist: tox; extra == "full"
 Requires-Dist: isort; extra == "full"
-Requires-Dist: mkdocs-section-index~=0.3; extra == "full"
-Requires-Dist: mkdocs-material~=9.5; extra == "full"
+Requires-Dist: colorama; extra == "full"
+Requires-Dist: PyMySQL; extra == "full"
+Requires-Dist: pymdown-extensions~=10.9; extra == "full"
+Requires-Dist: mkdocs~=1.5; extra == "full"
 Requires-Dist: pylint; extra == "full"
+Requires-Dist: mkdocs-literate-nav~=0.6; extra == "full"
+Requires-Dist: python-dotenv; extra == "full"
+Requires-Dist: mkdocs-section-index~=0.3; extra == "full"
+Requires-Dist: requests[socks]<3; extra == "full"
+Requires-Dist: mkdocs-git-revision-date-localized-plugin~=1.2; extra == "full"
+Requires-Dist: dnspython[DOH]~=2.6.0; extra == "full"
+Requires-Dist: zipp>=3.19.1; extra == "full"
+Requires-Dist: flake8; extra == "full"
 Provides-Extra: all
-Requires-Dist: requests[socks]<3; extra == "all"
-Requires-Dist: python-dotenv; extra == "all"
-Requires-Dist: pymdown-extensions~=10.9; extra == "all"
-Requires-Dist: PyYAML; extra == "all"
-Requires-Dist: tox; extra == "all"
-Requires-Dist: mkdocs-git-revision-date-localized-plugin~=1.2; extra == "all"
-Requires-Dist: mkdocs~=1.5; extra == "all"
 Requires-Dist: python-box[all]~=6.0.0; extra == "all"
-Requires-Dist: dnspython[DOH]~=2.6.0; extra == "all"
-Requires-Dist: psycopg2-binary; extra == "all"
-Requires-Dist: mkdocs-literate-nav~=0.6; extra == "all"
-Requires-Dist: zipp>=3.19.1; extra == "all"
-Requires-Dist: flake8; extra == "all"
-Requires-Dist: mkdocstrings[python]~=0.26; extra == "all"
-Requires-Dist: setuptools>=65.5.1; extra == "all"
-Requires-Dist: black; extra == "all"
+Requires-Dist: mkdocs-material~=9.5; extra == "all"
 Requires-Dist: packaging; extra == "all"
-Requires-Dist: coverage; extra == "all"
-Requires-Dist: alembic; extra == "all"
-Requires-Dist: SQLAlchemy~=2.0; extra == "all"
+Requires-Dist: inflection; extra == "all"
+Requires-Dist: black; extra == "all"
 Requires-Dist: domain2idna~=1.12.0; extra == "all"
-Requires-Dist: mkdocs-git-authors-plugin~=0.9; extra == "all"
+Requires-Dist: setuptools>=65.5.1; extra == "all"
+Requires-Dist: SQLAlchemy~=2.0; extra == "all"
+Requires-Dist: mkdocstrings[python]~=0.26; extra == "all"
+Requires-Dist: PyYAML; extra == "all"
+Requires-Dist: alembic; extra == "all"
 Requires-Dist: mkdocs-macros-plugin~=1.2; extra == "all"
-Requires-Dist: PyMySQL; extra == "all"
-Requires-Dist: colorama; extra == "all"
-Requires-Dist: mkdocs-gen-files~=0.5; extra == "all"
+Requires-Dist: coverage; extra == "all"
+Requires-Dist: mkdocs-git-authors-plugin~=0.9; extra == "all"
 Requires-Dist: shtab; extra == "all"
-Requires-Dist: inflection; extra == "all"
+Requires-Dist: mkdocs-gen-files~=0.5; extra == "all"
+Requires-Dist: tox; extra == "all"
+Requires-Dist: psycopg2-binary; extra == "all"
 Requires-Dist: isort; extra == "all"
-Requires-Dist: mkdocs-section-index~=0.3; extra == "all"
-Requires-Dist: mkdocs-material~=9.5; extra == "all"
+Requires-Dist: colorama; extra == "all"
+Requires-Dist: PyMySQL; extra == "all"
+Requires-Dist: pymdown-extensions~=10.9; extra == "all"
+Requires-Dist: mkdocs~=1.5; extra == "all"
 Requires-Dist: pylint; extra == "all"
+Requires-Dist: mkdocs-literate-nav~=0.6; extra == "all"
+Requires-Dist: python-dotenv; extra == "all"
+Requires-Dist: mkdocs-section-index~=0.3; extra == "all"
+Requires-Dist: requests[socks]<3; extra == "all"
+Requires-Dist: mkdocs-git-revision-date-localized-plugin~=1.2; extra == "all"
+Requires-Dist: dnspython[DOH]~=2.6.0; extra == "all"
+Requires-Dist: zipp>=3.19.1; extra == "all"
+Requires-Dist: flake8; extra == "all"
 
 ![image](https://raw.githubusercontent.com/PyFunceble/logo/dev/Green/HD/RM.png)
 

{pyfunceble_dev-4.3.0a13 → pyfunceble_dev-4.3.0a14}/PyFunceble/checker/availability/base.py

@@ -64,6 +64,7 @@ import PyFunceble.storage
 from PyFunceble.checker.availability.extras.base import ExtraRuleHandlerBase
 from PyFunceble.checker.availability.extras.dns import DNSRulesHandler
 from PyFunceble.checker.availability.extras.etoxic import EToxicHandler
+from PyFunceble.checker.availability.extras.external import ExternalRulesHandler
 from PyFunceble.checker.availability.extras.rules import ExtraRulesHandler
 from PyFunceble.checker.availability.extras.subject_switch import (
     SubjectSwitchRulesHandler,
@@ -174,6 +175,7 @@ class AvailabilityCheckerBase(CheckerBase):
             DNSRulesHandler(),
             EToxicHandler(),
             ExtraRulesHandler(),
+            ExternalRulesHandler(rulesets=PyFunceble.storage.SPECIAL_RULES),
         ]
         self.db_session = db_session
 

{pyfunceble_dev-4.3.0a13 → pyfunceble_dev-4.3.0a14}/PyFunceble/checker/availability/extras/base.py

@@ -328,7 +328,7 @@ class ExtraRuleHandlerBase:
         def handle_regex_match_mode(_req: requests.Response):
             matches2search_result = {}
 
-            for header, loc_matches in matches:
+            for header, loc_matches in matches.items():
                 matches2search_result[header] = False
 
                 if header not in _req.headers:

pyfunceble_dev-4.3.0a14/PyFunceble/checker/availability/extras/external.py

@@ -0,0 +1,341 @@
+"""
+The tool to check the availability or syntax of domain, IP or URL.
+
+::
+
+
+    ██████╗ ██╗   ██╗███████╗██╗   ██╗███╗   ██╗ ██████╗███████╗██████╗ ██╗     ███████╗
+    ██╔══██╗╚██╗ ██╔╝██╔════╝██║   ██║████╗  ██║██╔════╝██╔════╝██╔══██╗██║     ██╔════╝
+    ██████╔╝ ╚████╔╝ █████╗  ██║   ██║██╔██╗ ██║██║     █████╗  ██████╔╝██║     █████╗
+    ██╔═══╝   ╚██╔╝  ██╔══╝  ██║   ██║██║╚██╗██║██║     ██╔══╝  ██╔══██╗██║     ██╔══╝
+    ██║        ██║   ██║     ╚██████╔╝██║ ╚████║╚██████╗███████╗██████╔╝███████╗███████╗
+    ╚═╝        ╚═╝   ╚═╝      ╚═════╝ ╚═╝  ╚═══╝ ╚═════╝╚══════╝╚═════╝ ╚══════╝╚══════╝
+
+Provides the extra rules handler based on some DNS records.
+
+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://docs.pyfunceble.com
+
+Project homepage:
+    https://pyfunceble.github.io/
+
+License:
+::
+
+
+    Copyright 2017, 2018, 2019, 2020, 2022, 2023, 2024 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
+
+        https://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.
+"""
+
+# pylint: disable=line-too-long
+
+from typing import Optional
+
+from PyFunceble.checker.availability.extras.base import ExtraRuleHandlerBase
+from PyFunceble.checker.availability.status import AvailabilityCheckerStatus
+
+
+class ExternalRulesHandler(ExtraRuleHandlerBase):
+    """
+    Provides the external rules handler that is used to handle the external
+    provided rules.
+
+    Through this handler, end-user can provide their own rules to handle
+    the availability status of a subject.
+
+    :param status:
+        The previously gathered status.
+    :type status:
+        :class:`~PyFunceble.checker.availability.status.AvailabilityCheckerStatus`
+    """
+
+    rulesets: list = []
+    """
+    The rulesets to process.
+
+    If you want to switch from the status code, you should provide a dict
+    with the following structure:
+
+        {
+            "subject_pattern": ".*", // The pattern the subject should match.
+            "validation_type": "status_code", // Type of validation (status_code, headers, body, etc.)
+            "state_transition": "up", // "up" -> ACTIVE, "down" -> INACTIVE
+            "required_status_code": [404], // Status code to match.
+        }
+
+    If you want to switch from the headers, you should provide a dict
+
+        {
+            "subject_pattern": ".*", // The pattern the subject should match.
+            "validation_type": "headers", // Type of validation (status_code, headers, body, etc.)
+            "state_transition": "up", // "up" -> ACTIVE, "down" -> INACTIVE
+            "required_headers_patterns": { // Required, the headers to match.
+                "header_name": ["possible", "values"]
+            },
+        }
+
+    If you want to switch from the body, you should provide a dict
+
+        {
+            "subject_pattern": ".*", // The pattern the subject should match.
+            "validation_type": "body", // Type of validation (status_code, headers, body, etc.)
+            "state_transition": "up", // "up" -> ACTIVE, "down" -> INACTIVE
+            "required_body_patterns": ["regex1", "regex2"] // Required, the body patterns to match.
+        }
+
+    If you want to switch from a combination of headers and body, you should provide a dict
+
+        {
+            "subject_pattern": ".*", // The pattern the subject should match.
+            "validation_type": "headers+body", // Type of validation (status_code, headers, body, etc.)
+            "state_transition": "up", // "up" -> ACTIVE, "down" -> INACTIVE
+            "required_headers_patterns": { // Required, the headers to match.
+                "header_name": ["possible", "values"]
+            },
+            "required_body_patterns": ["regex1", "regex2"] // Required, the body patterns to match.
+        }
+
+    If you want to switch from a combination of all, you should provide a dict
+
+        {
+            "subject_pattern": ".*", // The pattern the subject should match.
+            "validation_type": "all", // Type of validation (status_code, headers, body, etc.)
+            "state_transition": "up", // "up" -> ACTIVE, "down" -> INACTIVE
+            "required_status_code": [404], // Optional, Status code to match.
+            "required_headers_patterns": { // Optional, the headers to match.
+                "header_name": ["possible", "values"]
+            },
+            "required_body_patterns": ["regex1", "regex2"] // Optional, the body patterns to match.
+        }
+
+    """
+
+    def __init__(
+        self,
+        status: Optional[AvailabilityCheckerStatus] = None,
+        *,
+        rulesets: list = None
+    ) -> None:
+        if rulesets is not None:
+            self.rulesets = rulesets
+
+        super().__init__(status)
+
+    def switch_from_status_code_rule(self, rule: dict) -> "ExternalRulesHandler":
+        """
+        Switch from the status code rule.
+
+        :param rule:
+            The rule to switch from.
+        :type rule: dict
+        """
+
+        required_keys = ["validation_type", "required_status_code"]
+
+        if any(x not in rule for x in required_keys):
+            return self
+
+        if rule["validation_type"] != "status_code":
+            return self
+
+        if all(
+            self.status.http_status_code != int(x) for x in rule["required_status_code"]
+        ):
+            return self
+
+        if rule["state_transition"] == "up":
+            return self.switch_to_up()
+
+        if rule["state_transition"] == "down":
+            return self.switch_to_down()
+
+        return self
+
+    def switch_from_headers_rule(self, rule: dict) -> "ExternalRulesHandler":
+        """
+        Switch from the headers rule.
+
+        :param rule:
+            The rule to switch from.
+        :type rule: dict
+        """
+
+        required_keys = ["validation_type", "required_headers_patterns"]
+
+        if any(x not in rule for x in required_keys):
+            return self
+
+        if rule["validation_type"] != "headers":
+            return self
+
+        if rule["state_transition"] == "up":
+            switch_method = self.switch_to_up
+
+        if rule["state_transition"] == "down":
+            switch_method = self.switch_to_down
+
+        if "required_headers_patterns" in rule and rule["required_headers_patterns"]:
+            # pylint: disable=possibly-used-before-assignment
+            self.do_on_header_match(
+                self.req_url,
+                rule["required_headers_patterns"],
+                method=switch_method,
+                strict=False,
+                allow_redirects=False,
+            )
+
+        return self
+
+    def switch_from_body_rule(self, rule: dict) -> "ExternalRulesHandler":
+        """
+        Switch from the body rule.
+
+        :param rule:
+            The rule to switch from.
+        :type rule: dict
+        """
+
+        required_keys = ["validation_type", "required_body_patterns"]
+
+        if any(x not in rule for x in required_keys):
+            return self
+
+        if rule["validation_type"] != "body":
+            return self
+
+        if rule["state_transition"] == "up":
+            switch_method = self.switch_to_up
+
+        if rule["state_transition"] == "down":
+            switch_method = self.switch_to_down
+
+        if "required_body_patterns" in rule and rule["required_body_patterns"]:
+            # pylint: disable=possibly-used-before-assignment
+            self.do_on_body_match(
+                self.req_url,
+                rule["required_body_patterns"],
+                method=switch_method,
+                strict=False,
+                allow_redirects=False,
+            )
+
+        return self
+
+    def switch_from_all_rule(self, rule: dict) -> "ExternalRulesHandler":
+        """
+        Switch from the all rule.
+
+        :param rule:
+            The rule to switch from.
+        :type rule: dict
+        """
+
+        required_keys = [
+            "validation_type",
+        ]
+
+        if any(x not in rule for x in required_keys):
+            return self
+
+        if rule["validation_type"] != "all":
+            return self
+
+        if rule["state_transition"] == "up":
+            switch_method = self.switch_to_up
+
+        if rule["state_transition"] == "down":
+            switch_method = self.switch_to_down
+
+        if (
+            "required_status_code" in rule
+            and rule["required_status_code"]
+            and any(
+                self.status.http_status_code == int(x)
+                for x in rule["required_status_code"]
+            )
+        ):
+            # pylint: disable=possibly-used-before-assignment
+            switch_method()
+
+        if "required_headers_patterns" in rule and rule["required_headers_patterns"]:
+            self.do_on_header_match(
+                self.req_url,
+                rule["required_headers_patterns"],
+                method=switch_method,
+                strict=False,
+                allow_redirects=False,
+            )
+
+        if "required_body_patterns" in rule and rule["required_body_patterns"]:
+            self.do_on_body_match(
+                self.req_url,
+                rule["required_body_patterns"],
+                method=switch_method,
+                strict=False,
+                allow_redirects=False,
+            )
+
+        return self
+
+    @ExtraRuleHandlerBase.ensure_status_is_given
+    @ExtraRuleHandlerBase.setup_status_before
+    @ExtraRuleHandlerBase.setup_status_after
+    def start(self) -> "ExternalRulesHandler":
+        """
+        Process the check and handling of the external rules for the given subject.
+        """
+
+        required_keys = ["subject_pattern", "validation_type", "state_transition"]
+
+        for rule in self.rulesets:
+            if any(x not in rule for x in required_keys):
+                continue
+
+            if not self.regex_helper.set_regex(rule["subject_pattern"]).match(
+                self.status.netloc, return_match=False
+            ):
+                continue
+
+            if rule["state_transition"] not in ["up", "down"]:
+                continue
+
+            if self.status.status_after_extra_rules:
+                # We already switched the status.
+                break
+
+            if rule["validation_type"] == "status_code":
+                self.switch_from_status_code_rule(rule)
+            elif rule["validation_type"] == "headers":
+                self.switch_from_headers_rule(rule)
+            elif rule["validation_type"] == "body":
+                self.switch_from_body_rule(rule)
+            elif rule["validation_type"] == "headers+body":
+                self.switch_from_headers_rule(rule)
+                self.switch_from_body_rule(rule)
+            elif rule["validation_type"] == "all":
+                self.switch_from_all_rule(rule)
+
+        return self

{pyfunceble_dev-4.3.0a13 → pyfunceble_dev-4.3.0a14}/PyFunceble/config/loader.py

@@ -651,6 +651,9 @@ class ConfigLoader:
         if "proxy" in config and config["proxy"]:
             PyFunceble.storage.PROXY = Box(config["proxy"])
 
+        if "special_rules" in config and config["special_rules"]:
+            PyFunceble.storage.SPECIAL_RULES = config["special_rules"]
+
         # Early load user agents to allow usage of defined user agents.
         UserAgentDataset().get_latest()
 
@@ -676,6 +679,7 @@ class ConfigLoader:
             PyFunceble.storage.PLATFORM = Box({})
             PyFunceble.storage.LINKS = Box({})
             PyFunceble.storage.PROXY = Box({})
+            PyFunceble.storage.SPECIAL_RULES = Box({})
         except (AttributeError, TypeError):  # pragma: no cover ## Safety.
             pass