SQL Query Log Formatter

sql_query_log_formatter.py

import logging
import sqlparse


class SQLFileHandler(logging.FileHandler):
    def emit(self, record):
        try:
            record.msg = sqlparse.format(record.msg, reindent=True, keyword_case='upper')
        except Exception:
            pass
        super().emit(record)


# Get logger
sql_logger = logging.getLogger('sqlalchemy.engine')
# Attach adapter
# Set level
sql_logger.setLevel(logging.INFO)
# Log formatter
log_formatter = logging.Formatter('[%(asctime)s] %(levelname)s: \n%(message)s')

file_handler = SQLFileHandler('logs/mylog.log')

file_handler.setFormatter(log_formatter)
sql_logger.addHandler(file_handler)