Reads in a minecraft server log ranking users by time spent

timespent_ranking.py

"""
Minecraft Server Log Time Spent Ranking

"""
__author__ = 'Gavin M. Roy'
__email__ = 'gavinmroy@gmail.com'
__since__ = '2012-01-05'

import datetime
import operator
import re
import string
import sys


_LOGIN = re.compile("(?P<date>\d\d\d\d-\d\d-\d\d) "
                    "(?P<time>\d\d:\d\d\:\d\d) \[INFO\] "
                    "(?P<nickname>\w*) \[/"
                    "(?P<ipaddress>[0-9]{1,3}\.[0-9]{1,3}\."
                                  "[0-9]{1,3}\.[0-9]{1,3}):"
                    "(?P<port>[0-9]{2,8})\] logged in with entity id "
                    "(?P<entity>[0-9]{1,20}) at \("
                    "(?P<x>[\-0-9\.]*), "
                    "(?P<y>[\-0-9\.]*), "
                    "(?P<z>[\-0-9\.]*)\)")
_LOGOUT = re.compile("(?P<date>\d\d\d\d-\d\d-\d\d) "
                    "(?P<time>\d\d:\d\d:\d\d) \[INFO\] "
                    "(?P<nickname>\w*) lost connection: disconnect."
                    "(?P<reason>\w*)")

# Globals that carry state across methods
users = dict()
last = dict()


def get_datetime(date, time):
    temp = '%s %s' % (date, time)
    return datetime.datetime.strptime(temp, '%Y-%m-%d %H:%M:%S')


def set_duration(nickname, start, end):
    global users
    delta = end - start
    if nickname not in users:
        users[nickname] = delta
    else:
        users[nickname] += delta


def read_file():
    if len(sys.argv) > 1:
        filename = sys.argv[1]
    else:
        filename = 'sever.log'
    with open(filename) as handle:
        return handle.read()


def process_line(line):
    global last
    matches = _LOGIN.match(line)
    if matches:
        login = matches.groupdict()
        last[login['nickname']] = get_datetime(login['date'], login['time'])
    else:
        matches = _LOGOUT.match(line)
        if matches:
            logout = matches.groupdict()
            set_duration(logout['nickname'],
                         last[logout['nickname']],
                         get_datetime(logout['date'], logout['time']))


def display_results():
    time_spent = sorted(users.iteritems(),
                         key=operator.itemgetter(1),
                         reverse=True)
    print
    print '%s%s' % (string.ljust('User ranking by time spent', 30),
                    string.rjust('(Hours:Minutes:Seconds)', 30))
    print ''.join(['-' for x in xrange(0, 60)])
    print
    for user in time_spent:
        print '%s%s' % (string.ljust(user[0], 30),
                        string.rjust(str(user[1]), 30))
    print


def main():

    # Read in the server log
    data = read_file().split('\n')

    # Iterate over each line
    for line in data:
        process_line(line)

    # Display the results
    display_results()

if __name__ == '__main__':
    main()