From a2491056d2bf855ce35b2cfca8ae2a50c8f815ba Mon Sep 17 00:00:00 2001
From: eht17 <eht17@inf.ufpr.br>
Date: Mon, 27 May 2019 11:12:08 -0300
Subject: [PATCH] add function to set logging level

---
 database/actions.py | 13 +++++++++++++
 manage.py           |  3 ++-
 2 files changed, 15 insertions(+), 1 deletion(-)

diff --git a/database/actions.py b/database/actions.py
index 9ba4c38..29b4962 100644
--- a/database/actions.py
+++ b/database/actions.py
@@ -40,6 +40,19 @@ protocol_logger.setLevel(settings.LOGGING_LEVEL)
 sqlalchemy_logger = logging.getLogger('sqlalchemy.engine')
 sqlalchemy_logger.setLevel(settings.LOGGING_LEVEL)
 
+def set_logging_level(logging_level):
+    '''Modify priority level required for output messages'''
+    logging_level = logging_level.upper()
+    logging_level = {
+        'INFO':logging.INFO,
+        'WARNING':logging.WARNING,
+    }[logging_level]
+
+    database_table_logger.setLevel(logging_level)
+    protocol_logger.setLevel(logging_level)
+    sqlalchemy_logger.setLevel(logging_level)
+
+
 def temporary_data(connection, file_name, table, year, offset=2,
                    delimiters=[';', '\\n', '"'], null=''):
     header = open(file_name, encoding="ISO-8859-9").readline().strip()
diff --git a/manage.py b/manage.py
index a4dba50..c671265 100755
--- a/manage.py
+++ b/manage.py
@@ -37,8 +37,9 @@ def insert(csv_file, table, year, sep=';', null='',notifybackup=None):
     if notifybackup:
         database.actions.generate_backup()
 @manager.command
-def create(table):
+def create(table, logging_level="WARNING"):
     '''Creates table using mapping protocols'''
+    database.actions.set_logging_level(logging_level)
     database.actions.create(table)
 
 @manager.command
-- 
GitLab