Fix gameheart plugin handling of teams and observers. Closes #174.

Author: GraylinKim

sc2reader/engine/plugins/gameheart.py

@@ -69,6 +69,9 @@ def fix_events(self, replay, start_frame):
     def fix_entities(self, replay, actual_players):
         # Change the players that aren't playing into observers
         for p in [p for p in replay.players if p.pid not in actual_players]:
+            # Fix the slot data to be accurate
+            p.slot_data['observe'] = 1
+            p.slot_data['team_id'] = None
             obs = Observer(p.sid, p.slot_data, p.uid, p.init_data, p.pid)
 
             # Because these obs start the game as players the client

sc2reader/objects.py

@@ -101,7 +101,9 @@ def __init__(self, sid, slot_data):
         self.handicap = slot_data['handicap']
 
         #: The entity's team number. None for observers
-        self.team_id = slot_data['team_id']+1
+        self.team_id = None
+        if slot_data['team_id'] is not None:
+            self.team_id = slot_data['team_id'] + 1
 
         #: A flag indicating if the person is a human or computer
         #: Really just a shortcut for isinstance(entity, User)