Merge branch 'master' into zykure/wip
[xonotic/xonstat.git] / xonstat / views / player.py
1 import datetime
2 import logging
3 import pyramid.httpexceptions
4 import sqlalchemy as sa
5 import sqlalchemy.sql.functions as func
6 import sqlalchemy.sql.expression as expr
7 from calendar import timegm
8 from collections import namedtuple
9 from webhelpers.paginate import Page
10 from xonstat.models import *
11 from xonstat.util import page_url, to_json, pretty_date, datetime_seconds
12 from xonstat.util import is_cake_day, verify_request
13 from xonstat.views.helpers import RecentGame, recent_games_q
14 from urllib import unquote
15
16 log = logging.getLogger(__name__)
17
18
19 def player_index_data(request):
20     if request.params.has_key('page'):
21         current_page = request.params['page']
22     else:
23         current_page = 1
24
25     try:
26         player_q = DBSession.query(Player).\
27                 filter(Player.player_id > 2).\
28                 filter(Player.active_ind == True).\
29                 filter(sa.not_(Player.nick.like('Anonymous Player%'))).\
30                 order_by(Player.player_id.desc())
31
32         players = Page(player_q, current_page, items_per_page=25, url=page_url)
33
34     except Exception as e:
35         players = None
36         raise e
37
38     return {'players':players
39            }
40
41
42 def player_index(request):
43     """
44     Provides a list of all the current players.
45     """
46     return player_index_data(request)
47
48
49 def player_index_json(request):
50     """
51     Provides a list of all the current players. JSON.
52     """
53     return [{'status':'not implemented'}]
54
55
56 def get_games_played(player_id):
57     """
58     Provides a breakdown by gametype of the games played by player_id.
59
60     Returns a list of namedtuples with the following members:
61         - game_type_cd
62         - games
63         - wins
64         - losses
65         - win_pct
66
67     The list itself is ordered by the number of games played
68     """
69     GamesPlayed = namedtuple('GamesPlayed', ['game_type_cd', 'games', 'wins',
70         'losses', 'win_pct'])
71
72     raw_games_played = DBSession.query('game_type_cd', 'wins', 'losses').\
73             from_statement(
74                 "SELECT game_type_cd, "
75                        "SUM(win) wins, "
76                        "SUM(loss) losses "
77                 "FROM   (SELECT g.game_id, "
78                                "g.game_type_cd, "
79                                "CASE "
80                                  "WHEN g.winner = pgs.team THEN 1 "
81                                  "WHEN pgs.scoreboardpos = 1 THEN 1 "
82                                  "ELSE 0 "
83                                "END win, "
84                                "CASE "
85                                  "WHEN g.winner = pgs.team THEN 0 "
86                                  "WHEN pgs.scoreboardpos = 1 THEN 0 "
87                                  "ELSE 1 "
88                                "END loss "
89                         "FROM   games g, "
90                                "player_game_stats pgs "
91                         "WHERE  g.game_id = pgs.game_id "
92                         "AND pgs.player_id = :player_id) win_loss "
93                 "GROUP  BY game_type_cd "
94             ).params(player_id=player_id).all()
95
96     games_played = []
97     overall_games = 0
98     overall_wins = 0
99     overall_losses = 0
100     for row in raw_games_played:
101         games = row.wins + row.losses
102         overall_games += games
103         overall_wins += row.wins
104         overall_losses += row.losses
105         win_pct = float(row.wins)/games * 100
106
107         games_played.append(GamesPlayed(row.game_type_cd, games, row.wins,
108             row.losses, win_pct))
109
110     try:
111         overall_win_pct = float(overall_wins)/overall_games * 100
112     except:
113         overall_win_pct = 0.0
114
115     games_played.append(GamesPlayed('overall', overall_games, overall_wins,
116         overall_losses, overall_win_pct))
117
118     # sort the resulting list by # of games played
119     games_played = sorted(games_played, key=lambda x:x.games)
120     games_played.reverse()
121     return games_played
122
123
124 def get_overall_stats(player_id):
125     """
126     Provides a breakdown of stats by gametype played by player_id.
127
128     Returns a dictionary of namedtuples with the following members:
129         - total_kills
130         - total_deaths
131         - k_d_ratio
132         - last_played (last time the player played the game type)
133         - last_played_epoch (same as above, but in seconds since epoch)
134         - last_played_fuzzy (same as above, but in relative date)
135         - total_playing_time (total amount of time played the game type)
136         - total_playing_time_secs (same as the above, but in seconds)
137         - total_pickups (ctf only)
138         - total_captures (ctf only)
139         - cap_ratio (ctf only)
140         - total_carrier_frags (ctf only)
141         - game_type_cd
142         - game_type_descr
143
144     The key to the dictionary is the game type code. There is also an
145     "overall" game_type_cd which sums the totals and computes the total ratios.
146     """
147     OverallStats = namedtuple('OverallStats', ['total_kills', 'total_deaths',
148         'k_d_ratio', 'last_played', 'last_played_epoch', 'last_played_fuzzy',
149         'total_playing_time', 'total_playing_time_secs', 'total_pickups', 'total_captures', 'cap_ratio',
150         'total_carrier_frags', 'game_type_cd', 'game_type_descr'])
151
152     raw_stats = DBSession.query('game_type_cd', 'game_type_descr',
153             'total_kills', 'total_deaths', 'last_played', 'total_playing_time',
154             'total_pickups', 'total_captures', 'total_carrier_frags').\
155             from_statement(
156                 "SELECT g.game_type_cd, "
157                        "gt.descr game_type_descr, "
158                        "Sum(pgs.kills)         total_kills, "
159                        "Sum(pgs.deaths)        total_deaths, "
160                        "Max(pgs.create_dt)     last_played, "
161                        "Sum(pgs.alivetime)     total_playing_time, "
162                        "Sum(pgs.pickups)       total_pickups, "
163                        "Sum(pgs.captures)      total_captures, "
164                        "Sum(pgs.carrier_frags) total_carrier_frags "
165                 "FROM   games g, "
166                        "cd_game_type gt, "
167                        "player_game_stats pgs "
168                 "WHERE  g.game_id = pgs.game_id "
169                   "AND  g.game_type_cd = gt.game_type_cd "
170                   "AND  pgs.player_id = :player_id "
171                 "GROUP  BY g.game_type_cd, game_type_descr "
172                 "UNION "
173                 "SELECT 'overall'              game_type_cd, "
174                        "'Overall'              game_type_descr, "
175                        "Sum(pgs.kills)         total_kills, "
176                        "Sum(pgs.deaths)        total_deaths, "
177                        "Max(pgs.create_dt)     last_played, "
178                        "Sum(pgs.alivetime)     total_playing_time, "
179                        "Sum(pgs.pickups)       total_pickups, "
180                        "Sum(pgs.captures)      total_captures, "
181                        "Sum(pgs.carrier_frags) total_carrier_frags "
182                 "FROM   player_game_stats pgs "
183                 "WHERE  pgs.player_id = :player_id "
184             ).params(player_id=player_id).all()
185
186     # to be indexed by game_type_cd
187     overall_stats = {}
188
189     for row in raw_stats:
190         # individual gametype ratio calculations
191         try:
192             k_d_ratio = float(row.total_kills)/row.total_deaths
193         except:
194             k_d_ratio = None
195
196         try:
197             cap_ratio = float(row.total_captures)/row.total_pickups
198         except:
199             cap_ratio = None
200
201         # everything else is untouched or "raw"
202         os = OverallStats(total_kills=row.total_kills,
203                 total_deaths=row.total_deaths,
204                 k_d_ratio=k_d_ratio,
205                 last_played=row.last_played,
206                 last_played_epoch=timegm(row.last_played.timetuple()),
207                 last_played_fuzzy=pretty_date(row.last_played),
208                 total_playing_time=row.total_playing_time,
209                 total_playing_time_secs=int(datetime_seconds(row.total_playing_time)),
210                 total_pickups=row.total_pickups,
211                 total_captures=row.total_captures,
212                 cap_ratio=cap_ratio,
213                 total_carrier_frags=row.total_carrier_frags,
214                 game_type_cd=row.game_type_cd,
215                 game_type_descr=row.game_type_descr)
216
217         overall_stats[row.game_type_cd] = os
218
219     # We have to edit "overall" stats to exclude deaths in CTS.
220     # Although we still want to record deaths, they shouldn't
221     # count towards the overall K:D ratio.
222     if 'cts' in overall_stats:
223         os = overall_stats['overall']
224
225         try:
226             k_d_ratio = float(os.total_kills)/(os.total_deaths - overall_stats['cts'].total_deaths)
227         except:
228             k_d_ratio = None
229
230         non_cts_deaths = os.total_deaths - overall_stats['cts'].total_deaths
231
232
233         overall_stats['overall'] = OverallStats(
234                 total_kills             = os.total_kills,
235                 total_deaths            = non_cts_deaths,
236                 k_d_ratio               = k_d_ratio,
237                 last_played             = os.last_played,
238                 last_played_epoch       = os.last_played_epoch,
239                 last_played_fuzzy       = os.last_played_fuzzy,
240                 total_playing_time      = os.total_playing_time,
241                 total_playing_time_secs = os.total_playing_time_secs,
242                 total_pickups           = os.total_pickups,
243                 total_captures          = os.total_captures,
244                 cap_ratio               = os.cap_ratio,
245                 total_carrier_frags     = os.total_carrier_frags,
246                 game_type_cd            = os.game_type_cd,
247                 game_type_descr         = os.game_type_descr)
248
249     return overall_stats
250
251
252 def get_fav_maps(player_id, game_type_cd=None):
253     """
254     Provides a breakdown of favorite maps by gametype.
255
256     Returns a dictionary of namedtuples with the following members:
257         - game_type_cd
258         - map_name (map name)
259         - map_id
260         - times_played
261
262     The favorite map is defined as the map you've played the most
263     for the given game_type_cd.
264
265     The key to the dictionary is the game type code. There is also an
266     "overall" game_type_cd which is the overall favorite map. This is
267     defined as the favorite map of the game type you've played the
268     most. The input parameter game_type_cd is for this.
269     """
270     FavMap = namedtuple('FavMap', ['map_name', 'map_id', 'times_played', 'game_type_cd'])
271
272     raw_favs = DBSession.query('game_type_cd', 'map_name',
273             'map_id', 'times_played').\
274             from_statement(
275                 "SELECT game_type_cd, "
276                        "name map_name, "
277                        "map_id, "
278                        "times_played "
279                 "FROM   (SELECT g.game_type_cd, "
280                                "m.name, "
281                                "m.map_id, "
282                                "Count(*) times_played, "
283                                "Row_number() "
284                                  "OVER ( "
285                                    "partition BY g.game_type_cd "
286                                    "ORDER BY Count(*) DESC, m.map_id ASC) rank "
287                         "FROM   games g, "
288                                "player_game_stats pgs, "
289                                "maps m "
290                         "WHERE  g.game_id = pgs.game_id "
291                                "AND g.map_id = m.map_id "
292                                "AND pgs.player_id = :player_id "
293                         "GROUP  BY g.game_type_cd, "
294                                   "m.map_id, "
295                                   "m.name) most_played "
296                 "WHERE  rank = 1 "
297                 "ORDER BY  times_played desc "
298             ).params(player_id=player_id).all()
299
300     fav_maps = {}
301     overall_fav = None
302     for row in raw_favs:
303         fv = FavMap(map_name=row.map_name,
304             map_id=row.map_id,
305             times_played=row.times_played,
306             game_type_cd=row.game_type_cd)
307
308         # if we aren't given a favorite game_type_cd
309         # then the overall favorite is the one we've
310         # played the most
311         if overall_fav is None:
312             fav_maps['overall'] = fv
313             overall_fav = fv.game_type_cd
314
315         # otherwise it is the favorite map from the
316         # favorite game_type_cd (provided as a param)
317         # and we'll overwrite the first dict entry
318         if game_type_cd == fv.game_type_cd:
319             fav_maps['overall'] = fv
320
321         fav_maps[row.game_type_cd] = fv
322
323     return fav_maps
324
325
326 def get_ranks(player_id):
327     """
328     Provides a breakdown of the player's ranks by game type.
329
330     Returns a dictionary of namedtuples with the following members:
331         - game_type_cd
332         - rank
333         - max_rank
334
335     The key to the dictionary is the game type code. There is also an
336     "overall" game_type_cd which is the overall best rank.
337     """
338     Rank = namedtuple('Rank', ['rank', 'max_rank', 'percentile', 'game_type_cd'])
339
340     raw_ranks = DBSession.query("game_type_cd", "rank", "max_rank").\
341             from_statement(
342                 "select pr.game_type_cd, pr.rank, overall.max_rank "
343                 "from player_ranks pr,  "
344                    "(select game_type_cd, max(rank) max_rank "
345                     "from player_ranks  "
346                     "group by game_type_cd) overall "
347                 "where pr.game_type_cd = overall.game_type_cd  "
348                 "and player_id = :player_id "
349                 "order by rank").\
350             params(player_id=player_id).all()
351
352     ranks = {}
353     found_top_rank = False
354     for row in raw_ranks:
355         rank = Rank(rank=row.rank,
356             max_rank=row.max_rank,
357             percentile=100 - 100*float(row.rank-1)/(row.max_rank-1),
358             game_type_cd=row.game_type_cd)
359
360
361         if not found_top_rank:
362             ranks['overall'] = rank
363             found_top_rank = True
364         elif rank.percentile > ranks['overall'].percentile:
365             ranks['overall'] = rank
366
367         ranks[row.game_type_cd] = rank
368
369     return ranks;
370
371
372 def get_elos(player_id):
373     """
374     Provides a breakdown of the player's elos by game type.
375
376     Returns a dictionary of namedtuples with the following members:
377         - player_id
378         - game_type_cd
379         - games
380         - elo
381
382     The key to the dictionary is the game type code. There is also an
383     "overall" game_type_cd which is the overall best rank.
384     """
385     raw_elos = DBSession.query(PlayerElo).filter_by(player_id=player_id).\
386             order_by(PlayerElo.elo.desc()).all()
387
388     elos = {}
389     found_max_elo = False
390     for row in raw_elos:
391         if not found_max_elo:
392             elos['overall'] = row
393             found_max_elo = True
394
395         elos[row.game_type_cd] = row
396
397     return elos
398
399
400 def get_recent_games(player_id, limit=10):
401     """
402     Provides a list of recent games for a player. Uses the recent_games_q helper.
403     """
404     # recent games played in descending order
405     rgs = recent_games_q(player_id=player_id, force_player_id=True).limit(limit).all()
406     recent_games = [RecentGame(row) for row in rgs]
407
408     return recent_games
409
410
411 def get_recent_weapons(player_id):
412     """
413     Returns the weapons that have been used in the past 90 days
414     and also used in 5 games or more.
415     """
416     cutoff = datetime.datetime.utcnow() - datetime.timedelta(days=90)
417     recent_weapons = []
418     for weapon in DBSession.query(PlayerWeaponStat.weapon_cd, func.count()).\
419             filter(PlayerWeaponStat.player_id == player_id).\
420             filter(PlayerWeaponStat.create_dt > cutoff).\
421             group_by(PlayerWeaponStat.weapon_cd).\
422             having(func.count() > 4).\
423             all():
424                 recent_weapons.append(weapon[0])
425
426     return recent_weapons
427
428
429 def get_accuracy_stats(player_id, weapon_cd, games):
430     """
431     Provides accuracy for weapon_cd by player_id for the past N games.
432     """
433     # Reaching back 90 days should give us an accurate enough average
434     # We then multiply this out for the number of data points (games) to
435     # create parameters for a flot graph
436     try:
437         raw_avg = DBSession.query(func.sum(PlayerWeaponStat.hit),
438                 func.sum(PlayerWeaponStat.fired)).\
439                 filter(PlayerWeaponStat.player_id == player_id).\
440                 filter(PlayerWeaponStat.weapon_cd == weapon_cd).\
441                 one()
442
443         avg = round(float(raw_avg[0])/raw_avg[1]*100, 2)
444
445         # Determine the raw accuracy (hit, fired) numbers for $games games
446         # This is then enumerated to create parameters for a flot graph
447         raw_accs = DBSession.query(PlayerWeaponStat.game_id,
448             PlayerWeaponStat.hit, PlayerWeaponStat.fired).\
449                 filter(PlayerWeaponStat.player_id == player_id).\
450                 filter(PlayerWeaponStat.weapon_cd == weapon_cd).\
451                 order_by(PlayerWeaponStat.game_id.desc()).\
452                 limit(games).\
453                 all()
454
455         # they come out in opposite order, so flip them in the right direction
456         raw_accs.reverse()
457
458         accs = []
459         for i in range(len(raw_accs)):
460             accs.append((raw_accs[i][0], round(float(raw_accs[i][1])/raw_accs[i][2]*100, 2)))
461     except:
462         accs = []
463         avg = 0.0
464
465     return (avg, accs)
466
467
468 def get_damage_stats(player_id, weapon_cd, games):
469     """
470     Provides damage info for weapon_cd by player_id for the past N games.
471     """
472     try:
473         raw_avg = DBSession.query(func.sum(PlayerWeaponStat.actual),
474                 func.sum(PlayerWeaponStat.hit)).\
475                 filter(PlayerWeaponStat.player_id == player_id).\
476                 filter(PlayerWeaponStat.weapon_cd == weapon_cd).\
477                 one()
478
479         avg = round(float(raw_avg[0])/raw_avg[1], 2)
480
481         # Determine the damage efficiency (hit, fired) numbers for $games games
482         # This is then enumerated to create parameters for a flot graph
483         raw_dmgs = DBSession.query(PlayerWeaponStat.game_id,
484             PlayerWeaponStat.actual, PlayerWeaponStat.hit).\
485                 filter(PlayerWeaponStat.player_id == player_id).\
486                 filter(PlayerWeaponStat.weapon_cd == weapon_cd).\
487                 order_by(PlayerWeaponStat.game_id.desc()).\
488                 limit(games).\
489                 all()
490
491         # they come out in opposite order, so flip them in the right direction
492         raw_dmgs.reverse()
493
494         dmgs = []
495         for i in range(len(raw_dmgs)):
496             # try to derive, unless we've hit nothing then set to 0!
497             try:
498                 dmg = round(float(raw_dmgs[i][1])/raw_dmgs[i][2], 2)
499             except:
500                 dmg = 0.0
501
502             dmgs.append((raw_dmgs[i][0], dmg))
503     except Exception as e:
504         dmgs = []
505         avg = 0.0
506
507     return (avg, dmgs)
508
509
510 def player_info_data(request):
511     player_id = int(request.matchdict['id'])
512     if player_id <= 2:
513         player_id = -1;
514
515     try:
516         player = DBSession.query(Player).filter_by(player_id=player_id).\
517                 filter(Player.active_ind == True).one()
518
519         games_played   = get_games_played(player_id)
520         overall_stats  = get_overall_stats(player_id)
521         fav_maps       = get_fav_maps(player_id)
522         elos           = get_elos(player_id)
523         ranks          = get_ranks(player_id)
524         recent_games   = get_recent_games(player_id)
525         recent_weapons = get_recent_weapons(player_id)
526         cake_day       = is_cake_day(player.create_dt)
527
528     except Exception as e:
529         raise pyramid.httpexceptions.HTTPNotFound
530
531         ## do not raise application exceptions here (only for debugging)
532         # raise e
533
534     return {'player':player,
535             'games_played':games_played,
536             'overall_stats':overall_stats,
537             'fav_maps':fav_maps,
538             'elos':elos,
539             'ranks':ranks,
540             'recent_games':recent_games,
541             'recent_weapons':recent_weapons,
542             'cake_day':cake_day,
543             }
544
545
546 def player_info(request):
547     """
548     Provides detailed information on a specific player
549     """
550     return player_info_data(request)
551
552
553 def player_info_json(request):
554     """
555     Provides detailed information on a specific player. JSON.
556     """
557
558     # All player_info fields are converted into JSON-formattable dictionaries
559     player_info = player_info_data(request)
560
561     player = player_info['player'].to_dict()
562
563     games_played = {}
564     for game in player_info['games_played']:
565         games_played[game.game_type_cd] = to_json(game)
566
567     overall_stats = {}
568     for gt,stats in player_info['overall_stats'].items():
569         overall_stats[gt] = to_json(stats)
570
571     elos = {}
572     for gt,elo in player_info['elos'].items():
573         elos[gt] = to_json(elo.to_dict())
574
575     ranks = {}
576     for gt,rank in player_info['ranks'].items():
577         ranks[gt] = to_json(rank)
578
579     fav_maps = {}
580     for gt,mapinfo in player_info['fav_maps'].items():
581         fav_maps[gt] = to_json(mapinfo)
582
583     recent_games = []
584     for game in player_info['recent_games']:
585         recent_games.append(to_json(game))
586
587     #recent_weapons = player_info['recent_weapons']
588
589     return [{
590         'player':           player,
591         'games_played':     games_played,
592         'overall_stats':    overall_stats,
593         'fav_maps':         fav_maps,
594         'elos':             elos,
595         'ranks':            ranks,
596         'recent_games':     recent_games,
597     #    'recent_weapons':   recent_weapons,
598         'recent_weapons':   ['not implemented'],
599     }]
600     #return [{'status':'not implemented'}]
601
602
603 def player_game_index_data(request):
604     player_id = request.matchdict['player_id']
605
606     game_type_cd = None
607     game_type_descr = None
608
609     if request.params.has_key('type'):
610         game_type_cd = request.params['type']
611         try:
612             game_type_descr = DBSession.query(GameType.descr).\
613                 filter(GameType.game_type_cd == game_type_cd).\
614                 one()[0]
615         except Exception as e:
616             pass
617
618     else:
619         game_type_cd = None
620         game_type_descr = None
621
622     if request.params.has_key('page'):
623         current_page = request.params['page']
624     else:
625         current_page = 1
626
627     try:
628         player = DBSession.query(Player).\
629                 filter_by(player_id=player_id).\
630                 filter(Player.active_ind == True).\
631                 one()
632
633         rgs_q = recent_games_q(player_id=player.player_id,
634             force_player_id=True, game_type_cd=game_type_cd)
635
636         games = Page(rgs_q, current_page, items_per_page=20, url=page_url)
637
638         # replace the items in the canned pagination class with more rich ones
639         games.items = [RecentGame(row) for row in games.items]
640
641         games_played = get_games_played(player_id)
642
643     except Exception as e:
644         player = None
645         games = None
646         game_type_cd = None
647         game_type_descr = None
648         games_played = None
649
650     return {
651             'player_id':player.player_id,
652             'player':player,
653             'games':games,
654             'game_type_cd':game_type_cd,
655             'game_type_descr':game_type_descr,
656             'games_played':games_played,
657            }
658
659
660 def player_game_index(request):
661     """
662     Provides an index of the games in which a particular
663     player was involved. This is ordered by game_id, with
664     the most recent game_ids first. Paginated.
665     """
666     return player_game_index_data(request)
667
668
669 def player_game_index_json(request):
670     """
671     Provides an index of the games in which a particular
672     player was involved. This is ordered by game_id, with
673     the most recent game_ids first. Paginated. JSON.
674     """
675     return [{'status':'not implemented'}]
676
677
678 def player_accuracy_data(request):
679     player_id = request.matchdict['id']
680     allowed_weapons = ['nex', 'rifle', 'shotgun', 'uzi', 'minstanex']
681     weapon_cd = 'nex'
682     games = 20
683
684     if request.params.has_key('weapon'):
685         if request.params['weapon'] in allowed_weapons:
686             weapon_cd = request.params['weapon']
687
688     if request.params.has_key('games'):
689         try:
690             games = request.params['games']
691
692             if games < 0:
693                 games = 20
694             if games > 50:
695                 games = 50
696         except:
697             games = 20
698
699     (avg, accs) = get_accuracy_stats(player_id, weapon_cd, games)
700
701     # if we don't have enough data for the given weapon
702     if len(accs) < games:
703         games = len(accs)
704
705     return {
706             'player_id':player_id,
707             'player_url':request.route_url('player_info', id=player_id),
708             'weapon':weapon_cd,
709             'games':games,
710             'avg':avg,
711             'accs':accs
712             }
713
714
715 def player_accuracy(request):
716     """
717     Provides the accuracy for the given weapon. (JSON only)
718     """
719     return player_accuracy_data(request)
720
721
722 def player_accuracy_json(request):
723     """
724     Provides a JSON response representing the accuracy for the given weapon.
725
726     Parameters:
727        weapon = which weapon to display accuracy for. Valid values are 'nex',
728                 'shotgun', 'uzi', and 'minstanex'.
729        games = over how many games to display accuracy. Can be up to 50.
730     """
731     return player_accuracy_data(request)
732
733
734 def player_damage_data(request):
735     player_id = request.matchdict['id']
736     allowed_weapons = ['grenadelauncher', 'electro', 'crylink', 'hagar',
737             'rocketlauncher', 'laser']
738     weapon_cd = 'rocketlauncher'
739     games = 20
740
741     if request.params.has_key('weapon'):
742         if request.params['weapon'] in allowed_weapons:
743             weapon_cd = request.params['weapon']
744
745     if request.params.has_key('games'):
746         try:
747             games = request.params['games']
748
749             if games < 0:
750                 games = 20
751             if games > 50:
752                 games = 50
753         except:
754             games = 20
755
756     (avg, dmgs) = get_damage_stats(player_id, weapon_cd, games)
757
758     # if we don't have enough data for the given weapon
759     if len(dmgs) < games:
760         games = len(dmgs)
761
762     return {
763             'player_id':player_id,
764             'player_url':request.route_url('player_info', id=player_id),
765             'weapon':weapon_cd,
766             'games':games,
767             'avg':avg,
768             'dmgs':dmgs
769             }
770
771
772 def player_damage_json(request):
773     """
774     Provides a JSON response representing the damage for the given weapon.
775
776     Parameters:
777        weapon = which weapon to display damage for. Valid values are
778          'grenadelauncher', 'electro', 'crylink', 'hagar', 'rocketlauncher',
779          'laser'.
780        games = over how many games to display damage. Can be up to 50.
781     """
782     return player_damage_data(request)
783
784
785 def player_hashkey_info_data(request):
786     # hashkey = request.matchdict['hashkey']
787
788     # the incoming hashkey is double quoted, and WSGI unquotes once...
789     # hashkey = unquote(hashkey)
790
791     # if using request verification to obtain the hashkey
792     (idfp, status) = verify_request(request)
793     log.debug("d0_blind_id verification: idfp={0} status={1}\n".format(idfp, status))
794
795     log.debug("\n----- BEGIN REQUEST BODY -----\n" + request.body +
796             "----- END REQUEST BODY -----\n\n")
797
798     # if config is to *not* verify requests and we get nothing back, this
799     # query will return nothing and we'll 404.
800     try:
801         player = DBSession.query(Player).\
802                 filter(Player.player_id == Hashkey.player_id).\
803                 filter(Player.active_ind == True).\
804                 filter(Hashkey.hashkey == idfp).one()
805
806         games_played      = get_games_played(player.player_id)
807         overall_stats     = get_overall_stats(player.player_id)
808         fav_maps          = get_fav_maps(player.player_id)
809         elos              = get_elos(player.player_id)
810         ranks             = get_ranks(player.player_id)
811         most_recent_game  = get_recent_games(player.player_id, 1)[0]
812
813     except Exception as e:
814         raise pyramid.httpexceptions.HTTPNotFound
815
816     return {'player':player,
817             'hashkey':idfp,
818             'games_played':games_played,
819             'overall_stats':overall_stats,
820             'fav_maps':fav_maps,
821             'elos':elos,
822             'ranks':ranks,
823             'most_recent_game':most_recent_game,
824             }
825
826
827 def player_hashkey_info_json(request):
828     """
829     Provides detailed information on a specific player. JSON.
830     """
831
832     # All player_info fields are converted into JSON-formattable dictionaries
833     player_info = player_hashkey_info_data(request)
834
835     player = player_info['player'].to_dict()
836
837     games_played = {}
838     for game in player_info['games_played']:
839         games_played[game.game_type_cd] = to_json(game)
840
841     overall_stats = {}
842     for gt,stats in player_info['overall_stats'].items():
843         overall_stats[gt] = to_json(stats)
844
845     elos = {}
846     for gt,elo in player_info['elos'].items():
847         elos[gt] = to_json(elo.to_dict())
848
849     ranks = {}
850     for gt,rank in player_info['ranks'].items():
851         ranks[gt] = to_json(rank)
852
853     fav_maps = {}
854     for gt,mapinfo in player_info['fav_maps'].items():
855         fav_maps[gt] = to_json(mapinfo)
856
857     most_recent_game = to_json(player_info['most_recent_game'])
858
859     return [{
860         'version':          1,
861         'player':           player,
862         'games_played':     games_played,
863         'overall_stats':    overall_stats,
864         'fav_maps':         fav_maps,
865         'elos':             elos,
866         'ranks':            ranks,
867         'most_recent_game': most_recent_game,
868     }]
869
870
871 def player_hashkey_info_text(request):
872     """
873     Provides detailed information on a specific player. Plain text.
874     """
875     # UTC epoch
876     now = timegm(datetime.datetime.utcnow().timetuple())
877
878     # All player_info fields are converted into JSON-formattable dictionaries
879     player_info = player_hashkey_info_data(request)
880
881     # gather all of the data up into aggregate structures
882     player = player_info['player']
883     games_played = player_info['games_played']
884     overall_stats = player_info['overall_stats']
885     elos = player_info['elos']
886     ranks = player_info['ranks']
887     fav_maps = player_info['fav_maps']
888     most_recent_game = player_info['most_recent_game']
889
890     # one-offs for things needing conversion for text/plain
891     player_joined = timegm(player.create_dt.timetuple())
892     player_joined_dt = player.create_dt
893     alivetime = int(datetime_seconds(overall_stats['overall'].total_playing_time))
894
895     # this is a plain text response, if we don't do this here then
896     # Pyramid will assume html
897     request.response.content_type = 'text/plain'
898
899     return {
900         'version':          1,
901         'now':              now,
902         'player':           player,
903         'hashkey':          player_info['hashkey'],
904         'player_joined':    player_joined,
905         'player_joined_dt': player_joined_dt,
906         'games_played':     games_played,
907         'overall_stats':    overall_stats,
908         'alivetime':        alivetime,
909         'fav_maps':         fav_maps,
910         'elos':             elos,
911         'ranks':            ranks,
912         'most_recent_game': most_recent_game,
913     }
914
915
916 def player_elo_info_data(request):
917     """
918     Provides elo information on a specific player. Raw data is returned.
919     """
920     (idfp, status) = verify_request(request)
921     log.debug("d0_blind_id verification: idfp={0} status={1}\n".format(idfp, status))
922
923     log.debug("\n----- BEGIN REQUEST BODY -----\n" + request.body +
924             "----- END REQUEST BODY -----\n\n")
925
926     hashkey = request.matchdict['hashkey']
927
928     # the incoming hashkey is double quoted, and WSGI unquotes once...
929     hashkey = unquote(hashkey)
930
931     try:
932         player = DBSession.query(Player).\
933                 filter(Player.player_id == Hashkey.player_id).\
934                 filter(Player.active_ind == True).\
935                 filter(Hashkey.hashkey == hashkey).one()
936
937         elos = get_elos(player.player_id)
938
939     except Exception as e:
940         log.debug(e)
941         raise pyramid.httpexceptions.HTTPNotFound
942
943     return {
944         'hashkey':hashkey,
945         'player':player,
946         'elos':elos,
947     }
948
949
950 def player_elo_info_json(request):
951     """
952     Provides elo information on a specific player. JSON.
953     """
954     elo_info = player_elo_info_data(request)
955
956     player = player_info['player'].to_dict()
957
958     elos = {}
959     for gt, elo in elo_info['elos'].items():
960         elos[gt] = to_json(elo.to_dict())
961
962     return [{
963         'version':          1,
964         'player':           player,
965         'elos':             elos,
966     }]
967
968
969 def player_elo_info_text(request):
970     """
971     Provides elo information on a specific player. Plain text.
972     """
973     # UTC epoch
974     now = timegm(datetime.datetime.utcnow().timetuple())
975
976     # All player_info fields are converted into JSON-formattable dictionaries
977     elo_info = player_elo_info_data(request)
978
979     # this is a plain text response, if we don't do this here then
980     # Pyramid will assume html
981     request.response.content_type = 'text/plain'
982
983     return {
984         'version':          1,
985         'now':              now,
986         'hashkey':          elo_info['hashkey'],
987         'player':           elo_info['player'],
988         'elos':             elo_info['elos'],
989     }
990
991
992 def player_captimes_data(request):
993     player_id = int(request.matchdict['player_id'])
994     if player_id <= 2:
995         player_id = -1;
996
997     if request.params.has_key('page'):
998         current_page = request.params['page']
999     else:
1000         current_page = 1
1001
1002     PlayerCaptimes = namedtuple('PlayerCaptimes', ['fastest_cap',
1003             'create_dt', 'create_dt_epoch', 'create_dt_fuzzy',
1004             'player_id', 'game_id', 'map_id', 'map_name', 'server_id', 'server_name'])
1005
1006     player = DBSession.query(Player).filter_by(player_id=player_id).one()
1007
1008     #pct_q = DBSession.query('fastest_cap', 'create_dt', 'player_id', 'game_id', 'map_id',
1009     #            'map_name', 'server_id', 'server_name').\
1010     #        from_statement(
1011     #            "SELECT ct.fastest_cap, "
1012     #                   "ct.create_dt, "
1013     #                   "ct.player_id, "
1014     #                   "ct.game_id, "
1015     #                   "ct.map_id, "
1016     #                   "m.name map_name, "
1017     #                   "g.server_id, "
1018     #                   "s.name server_name "
1019     #            "FROM   player_map_captimes ct, "
1020     #                   "games g, "
1021     #                   "maps m, "
1022     #                   "servers s "
1023     #            "WHERE  ct.player_id = :player_id "
1024     #              "AND  g.game_id = ct.game_id "
1025     #              "AND  g.server_id = s.server_id "
1026     #              "AND  m.map_id = ct.map_id "
1027     #            #"ORDER  BY ct.fastest_cap "
1028     #            "ORDER  BY ct.create_dt desc"
1029     #        ).params(player_id=player_id)
1030
1031     try:
1032         pct_q = DBSession.query(PlayerCaptime.fastest_cap, PlayerCaptime.create_dt,
1033                 PlayerCaptime.player_id, PlayerCaptime.game_id, PlayerCaptime.map_id,
1034                 Map.name.label('map_name'), Game.server_id, Server.name.label('server_name')).\
1035                 filter(PlayerCaptime.player_id==player_id).\
1036                 filter(PlayerCaptime.game_id==Game.game_id).\
1037                 filter(PlayerCaptime.map_id==Map.map_id).\
1038                 filter(Game.server_id==Server.server_id).\
1039                 order_by(expr.asc(PlayerCaptime.fastest_cap))  # or PlayerCaptime.create_dt
1040
1041         player_captimes = Page(pct_q, current_page, items_per_page=20, url=page_url)
1042
1043         # replace the items in the canned pagination class with more rich ones
1044         player_captimes.items = [PlayerCaptimes(
1045                 fastest_cap=row.fastest_cap,
1046                 create_dt=row.create_dt,
1047                 create_dt_epoch=timegm(row.create_dt.timetuple()),
1048                 create_dt_fuzzy=pretty_date(row.create_dt),
1049                 player_id=row.player_id,
1050                 game_id=row.game_id,
1051                 map_id=row.map_id,
1052                 map_name=row.map_name,
1053                 server_id=row.server_id,
1054                 server_name=row.server_name
1055                 ) for row in player_captimes.items]
1056
1057     except Exception as e:
1058         player = None
1059         player_captimes = None
1060
1061     return {
1062             'player_id':player_id,
1063             'player':player,
1064             'captimes':player_captimes,
1065             #'player_url':request.route_url('player_info', id=player_id),
1066         }
1067
1068
1069 def player_captimes(request):
1070     return player_captimes_data(request)
1071
1072
1073 def player_captimes_json(request):
1074     return player_captimes_data(request)
1075
1076
1077 def player_weaponstats_data_json(request):
1078     player_id = request.matchdict["id"]
1079     if player_id <= 2:
1080         player_id = -1;
1081
1082     game_type_cd = request.params.get("game_type", None)
1083     if game_type_cd == "overall":
1084         game_type_cd = None
1085
1086     limit = 20
1087     if request.params.has_key("limit"):
1088         limit = int(request.params["limit"])
1089
1090         if limit < 0:
1091             limit = 20
1092         if limit > 50:
1093             limit = 50
1094
1095     games_raw = DBSession.query(sa.distinct(Game.game_id)).\
1096         filter(Game.game_id == PlayerWeaponStat.game_id).\
1097         filter(PlayerWeaponStat.player_id == player_id)
1098
1099     if game_type_cd is not None:
1100         games_raw = games_raw.filter(Game.game_type_cd == game_type_cd)
1101
1102     games_raw = games_raw.order_by(Game.game_id.desc()).limit(limit).all()
1103
1104     weapon_stats_raw = DBSession.query(PlayerWeaponStat).\
1105         filter(PlayerWeaponStat.player_id == player_id).\
1106         filter(PlayerWeaponStat.game_id.in_(games_raw)).all()
1107
1108     # NVD3 expects data points for all weapons used across the
1109     # set of games *for each* point on the x axis. This means populating
1110     # zero-valued weapon stat entries for games where a weapon was not
1111     # used in that game, but was used in another game for the set
1112     games_to_weapons = {}
1113     weapons_used = {}
1114     sum_avgs = {}
1115     for ws in weapon_stats_raw:
1116         if ws.game_id not in games_to_weapons:
1117             games_to_weapons[ws.game_id] = [ws.weapon_cd]
1118         else:
1119             games_to_weapons[ws.game_id].append(ws.weapon_cd)
1120
1121         weapons_used[ws.weapon_cd] = weapons_used.get(ws.weapon_cd, 0) + 1
1122         sum_avgs[ws.weapon_cd] = sum_avgs.get(ws.weapon_cd, 0) + float(ws.hit)/float(ws.fired)
1123
1124     for game_id in games_to_weapons.keys():
1125         for weapon_cd in set(weapons_used.keys()) - set(games_to_weapons[game_id]):
1126             weapon_stats_raw.append(PlayerWeaponStat(player_id=player_id,
1127                 game_id=game_id, weapon_cd=weapon_cd))
1128
1129     # averages for the weapons used in the range
1130     avgs = {}
1131     for w in weapons_used.keys():
1132         avgs[w] = round(sum_avgs[w]/float(weapons_used[w])*100, 2)
1133
1134     weapon_stats_raw = sorted(weapon_stats_raw, key = lambda x: x.game_id)
1135     games            = sorted(games_to_weapons.keys())
1136     weapon_stats     = [ws.to_dict() for ws in weapon_stats_raw]
1137
1138     return {
1139         "weapon_stats": weapon_stats,
1140         "weapons_used": weapons_used.keys(),
1141         "games": games,
1142         "averages": avgs,
1143     }
1144