import sys, pickle, time from operator import itemgetter class Container: def __init__(self): self.total_connections = 0 self.country_count = {} self.last_10 = [] def copy_from_other(self, other): try: self.total_connections = other.total_connections self.country_count = other.country_count self.last_10 = other.last_10 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 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("\n") outf.write("

Frodo-Wii network statistics

\n") outf.write("The total number of connections is %d

\n" % (self.container.total_connections)) outf.write("\n") outf.write("\n" % (len(self.container.last_10)) ) cnt = 0 for item in self.container.last_10: images = "" if cnt % 4 == 0: cur = cnt images = "" % (cur, cur + 1, cur + 2, cur + 3) outf.write("%s\n" % (item, images) ) cnt = cnt + 1 outf.write("") outf.write("

Last %d connections

 

Random connection screenshots

%s 
 
\n") outf.write("

\n") outf.write("\n") count = 1 n_countries = len(sorted_countries) for i in range(0, n_countries / 2): c1, n1 = sorted_countries[i] c2, n2 = sorted_countries[i + n_countries / 2] outf.write("\n" % (count, c1, n1, count + n_countries / 2, c2, n2) ) count = count + 1 outf.write("

Country list

%3d %s (%d)   %3d %s (%d)
\n") outf.write("\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 add_connection(who, country): g_stat.add_connection(who, country) 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") save("/tmp/vobb") hg = HtmlGenerator(g_stat) hg.generate(sys.stdout)