Adds experimental map support.

Author: GraylinKim

sc2reader/objects.py

@@ -1,9 +1,14 @@
 from __future__ import absolute_import
 
+import urllib2
 import hashlib
 
+from cStringIO import StringIO
+
 from collections import namedtuple
 
+from mpyq import MPQArchive
+
 from sc2reader.constants import *
 from sc2reader.utils import PersonDict, AttributeDict
 
@@ -210,3 +215,28 @@ def format(self, format_string):
 
     def __repr__(self):
         return str(self)
+
+class Map(object):
+    url_template = 'http://{0}.depot.battle.net:1119/{1}.s2ma'
+
+    def __init__(self, gateway, map_hash, map_name=''):
+        self.hash = map_hash.encode('hex')
+        self.gateway = gateway
+        self.url = Map.url_template.format(self.gateway, self.hash)
+        self.name = map_name
+
+    def load(self):
+        print "Fetching map: {0}".format(self.url);
+        self.file = urllib2.urlopen(self.url).read()
+        print "Map Received"
+        self.archive = MPQArchive(StringIO(self.file))
+        self.minimap = self.archive.read_file('Minimap.tga')
+        self.game_strings = self.archive.read_file('enUS.SC2Data\LocalizedData\GameStrings.txt')
+        for line in self.game_strings.split('\r\n'):
+            parts = line.split('=')
+            if parts[0] == 'DocInfo/Name':
+                self.name = parts[1]
+            elif parts[0] == 'DocInfo/Author':
+                self.author = parts[1]
+            elif parts[0] == 'DocInfo/DescLong':
+                self.description = parts[1]

sc2reader/replay.py

@@ -5,7 +5,7 @@
 from datetime import datetime
 from collections import defaultdict
 from sc2reader.constants import REGIONS, LOCALIZED_RACES, GAME_SPEED_FACTOR
-from sc2reader.objects import Player, Observer, Team
+from sc2reader.objects import Player, Observer, Team, Map
 
 from sc2reader import utils
 
@@ -42,7 +42,7 @@ class Replay(object):
     is_private = bool()
 
     #: The name of the map the game was played on
-    map = str()
+    map = None
 
     #: The gateway the game was played on: us, eu, sea, etc
     gateway = str()
@@ -151,6 +151,7 @@ def load_details(self):
             initData = self.raw_data['replay.initData']
             if initData.map_data:
                 self.gateway = initData.map_data[0].gateway
+                self.map = Map(self.gateway, initData.map_data[-1].map_hash)
 
                 #Expand this special case mapping
                 if self.gateway == 'sg':
@@ -173,7 +174,8 @@ def load_details(self):
         if 'replay.details' in self.raw_data:
             details = self.raw_data['replay.details']
 
-            self.map = details.map
+            if self.map:
+                self.map.name = details.map
 
             self.windows_timestamp = details.file_time-details.utc_adjustment
             self.unix_timestamp = utils.windows_to_unix(self.windows_timestamp)
@@ -185,6 +187,8 @@ def load_details(self):
             self.start_time = datetime.utcfromtimestamp(self.unix_timestamp-self.real_length.seconds)
             self.date = self.end_time #backwards compatibility
 
+    def load_map(self):
+        self.map.load()
 
     def load_players(self):
         #If we don't at least have details and attributes_events we can go no further