Adds initial s2gs support via load_summary.

Author: GraylinKim

sc2reader/__init__.py

@@ -26,6 +26,8 @@
 load_replay = __defaultSC2Reader.load_replay
 load_maps = __defaultSC2Reader.load_maps
 load_map = __defaultSC2Reader.load_map
+load_summaries = __defaultSC2Reader.load_summaries
+load_summary = __defaultSC2Reader.load_summary
 
 configure = __defaultSC2Reader.configure
 reset = __defaultSC2Reader.reset

sc2reader/factories.py

@@ -14,7 +14,7 @@
 from sc2reader import exceptions
 from sc2reader import utils
 from sc2reader import log_utils
-from sc2reader.resources import Replay, Map
+from sc2reader.resources import Replay, Map, Summary
 
 class SC2Factory(object):
     """
@@ -175,6 +175,40 @@ def load_resource(self, resource, options=None, **new_options):
 
         return (resource, resource_name)
 
+    def load_summaries(self, gs, options=None, **new_options):
+        """
+        Loads a collection of replays. See load_resources for detailed parameter
+        documentation.
+
+        :rtype: generator(:class:`Map`)
+        """
+        for s in self.load_resources(gs, self.load_summary, options=options, **new_options):
+            yield s
+
+    def load_summary(self, summary_file, options=None, **new_options):
+        """
+        Loads the specified summary using the current factory settings with the
+        specified overrides.
+
+        :param summary_file: An open file object or path/url to a single file
+
+        :param None options: When options are passed directly into the options
+            parameter the current factory settings are ignored and only the
+            specified options are used during replay load.
+
+        :param new_options: Options values to override current factory settings
+            while loading this map.
+
+        :rtype: :class:`Replay`
+        """
+        options = options or utils.merged_dict(self.options, new_options)
+        resource, name = self.load_resource(summary_file, options=options)
+        s = Summary(resource, name, **options)
+
+        # Load summary procedure here!
+        #
+
+        return s
 
     def load_maps(self, maps, options=None, **new_options):
         """

sc2reader/resources.py

@@ -1,5 +1,6 @@
 from __future__ import absolute_import
 
+import zlib
 import hashlib
 from datetime import datetime
 from StringIO import StringIO
@@ -23,7 +24,6 @@ def __init__(self, file_object, filename=None, **options):
         self.filehash = hashlib.sha256(file_object.read()).hexdigest()
         file_object.seek(0)
 
-
 class Replay(Resource):
 
     #: Fully qualified filename of the replay file represented.
@@ -394,4 +394,18 @@ def read_game_strings(self):
             elif parts[0] == 'DocInfo/Author':
                 self.author = parts[1]
             elif parts[0] == 'DocInfo/DescLong':
-                self.description = parts[1]
+                self.description = parts[1]
+
+class Summary(Resource):
+    url_template = 'http://{0}.depot.battle.net:1119/{1}.s2ma'
+
+    def __init__(self, summary_file, filename=None, **options):
+        super(Summary, 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)