from datetime import datetime
import sqlalchemy as sa
import sqlalchemy.sql.functions as func
+from sqlalchemy import distinct
from pyramid.paster import bootstrap
from xonstat.models import *
+from xonstat.util import datetime_seconds
-from render import Skin
+from skin import Skin
+from playerdata import PlayerData
-# maximal number of query results (for testing, set to 0 to get all)
-#NUM_PLAYERS = 100
+# maximal number of query results (for testing, set to None to get all)
+NUM_PLAYERS = None
-skin_classic = Skin(
- bg = "asfalt",
+# we look for players who have activity within the past DELTA hours
+DELTA = 6
+
+
+# classic skin WITHOUT NAME - writes PNGs into "output//###.png"
+skin_classic = Skin( "",
+ bg = "broken_noise",
+ overlay = "overlay_classic",
)
-skin_archer = Skin(
+# more fancy skin [** WIP **]- writes PNGs into "output/archer/###.png"
+skin_archer = Skin( "archer",
bg = "background_archer-v1",
- overlay = "",
+ overlay = None,
)
-skin_minimal = Skin(
+# minimal skin - writes PNGs into "output/minimal/###.png"
+skin_minimal = Skin( "minimal",
bg = None,
bgcolor = (0.04, 0.04, 0.04, 1.0),
overlay = "overlay_minimal",
width = 560,
height = 40,
- num_gametypes = 4,
- gametype_pos = (25,30),
- gametype_text = "%s :",
- gametype_width = 120,
+ nick_fontsize = 16,
+ nick_pos = (36,16),
+ num_gametypes = 3,
+ nick_maxwidth = 300,
+ gametype_pos = (70,30),
+ gametype_color = (0.0, 0.0, 0.0),
+ gametype_text = "%s:",
+ gametype_width = 100,
gametype_fontsize = 10,
+ gametype_align = -1,
+ gametype_upper = False,
elo_pos = (75,30),
elo_text = "Elo %.0f",
- elo_color = (1.0, 1.0, 0.6),
+ elo_color = (0.7, 0.7, 0.7),
+ elo_align = 1,
rank_pos = None,
- nostats_pos = (80,30),
- nostats_fontsize = 10,
- nostats_angle = 0,
- nostats_text = "no stats!",
+ nostats_pos = None,
+ #nostats_pos = (75,30),
+ #nostats_fontsize = 10,
+ #nostats_angle = 0,
+ #nostats_text = "no stats yet!",
+ #nostats_color = (0.7, 0.4, 0.4),
kdr_pos = (392,15),
+ kdr_fontsize = 10,
+ kdr_colortop = (0.6, 0.8, 0.6),
+ kdr_colormid = (0.6, 0.6, 0.6),
+ kdr_colorbot = (0.8, 0.6, 0.6),
kills_pos = None,
deaths_pos = None,
winp_pos = (508,15),
+ winp_fontsize = 10,
+ winp_colortop = (0.6, 0.8, 0.8),
+ winp_colormid = (0.6, 0.6, 0.6),
+ winp_colorbot = (0.8, 0.8, 0.6),
wins_pos = None,
loss_pos = None,
- ptime_pos = (458,30),
- ptime_color = (0.8, 0.8, 0.9),
+ ptime_pos = (451,30),
+ ptime_color = (0.7, 0.7, 0.7),
)
+
# parse cmdline parameters (for testing)
-skin = skin_classic
-if len(sys.argv) > 1:
- arg = sys.argv[1].lower()
- if arg == "classic":
- skin = skin_classic
- elif arg == "minimal":
- skin = skin_minimal
- elif arg == "archer":
- skin = skin_archer
+skins = []
+for arg in sys.argv[1:]:
+ if arg.startswith("-"):
+ arg = arg[1:]
+ if arg == "force":
+ DELTA = 2**24 # large enough to enforce update, and doesn't result in errors
+ elif arg == "test":
+ NUM_PLAYERS = 100
+ else:
+ print """Usage: gen_badges.py [options] [skin list]
+ Options:
+ -force Force updating all badges (delta = 2^24)
+ -test Limit number of players to 100 (for testing)
+ -help Show this help text
+ Skin list:
+ Space-separated list of skins to use when creating badges.
+ Available skins: classic, minimal, archer
+ If no skins are given, classic and minmal will be used by default.
+ NOTE: Output directories must exists before running the program!
+"""
+ sys.exit(-1)
+ else:
+ if arg == "classic":
+ skins.append(skin_classic)
+ elif arg == "minimal":
+ skins.append(skin_minimal)
+ elif arg == "archer":
+ skins.append(skin_archer)
+
+if len(skins) == 0:
+ skins = [ skin_classic, skin_minimal ]
# environment setup
env = bootstrap('../../../development.ini')
req.matchdict = {'id':3}
print "Requesting player data from db ..."
+cutoff_dt = datetime.utcnow() - timedelta(hours=DELTA)
start = datetime.now()
players = []
-if locals().has_key('NUM_PLAYERS'):
- players = DBSession.query(Player).\
+if NUM_PLAYERS:
+ players = DBSession.query(distinct(Player.player_id)).\
filter(Player.player_id == PlayerElo.player_id).\
+ filter(Player.player_id == PlayerGameStat.player_id).\
+ filter(PlayerGameStat.create_dt > cutoff_dt).\
filter(Player.nick != None).\
filter(Player.player_id > 2).\
filter(Player.active_ind == True).\
limit(NUM_PLAYERS).all()
else:
- players = DBSession.query(Player).\
+ players = DBSession.query(distinct(Player.player_id)).\
filter(Player.player_id == PlayerElo.player_id).\
+ filter(Player.player_id == PlayerGameStat.player_id).\
+ filter(PlayerGameStat.create_dt > cutoff_dt).\
filter(Player.nick != None).\
filter(Player.player_id > 2).\
filter(Player.active_ind == True).\
all()
-stop = datetime.now()
-td = stop-start
-total_seconds = (td.microseconds + (td.seconds + td.days * 24 * 3600) * 10**6) / 10**6
-print "Query took %.2f seconds" % (total_seconds)
+playerdata = PlayerData()
+
+if len(players) > 0:
+ stop = datetime.now()
+ td = stop-start
+ print "Query took %.2f seconds" % (datetime_seconds(td))
+
+ print "Creating badges for %d players ..." % len(players)
+ start = datetime.now()
+ data_time, render_time = 0,0
+ for player_id in players:
+ req.matchdict['id'] = player_id
+
+ sstart = datetime.now()
+ playerdata.get_data(player_id)
+ sstop = datetime.now()
+ td = sstop-sstart
+ data_time += datetime_seconds(td)
+
+ sstart = datetime.now()
+ for sk in skins:
+ sk.render_image(playerdata, "output/%s/%d.png" % (str(sk), player_id[0]))
+ sstop = datetime.now()
+ td = sstop-sstart
+ render_time += datetime_seconds(td)
+
+ stop = datetime.now()
+ td = stop-start
+ total_seconds = datetime_seconds(td)
+ print "Creating the badges took %.1f seconds (%.3f s per player)" % (total_seconds, total_seconds/float(len(players)))
+ print "Total time for rendering images: %.3f s" % render_time
+ print "Total time for getting data: %.3f s" % data_time
-print "Creating badges for %d players ..." % len(players)
-start = datetime.now()
-data_time, render_time = 0,0
-for player in players:
- req.matchdict['id'] = player.player_id
-
- sstart = datetime.now()
- skin.get_data(player)
- sstop = datetime.now()
- td = sstop-sstart
- total_seconds = (td.microseconds + (td.seconds + td.days * 24 * 3600) * 10**6) / 10**6
- data_time += total_seconds
-
- sstart = datetime.now()
- skin.render_image("output/%d.png" % player.player_id)
- sstop = datetime.now()
- td = sstop-sstart
- total_seconds = (td.microseconds + (td.seconds + td.days * 24 * 3600) * 10**6) / 10**6
- render_time += total_seconds
-
-stop = datetime.now()
-td = stop-start
-total_seconds = (td.microseconds + (td.seconds + td.days * 24 * 3600) * 10**6) / 10**6
-print "Creating the badges took %.1f seconds (%.3f s per player)" % (total_seconds, total_seconds/float(len(players)))
-print "Total time for redering images: %.3f s" % render_time
-print "Total time for getting data: %.3f s" % data_time
+else:
+ print "No active players found!"