|
|
|
#!/usr/bin/python3
|
|
|
|
# -*- coding: utf-8 -*-
|
|
|
|
# key-mapper - GUI for device specific keyboard mappings
|
|
|
|
# Copyright (C) 2021 sezanzeb <proxima@sezanzeb.de>
|
|
|
|
#
|
|
|
|
# This file is part of key-mapper.
|
|
|
|
#
|
|
|
|
# key-mapper 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.
|
|
|
|
#
|
|
|
|
# key-mapper 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 key-mapper. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
|
|
|
|
|
|
|
|
"""Logging setup for key-mapper."""
|
|
|
|
|
|
|
|
|
|
|
|
import os
|
|
|
|
import time
|
|
|
|
import logging
|
|
|
|
import pkg_resources
|
|
|
|
|
|
|
|
|
|
|
|
SPAM = 5
|
|
|
|
|
|
|
|
start = time.time()
|
|
|
|
|
|
|
|
previous_key_spam = None
|
|
|
|
|
|
|
|
COMMIT_HASH = '12ff3df22e47a2e8b7be2811b300512c2d597725' # overwritten in setup.py
|
|
|
|
|
|
|
|
|
|
|
|
def spam(self, message, *args, **kwargs):
|
|
|
|
"""Log a more-verbose message than debug."""
|
|
|
|
# pylint: disable=protected-access
|
|
|
|
if self.isEnabledFor(SPAM):
|
|
|
|
# https://stackoverflow.com/a/13638084
|
|
|
|
self._log(SPAM, message, args, **kwargs)
|
|
|
|
|
|
|
|
|
|
|
|
def key_spam(self, key, msg, *args):
|
|
|
|
"""Log a spam message custom tailored to keycode_mapper.
|
|
|
|
|
|
|
|
Parameters
|
|
|
|
----------
|
|
|
|
key : tuple
|
|
|
|
anything that can be string formatted, but usually a tuple of
|
|
|
|
(type, code, value) tuples
|
|
|
|
"""
|
|
|
|
# pylint: disable=protected-access
|
|
|
|
if not self.isEnabledFor(SPAM):
|
|
|
|
return
|
|
|
|
|
|
|
|
global previous_key_spam
|
|
|
|
|
|
|
|
msg = msg % args
|
|
|
|
str_key = str(key)
|
|
|
|
str_key = str_key.replace(',)', ')')
|
|
|
|
spacing = ' ' + '-' * max(0, 30 - len(str_key))
|
|
|
|
if len(spacing) == 1:
|
|
|
|
spacing = ''
|
|
|
|
msg = f'{str_key}{spacing} {msg}'
|
|
|
|
|
|
|
|
if msg == previous_key_spam:
|
|
|
|
# avoid some super spam from EV_ABS events
|
|
|
|
return
|
|
|
|
|
|
|
|
previous_key_spam = msg
|
|
|
|
|
|
|
|
self._log(SPAM, msg, args=None)
|
|
|
|
|
|
|
|
|
|
|
|
logging.addLevelName(SPAM, "SPAM")
|
|
|
|
logging.Logger.spam = spam
|
|
|
|
logging.Logger.key_spam = key_spam
|
|
|
|
|
|
|
|
start = time.time()
|
|
|
|
|
|
|
|
LOG_PATH = os.path.expanduser('~/.log/key-mapper')
|
|
|
|
|
|
|
|
|
|
|
|
class Formatter(logging.Formatter):
|
|
|
|
"""Overwritten Formatter to print nicer logs."""
|
|
|
|
def format(self, record):
|
|
|
|
"""Overwritten format function."""
|
|
|
|
# pylint: disable=protected-access
|
|
|
|
debug = is_debug()
|
|
|
|
if record.levelno == logging.INFO and not debug:
|
|
|
|
# if not launched with --debug, then don't print "INFO:"
|
|
|
|
self._style._fmt = '%(message)s'
|
|
|
|
else:
|
|
|
|
# see https://en.wikipedia.org/wiki/ANSI_escape_code#3/4_bit
|
|
|
|
# for those numbers
|
|
|
|
color = {
|
|
|
|
logging.WARNING: 33,
|
|
|
|
logging.ERROR: 31,
|
|
|
|
logging.FATAL: 31,
|
|
|
|
logging.DEBUG: 36,
|
|
|
|
SPAM: 34,
|
|
|
|
logging.INFO: 32,
|
|
|
|
}.get(record.levelno, 0)
|
|
|
|
|
|
|
|
# if this runs in a separate process, write down the pid
|
|
|
|
# to debug exit codes and such
|
|
|
|
pid = ''
|
|
|
|
if os.getpid() != logger.main_pid:
|
|
|
|
pid = f'pid {os.getpid()}, '
|
|
|
|
|
|
|
|
if debug:
|
|
|
|
delta = f'{str(time.time() - start)[:7]}'
|
|
|
|
self._style._fmt = ( # noqa
|
|
|
|
f'\033[{color}m' # color
|
|
|
|
f'{delta} '
|
|
|
|
'\033[1m' # bold
|
|
|
|
f'%(levelname)s'
|
|
|
|
'\033[0m' # end style
|
|
|
|
f'\033[{color}m' # color
|
|
|
|
f': {pid}%(filename)s, line %(lineno)d, %(message)s'
|
|
|
|
'\033[0m' # end style
|
|
|
|
)
|
|
|
|
else:
|
|
|
|
self._style._fmt = ( # noqa
|
|
|
|
f'\033[{color}m%(levelname)s\033[0m: %(message)s'
|
|
|
|
)
|
|
|
|
return super().format(record)
|
|
|
|
|
|
|
|
|
|
|
|
logger = logging.getLogger()
|
|
|
|
handler = logging.StreamHandler()
|
|
|
|
handler.setFormatter(Formatter())
|
|
|
|
logger.addHandler(handler)
|
|
|
|
logger.setLevel(logging.INFO)
|
|
|
|
logging.getLogger('asyncio').setLevel(logging.WARNING)
|
|
|
|
logger.main_pid = os.getpid()
|
|
|
|
|
|
|
|
try:
|
|
|
|
name = pkg_resources.require('key-mapper')[0].project_name
|
|
|
|
version = pkg_resources.require('key-mapper')[0].version
|
|
|
|
evdev_version = pkg_resources.require('evdev')[0].version
|
|
|
|
except pkg_resources.DistributionNotFound as error:
|
|
|
|
name = 'key-mapper'
|
|
|
|
version = ''
|
|
|
|
evdev_version = None
|
|
|
|
logger.info('Could not figure out the version')
|
|
|
|
logger.debug(error)
|
|
|
|
|
|
|
|
|
|
|
|
def is_debug():
|
|
|
|
"""True, if the logger is currently in DEBUG or SPAM mode."""
|
|
|
|
return logger.level <= logging.DEBUG
|
|
|
|
|
|
|
|
|
|
|
|
def log_info():
|
|
|
|
"""Log version and name to the console"""
|
|
|
|
# read values from setup.py
|
|
|
|
logger.info(
|
|
|
|
'%s %s %s https://github.com/sezanzeb/key-mapper',
|
|
|
|
name, version, COMMIT_HASH
|
|
|
|
)
|
|
|
|
|
|
|
|
if evdev_version:
|
|
|
|
logger.info('python-evdev %s', evdev_version)
|
|
|
|
|
|
|
|
if is_debug():
|
|
|
|
logger.warning(
|
|
|
|
'Debug level will log all your keystrokes! Do not post this '
|
|
|
|
'output in the internet if you typed in sensitive or private '
|
|
|
|
'information with your device!'
|
|
|
|
)
|
|
|
|
|
|
|
|
logger.debug('pid %s', os.getpid())
|
|
|
|
|
|
|
|
|
|
|
|
def update_verbosity(debug):
|
|
|
|
"""Set the logging verbosity according to the settings object.
|
|
|
|
|
|
|
|
Also enable rich tracebacks in debug mode.
|
|
|
|
"""
|
|
|
|
# pylint really doesn't like what I'm doing with rich.traceback here
|
|
|
|
# pylint: disable=broad-except,import-error,import-outside-toplevel
|
|
|
|
|
|
|
|
if debug:
|
|
|
|
logger.setLevel(SPAM)
|
|
|
|
|
|
|
|
try:
|
|
|
|
from rich.traceback import install
|
|
|
|
install(show_locals=True)
|
|
|
|
logger.debug('Using rich.traceback')
|
|
|
|
except Exception as error:
|
|
|
|
# since this is optional, just skip all exceptions
|
|
|
|
if not isinstance(error, ImportError):
|
|
|
|
logger.debug('Cannot use rich.traceback: %s', error)
|
|
|
|
else:
|
|
|
|
logger.setLevel(logging.INFO)
|
|
|
|
|
|
|
|
|
|
|
|
def add_filehandler(path=LOG_PATH):
|
|
|
|
"""Clear the existing logfile and start logging to it."""
|
|
|
|
logger.info('This output is also stored in "%s"', LOG_PATH)
|
|
|
|
|
|
|
|
log_path = os.path.expanduser(path)
|
|
|
|
log_file = os.path.join(log_path, 'log')
|
|
|
|
|
|
|
|
os.makedirs(log_path, exist_ok=True)
|
|
|
|
|
|
|
|
if os.path.exists(log_file):
|
|
|
|
# keep the log path small, start from scratch each time
|
|
|
|
os.remove(log_file)
|
|
|
|
|
|
|
|
file_handler = logging.FileHandler(log_file)
|
|
|
|
file_handler.setFormatter(Formatter())
|
|
|
|
|
|
|
|
logger.info('Logging to "%s"', log_file)
|
|
|
|
|
|
|
|
logger.addHandler(file_handler)
|
|
|
|
|
|
|
|
return os.path.join(log_path, log_file)
|