feat: added return to log message

Author: jlerat

src/hydrodiy/io/iutils.py

@@ -140,19 +140,27 @@ def add_tab(self, msg, ntab):
         tab_space = " " * self.tab_length * ntab
         return tab_space + msg
 
-    def error(self, msg, ntab=0, *args, **kwargs):
+    def add_ret(self, nret):
+        for i in range(nret):
+            self._logger.info("")
+
+    def error(self, msg, ntab=0, nret=0, *args, **kwargs):
+        self.add_ret(nret)
         msg = self.add_tab(msg, ntab)
         return self._logger.error(msg, *args, **kwargs)
 
-    def info(self, msg, ntab=0, *args, **kwargs):
+    def info(self, msg, ntab=0, nret=0, *args, **kwargs):
+        self.add_ret(nret)
         msg = self.add_tab(msg, ntab)
         return self._logger.info(msg, *args, **kwargs)
 
-    def warning(self, msg, ntab=0, *args, **kwargs):
+    def warning(self, msg, ntab=0, nret=0, *args, **kwargs):
+        self.add_ret(nret)
         msg = self.add_tab(msg, ntab)
         return self._logger.warning(msg, *args, **kwargs)
 
-    def critical(self, msg, ntab=0, *args, **kwargs):
+    def critical(self, msg, ntab=0, nret=0, *args, **kwargs):
+        self.add_ret(nret)
         msg = self.add_tab(msg, ntab)
         return self._logger.critical(msg, *args, **kwargs)
 
@@ -236,19 +244,23 @@ def get_message(self, msg, ntab):
             return "{{ {0} }} {1}{2}".format(self.context, tab, msg)
         return msg
 
-    def error(self, msg, ntab=0, *args, **kwargs):
+    def error(self, msg, ntab=0, nret=0, *args, **kwargs):
+        self.add_ret(nret)
         msg = self.get_message(msg, ntab)
         return self._logger.error(msg, *args, **kwargs)
 
-    def info(self, msg, ntab=0, *args, **kwargs):
+    def info(self, msg, ntab=0, nret=0, *args, **kwargs):
+        self.add_ret(nret)
         msg = self.get_message(msg, ntab)
         return self._logger.info(msg, *args, **kwargs)
 
-    def warning(self, msg, ntab=0, *args, **kwargs):
+    def warning(self, msg, ntab=0, nret=0, *args, **kwargs):
+        self.add_ret(nret)
         msg = self.get_message(msg, ntab)
         return self._logger.warning(msg, *args, **kwargs)
 
-    def critical(self, msg, ntab=0, *args, **kwargs):
+    def critical(self, msg, ntab=0, nret=0, *args, **kwargs):
+        self.add_ret(nret)
         msg = self.get_message(msg, ntab)
         return self._logger.critical(msg, *args, **kwargs)
 

src/hydrodiy/io/tests/test_hyio_iutils.py

@@ -152,8 +152,8 @@ def test_get_logger():
                     flog=flog2)
 
     mess = ["flog2 A", "flog2 B"]
-    logger2.warning(mess[0])
-    logger2.critical(mess[1])
+    logger2.warning(mess[0], nret=2)
+    logger2.critical(mess[1], ntab=2)
     logger2.completed()
 
     assert flog2.exists()
@@ -162,7 +162,7 @@ def test_get_logger():
         txt = fl.readlines()
     expected = ["@@@ Process started @@@",
                 logger2.separator_charac*logger2.separator_length,
-                ""] +\
+                "", "", ""] +\
                mess +\
                ["",
                 logger2.separator_charac*logger2.separator_length,