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