utils.debug: Allow passing logger to log_time.

qutebrowser/utils/debug.py

@@ -21,6 +21,7 @@
 
 import re
 import inspect
+import logging
 import functools
 import datetime
 import contextlib
@@ -229,9 +230,11 @@ def log_time(logger, action='operation'):
     """Log the time the operation in the with-block takes.
 
     Args:
-        logger: The logging.Logger to use for logging.
+        logger: The logging.Logger to use for logging, or a logger name.
         action: A description of what's being done.
     """
+    if isinstance(logger, str):
+        logger = logging.getLogger(logger)
     started = datetime.datetime.now()
     try:
         yield

tests/unit/utils/test_debug.py

@@ -81,11 +81,13 @@ def test_log_signals(caplog, signal_obj):
     assert records[1].msg == "Signal in <repr>: signal2('foo', 'bar')"
 
 
-def test_log_time(caplog):
+class TestLogTime:
+
+    def test_duration(self, caplog):
         logger_name = 'qt-tests'
 
         with caplog.atLevel(logging.DEBUG, logger_name):
-        with debug.log_time(logging.getLogger(logger_name), action='foobar'):
+            with debug.log_time(logger_name, action='foobar'):
                 time.sleep(0.1)
 
             records = caplog.records()
@@ -98,6 +100,16 @@ def test_log_time(caplog):
             duration = float(match.group(1))
             assert 0 < duration < 5
 
+    def test_logger(self, caplog):
+        """Test with an explicit logger instead of a name."""
+        logger_name = 'qt-tests'
+
+        with caplog.atLevel(logging.DEBUG, logger_name):
+            with debug.log_time(logging.getLogger(logger_name)):
+                pass
+
+        assert len(caplog.records()) == 1
+
 
 class TestQEnumKey: