Fixed error with non-string events

Author: SebiSebi

friendlylog/colored_logger.py

@@ -26,7 +26,7 @@ def _colorize(msg, loglevel):
         if loglevel not in [DEBUG, INFO, WARNING, ERROR, CRITICAL]:
             raise RuntimeError("{} should be oneof {}.".format(
                 loglevel, [DEBUG, INFO, WARNING, ERROR, CRITICAL]))  # pragma: no cover
-        msg = str(loglevel).upper() + ": " + msg
+        msg = str(loglevel).upper() + ": " + str(msg)
 
         if loglevel == DEBUG:
             return "{}{}{}{}{}".format(fg(14), attr(1), msg, attr(21), attr(0))  # noqa: E501

friendlylog/simple_logger.py

@@ -25,7 +25,7 @@ def _process(msg, loglevel):
         if loglevel not in [DEBUG, INFO, WARNING, ERROR, CRITICAL]:
             raise RuntimeError("{} should be oneof {}.".format(
                 loglevel, [DEBUG, INFO, WARNING, ERROR, CRITICAL]))  # pragma: no cover
-        msg = str(loglevel).upper() + ": " + msg
+        msg = str(loglevel).upper() + ": " + str(msg)
 
         return msg
 

tests/test_colored_logger.py

@@ -201,6 +201,15 @@ def test_terminal_logging(self):
         # 118 (the length without colors) + 4 coloring characters.
         self.assertGreaterEqual(len(self.last_line()), 118 + 4)
 
+    def test_non_str_logging(self):
+        logger.info(10)
+        self.assertIn("10", self.last_line())
+
+        # Those should not throw any error.
+        logger.debug([10, 20, 30])
+        logger.critical({})
+        logger.warning(set([-1, 4]))
+
 
 if __name__ == '__main__':
     unittest.main()

tests/test_simple_logger.py

@@ -195,6 +195,15 @@ def count_in(array, substr):
             for level in ["DEBUG", "INFO", "WARNING", "ERROR", "CRITICAL"]:
                 self.assertEqual(count_in(log, level + ": " + msg), 11)
 
+    def test_non_str_logging(self):
+        logger.info(-412413)
+        self.assertIn("-412413", self.last_line())
+
+        # Those should not throw any error.
+        logger.debug([101, 40, 35])
+        logger.critical({})
+        logger.warning(set([-1, 4, 10, -100]))
+
 
 if __name__ == '__main__':
     unittest.main()