49 lines
1.4 KiB
Python
49 lines
1.4 KiB
Python
|
import argparse
|
||
|
|
import json
|
||
|
|
from os.path import abspath, exists
|
||
|
|
|
||
|
|
from tsstats.config import parse_config
|
||
|
|
from tsstats.exceptions import ConfigNotFound
|
||
|
|
from tsstats.log import parse_logs
|
||
|
|
from tsstats.template import render_template
|
||
|
|
|
||
|
|
|
||
|
|
def main(config_path='config.ini', id_map_path='id_map.json',
|
||
|
|
debug=False, debugfile=False):
|
||
|
|
# check cmdline-args
|
||
|
|
config_path = abspath(config_path)
|
||
|
|
id_map_path = abspath(id_map_path)
|
||
|
|
|
||
|
|
if not exists(config_path):
|
||
|
|
raise ConfigNotFound(config_path)
|
||
|
|
|
||
|
|
if exists(id_map_path):
|
||
|
|
# read id_map
|
||
|
|
id_map = json.load(open(id_map_path))
|
||
|
|
else:
|
||
|
|
id_map = {}
|
||
|
|
|
||
|
|
log_path, output_path = parse_config(config_path)
|
||
|
|
clients = parse_logs(log_path, ident_map=id_map, file_log=debugfile)
|
||
|
|
render_template(clients, output=output_path, debug=debug)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
parser = argparse.ArgumentParser(
|
||
|
|
description='A simple Teamspeak stats-generator - based on server-logs'
|
||
|
|
)
|
||
|
|
parser.add_argument(
|
||
|
|
'--config', type=str, help='path to config', default='config.ini'
|
||
|
|
)
|
||
|
|
parser.add_argument(
|
||
|
|
'--idmap', type=str, help='path to id_map', default='id_map.json'
|
||
|
|
)
|
||
|
|
parser.add_argument(
|
||
|
|
'--debug', help='debug mode', action='store_true'
|
||
|
|
)
|
||
|
|
parser.add_argument(
|
||
|
|
'--debugfile', help='write debug-log to file', action='store_true'
|
||
|
|
)
|
||
|
|
args = parser.parse_args()
|
||
|
|
main(args.config, args.idmap, args.debug, args.debugfile)
|