feat: Provide function to enable/configure logger

src/aria2p/__init__.py

@@ -8,6 +8,8 @@
 please refer to the README.md included in this package to get the link to the official documentation.
 """
 
+import sys
+
 from loguru import logger
 
 from .api import API
@@ -18,4 +20,21 @@
 
 logger.disable("aria2p")
 
-__all__ = ["API", "ClientException", "Client", "Download", "BitTorrent", "File", "Options", "Stats"]
+
+def enable_logger(sink=sys.stderr, level="WARNING"):
+    """
+    Enable the logging of messages.
+
+    Configure the ``logger`` variable imported from ``loguru``.
+
+    Args:
+        sink (file): An opened file pointer, or stream handler. Default to standard error.
+        level (str): The log level to use. Possible values are TRACE, DEBUG, INFO, WARNING, ERROR, CRITICAL.
+            Default to WARNING.
+    """
+    logger.remove()
+    logger.configure(handlers=[{"sink": sink, "level": level}])
+    logger.enable("aria2p")
+
+
+__all__ = ["API", "ClientException", "Client", "Download", "BitTorrent", "File", "Options", "Stats", "enable_logger"]

src/aria2p/cli.py

@@ -22,7 +22,7 @@
 import requests
 from loguru import logger
 
-from aria2p import Download
+from aria2p import Download, enable_logger
 
 from .api import API
 from .client import DEFAULT_HOST, DEFAULT_PORT, Client, ClientException
@@ -36,9 +36,7 @@ def main(args=None):
     args = parser.parse_args(args=args)
     kwargs = args.__dict__
 
-    logger.remove()
-    logger.configure(handlers=[{"sink": sys.stderr, "level": kwargs.pop("log_level")}])
-    logger.enable("aria2p")
+    enable_logger(kwargs.pop("log_level"))
 
     check_args(parser, args)