PyFunceble-dev 4.2.29__py3-none-any.whl → 4.3.0a1__py3-none-any.whl

PyFunceble/cli/entry_points/pyfunceble/cli.py

@@ -1197,6 +1197,28 @@ def get_default_group_data() -> List[Tuple[List[str], dict]]:
                 "version": "%(prog)s " + PyFunceble.storage.PROJECT_VERSION,
             },
         ),
+        (
+            [
+                "--config-file",
+            ],
+            {
+                "dest": "config_file",
+                "type": str,
+                "help": "Sets the configuration file to use. It can be a\n"
+                "local or remote file. Please note that this configuration can be\n"
+                "overwritten by your overwrite configuration file.",
+            },
+        ),
+        (
+            [
+                "--config-dir",
+            ],
+            {
+                "dest": "config_dir",
+                "type": os.path.realpath,
+                "help": "Sets the configuration directory to use.",
+            },
+        ),
     ]
 
 

PyFunceble/cli/filesystem/dir_structure/base.py

@@ -59,7 +59,6 @@ except ImportError:  # pragma: no cover ## Retro compatibility
     import importlib_resources as package_resources
 
 
-import PyFunceble.storage
 from PyFunceble.cli.filesystem.cleanup import FilesystemCleanup
 from PyFunceble.cli.filesystem.dir_base import FilesystemDirBase
 
@@ -78,7 +77,7 @@ class DirectoryStructureBase(FilesystemDirBase):
     ) -> None:
         with package_resources.path(
             "PyFunceble.data.infrastructure",
-            PyFunceble.storage.DISTRIBUTED_DIR_STRUCTURE_FILENAME,
+            "dir_structure_production.json",
         ) as file_path:
             self.std_source_file = str(file_path)
 

PyFunceble/cli/migrators/base.py

@@ -54,6 +54,7 @@ from typing import Optional
 
 from sqlalchemy.orm import Session
 
+import PyFunceble.storage
 from PyFunceble.cli.continuous_integration.base import ContinuousIntegrationBase
 
 
@@ -66,11 +67,19 @@ class MigratorBase:
     continuous_integration: Optional[ContinuousIntegrationBase] = None
     db_session: Optional[Session] = None
 
+    _config_dir: Optional[str] = None
     print_action_to_stdout: bool = False
 
-    def __init__(self, print_action_to_stdout: bool = False) -> None:
+    def __init__(
+        self, print_action_to_stdout: bool = False, *, config_dir: Optional[str] = None
+    ) -> None:
         self.print_action_to_stdout = print_action_to_stdout
 
+        if config_dir is not None:
+            self._config_dir = config_dir
+        else:
+            self._config_dir = PyFunceble.storage.CONFIG_DIRECTORY
+
         self.__post_init__()
 
     def __post_init__(self) -> None:
@@ -78,6 +87,43 @@ class MigratorBase:
         A method to be called (automatically) after the __init__ execution.
         """
 
+    @property
+    def config_dir(self) -> Optional[str]:
+        """
+        Provides the current state of the :code:`_config_dir` attribute.
+        """
+
+        return self._config_dir
+
+    @config_dir.setter
+    def config_dir(self, value: str) -> None:
+        """
+        Sets the configuration directory.
+
+        :param value:
+            The value to set.
+
+        :raise TypeError:
+            When value is not a :py:class:`str`.
+        """
+
+        if not isinstance(value, str):
+            raise TypeError(f"<value> should be {str}, {type(value)} given.")
+
+        self._config_dir = value
+
+    def set_config_dir(self, value: str) -> "MigratorBase":
+        """
+        Sets the configuration directory.
+
+        :param value:
+            The value to set.
+        """
+
+        self.config_dir = value
+
+        return self
+
     def start(self) -> "MigratorBase":
         """
         Starts the migration.

PyFunceble/cli/migrators/csv_file/inactive_source_delete.py

@@ -77,7 +77,7 @@ class InactiveDatasetDeleteSourceColumnMigrator(CSVFileMigratorBase):
 
     def __post_init__(self) -> None:
         self.source_file = os.path.join(
-            PyFunceble.storage.CONFIG_DIRECTORY, PyFunceble.cli.storage.INACTIVE_DB_FILE
+            self.config_dir, PyFunceble.cli.storage.INACTIVE_DB_FILE
         )
 
         return super().__post_init__()

PyFunceble/cli/migrators/csv_file/whois_registrar_add.py

@@ -71,7 +71,7 @@ class WhoisDatasetAddRegistrarColumnMigrator(CSVFileMigratorBase):
 
     def __post_init__(self) -> None:
         self.source_file = os.path.join(
-            PyFunceble.storage.CONFIG_DIRECTORY, PyFunceble.cli.storage.WHOIS_DB_FILE
+            self.config_dir, PyFunceble.cli.storage.WHOIS_DB_FILE
         )
 
         return super().__post_init__()

PyFunceble/cli/migrators/file_cleanup/hashes_file.py

@@ -64,7 +64,7 @@ class HashesFileCleanupMigrator(FileClenupMigratorBase):
 
     def __post_init__(self) -> None:
         self.source_file = os.path.join(
-            PyFunceble.storage.CONFIG_DIRECTORY, PyFunceble.cli.storage.HASHES_FILENAME
+            self.config_dir, PyFunceble.cli.storage.HASHES_FILENAME
         )
 
         return super().__post_init__()

PyFunceble/cli/migrators/file_cleanup/mining_file.py

@@ -64,7 +64,7 @@ class MiningFileCleanupMigrator(FileClenupMigratorBase):
 
     def __post_init__(self) -> None:
         self.source_file = os.path.join(
-            PyFunceble.storage.CONFIG_DIRECTORY, PyFunceble.cli.storage.MINING_OLD_FILE
+            self.config_dir, PyFunceble.cli.storage.MINING_OLD_FILE
         )
 
         return super().__post_init__()

PyFunceble/cli/migrators/file_cleanup/production_config_file.py

@@ -65,7 +65,7 @@ class ProductionConfigFileCleanupMigrator(FileClenupMigratorBase):
 
     def __post_init__(self) -> None:
         self.source_file = os.path.join(
-            PyFunceble.storage.CONFIG_DIRECTORY,
+            self.config_dir,
             PyFunceble.storage.DISTRIBUTED_CONFIGURATION_FILENAME,
         )
 

PyFunceble/cli/migrators/json2csv/inactive.py

@@ -75,7 +75,7 @@ class InactiveJSON2CSVMigrator(JSON2CSVMigratorBase):
 
     def __post_init__(self) -> None:
         self.source_file = os.path.join(
-            PyFunceble.storage.CONFIG_DIRECTORY,
+            self.config_dir,
             PyFunceble.cli.storage.INACTIVE_DB_OLD_FILE,
         )
 

PyFunceble/cli/migrators/json2csv/whois.py

@@ -73,7 +73,7 @@ class WhoisJSON2CSVMigrator(JSON2CSVMigratorBase):
 
     def __post_init__(self) -> None:
         self.source_file = os.path.join(
-            PyFunceble.storage.CONFIG_DIRECTORY,
+            self.config_dir,
             PyFunceble.cli.storage.WHOIS_DB_OLD_FILE,
         )
 

PyFunceble/cli/scripts/production.py

@@ -84,10 +84,6 @@ class ProductionPrep:
         you should run this only if your are developing PyFunceble.
     """
 
-    VERSION_FILE_PATH: str = os.path.join(
-        PyFunceble.storage.CONFIG_DIRECTORY,
-        PyFunceble.cli.storage.DISTRIBUTED_VERSION_FILENAME,
-    )
     AVAILABLE_BRANCHES: List[str] = ["dev", "master"]
 
     regex_helper: RegexHelper = RegexHelper()
@@ -101,15 +97,23 @@ class ProductionPrep:
     """
 
     _branch: Optional[str] = None
+    _config_dir: Optional[str] = None
 
     previous_version: Optional[str] = None
     """
     Provides the previous version (from :code:`version_file_content`)
     """
 
-    def __init__(self, branch: Optional[str] = None) -> None:
+    def __init__(
+        self, branch: Optional[str] = None, *, config_dir: Optional[str] = None
+    ) -> None:
+        if config_dir is not None:
+            self.config_dir = config_dir
+        else:
+            self.config_dir = PyFunceble.storage.CONFIG_DIRECTORY
+
         self.version_file_content = self.dict_helper.from_yaml_file(
-            self.VERSION_FILE_PATH
+            self.version_file_path
         )
 
         self.previous_version = copy.deepcopy(
@@ -138,6 +142,54 @@ class ProductionPrep:
 
         return wrapper
 
+    @property
+    def config_dir(self) -> Optional[str]:
+        """
+        Provides the current state of the :code:`_config_dir` attribute.
+        """
+
+        return self._config_dir
+
+    @config_dir.setter
+    def config_dir(self, value: str) -> None:
+        """
+        Sets the configuration directory.
+
+        :param value:
+            The value to set.
+
+        :raise TypeError:
+            When value is not a :py:class:`str`.
+        """
+
+        if not isinstance(value, str):
+            raise TypeError(f"<value> should be {str}, {type(value)} given.")
+
+        self._config_dir = value
+
+    def set_config_dir(self, value: str) -> "ProductionPrep":
+        """
+        Sets the configuration directory.
+
+        :param value:
+            The value to set.
+        """
+
+        self.config_dir = value
+
+        return self
+
+    @property
+    def version_file_path(self) -> str:
+        """
+        Provides the path to the version file.
+        """
+
+        return os.path.join(
+            self.config_dir,
+            PyFunceble.cli.storage.DISTRIBUTED_VERSION_FILENAME,
+        )
+
     @property
     def branch(self) -> Optional[str]:
         """
@@ -270,8 +322,7 @@ class ProductionPrep:
 
         return self
 
-    @staticmethod
-    def update_code_format() -> "ProductionPrep":
+    def update_code_format(self) -> "ProductionPrep":
         """
         Updates the format of the source code using black.
         """
@@ -304,16 +355,14 @@ class ProductionPrep:
         isort_config = isort.settings.Config(settings_file="setup.cfg")
 
         files = [
-            os.path.join(PyFunceble.storage.CONFIG_DIRECTORY, "setup.py"),
+            os.path.join(self.config_dir, "setup.py"),
         ]
 
         for file in files:
             format_file(file, isort_config)
 
         for root, _, files in os.walk(
-            os.path.join(
-                PyFunceble.storage.CONFIG_DIRECTORY, PyFunceble.storage.PROJECT_NAME
-            )
+            os.path.join(self.config_dir, PyFunceble.storage.PROJECT_NAME)
         ):
             if "__pycache__" in root:
                 continue
@@ -324,9 +373,7 @@ class ProductionPrep:
 
                 format_file(os.path.join(root, file), isort_config)
 
-        for root, _, files in os.walk(
-            os.path.join(PyFunceble.storage.CONFIG_DIRECTORY, "tests")
-        ):
+        for root, _, files in os.walk(os.path.join(self.config_dir, "tests")):
             if "__pycache__" in root:
                 continue
 
@@ -346,12 +393,10 @@ class ProductionPrep:
             ".keep",
         ]
 
-        self.update_urls(os.path.join(PyFunceble.storage.CONFIG_DIRECTORY, "setup.py"))
+        self.update_urls(os.path.join(self.config_dir, "setup.py"))
 
         for root, _, files in os.walk(
-            os.path.join(
-                PyFunceble.storage.CONFIG_DIRECTORY, PyFunceble.storage.PROJECT_NAME
-            )
+            os.path.join(self.config_dir, PyFunceble.storage.PROJECT_NAME)
         ):
             if "__pycache__" in root:
                 continue
@@ -362,9 +407,7 @@ class ProductionPrep:
 
                 self.update_urls(os.path.join(root, file))
 
-        for root, _, files in os.walk(
-            os.path.join(PyFunceble.storage.CONFIG_DIRECTORY, "tests")
-        ):
+        for root, _, files in os.walk(os.path.join(self.config_dir, "tests")):
             if "__pycache__" in root:
                 continue
 
@@ -409,9 +452,7 @@ class ProductionPrep:
                 ),
             ]
 
-        self.file_helper.set_path(
-            os.path.join(PyFunceble.storage.CONFIG_DIRECTORY, "setup.py")
-        )
+        self.file_helper.set_path(os.path.join(self.config_dir, "setup.py"))
 
         if not self.file_helper.exists():
             raise FileNotFoundError(self.file_helper.path)
@@ -453,7 +494,7 @@ class ProductionPrep:
         )
 
         self.dict_helper.set_subject(self.version_file_content).to_yaml_file(
-            self.VERSION_FILE_PATH
+            self.version_file_path
         )
 
         PyFunceble.facility.Logger.info(

PyFunceble/cli/system/integrator.py

@@ -52,6 +52,8 @@ License:
 """
 
 import os
+import sys
+import traceback
 
 import colorama
 
@@ -59,7 +61,9 @@ import PyFunceble.cli.facility
 import PyFunceble.cli.factory
 import PyFunceble.cli.storage
 import PyFunceble.facility
+import PyFunceble.helpers.exceptions
 import PyFunceble.storage
+from PyFunceble.cli.continuous_integration.exceptions import StopExecution
 from PyFunceble.cli.system.base import SystemBase
 from PyFunceble.helpers.dict import DictHelper
 
@@ -250,23 +254,56 @@ class SystemIntegrator(SystemBase):
         Starts a group of actions provided by this interface.
         """
 
-        if hasattr(self.args, "output_location") and self.args.output_location:
-            PyFunceble.cli.storage.OUTPUT_DIRECTORY = os.path.realpath(
-                os.path.join(
-                    self.args.output_location,
-                    PyFunceble.cli.storage.OUTPUTS.parent_directory,
-                )
-            )
+        try:
+            self.init_logger()
 
-        self.init_logger()
+            if hasattr(self.args, "output_location") and self.args.output_location:
+                PyFunceble.cli.storage.OUTPUT_DIRECTORY = os.path.realpath(
+                    os.path.join(
+                        self.args.output_location,
+                        PyFunceble.cli.storage.OUTPUTS.parent_directory,
+                    )
+                )
 
-        PyFunceble.facility.Logger.debug("Given arguments:\n%r", self.args)
+            if hasattr(self.args, "config_dir") and self.args.config_dir:
+                PyFunceble.facility.ConfigLoader.set_config_dir(self.args.config_dir)
+                PyFunceble.storage.CONFIG_DIRECTORY = self.args.config_dir
+
+            if hasattr(self.args, "config_file") and self.args.config_file:
+                PyFunceble.facility.ConfigLoader.set_remote_config_location(
+                    self.args.config_file
+                ).reload()
+
+            PyFunceble.facility.Logger.debug("Given arguments:\n%r", self.args)
+
+            self.inject_into_config()
+            self.check_config()
+            self.check_deprecated()
+
+            PyFunceble.cli.facility.CredentialLoader.start()
+            PyFunceble.cli.factory.DBSession.init_db_sessions()
+        except (KeyboardInterrupt, StopExecution):
+            pass
+        except Exception as exception:  # pylint: disable=broad-except
+            PyFunceble.facility.Logger.critical(
+                "Fatal error.",
+                exc_info=True,
+            )
+            if isinstance(exception, PyFunceble.helpers.exceptions.UnableToDownload):
+                message = (
+                    f"{colorama.Fore.RED}{colorama.Style.BRIGHT}Unable to download "
+                    f"{exception}"
+                )
+            else:
+                message = (
+                    f"{colorama.Fore.RED}{colorama.Style.BRIGHT}Fatal Error: "
+                    f"{exception}"
+                )
+            print(message)
 
-        self.inject_into_config()
-        self.check_config()
-        self.check_deprecated()
+            if PyFunceble.facility.Logger.authorized:
+                print(traceback.format_exc())
 
-        PyFunceble.cli.facility.CredentialLoader.start()
-        PyFunceble.cli.factory.DBSession.init_db_sessions()
+            sys.exit(1)
 
         return self

PyFunceble/cli/system/launcher.py

@@ -74,6 +74,7 @@ import PyFunceble.cli.utils.ascii_logo
 import PyFunceble.cli.utils.sort
 import PyFunceble.cli.utils.stdout
 import PyFunceble.facility
+import PyFunceble.helpers.exceptions
 import PyFunceble.storage
 from PyFunceble.checker.syntax.url import URLSyntaxChecker
 from PyFunceble.cli.continuous_integration.base import ContinuousIntegrationBase
@@ -1138,12 +1139,20 @@ class SystemLauncher(SystemBase):
                 "Fatal error.",
                 exc_info=True,
             )
-            print(
-                f"{colorama.Fore.RED}{colorama.Style.BRIGHT}Fatal Error: "
-                f"{exception}"
-            )
+            if isinstance(exception, PyFunceble.helpers.exceptions.UnableToDownload):
+                message = (
+                    f"{colorama.Fore.RED}{colorama.Style.BRIGHT}Unable to download "
+                    f"{exception}"
+                )
+            else:
+                message = (
+                    f"{colorama.Fore.RED}{colorama.Style.BRIGHT}Fatal Error: "
+                    f"{exception}"
+                )
+            print(message)
 
-            print(traceback.format_exc())
+            if PyFunceble.facility.Logger.authorized:
+                print(traceback.format_exc())
             sys.exit(1)
 
         PyFunceble.cli.utils.stdout.print_thanks()