frodo-wii/network-broker/stats.py

import sys, pickle, time
from operator import itemgetter

class Container:
    def __init__(self):
        self.total_connections = 0
        self.country_count = {}
        self.last_10 = []
        self.messages = []
        self.nr_active = 0
        self.nr_waiting = 0

    def set_nr_active(self, nr_active):
        self.nr_active = nr_active

    def set_nr_waiting(self, nr_waiting):
        self.nr_waiting = nr_waiting

    def copy_from_other(self, other):
        try:
            self.nr_active = 0 # Always reset
            self.nr_waiting = 0
            self.total_connections = other.total_connections
            self.country_count = other.country_count
            self.last_10 = other.last_10
            self.messages = other.messages
        except:
            pass

    def add_country(self, country):
        try:
            cur = self.country_count[country]
        except KeyError, e:
            cur = 0

        self.country_count[country] = cur + 1

    def set_messages(self, messages):
        self.messages = messages

    def add_connection(self, who, country):
        time_now = time.strftime("%Y-%m-%d %H:%M", time.gmtime())
        s = "%s - %s (%s)" % (time_now, who, country)
        self.total_connections = self.total_connections + 1
        self.last_10 = [s] + self.last_10[:9]
        self.add_country(country)

class HtmlGenerator:
    def __init__(self, container):
        self.container = container

    def generate(self, outf):
        sorted_countries = sorted(self.container.country_count.items(),
                                  reverse=True, key=itemgetter(1))

        outf.write("<html><body>\n")
        outf.write("<H2>Frodo-Wii network statistics</H2>\n")
        outf.write("The total number of connections is <b>%d</b><br><br>\n" % (self.container.total_connections))
        outf.write("There are currently <b>%d</b> players waiting for connections and <b>%d</b> players playing<br><br>\n" % (self.container.nr_waiting, self.container.nr_active))
        outf.write("<TABLE border=\"0\" cellpadding=\"0\">\n")
        outf.write("<TR><TD><H3>Last %d connections</H3></TD><TD>&nbsp;</TD<TD colspan=4><H3>Random connection screenshots</TD></TR>\n" %
                   (len(self.container.last_10)) )

        cnt = 0
        for item in self.container.last_10:
            images = ""
            if cnt % 3 == 0:
                cur = cnt
                images = "<TH ROWSPAN=3><IMG SRC=\"images/%d.png\"></TH><TH ROWSPAN=3><IMG SRC=\"images/%d.png\"></TH><TH ROWSPAN=3><IMG SRC=\"images/%d.png\"></TH>" % (cur, cur + 1, cur + 2)
            outf.write("<TR><TD>%s</TD><TD>&nbsp;</TD>%s</TR>\n" % (item, images) )
            cnt = cnt + 1
            if cnt >= 9:
                break
        outf.write("<TR><TD>&nbsp;</TD></TR>")
        outf.write("</TABLE>\n")

        outf.write("<br><br><TABLE border=\"0\" cellpadding=\"0\">\n")
        outf.write("<TR><TD colspan=3><H3>List of countries</H3></TD></TR>\n")
        count = 1

        outf.write("<H3>Last server messages</H3>\n")
        outf.write("<UL>\n")
        for msg in self.container.messages:
            outf.write("<LI>%s</LI>\n" % (msg))
        outf.write("</UL><br>")

        n_countries = len(sorted_countries)
        for i in range(0, n_countries / 3):
            c1, n1 = sorted_countries[i]
            c2, n2 = sorted_countries[i + n_countries / 3]
            c3, n3 = sorted_countries[i + (n_countries / 3) * 2]
            outf.write("<TR><TD><b>%3d</b></TD><TD>&nbsp;</TD><TD>%s (%d)</TD><TD>&nbsp;</TD><TD>&nbsp;</TD><TD>&nbsp;</TD><TD><b>%3d</b></TD><TD>&nbsp;</TD><TD>%s (%d)</TD><TD>&nbsp;</TD><TD>&nbsp;</TD><TD>&nbsp;</TD><TD><b>%3d</b></TD><TD>&nbsp;</TD><TD>%s (%d)</TD></TR>\n" %
                       (count, c1, n1, count + n_countries / 3, c2, n2, count + (n_countries / 3) * 2, c3, n3) )
            count = count + 1
        # Output the last if it's odd
        for i in range(0, n_countries % 3):
            c1, n1 = sorted_countries[n_countries - i - 1]
            outf.write("<TR><TD></TD><TD></TD><TD></TD><TD></TD><TD></TD><TD></TD><TD></TD><TD></TD><TD></TD><TD></TD><TD></TD><TD></TD><TD><b>%3d</b></TD><TD>&nbsp;</TD><TD>%s (%d)</TD></TR>\n" %
                       (n_countries - n_countries % 3 + i + 1, c1, n1) )
            
            
        outf.write("</TABLE>\n")
        outf.write("</body></html>\n")


g_stat = None
def save(filename):
    global g_stat

    of = open(filename, "wb")
    pickle.dump(g_stat, of)
    of.close()

def generate_html(filename):
    of = open(filename, "wb")
    hg = HtmlGenerator(g_stat)
    hg.generate(of)
    of.close()


def load(filename):
    global g_stat

    g_stat = Container()
    try:
        of = open(filename, "r")
        other = pickle.load(of)
        g_stat.copy_from_other(other)
        of.close()
    except:
        pass

def update_peer_nr(waiting, active):
    g_stat.set_nr_waiting(waiting)
    g_stat.set_nr_active(active)

def add_connection(who, country):
    g_stat.add_connection(who, country)

def set_messages(messages):
    g_stat.set_messages(messages)

if __name__ == "__main__":
    load("/tmp/vobb")
    for i in range(0, 10):
        add_connection("MABOO", "Unknown country")
    add_connection("SIMONK", "Sweden")
    add_connection("SIMONK", "Sweden")
    add_connection("SIMONK", "Sweden")
    add_connection("Linda", "Germany")
    add_connection("Linda", "Germany")
    set_messages(["Hej", "Who framed Roger Rabbit?", "IK+ at 19.00 CET?"])
    save("/tmp/vobb")

    hg = HtmlGenerator(g_stat)
    hg.generate(sys.stdout)
Add timestamp 2009-04-08 18:33:10 +02:00			`import sys, pickle, time`
Produce HTML output from the server 2009-03-31 19:16:10 +02:00			`from operator import itemgetter`

			`class Container:`
			`def __init__(self):`
Count the total number of connections 2009-03-31 19:27:47 +02:00			`self.total_connections = 0`
Produce HTML output from the server 2009-03-31 19:16:10 +02:00			`self.country_count = {}`
			`self.last_10 = []`
Write out server messages on the stats page 2010-02-28 09:52:34 +01:00			`self.messages = []`
Generate statistics on current number of active and waiting players 2010-02-25 12:57:25 +01:00			`self.nr_active = 0`
			`self.nr_waiting = 0`

			`def set_nr_active(self, nr_active):`
			`self.nr_active = nr_active`

			`def set_nr_waiting(self, nr_waiting):`
			`self.nr_waiting = nr_waiting`
Produce HTML output from the server 2009-03-31 19:16:10 +02:00
Add timestamp 2009-04-08 18:33:10 +02:00			`def copy_from_other(self, other):`
			`try:`
Write out server messages on the stats page 2010-02-28 09:52:34 +01:00			`self.nr_active = 0 # Always reset`
			`self.nr_waiting = 0`
Add timestamp 2009-04-08 18:33:10 +02:00			`self.total_connections = other.total_connections`
			`self.country_count = other.country_count`
			`self.last_10 = other.last_10`
Write out server messages on the stats page 2010-02-28 09:52:34 +01:00			`self.messages = other.messages`
Add timestamp 2009-04-08 18:33:10 +02:00			`except:`
			`pass`

Produce HTML output from the server 2009-03-31 19:16:10 +02:00			`def add_country(self, country):`
			`try:`
			`cur = self.country_count[country]`
			`except KeyError, e:`
			`cur = 0`

			`self.country_count[country] = cur + 1`

Write out server messages on the stats page 2010-02-28 09:52:34 +01:00			`def set_messages(self, messages):`
			`self.messages = messages`

Produce HTML output from the server 2009-03-31 19:16:10 +02:00			`def add_connection(self, who, country):`
Add timestamp 2009-04-08 18:33:10 +02:00			`time_now = time.strftime("%Y-%m-%d %H:%M", time.gmtime())`
			`s = "%s - %s (%s)" % (time_now, who, country)`
Count the total number of connections 2009-03-31 19:27:47 +02:00			`self.total_connections = self.total_connections + 1`
Produce HTML output from the server 2009-03-31 19:16:10 +02:00			`self.last_10 = [s] + self.last_10[:9]`
			`self.add_country(country)`

			`class HtmlGenerator:`
			`def __init__(self, container):`
			`self.container = container`

			`def generate(self, outf):`
			`sorted_countries = sorted(self.container.country_count.items(),`
			`reverse=True, key=itemgetter(1))`

			`outf.write("<html><body>\n")`
Count the total number of connections 2009-03-31 19:27:47 +02:00			`outf.write("<H2>Frodo-Wii network statistics</H2>\n")`
Better HTML 2009-11-01 17:12:22 +01:00			`outf.write("The total number of connections is <b>%d</b><br><br>\n" % (self.container.total_connections))`
Generate statistics on current number of active and waiting players 2010-02-25 12:57:25 +01:00			`outf.write("There are currently <b>%d</b> players waiting for connections and <b>%d</b> players playing<br><br>\n" % (self.container.nr_waiting, self.container.nr_active))`
Correct HTML generation (not finished) 2009-11-01 12:15:37 +01:00			`outf.write("<TABLE border=\"0\" cellpadding=\"0\">\n")`
Heading 2009-11-02 07:35:31 +01:00			`outf.write("<TR><TD><H3>Last %d connections</H3></TD><TD> </TD<TD colspan=4><H3>Random connection screenshots</TD></TR>\n" %`
Correct HTML generation (not finished) 2009-11-01 12:15:37 +01:00			`(len(self.container.last_10)) )`
Produce HTML output from the server 2009-03-31 19:16:10 +02:00
Correct HTML generation (not finished) 2009-11-01 12:15:37 +01:00			`cnt = 0`
			`for item in self.container.last_10:`
			`images = ""`
Fix image counting 2010-03-14 09:32:53 +01:00			`if cnt % 3 == 0:`
Correct HTML generation (not finished) 2009-11-01 12:15:37 +01:00			`cur = cnt`
Fix stats generation 2010-03-14 09:39:01 +01:00			`images = "<TH ROWSPAN=3><IMG SRC=\"images/%d.png\"></TH><TH ROWSPAN=3><IMG SRC=\"images/%d.png\"></TH><TH ROWSPAN=3><IMG SRC=\"images/%d.png\"></TH>" % (cur, cur + 1, cur + 2)`
Better HTML 2009-11-01 17:12:22 +01:00			`outf.write("<TR><TD>%s</TD><TD> </TD>%s</TR>\n" % (item, images) )`
Correct HTML generation (not finished) 2009-11-01 12:15:37 +01:00			`cnt = cnt + 1`
Fix stats generation 2010-03-14 09:39:01 +01:00			`if cnt >= 9:`
			`break`
Better HTML 2009-11-01 17:12:22 +01:00			`outf.write("<TR><TD> </TD></TR>")`
Correct HTML generation (not finished) 2009-11-01 12:15:37 +01:00			`outf.write("</TABLE>\n")`

Better HTML 2009-11-01 17:12:22 +01:00			`outf.write("<br><br><TABLE border=\"0\" cellpadding=\"0\">\n")`
Make that three columns instead 2009-11-03 21:36:41 +01:00			`outf.write("<TR><TD colspan=3><H3>List of countries</H3></TD></TR>\n")`
Produce HTML output from the server 2009-03-31 19:16:10 +02:00			`count = 1`
Keep countries in two columns 2009-11-03 21:21:52 +01:00
Write out server messages on the stats page 2010-02-28 09:52:34 +01:00			`outf.write("<H3>Last server messages</H3>\n")`
Use UL/LI 2010-02-28 09:56:45 +01:00			`outf.write("<UL>\n")`
Write out server messages on the stats page 2010-02-28 09:52:34 +01:00			`for msg in self.container.messages:`
Use UL/LI 2010-02-28 09:56:45 +01:00			`outf.write("<LI>%s</LI>\n" % (msg))`
			`outf.write("</UL><br>")`
Write out server messages on the stats page 2010-02-28 09:52:34 +01:00
Keep countries in two columns 2009-11-03 21:21:52 +01:00			`n_countries = len(sorted_countries)`
Make that three columns instead 2009-11-03 21:36:41 +01:00			`for i in range(0, n_countries / 3):`
Keep countries in two columns 2009-11-03 21:21:52 +01:00			`c1, n1 = sorted_countries[i]`
Make that three columns instead 2009-11-03 21:36:41 +01:00			`c2, n2 = sorted_countries[i + n_countries / 3]`
			`c3, n3 = sorted_countries[i + (n_countries / 3) * 2]`
			`outf.write("<TR><TD><b>%3d</b></TD><TD> </TD><TD>%s (%d)</TD><TD> </TD><TD> </TD><TD> </TD><TD><b>%3d</b></TD><TD> </TD><TD>%s (%d)</TD><TD> </TD><TD> </TD><TD> </TD><TD><b>%3d</b></TD><TD> </TD><TD>%s (%d)</TD></TR>\n" %`
			`(count, c1, n1, count + n_countries / 3, c2, n2, count + (n_countries / 3) * 2, c3, n3) )`
Produce HTML output from the server 2009-03-31 19:16:10 +02:00			`count = count + 1`
Make that three columns instead 2009-11-03 21:36:41 +01:00			`# Output the last if it's odd`
			`for i in range(0, n_countries % 3):`
			`c1, n1 = sorted_countries[n_countries - i - 1]`
			`outf.write("<TR><TD></TD><TD></TD><TD></TD><TD></TD><TD></TD><TD></TD><TD></TD><TD></TD><TD></TD><TD></TD><TD></TD><TD></TD><TD><b>%3d</b></TD><TD> </TD><TD>%s (%d)</TD></TR>\n" %`
			`(n_countries - n_countries % 3 + i + 1, c1, n1) )`


Better HTML 2009-11-01 17:12:22 +01:00			`outf.write("</TABLE>\n")`
Produce HTML output from the server 2009-03-31 19:16:10 +02:00			`outf.write("</body></html>\n")`


			`g_stat = None`
			`def save(filename):`
			`global g_stat`

			`of = open(filename, "wb")`
			`pickle.dump(g_stat, of)`
			`of.close()`

			`def generate_html(filename):`
			`of = open(filename, "wb")`
			`hg = HtmlGenerator(g_stat)`
			`hg.generate(of)`
			`of.close()`


			`def load(filename):`
			`global g_stat`

Add timestamp 2009-04-08 18:33:10 +02:00			`g_stat = Container()`
Produce HTML output from the server 2009-03-31 19:16:10 +02:00			`try:`
			`of = open(filename, "r")`
Add timestamp 2009-04-08 18:33:10 +02:00			`other = pickle.load(of)`
			`g_stat.copy_from_other(other)`
Produce HTML output from the server 2009-03-31 19:16:10 +02:00			`of.close()`
			`except:`
Add timestamp 2009-04-08 18:33:10 +02:00			`pass`
Produce HTML output from the server 2009-03-31 19:16:10 +02:00
Generate statistics on current number of active and waiting players 2010-02-25 12:57:25 +01:00			`def update_peer_nr(waiting, active):`
			`g_stat.set_nr_waiting(waiting)`
			`g_stat.set_nr_active(active)`

Produce HTML output from the server 2009-03-31 19:16:10 +02:00			`def add_connection(who, country):`
			`g_stat.add_connection(who, country)`

Write out server messages on the stats page 2010-02-28 09:52:34 +01:00			`def set_messages(messages):`
			`g_stat.set_messages(messages)`

Produce HTML output from the server 2009-03-31 19:16:10 +02:00			`if __name__ == "__main__":`
			`load("/tmp/vobb")`
			`for i in range(0, 10):`
			`add_connection("MABOO", "Unknown country")`
			`add_connection("SIMONK", "Sweden")`
			`add_connection("SIMONK", "Sweden")`
			`add_connection("SIMONK", "Sweden")`
			`add_connection("Linda", "Germany")`
			`add_connection("Linda", "Germany")`
Write out server messages on the stats page 2010-02-28 09:52:34 +01:00			`set_messages(["Hej", "Who framed Roger Rabbit?", "IK+ at 19.00 CET?"])`
Produce HTML output from the server 2009-03-31 19:16:10 +02:00			`save("/tmp/vobb")`

			`hg = HtmlGenerator(g_stat)`
			`hg.generate(sys.stdout)`