4 from datetime import datetime
5 import sqlalchemy as sa
6 import sqlalchemy.sql.functions as func
7 from sqlalchemy import distinct
8 from pyramid.paster import bootstrap
9 from xonstat.models import *
10 from xonstat.util import datetime_seconds
13 from playerdata import PlayerData
16 # maximal number of query results (for testing, set to None to get all)
19 # we look for players who have activity within the past DELTA hours
23 # classic skin WITHOUT NAME - writes PNGs into "output//###.png"
24 skin_classic = Skin( "",
26 overlay = "overlay_classic",
29 # more fancy skin [** WIP **]- writes PNGs into "output/archer/###.png"
30 skin_archer = Skin( "archer",
31 bg = "background_archer-v1",
35 # minimal skin - writes PNGs into "output/minimal/###.png"
36 skin_minimal = Skin( "minimal",
38 bgcolor = (0.04, 0.04, 0.04, 1.0),
39 overlay = "overlay_minimal",
46 gametype_pos = (70,30),
47 gametype_color = (0.0, 0.0, 0.0),
48 gametype_text = "%s:",
50 gametype_fontsize = 10,
52 gametype_upper = False,
54 elo_text = "Elo %.0f",
55 elo_color = (0.7, 0.7, 0.7),
59 #nostats_pos = (75,30),
60 #nostats_fontsize = 10,
62 #nostats_text = "no stats yet!",
63 #nostats_color = (0.7, 0.4, 0.4),
66 kdr_colortop = (0.6, 0.8, 0.6),
67 kdr_colormid = (0.6, 0.6, 0.6),
68 kdr_colorbot = (0.8, 0.6, 0.6),
73 winp_colortop = (0.6, 0.8, 0.8),
74 winp_colormid = (0.6, 0.6, 0.6),
75 winp_colorbot = (0.8, 0.8, 0.6),
79 ptime_color = (0.7, 0.7, 0.7),
83 # parse cmdline parameters (for testing)
86 for arg in sys.argv[1:]:
87 if arg.startswith("-"):
90 DELTA = 2**24 # large enough to enforce update, and doesn't result in errors
94 print """Usage: gen_badges.py [options] [skin list]
96 -force Force updating all badges (delta = 2^24)
97 -test Limit number of players to 100 (for testing)
98 -help Show this help text
100 Space-separated list of skins to use when creating badges.
101 Available skins: classic, minimal, archer
102 If no skins are given, classic and minmal will be used by default.
103 NOTE: Output directories must exists before running the program!
108 skins.append(skin_classic)
109 elif arg == "minimal":
110 skins.append(skin_minimal)
111 elif arg == "archer":
112 skins.append(skin_archer)
115 skins = [ skin_classic, skin_minimal ]
118 env = bootstrap('../../../development.ini')
120 req.matchdict = {'id':3}
122 print "Requesting player data from db ..."
123 cutoff_dt = datetime.utcnow() - timedelta(hours=DELTA)
124 start = datetime.now()
127 players = DBSession.query(distinct(Player.player_id)).\
128 filter(Player.player_id == PlayerElo.player_id).\
129 filter(Player.player_id == PlayerGameStat.player_id).\
130 filter(PlayerGameStat.create_dt > cutoff_dt).\
131 filter(Player.nick != None).\
132 filter(Player.player_id > 2).\
133 filter(Player.active_ind == True).\
134 limit(NUM_PLAYERS).all()
136 players = DBSession.query(distinct(Player.player_id)).\
137 filter(Player.player_id == PlayerElo.player_id).\
138 filter(Player.player_id == PlayerGameStat.player_id).\
139 filter(PlayerGameStat.create_dt > cutoff_dt).\
140 filter(Player.nick != None).\
141 filter(Player.player_id > 2).\
142 filter(Player.active_ind == True).\
145 playerdata = PlayerData()
148 stop = datetime.now()
150 print "Query took %.2f seconds" % (datetime_seconds(td))
152 print "Creating badges for %d players ..." % len(players)
153 start = datetime.now()
154 data_time, render_time = 0,0
155 for player_id in players:
156 req.matchdict['id'] = player_id
158 sstart = datetime.now()
159 playerdata.get_data(player_id)
160 sstop = datetime.now()
162 data_time += datetime_seconds(td)
164 sstart = datetime.now()
166 sk.render_image(playerdata, "output/%s/%d.png" % (str(sk), player_id[0]))
167 sstop = datetime.now()
169 render_time += datetime_seconds(td)
171 stop = datetime.now()
173 total_seconds = datetime_seconds(td)
174 print "Creating the badges took %.1f seconds (%.3f s per player)" % (total_seconds, total_seconds/float(len(players)))
175 print "Total time for rendering images: %.3f s" % render_time
176 print "Total time for getting data: %.3f s" % data_time
179 print "No active players found!"