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