Work on parsing GameSummary

Author: Prillan

sc2reader/objects.py

@@ -207,3 +207,31 @@ def format(self, format_string):
 
     def __repr__(self):
         return str(self)
+
+class PlayerSummary():
+    """
+    A class to represent a player in the game summary (.s2gs)
+    """
+    #: The index of the player in the game
+    pid = int()
+
+    #: The race the player used
+    race = str()
+
+    #: Battle.Net id of the player
+    bnetid = int()
+
+    #: Subregion id of player
+    subregion = int()
+
+    #: unknown1
+    unknown1 = int()
+    
+    #: unknown2
+    unknown2 = dict()
+
+    def __init__(self, pid):
+        self.pid = pid
+        
+    def __str__(self):
+        return '{} - {}/{}/'.format(self.race, self.subregion, self.bnetid)

sc2reader/resources.py

@@ -11,7 +11,7 @@
 
 from sc2reader import utils
 from sc2reader import log_utils
-from sc2reader.objects import Player, Observer, Team
+from sc2reader.objects import Player, Observer, Team, PlayerSummary
 from sc2reader.constants import REGIONS, LOCALIZED_RACES, GAME_SPEED_FACTOR
 
 
@@ -407,20 +407,42 @@ def read_game_strings(self):
 
 class GameSummary(Resource):
     url_template = 'http://{0}.depot.battle.net:1119/{1}.s2gs'
+
+    race_map = {
+        'Zerg' : 'Zerg',
+        'Terr' : 'Terran',
+        'Prot' : 'Protoss',
+        'RAND' : 'Random'
+        }
+
+
+    #: Players, a list of :class`PlayerSummary` from the game
+    players = list()
+
     def __init__(self, summary_file, filename=None, **options):
         super(GameSummary, self).__init__(summary_file, filename,**options)
         self.data = zlib.decompress(summary_file.read()[16:])
         self.parts = list()
         buffer = utils.ReplayBuffer(self.data)
         while buffer.left:
             part = buffer.read_data_struct()
-#            print str(part)+"\n\n\n"
             self.parts.append(part)
-#        print len(self.parts)
-#        pprint.PrettyPrinter(indent=2).pprint(self.parts)
-        for index, name in s2gsmap:
-            for player in [0, 1]:
-                print "Player", player, name, self.parts[3][0][index][1][player][0][0]
+
+        #Parse players, 16 is the maximum amount of players
+        for i in range(16):
+            player = None
+            # Check if player, break if not
+            if self.parts[0][3][i][2] == '\x00\x00\x00\x00':
+                break
+            player_struct = self.parts[0][3][i]
+
+            player = PlayerSummary(i)
+            player.race = self.race_map[player_struct[2]]
+            player.bnetid = player_struct[0][1][0][3]
+            player.subregion = player_struct[0][1][0][2]
+
+            self.players.append(player)
+        
 
 class MapInfo(Resource):
     url_template = 'http://{0}.depot.battle.net:1119/{1}.s2mi'