92 lines
2.7 KiB
Python
Executable File
92 lines
2.7 KiB
Python
Executable File
"""
|
|
Onionr - Private P2P Communication
|
|
|
|
We use built in logging but with a custom formatter for colors and such
|
|
"""
|
|
import logging
|
|
|
|
from filepaths import log_file
|
|
"""
|
|
This program is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation, either version 3 of the License, or
|
|
(at your option) any later version.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
"""
|
|
|
|
# credit: https://stackoverflow.com/a/384076
|
|
# license: https://creativecommons.org/licenses/by-sa/4.0/
|
|
class ConsoleFormatter(logging.Formatter):
|
|
|
|
grey = "\x1b[38;20m"
|
|
green = "\x1b[38;5;82m"
|
|
yellow = "\x1b[33;20m"
|
|
red = "\x1b[31;20m"
|
|
bold_red = "\x1b[31;1m"
|
|
reset = "\x1b[0m"
|
|
format_default = "%(levelname)s - %(message)s (%(filename)s:%(lineno)d)"
|
|
format_info = "%(message)s - (%(filename)s:%(lineno)d)"
|
|
|
|
FORMATS = {
|
|
logging.DEBUG: grey + format_default + reset,
|
|
logging.INFO: green + format_info + reset,
|
|
logging.WARNING: yellow + format_default + reset,
|
|
logging.ERROR: red + format_default + reset,
|
|
logging.CRITICAL: bold_red + format_default + reset
|
|
}
|
|
|
|
def format(self, record):
|
|
log_fmt = self.FORMATS.get(record.levelno)
|
|
formatter = logging.Formatter(log_fmt)
|
|
return formatter.format(record)
|
|
|
|
class FileFormatter(logging.Formatter):
|
|
|
|
|
|
format_default = "%(levelname)s - %(message)s (%(filename)s:%(lineno)d)"
|
|
format_info = "%(message)s - (%(filename)s:%(lineno)d)"
|
|
|
|
FORMATS = {
|
|
logging.DEBUG: format_default,
|
|
logging.INFO: format_info,
|
|
logging.WARNING: format_default,
|
|
logging.ERROR: format_default,
|
|
logging.CRITICAL: format_default
|
|
}
|
|
|
|
def format(self, record):
|
|
log_fmt = self.FORMATS.get(record.levelno)
|
|
formatter = logging.Formatter(log_fmt)
|
|
return formatter.format(record)
|
|
|
|
|
|
#logging.basicConfig(level=logging.ERROR, format='%(message)s ')
|
|
log = logging.getLogger('onionr')
|
|
log.setLevel(logging.INFO)
|
|
|
|
ch = logging.StreamHandler()
|
|
ch.setLevel(logging.INFO)
|
|
|
|
ch.setFormatter(ConsoleFormatter())
|
|
|
|
|
|
def enable_file_logging():
|
|
fh = logging.FileHandler(log_file)
|
|
fh.setLevel(logging.INFO)
|
|
fh.setFormatter(FileFormatter())
|
|
log.addHandler(fh)
|
|
|
|
def disable_console_logging():
|
|
log.removeHandler(ch)
|
|
|
|
def enable_console_logging():
|
|
log.addHandler(ch)
|
|
enable_console_logging()
|