add levels output

Author: welir

logs.py

@@ -17,6 +17,7 @@
     plfm = 'windows'
 
 
+
 class Log:
     file_path = os.getcwd()
     file_name = 'file_log'
@@ -56,15 +57,23 @@ def getFullPath(self):
     def getLogger(self):
         return self.logger
 
-    def addToLog(self, s):
+    def addToLog(self, s, type_output='info'):
         if plfm == 'windows':
             file_ = self.file_path + "\\logs\\" + self.file_name + "_" + strftime('%d_%m_%Y', gmtime()) + ".txt"
         else:
             file_ = self.file_path + "/logs/" + self.file_name + "_" + strftime('%d_%m_%Y', gmtime()) + ".txt"
 
         if not os.path.isfile(file_):
             self.init_logging()
-        self.getLogger().info(s)
+
+        output_level = {'info':    0, 'warning':   1, 'debug':  2}
+
+        if output_level[type_output] == 0:
+            self.getLogger().info(s)
+        elif output_level[type_output] == 1:
+            self.getLogger().warning(s)
+        else:
+            self.getLogger().debug(s)
 
     def initLogFolder(self):
         try:

test_loger/logs.py

@@ -3,4 +3,8 @@
 
 Log = Log(os.getcwd(), '_hello_world_logs')
 
-Log.addToLog('Hello world')
+Log.addToLog('Hello world', 'info')
+
+Log.addToLog('Hello world', 'warning')
+
+Log.addToLog('Hello world', 'debug')