xonstat/batch/badges/gen_badges.py

   1 from mako.template import Template
   2 from os import system
   3 from pyramid.paster import bootstrap
   4 from xonstat.models import *
   5 from xonstat.views.player import player_info_data
   6
   7 # converter params
   8 CONVERTER = "/home/ant/xonotic/xonstat/xonstat/batch/badges/wkhtmltoimage-amd64 --crop-x 10 --crop-y 10 --width 560 --height 70 -f png"
   9
  10 # environment setup
  11 env = bootstrap('../../../development.ini.home')
  12 req = env['request']
  13 req.matchdict = {'id':3}
  14
  15 # template setup
  16 t = Template(filename = 'templates/badge.mako')
  17
  18 players = DBSession.query(Player).\
  19         filter(Player.player_id == PlayerElo.player_id).\
  20         filter(Player.nick != None).\
  21         filter(Player.player_id > 2).\
  22         filter(Player.active_ind == True).all()
  23
  24 for player in players:
  25     req.matchdict['id'] = player.player_id
  26     data = player_info_data(req)
  27     rt = t.render(player=data['player'], elos_display=data['elos_display'],
  28             total_stats=data['total_stats'], total_games=data['total_games'],
  29             games_breakdown=data['games_breakdown'])
  30     f = open("output/%s.html" % player.player_id, 'w')
  31     f.write(rt)
  32     f.close()
  33
  34     cmd = "%s %s %s" % (CONVERTER, "output/%s.html" % player.player_id,
  35             "output/%s.png" % player.player_id)
  36     system(cmd)