145 lines
4.4 KiB
Python
145 lines
4.4 KiB
Python
# -*- coding: utf-8 -*-
|
|
import argparse
|
|
import json
|
|
import logging
|
|
from os.path import join as pathjoin
|
|
from os.path import abspath, exists, isdir
|
|
from time import time
|
|
|
|
from tsstats import config
|
|
from tsstats.exceptions import InvalidConfiguration
|
|
from tsstats.log import parse_logs
|
|
from tsstats.logger import file_handler, stream_handler
|
|
from tsstats.template import render_servers
|
|
from tsstats.utils import transform_pretty_identmap
|
|
|
|
logger = logging.getLogger('tsstats')
|
|
|
|
|
|
def cli():
|
|
parser = argparse.ArgumentParser(
|
|
description='A simple Teamspeak stats-generator,'
|
|
' based solely on server-logs',
|
|
argument_default=argparse.SUPPRESS
|
|
)
|
|
parser.add_argument(
|
|
'-c', '--config',
|
|
type=str, help='path to config'
|
|
)
|
|
parser.add_argument(
|
|
'--idmap', type=str, help='path to id_map'
|
|
)
|
|
parser.add_argument(
|
|
'-l', '--log',
|
|
type=str, help='path to your logfile(s). '
|
|
'pass a directory to use all logfiles inside it'
|
|
)
|
|
parser.add_argument(
|
|
'-o', '--output',
|
|
type=str, help='path to the output-file'
|
|
)
|
|
parser.add_argument(
|
|
'-d', '--debug',
|
|
help='debug mode', action='store_true'
|
|
)
|
|
parser.add_argument(
|
|
'-ds', '--debugstdout',
|
|
help='write debug output to stdout', action='store_true'
|
|
)
|
|
parser.add_argument(
|
|
'-nod', '--noonlinedc',
|
|
help='don\'t add connect until now to onlinetime',
|
|
action='store_false', dest='onlinedc'
|
|
)
|
|
parser.add_argument(
|
|
'-t', '--template',
|
|
type=str, help='path to custom template'
|
|
)
|
|
parser.add_argument(
|
|
'-dtf', '--datetimeformat',
|
|
type=str, help='format of date/time-values (datetime.strftime)'
|
|
)
|
|
parser.add_argument(
|
|
'-otth', '--onlinetimethreshold',
|
|
type=int, help='threshold for displaying onlinetime (in seconds)'
|
|
)
|
|
parser.add_argument(
|
|
'-lsa', '--lastseenabsolute',
|
|
help='render last seen timestamp absolute (instead of relative)',
|
|
action='store_false', dest='lastseenrelative'
|
|
)
|
|
parser.add_argument(
|
|
'-dc', '--disablecache',
|
|
help='disable caching feature',
|
|
action='store_false', dest='cache'
|
|
)
|
|
parser.add_argument(
|
|
'-cp', '--cachepath',
|
|
type=str, help='Path for cache location'
|
|
)
|
|
options = parser.parse_args()
|
|
if 'config' in options:
|
|
configuration = config.load(options.config)
|
|
else:
|
|
configuration = config.load()
|
|
for option, value in vars(options).items():
|
|
configuration.set('General', option, str(value))
|
|
main(configuration)
|
|
|
|
|
|
def main(configuration):
|
|
start_time = time()
|
|
# setup logging
|
|
if configuration.getboolean('General', 'debug'):
|
|
logger.setLevel(logging.DEBUG)
|
|
if configuration.getboolean('General', 'debugstdout'):
|
|
stream_handler.setLevel(logging.DEBUG)
|
|
else:
|
|
logger.addHandler(file_handler)
|
|
|
|
# attach handlers
|
|
logger.addHandler(stream_handler)
|
|
|
|
idmap = configuration.get('General', 'idmap')
|
|
if idmap:
|
|
idmap = abspath(idmap)
|
|
if not exists(idmap):
|
|
logger.fatal('identmap not found (%s)', idmap)
|
|
# read id_map
|
|
identmap = json.load(open(idmap))
|
|
else:
|
|
identmap = None
|
|
if isinstance(identmap, list):
|
|
identmap = transform_pretty_identmap(identmap)
|
|
|
|
log = configuration.get('General', 'log')
|
|
if not log:
|
|
raise InvalidConfiguration('log or output missing')
|
|
if isdir(log):
|
|
log = pathjoin(log, '*.log')
|
|
|
|
servers = parse_logs(
|
|
log, ident_map=identmap,
|
|
online_dc=configuration.getboolean('General', 'onlinedc'),
|
|
cache_path=pathjoin(
|
|
configuration.get('General', 'cachepath'), 'tsstats.pickle'
|
|
) if configuration.getboolean('General', 'cache') else None
|
|
)
|
|
render_servers(
|
|
sorted(servers, key=lambda s: s.sid),
|
|
output=abspath(configuration.get('General', 'output')),
|
|
template=configuration.get('General', 'template'),
|
|
datetime_fmt=configuration.get('General', 'datetimeformat'),
|
|
onlinetime_threshold=int(configuration.get(
|
|
'General', 'onlinetimethreshold'
|
|
)),
|
|
lastseen_relative=configuration.getboolean(
|
|
'General', 'lastseenrelative'
|
|
)
|
|
)
|
|
logger.info('Finished after %s seconds', time() - start_time)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
cli()
|