stats: hsave: don't choke on 'DClasses' tag

ec429 · Jul 18, 2014 · bee9dcb · bee9dcb
1 parent 41cfe2b
commit bee9dcb
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/stats/hsave.py b/stats/hsave.py
@@ -29,6 +29,7 @@ def __init__(self): pass
 	def parse(cls, f, read_init=True, check_integrity=False):
 		self = cls()
 		self.read_init = read_init
+		self.dclasses = None
 		self.difficulty = {}
 		stage = None
 		nosplit = False
@@ -118,6 +119,9 @@ def handle(self, tag, rest):
 		if tag == 'HARR':
 			self.ver = map(int, rest.split('.'))
 			return None, False
+		if tag == 'DClasses':
+			self.dclasses = int(rest)
+			return None, False
 		if tag == 'Difficulty':
 			dclass, level = map(int, rest.split(','))
 			self.difficulty[dclass] = level