From a0a2365240cdf0c04cbf1540a085b298d0ed1368 Mon Sep 17 00:00:00 2001
From: Ant Zucaro <azucaro@gmail.com>
Date: Sun, 1 Sep 2013 15:05:03 -0400
Subject: [PATCH] Zero out missing values. This one works!

---
 xonstat/models.py       |  5 ++++-
 xonstat/views/player.py | 19 +++++++++++++------
 2 files changed, 17 insertions(+), 7 deletions(-)

diff --git a/xonstat/models.py b/xonstat/models.py
index 8ecb0d2..0d469e1 100644
--- a/xonstat/models.py
+++ b/xonstat/models.py
@@ -178,7 +178,10 @@ class PlayerAchievement(object):
 
 
 class PlayerWeaponStat(object):
-    def __init__(self):
+    def __init__(self, player_id=None, game_id=None, weapon_cd=None):
+        self.player_id = player_id
+        self.game_id = game_id
+        self.weapon_cd = weapon_cd
         self.fired = 0
         self.max = 0
         self.hit = 0
diff --git a/xonstat/views/player.py b/xonstat/views/player.py
index baea252..421eb7a 100644
--- a/xonstat/views/player.py
+++ b/xonstat/views/player.py
@@ -1063,7 +1063,7 @@ def player_damage_data_v2(request):
 
     games_raw = DBSession.query(sa.distinct(Game.game_id)).\
         filter(Game.game_id == PlayerWeaponStat.game_id).\
-        filter(PlayerWeaponStat.player_id == 6)
+        filter(PlayerWeaponStat.player_id == player_id)
 
     if game_type_cd is not None:
         games_raw = games_raw.filter(Game.game_type_cd == game_type_cd)
@@ -1071,11 +1071,9 @@ def player_damage_data_v2(request):
     games_raw = games_raw.order_by(Game.game_id.desc()).limit(limit).all()
 
     weapon_stats_raw = DBSession.query(PlayerWeaponStat).\
-        filter(PlayerWeaponStat.player_id == 6).\
+        filter(PlayerWeaponStat.player_id == player_id).\
         filter(PlayerWeaponStat.game_id.in_(games_raw)).all()
 
-    weapon_stats = [ws.to_dict() for ws in weapon_stats_raw]
-
     # NVD3 expects data points for all weapons used across the
     # set of games *for each* point on the x axis. This means populating
     # zero-valued weapon stat entries for games where a weapon was not
@@ -1091,7 +1089,16 @@ def player_damage_data_v2(request):
         if ws.weapon_cd not in weapons_used:
             weapons_used.append(ws.weapon_cd)
 
-    games = sorted(games_to_weapons.keys())
+    for game_id in games_to_weapons.keys():
+        for weapon_cd in set(weapons_used) - set(games_to_weapons[game_id]):
+            log.debug("Inserting zero value for game_id {0} weapon {1}".format(game_id, weapon_cd))
+            weapon_stats_raw.append(PlayerWeaponStat(player_id=player_id,
+                game_id=game_id, weapon_cd=weapon_cd))
+
+
+    weapon_stats_raw = sorted(weapon_stats_raw, key=lambda x: x.game_id)
+    games        = sorted(games_to_weapons.keys())
+    weapon_stats = [ws.to_dict() for ws in weapon_stats_raw]
 
     log.debug(games_to_weapons)
     log.debug(weapons_used)
@@ -1099,7 +1106,7 @@ def player_damage_data_v2(request):
 
     return {
         "weapon_stats": weapon_stats,
-        "weapons_used": weapon_stats,
+        "weapons_used": weapons_used,
         "games": games,
     }
 
-- 
2.39.2