xonstat/batch/badges/gen_badges.py

   1 #-*- coding: utf-8 -*-
   2
   3 import sys
   4 from datetime import datetime
   5 import sqlalchemy as sa
   6 import sqlalchemy.sql.functions as func
   7 from pyramid.paster import bootstrap
   8 from xonstat.models import *
   9
  10 from render import Skin
  11
  12
  13 # maximal number of query results (for testing, set to 0 to get all)
  14 #NUM_PLAYERS = 100
  15
  16
  17 skin_classic = Skin(
  18         bg_h            = "asfalt",
  19     )
  20
  21 skin_archer = Skin(
  22         bg              = "background_archer-v1",
  23         overlay         = "",
  24     )
  25
  26 skin_minimal = Skin(
  27         bg              = None,
  28         bgcolor         = (0.04, 0.04, 0.04, 1.0),
  29         overlay         = "overlay_minimal",
  30         width           = 560,
  31         height          = 40,
  32         num_gametypes   = 4,
  33         gametype_pos    = (25,30),
  34         gametype_text   = "%s :",
  35         gametype_width  = 120,
  36         gametype_fontsize = 10,
  37         elo_pos         = (75,30),
  38         elo_text        = "Elo %.0f",
  39         elo_color       = (1.0, 1.0, 0.6),
  40         rank_pos        = None,
  41         nostats_pos     = (80,30),
  42         nostats_fontsize = 10,
  43         nostats_angle   = 0,
  44         nostats_text    = "no stats!",
  45         kdr_pos         = (392,15),
  46         kills_pos       = None,
  47         deaths_pos      = None,
  48         winp_pos        = (508,15),
  49         wins_pos        = None,
  50         loss_pos        = None,
  51         ptime_pos       = (458,30),
  52         ptime_color     = (0.8, 0.8, 0.9),
  53
  54     )
  55
  56 # parse cmdline parameters (for testing)
  57 skin = skin_classic
  58 if len(sys.argv) > 1:
  59     arg = sys.argv[1].lower()
  60     if arg == "classic":
  61         skin = skin_classic
  62     elif arg == "minimal":
  63         skin = skin_minimal
  64     elif arg == "archer":
  65         skin = skin_archer
  66
  67
  68 # environment setup
  69 env = bootstrap('../../../development.ini')
  70 req = env['request']
  71 req.matchdict = {'id':3}
  72
  73 print "Requesting player data from db ..."
  74 start = datetime.now()
  75 players = []
  76 if locals().has_key('NUM_PLAYERS'):
  77     players = DBSession.query(Player).\
  78             filter(Player.player_id == PlayerElo.player_id).\
  79             filter(Player.nick != None).\
  80             filter(Player.player_id > 2).\
  81             filter(Player.active_ind == True).\
  82             limit(NUM_PLAYERS).all()
  83 else:
  84     players = DBSession.query(Player).\
  85             filter(Player.player_id == PlayerElo.player_id).\
  86             filter(Player.nick != None).\
  87             filter(Player.player_id > 2).\
  88             filter(Player.active_ind == True).\
  89             all()
  90
  91 stop = datetime.now()
  92 td = stop-start
  93 total_seconds = (td.microseconds + (td.seconds + td.days * 24 * 3600) * 10**6) / 10**6
  94 print "Query took %.2f seconds" % (total_seconds)
  95
  96 print "Creating badges for %d players ..." % len(players)
  97 start = datetime.now()
  98 data_time, render_time = 0,0
  99 for player in players:
 100     req.matchdict['id'] = player.player_id
 101
 102     sstart = datetime.now()
 103     skin.get_data(player)
 104     sstop = datetime.now()
 105     td = sstop-sstart
 106     total_seconds = (td.microseconds + (td.seconds + td.days * 24 * 3600) * 10**6) / 10**6
 107     data_time += total_seconds
 108
 109     sstart = datetime.now()
 110     skin.render_image("output/%d.png" % player.player_id)
 111     sstop = datetime.now()
 112     td = sstop-sstart
 113     total_seconds = (td.microseconds + (td.seconds + td.days * 24 * 3600) * 10**6) / 10**6
 114     render_time += total_seconds
 115
 116 stop = datetime.now()
 117 td = stop-start
 118 total_seconds = (td.microseconds + (td.seconds + td.days * 24 * 3600) * 10**6) / 10**6
 119 print "Creating the badges took %.1f seconds (%.3f s per player)" % (total_seconds, total_seconds/float(len(players)))
 120 print "Total time for redering images: %.3f s" % render_time
 121 print "Total time for getting data: %.3f s" % data_time
 122