Added output parameter to log() method (#1)

sevketcaba · Nov 1, 2023 · 91e5956 · 91e5956
1 parent 0dd13e6
commit 91e5956
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/uglylogger/logger.py b/src/uglylogger/logger.py
@@ -408,6 +408,7 @@ def log(
         msg: typing.Any,
         color: LogColor | None = None,
         level: LogLevel = LogLevel.DEBUG,
+        output: LogOutput = LogOutput.ALL,
     ):
         """Logs both to the file and to the console
 
@@ -416,9 +417,12 @@ def log(
             color (LogColor | None, optional): Color to overwrite,
                 otherwise uses color by the LogLevel. Defaults to None.
             level (LogLevel, optional): Defaults to LogLevel.DEBUG.
+            output (LogOutput, optional): Defaults to LogOutput.ALL.
         """
-        self.console(msg, color, level)
-        self.file(msg, level)
+        if LogOutput.CONSOLE in output:
+            self.console(msg, color, level)
+        if LogOutput.FILE in output:
+            self.file(msg, level)
 
     def debug(
         self,