starting to parse interesting things out of the s2gs

Author: dsjoerg

sc2reader/resources.py

@@ -1,6 +1,7 @@
 from __future__ import absolute_import
 
 import zlib
+import pprint
 import hashlib
 from datetime import datetime
 from StringIO import StringIO
@@ -396,18 +397,30 @@ def read_game_strings(self):
             elif parts[0] == 'DocInfo/DescLong':
                 self.description = parts[1]
 
+s2gsmap = [[4, "Average Unspent Resources"],
+           [5, "Resource Collection Rate"],
+           [6, "Workers Created"],
+           [7, "Units Trained"],
+           [8, "Killed Unit Count"],
+           [9, "Structure Built"],
+          ]
+
 class GameSummary(Resource):
-    url_template = 'http://{0}.depot.battle.net:1119/{1}.s2ma'
+    url_template = 'http://{0}.depot.battle.net:1119/{1}.s2gs'
     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(buffer.read_data_struct())
-        print len(self.parts)
+#            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]
 
 class MatchInfo(Resource):
     url_template = 'http://{0}.depot.battle.net:1119/{1}.s2ma'
@@ -420,4 +433,4 @@ class MatchHistory(Resource):
     url_template = 'http://{0}.depot.battle.net:1119/{1}.s2ma'
     def __init__(self, history_file, filename=None, **options):
         super(MatchHistory, self).__init__(history_file, filename,**options)
-        self.data = utils.ReplayBuffer(history_file).read_data_struct()
+        self.data = utils.ReplayBuffer(history_file).read_data_struct()