Adds support for replay.is_ladder and replay.is_private boolean flags and updates tests

Author: GraylinKim

sc2reader/parsers.py

@@ -98,7 +98,12 @@ def load(self, replay, filecontents):
 
         #Get global settings first
         replay.speed = data[16]['Game Speed']
+        
+        #TODO: Do we need the category variable at this point?
         replay.category = data[16]['Category']
+        replay.is_ladder = (replay.category == "Ladder")
+        replay.is_private = (replay.category == "Private")
+        
         replay.type = data[16]['Game Type']
 
         #Set player attributes as available,  requires already populated player list

sc2reader/replay.py

@@ -15,6 +15,8 @@ def __init__(self, replay, partial_parse=True, full_parse=True):
         self.build = ""
         self.type = ""
         self.category = ""
+        self.is_ladder = False
+        self.is_private = False
         self.map = ""
         self.realm = ""
         self.events = list()

test_replays/test_all.py

@@ -40,8 +40,9 @@ def test_1():
     assert replay.release_string == "1.2.2.17811"
     assert replay.speed == "Faster"
     assert replay.type == "1v1"
-    # TODO: library should probably provide helper functions and/or constants to check this
-    assert replay.category == "Ladder"
+
+    assert replay.is_ladder == True
+    assert replay.is_private == False
 
     assert len(replay.players) == 2
     assert replay.player[1].name == "Emperor"
@@ -100,8 +101,8 @@ def test_1():
 def test_private_category():
     replay = Replay("test_replays/build17811/2.sc2replay")
 
-    # TODO: library should probably provide helper functions and/or constants to check this
-    assert replay.category == "Private"
+    assert replay.is_private == True
+    assert replay.is_ladder == False
 
 def test_3v3():
     replay = Replay("test_replays/build17811/3.sc2replay")