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("Last %d connections
\n" % (len(self.container.last_10)) )
for item in self.container.last_10:
outf.write("%s
\n" % (item) )
outf.write("Country list
\n")
count = 1
for country, num in sorted_countries:
outf.write("%3d. %s (%d)
\n" % (count, country, num) )
count = count + 1
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)