The player JSON view should use to_dict() instead.
[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 = [g.to_dict() for g in player_info['recent_games']]
568
569     return [{
570         'player':           player,
571         'games_played':     games_played,
572         'overall_stats':    overall_stats,
573         'fav_maps':         fav_maps,
574         'elos':             elos,
575         'ranks':            ranks,
576         'recent_games':     recent_games,
577     }]
578
579
580 def player_game_index_data(request):
581     try:
582         player_id = int(request.matchdict['player_id'])
583     except:
584         player_id = -1
585
586     game_type_cd = None
587     game_type_descr = None
588
589     if request.params.has_key('type'):
590         game_type_cd = request.params['type']
591         try:
592             game_type_descr = DBSession.query(GameType.descr).\
593                 filter(GameType.game_type_cd == game_type_cd).\
594                 one()[0]
595         except Exception as e:
596             pass
597
598     else:
599         game_type_cd = None
600         game_type_descr = None
601
602     if request.params.has_key('page'):
603         current_page = request.params['page']
604     else:
605         current_page = 1
606
607     try:
608         player = DBSession.query(Player).\
609                 filter_by(player_id=player_id).\
610                 filter(Player.active_ind == True).\
611                 one()
612
613         rgs_q = recent_games_q(player_id=player.player_id,
614             force_player_id=True, game_type_cd=game_type_cd)
615
616         games = Page(rgs_q, current_page, items_per_page=20, url=page_url)
617
618         # replace the items in the canned pagination class with more rich ones
619         games.items = [RecentGame(row) for row in games.items]
620
621         games_played = get_games_played(player_id)
622
623     except Exception as e:
624         raise e
625         player = None
626         games = None
627         game_type_cd = None
628         game_type_descr = None
629         games_played = None
630
631     return {
632             'player_id':player.player_id,
633             'player':player,
634             'games':games,
635             'game_type_cd':game_type_cd,
636             'game_type_descr':game_type_descr,
637             'games_played':games_played,
638            }
639
640
641 def player_game_index(request):
642     """
643     Provides an index of the games in which a particular
644     player was involved. This is ordered by game_id, with
645     the most recent game_ids first. Paginated.
646     """
647     return player_game_index_data(request)
648
649
650 def player_game_index_json(request):
651     """
652     Provides an index of the games in which a particular
653     player was involved. This is ordered by game_id, with
654     the most recent game_ids first. Paginated. JSON.
655     """
656     return [{'status':'not implemented'}]
657
658
659 def player_accuracy_data(request):
660     player_id = request.matchdict['id']
661     allowed_weapons = ['nex', 'rifle', 'shotgun', 'uzi', 'minstanex']
662     weapon_cd = 'nex'
663     games = 20
664
665     if request.params.has_key('weapon'):
666         if request.params['weapon'] in allowed_weapons:
667             weapon_cd = request.params['weapon']
668
669     if request.params.has_key('games'):
670         try:
671             games = request.params['games']
672
673             if games < 0:
674                 games = 20
675             if games > 50:
676                 games = 50
677         except:
678             games = 20
679
680     (avg, accs) = get_accuracy_stats(player_id, weapon_cd, games)
681
682     # if we don't have enough data for the given weapon
683     if len(accs) < games:
684         games = len(accs)
685
686     return {
687             'player_id':player_id,
688             'player_url':request.route_url('player_info', id=player_id),
689             'weapon':weapon_cd,
690             'games':games,
691             'avg':avg,
692             'accs':accs
693             }
694
695
696 def player_accuracy(request):
697     """
698     Provides the accuracy for the given weapon. (JSON only)
699     """
700     return player_accuracy_data(request)
701
702
703 def player_accuracy_json(request):
704     """
705     Provides a JSON response representing the accuracy for the given weapon.
706
707     Parameters:
708        weapon = which weapon to display accuracy for. Valid values are 'nex',
709                 'shotgun', 'uzi', and 'minstanex'.
710        games = over how many games to display accuracy. Can be up to 50.
711     """
712     return player_accuracy_data(request)
713
714
715 def player_damage_data(request):
716     player_id = request.matchdict['id']
717     allowed_weapons = ['grenadelauncher', 'electro', 'crylink', 'hagar',
718             'rocketlauncher', 'laser']
719     weapon_cd = 'rocketlauncher'
720     games = 20
721
722     if request.params.has_key('weapon'):
723         if request.params['weapon'] in allowed_weapons:
724             weapon_cd = request.params['weapon']
725
726     if request.params.has_key('games'):
727         try:
728             games = request.params['games']
729
730             if games < 0:
731                 games = 20
732             if games > 50:
733                 games = 50
734         except:
735             games = 20
736
737     (avg, dmgs) = get_damage_stats(player_id, weapon_cd, games)
738
739     # if we don't have enough data for the given weapon
740     if len(dmgs) < games:
741         games = len(dmgs)
742
743     return {
744             'player_id':player_id,
745             'player_url':request.route_url('player_info', id=player_id),
746             'weapon':weapon_cd,
747             'games':games,
748             'avg':avg,
749             'dmgs':dmgs
750             }
751
752
753 def player_damage_json(request):
754     """
755     Provides a JSON response representing the damage for the given weapon.
756
757     Parameters:
758        weapon = which weapon to display damage for. Valid values are
759          'grenadelauncher', 'electro', 'crylink', 'hagar', 'rocketlauncher',
760          'laser'.
761        games = over how many games to display damage. Can be up to 50.
762     """
763     return player_damage_data(request)
764
765
766 def player_hashkey_info_data(request):
767     # hashkey = request.matchdict['hashkey']
768
769     # the incoming hashkey is double quoted, and WSGI unquotes once...
770     # hashkey = unquote(hashkey)
771
772     # if using request verification to obtain the hashkey
773     (idfp, status) = verify_request(request)
774     log.debug("d0_blind_id verification: idfp={0} status={1}\n".format(idfp, status))
775
776     log.debug("\n----- BEGIN REQUEST BODY -----\n" + request.body +
777             "----- END REQUEST BODY -----\n\n")
778
779     # if config is to *not* verify requests and we get nothing back, this
780     # query will return nothing and we'll 404.
781     try:
782         player = DBSession.query(Player).\
783                 filter(Player.player_id == Hashkey.player_id).\
784                 filter(Player.active_ind == True).\
785                 filter(Hashkey.hashkey == idfp).one()
786
787         games_played      = get_games_played(player.player_id)
788         overall_stats     = get_overall_stats(player.player_id)
789         fav_maps          = get_fav_maps(player.player_id)
790         elos              = get_elos(player.player_id)
791         ranks             = get_ranks(player.player_id)
792         most_recent_game  = get_recent_games(player.player_id, 1)[0]
793
794     except Exception as e:
795         raise pyramid.httpexceptions.HTTPNotFound
796
797     return {'player':player,
798             'hashkey':idfp,
799             'games_played':games_played,
800             'overall_stats':overall_stats,
801             'fav_maps':fav_maps,
802             'elos':elos,
803             'ranks':ranks,
804             'most_recent_game':most_recent_game,
805             }
806
807
808 def player_hashkey_info_json(request):
809     """
810     Provides detailed information on a specific player. JSON.
811     """
812
813     # All player_info fields are converted into JSON-formattable dictionaries
814     player_info = player_hashkey_info_data(request)
815
816     player = player_info['player'].to_dict()
817
818     games_played = {}
819     for game in player_info['games_played']:
820         games_played[game.game_type_cd] = to_json(game)
821
822     overall_stats = {}
823     for gt,stats in player_info['overall_stats'].items():
824         overall_stats[gt] = to_json(stats)
825
826     elos = {}
827     for gt,elo in player_info['elos'].items():
828         elos[gt] = to_json(elo.to_dict())
829
830     ranks = {}
831     for gt,rank in player_info['ranks'].items():
832         ranks[gt] = to_json(rank)
833
834     fav_maps = {}
835     for gt,mapinfo in player_info['fav_maps'].items():
836         fav_maps[gt] = to_json(mapinfo)
837
838     most_recent_game = to_json(player_info['most_recent_game'])
839
840     return [{
841         'version':          1,
842         'player':           player,
843         'games_played':     games_played,
844         'overall_stats':    overall_stats,
845         'fav_maps':         fav_maps,
846         'elos':             elos,
847         'ranks':            ranks,
848         'most_recent_game': most_recent_game,
849     }]
850
851
852 def player_hashkey_info_text(request):
853     """
854     Provides detailed information on a specific player. Plain text.
855     """
856     # UTC epoch
857     now = timegm(datetime.datetime.utcnow().timetuple())
858
859     # All player_info fields are converted into JSON-formattable dictionaries
860     player_info = player_hashkey_info_data(request)
861
862     # gather all of the data up into aggregate structures
863     player = player_info['player']
864     games_played = player_info['games_played']
865     overall_stats = player_info['overall_stats']
866     elos = player_info['elos']
867     ranks = player_info['ranks']
868     fav_maps = player_info['fav_maps']
869     most_recent_game = player_info['most_recent_game']
870
871     # one-offs for things needing conversion for text/plain
872     player_joined = timegm(player.create_dt.timetuple())
873     player_joined_dt = player.create_dt
874     alivetime = int(datetime_seconds(overall_stats['overall'].total_playing_time))
875
876     # this is a plain text response, if we don't do this here then
877     # Pyramid will assume html
878     request.response.content_type = 'text/plain'
879
880     return {
881         'version':          1,
882         'now':              now,
883         'player':           player,
884         'hashkey':          player_info['hashkey'],
885         'player_joined':    player_joined,
886         'player_joined_dt': player_joined_dt,
887         'games_played':     games_played,
888         'overall_stats':    overall_stats,
889         'alivetime':        alivetime,
890         'fav_maps':         fav_maps,
891         'elos':             elos,
892         'ranks':            ranks,
893         'most_recent_game': most_recent_game,
894     }
895
896
897 def player_elo_info_data(request):
898     """
899     Provides elo information on a specific player. Raw data is returned.
900     """
901     (idfp, status) = verify_request(request)
902     log.debug("d0_blind_id verification: idfp={0} status={1}\n".format(idfp, status))
903
904     log.debug("\n----- BEGIN REQUEST BODY -----\n" + request.body +
905             "----- END REQUEST BODY -----\n\n")
906
907     hashkey = request.matchdict['hashkey']
908
909     # the incoming hashkey is double quoted, and WSGI unquotes once...
910     hashkey = unquote(hashkey)
911
912     try:
913         player = DBSession.query(Player).\
914                 filter(Player.player_id == Hashkey.player_id).\
915                 filter(Player.active_ind == True).\
916                 filter(Hashkey.hashkey == hashkey).one()
917
918         elos = get_elos(player.player_id)
919
920     except Exception as e:
921         log.debug(e)
922         raise pyramid.httpexceptions.HTTPNotFound
923
924     return {
925         'hashkey':hashkey,
926         'player':player,
927         'elos':elos,
928     }
929
930
931 def player_elo_info_json(request):
932     """
933     Provides elo information on a specific player. JSON.
934     """
935     elo_info = player_elo_info_data(request)
936
937     player = player_info['player'].to_dict()
938
939     elos = {}
940     for gt, elo in elo_info['elos'].items():
941         elos[gt] = to_json(elo.to_dict())
942
943     return [{
944         'version':          1,
945         'player':           player,
946         'elos':             elos,
947     }]
948
949
950 def player_elo_info_text(request):
951     """
952     Provides elo information on a specific player. Plain text.
953     """
954     # UTC epoch
955     now = timegm(datetime.datetime.utcnow().timetuple())
956
957     # All player_info fields are converted into JSON-formattable dictionaries
958     elo_info = player_elo_info_data(request)
959
960     # this is a plain text response, if we don't do this here then
961     # Pyramid will assume html
962     request.response.content_type = 'text/plain'
963
964     return {
965         'version':          1,
966         'now':              now,
967         'hashkey':          elo_info['hashkey'],
968         'player':           elo_info['player'],
969         'elos':             elo_info['elos'],
970     }
971
972
973 def player_captimes_data(request):
974     player_id = int(request.matchdict['player_id'])
975     if player_id <= 2:
976         player_id = -1;
977
978     current_page = request.params.get("page", 1)
979
980     try:
981         player = DBSession.query(Player).filter_by(player_id=player_id).one()
982
983         pct_q = DBSession.query(PlayerCaptime.fastest_cap, PlayerCaptime.create_dt,
984                 PlayerCaptime.player_id, PlayerCaptime.game_id, PlayerCaptime.map_id,
985                 Map.name.label('map_name'), Game.server_id, Server.name.label('server_name')).\
986                 filter(PlayerCaptime.player_id==player_id).\
987                 filter(PlayerCaptime.game_id==Game.game_id).\
988                 filter(PlayerCaptime.map_id==Map.map_id).\
989                 filter(Game.server_id==Server.server_id).\
990                 order_by(expr.desc(PlayerCaptime.create_dt))
991
992     except Exception as e:
993         raise pyramid.httpexceptions.HTTPNotFound
994
995     captimes = Page(pct_q, current_page, items_per_page=20, url=page_url)
996
997     # replace the items in the canned pagination class with more rich ones
998     captimes.items = [PlayerCapTime(row) for row in captimes.items]
999
1000     return {
1001             "player_id" : player_id,
1002             "player"    : player,
1003             "captimes"  : captimes,
1004         }
1005
1006
1007 def player_captimes(request):
1008     return player_captimes_data(request)
1009
1010
1011 def player_captimes_json(request):
1012     data = player_captimes_data(request)
1013     page = request.params.get("page", 1)
1014
1015     # perform any necessary JSON conversions
1016     player_id = data["player_id"]
1017     player = data["player"].to_dict()
1018     captimes = [ct.to_dict() for ct in data["captimes"].items]
1019
1020     return {
1021             "player"    : player,
1022             "captimes"  : captimes,
1023             "page"      : page,
1024             }
1025
1026
1027 def player_weaponstats_data_json(request):
1028     player_id = int(request.matchdict["id"])
1029     if player_id <= 2:
1030         player_id = -1;
1031
1032     game_type_cd = request.params.get("game_type", None)
1033     if game_type_cd == "overall":
1034         game_type_cd = None
1035
1036     limit = 20
1037     if request.params.has_key("limit"):
1038         limit = int(request.params["limit"])
1039
1040         if limit < 0:
1041             limit = 20
1042         if limit > 50:
1043             limit = 50
1044
1045
1046     # the game_ids of the most recently played ones 
1047     # of the given game type is used for a subquery
1048     games_list = DBSession.query(Game.game_id).\
1049             filter(Game.players.contains([player_id]))
1050
1051     if game_type_cd is not None:
1052         games_list = games_list.filter(Game.game_type_cd == game_type_cd)
1053
1054     games_list = games_list.order_by(Game.game_id.desc()).limit(limit)
1055
1056     weapon_stats_raw = DBSession.query(PlayerWeaponStat).\
1057         filter(PlayerWeaponStat.player_id == player_id).\
1058         filter(PlayerWeaponStat.game_id.in_(games_list)).\
1059         all()
1060
1061     games_to_weapons = {}
1062     weapons_used = {}
1063     sum_avgs = {}
1064     for ws in weapon_stats_raw:
1065         if ws.game_id not in games_to_weapons:
1066             games_to_weapons[ws.game_id] = [ws.weapon_cd]
1067         else:
1068             games_to_weapons[ws.game_id].append(ws.weapon_cd)
1069
1070         weapons_used[ws.weapon_cd] = weapons_used.get(ws.weapon_cd, 0) + 1
1071         sum_avgs[ws.weapon_cd] = sum_avgs.get(ws.weapon_cd, 0) + float(ws.hit)/float(ws.fired)
1072
1073     # Creating zero-valued weapon stat entries for games where a weapon was not
1074     # used in that game, but was used in another game for the set. This makes 
1075     # the charts look smoother
1076     for game_id in games_to_weapons.keys():
1077         for weapon_cd in set(weapons_used.keys()) - set(games_to_weapons[game_id]):
1078             weapon_stats_raw.append(PlayerWeaponStat(player_id=player_id,
1079                 game_id=game_id, weapon_cd=weapon_cd))
1080
1081     # averages for the weapons used in the range
1082     avgs = {}
1083     for w in weapons_used.keys():
1084         avgs[w] = round(sum_avgs[w]/float(weapons_used[w])*100, 2)
1085
1086     weapon_stats_raw = sorted(weapon_stats_raw, key = lambda x: x.game_id)
1087     games            = sorted(games_to_weapons.keys())
1088     weapon_stats     = [ws.to_dict() for ws in weapon_stats_raw]
1089
1090     return {
1091         "weapon_stats": weapon_stats,
1092         "weapons_used": weapons_used.keys(),
1093         "games": games,
1094         "averages": avgs,
1095     }
1096