Change behaviour of no_save flag

Author: ccmorenos

pretty_verbose/__init__.py

@@ -3,12 +3,12 @@
 from pretty_verbose.messages_classes import VerboseMessages
 from pretty_verbose.processes_classes import Process, Task
 from pretty_verbose.error_classes import (
-    RunningError, LoggerError, LoggerErrorBase
+    RunningError, LoggerErrorBase, LoggerError, MissingLogFolderError
 )
 
 __all__ = [
     "VerboseMessages",
     "Task", "Process",
     "Logger",
-    "RunningError", "LoggerError", "LoggerErrorBase"
+    "RunningError", "LoggerError", "LoggerErrorBase", "MissingLogFolderError"
 ]

pretty_verbose/logger_classes.py

@@ -1,6 +1,7 @@
 """Classes of the logger."""
 from pathlib import Path
 
+from pretty_verbose.error_classes import MissingLogFolderError
 from pretty_verbose.processes_classes import Process
 
 
@@ -52,15 +53,16 @@ def __init__(
         )
 
         # Check if the directory exists.
-        if not self.log_dir.exists():
+        if not self.log_dir.exists() and not no_save:
             self.warning(f"The log dir '{self.log_dir}' does not exist!")
 
             if self.confirm("Do you want to create it?"):
                 self.log_dir.mkdir()
             else:
                 self.error(
-                    "Logger folder does not exist, exiting...",
-                    err_id=101, err_str="MISSING FOLDER"
+                    f"Logger folder '{self.log_dir}' does not exist",
+                    "exiting program...",
+                    err_class=MissingLogFolderError
                 )
 
         # Update no save.

pretty_verbose/messages_classes.py

@@ -6,7 +6,7 @@
 from pathlib import Path
 
 from pretty_verbose.constants import colors
-from pretty_verbose import RunningError, LoggerErrorBase
+from pretty_verbose import RunningError, LoggerErrorBase, MissingLogFolderError
 
 
 class VerboseMessages:
@@ -66,11 +66,9 @@ def __init__(
 
         # No save flag.
         self.no_save = no_save
-        if self.no_save:
-            return
-
-        # Init the log DataFrame.
-        self.start_log(overwrite)
+        if not self.no_save:
+            # Init the log DataFrame.
+            self.start_log(overwrite)
 
     def start_log(self, overwrite):
         """
@@ -85,6 +83,24 @@ def start_log(self, overwrite):
             Flag indicating whether overwrite or not the log file.
 
         """
+        # Check if the directory exists.
+        if not self.filename.parent.exists():
+            no_save, self.no_save = self.no_save, True
+            p_folder = self.filename.parent
+            self.warning(
+                f"The log dir '{p_folder}' does not exist!"
+            )
+
+            if self.confirm("Do you want to create it?"):
+                self.filename.parent.mkdir(parents=True)
+                self.no_save = no_save
+            else:
+                self.error(
+                    f"Logger folder '{p_folder}' does not exist",
+                    "exiting program...",
+                    err_class=MissingLogFolderError
+                )
+
         if not self.filename.exists() or overwrite:
             with open(
                 self.filename, "w", newline="", encoding="utf-8"
@@ -94,7 +110,7 @@ def start_log(self, overwrite):
                     ["message_type", "n_datetime", "message"]
                 )
 
-    def add_message(self, message_type, right_now, message):
+    def __add_message(self, message_type, right_now, message):
         """
         Add a new row to the log.
 
@@ -110,8 +126,6 @@ def add_message(self, message_type, right_now, message):
             Message text.
 
         """
-        if self.no_save:
-            return
         with open(self.filename, "a", newline="", encoding="utf-8") as file:
             log_messages = csv.writer(file, delimiter=self.sep)
             log_messages.writerow([message_type, right_now, message])
@@ -177,7 +191,8 @@ def get_terminal_columns(self):
             return 80
 
     def log(
-        self, min_level, name, color, *message, decorator=" ", end="\n"
+        self, min_level, name, color, *message, decorator=" ", end="\n",
+        skip_save=False
     ):
         """
         Print a log message with name, color and decorator.
@@ -202,6 +217,9 @@ def log(
         end: Str. Default: "\n".
             End of the line for the printing.
 
+        skip_save: Bool. Default: False.
+            Skip saving the log to a file.
+
         """
         if self.level >= min_level:
             # Get time in the given color.
@@ -218,8 +236,11 @@ def log(
                 end=end
             )
 
+            if self.no_save or skip_save:
+                return
+
             # Add message to log file.
-            self.add_message(name, now, f"{message}")
+            self.__add_message(name, now, f"{message}")
 
     def error(self, *message, err_id=0, err_str="", err_class=None):
         """