mirror of https://github.com/dirtbags/moth.git
87 lines
2.0 KiB
Python
Executable File
87 lines
2.0 KiB
Python
Executable File
#! /usr/bin/env python3
|
|
|
|
import fcntl
|
|
import time
|
|
import os
|
|
import tempfile
|
|
from . import points
|
|
from . import teams
|
|
from . import config
|
|
|
|
pngout = config.datafile('histogram.png')
|
|
|
|
def main(s=None):
|
|
scores = {}
|
|
now = 0
|
|
|
|
if not s:
|
|
s = points.Storage()
|
|
|
|
plotparts = []
|
|
|
|
catscores = {}
|
|
for cat in s.categories():
|
|
score = s.cat_points(cat)
|
|
if score:
|
|
catscores[cat] = score
|
|
|
|
scoresfile = tempfile.NamedTemporaryFile('w')
|
|
fn = scoresfile.name
|
|
i = 2
|
|
for team in s.teams:
|
|
plotparts.append('"%s" using 1:%d with lines linewidth 2 linetype rgb "#%s"' % (fn, i, teams.color(team)))
|
|
scores[team] = 0
|
|
i += 1
|
|
|
|
def write_scores(t):
|
|
scoresfile.write('%d' % t)
|
|
for team in s.teams:
|
|
scoresfile.write('\t%f' % (scores[team]))
|
|
scoresfile.write('\n')
|
|
|
|
for when, cat, team, score in s.log:
|
|
if not cat in catscores:
|
|
continue
|
|
if when > now:
|
|
if now:
|
|
write_scores(now)
|
|
now = when
|
|
pct = score / catscores[cat]
|
|
scores[team] += pct
|
|
#print('%d [%s] [%s] %d' % (when, cat, team, points))
|
|
|
|
write_scores(when)
|
|
scoresfile.flush()
|
|
|
|
instructions = tempfile.NamedTemporaryFile('w')
|
|
instructions.write('''
|
|
set style data lines
|
|
set xdata time
|
|
set timefmt "%%s"
|
|
set format ""
|
|
set border 3
|
|
set xtics nomirror
|
|
set ytics nomirror
|
|
set nokey
|
|
set terminal png transparent size 1200,400 x000000 xffffff
|
|
set output "%(pngout)s,tmp"
|
|
plot %(plot)s\n''' % {'plot': ','.join(plotparts),
|
|
'pngout': pngout})
|
|
instructions.flush()
|
|
|
|
lock = open('%s,lock' % pngout, 'a')
|
|
try:
|
|
fcntl.flock(lock, fcntl.LOCK_EX | fcntl.LOCK_NB)
|
|
except:
|
|
return
|
|
|
|
try:
|
|
gp = os.system('gnuplot %s 2>/dev/null </dev/null' % instructions.name)
|
|
os.rename("%s,tmp" % pngout, pngout)
|
|
finally:
|
|
fcntl.flock(lock, fcntl.LOCK_UN)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|