1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83
|
# localslackirc
# Copyright (C) 2020 Salvo "LtWorf" Tomaselli
#
# localslackirc 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 <http://www.gnu.org/licenses/>.
#
# author Salvo "LtWorf" Tomaselli <tiposchi@tiscali.it>
from os import isatty
import sys
from syslog import LOG_INFO, LOG_DEBUG, syslog, LOG_ERR, LOG_WARNING
from syslog import openlog as _openlog
__all__ = [
'log',
'openlog',
'set_debug',
'debug',
'error',
'abort',
'warning',
]
tty = isatty(1) and isatty(2)
debug_enabled = False
def openlog(suffix: str) -> None:
"""
Opens the syslog connection if needed
otherwise does nothing.
"""
if tty:
return
if suffix:
suffix = f'-{suffix}'
_openlog(f'localslackirc{suffix}')
def log(*args, log_level=LOG_INFO) -> None:
"""
Logs to stdout or to syslog depending on if
running with a terminal attached.
"""
if tty:
if log_level in {LOG_DEBUG, LOG_INFO}:
f = sys.stdout
else:
f = sys.stderr
print(*args,file=f)
return
syslog(log_level, ' '.join(str(i) for i in args))
def set_debug(state: str|bool) -> None:
global debug_enabled
debug_enabled = bool(state)
def debug(*args) -> None:
if debug_enabled:
log(*args, log_level=LOG_DEBUG)
def error(*args) -> None:
log(*args, log_level=LOG_ERR)
def warning(*args) -> None:
log(*args, log_level=LOG_WARNING)
def abort(*args) -> None:
log(*args, log_level=LOG_ERR)
sys.exit(1)
|