Adds a few hashes to the team and replay objects.

These can be used in conjuction with the start/end_time attributes
to mark duplicate replay files and multiple replay versions of the
same game from different participants.

Author: GraylinKim

sc2reader/objects.py

@@ -1,5 +1,7 @@
 from __future__ import absolute_import
 
+import hashlib
+
 from collections import namedtuple
 
 from sc2reader.constants import *
@@ -25,6 +27,9 @@ class Team(object):
     :param interger number: The team number as recorded in the replay
     """
 
+    #: A unique hash identifying the team of players
+    hash = str()
+
     #: The team number as recorded in the replay
     number = int()
 
@@ -48,6 +53,10 @@ def __init__(self,number):
     def __iter__(self):
         return self.players.__iter__()
 
+    @property
+    def hash(self):
+        raw_hash = ','.join(sorted(p.url for p in self.players))
+        return hashlib.sha256(raw_hash).hexdigest()
 
 
 class Attribute(object):

sc2reader/replay.py

@@ -1,5 +1,7 @@
 from __future__ import absolute_import
 
+import hashlib
+
 from datetime import datetime
 from collections import defaultdict
 from sc2reader.constants import REGIONS, LOCALIZED_RACES, GAME_SPEED_FACTOR
@@ -97,6 +99,10 @@ def __init__(self, replay_file, **options):
         self.raw_data = dict()
         self.listeners = defaultdict(list)
 
+        replay_file.seek(0)
+        self.file_hash = hashlib.sha256(replay_file.read()).hexdigest()
+        replay_file.seek(0)
+
         self.filename = getattr(replay_file,'name', 'Unavailable')
         self.__dict__.update(utils.read_header(replay_file))
         self.archive = utils.open_archive(replay_file)
@@ -308,6 +314,9 @@ def load_players(self):
             else:
                 raise ValueError("Get Recorder algorithm is broken!")
 
+        player_names = sorted(map(lambda p: p.name, self.people))
+        hash_input = self.gateway+":"+','.join(player_names)
+        self.people_hash = hashlib.sha256(hash_input).hexdigest()
 
     def load_events(self, datapack=None):
         self.datapack = datapack