mirror of
https://github.com/Thor77/TeamspeakStats.git
synced 2025-09-22 20:18:45 -04:00
bc547228b8
This adds support for a more expressive (albeit more verbose) IdentMap structure. It makes it easier to annotate the structure with additional data (such as names to associate with the IDs), to assist with maintaining the IdentMap.
103 lines
3 KiB
Python
103 lines
3 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
import argparse
|
|
import json
|
|
import logging
|
|
from os.path import abspath, exists
|
|
|
|
from tsstats import config
|
|
from tsstats.exceptions import InvalidConfiguration
|
|
from tsstats.log import parse_logs
|
|
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)'
|
|
)
|
|
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(
|
|
'-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)'
|
|
)
|
|
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):
|
|
if configuration.getboolean('General', 'debug'):
|
|
logger.setLevel(logging.DEBUG)
|
|
|
|
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')
|
|
|
|
servers = parse_logs(
|
|
log, ident_map=identmap,
|
|
online_dc=configuration.getboolean('General', 'onlinedc')
|
|
)
|
|
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'))
|
|
)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
cli()
|