improve logging

Author: arch

funscript_editor/__main__.py

@@ -10,6 +10,7 @@ def main():
     parser = argparse.ArgumentParser()
     parser.add_argument("--generator", action = 'store_true', help = "Run only the generator")
     parser.add_argument("--multiaxis", action = 'store_true', help = "Show options for multiaxis output")
+    parser.add_argument("--logs", action = 'store_true', help = "Enable logging")
     parser.add_argument("-i", "--input", type = str, help = "Video File")
     parser.add_argument("-o", "--output", type = str, default = "/tmp/funscript_actions.json", help = "Output Path")
     parser.add_argument("-s", "--start", type = float, default = 0.0, help = "Start Time in Milliseconds")
@@ -20,4 +21,4 @@ def main():
         os.environ['PATH'] = os.getcwd() + os.sep + os.environ['PATH']
 
     if not args.generator: show_editor()
-    else: generate_funscript(args.input, args.start, args.end, args.output, args.multiaxis)
+    else: generate_funscript(args.input, args.start, args.end, args.output, args.multiaxis, args.logs)

funscript_editor/api.py

@@ -27,7 +27,8 @@ def generate_funscript(
         start_time: float,
         end_time :float,
         output_file: str,
-        include_multiaxis_options: bool = False) -> None:
+        include_multiaxis_options: bool = False,
+        enable_logging: bool = False) -> None:
     """ Generate a funscript with minimal UI
 
     Args:
@@ -37,7 +38,10 @@ def generate_funscript(
         output_file (str): path for the output file
         include_multiaxis_options (bool): include options for multiaxis output
     """
-    logging.setup_logging(silent=True)
+    if enable_logging:
+        logging.setup_logging(silent=True)
+    else:
+        logging.disable_logging()
     logger = logging.getLogger(__name__)
     logger.info("Python Funscript Generator %s", VERSION)
     logger.info("Startup Path: %s", str(os.getcwd()))

funscript_editor/utils/logging.py

@@ -138,7 +138,7 @@ def get_logfiles_paths() -> list:
 
 
 def setup_logging(
-        silent=False,
+        silent: bool = False,
         default_level :int = logging.INFO,
         env_key :str = 'LOG_CFG') -> None:
     """ Logging Setup
@@ -173,3 +173,7 @@ def setup_logging(
         if not silent:
             coloredlogs.install(level=default_level)
         print('Failed to load configuration file. Using default configs')
+
+
+def disable_logging():
+    logging.disable(logging.CRITICAL)