Skip to content

Commit 5b0d5fa

Browse files
committed
Makes OrderedDict import from ordereddict if it isn't present in the collections module. Also adds a new configuration option for parsing a replay while skiping the events information.
1 parent 3f1232a commit 5b0d5fa

File tree

1 file changed

+21
-1
lines changed

1 file changed

+21
-1
lines changed

sc2reader/config.py

Lines changed: 21 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,7 @@
1-
from ordereddict import OrderedDict
1+
try:
2+
from collections import OrderedDict
3+
except ImportError:
4+
from ordereddict import OrderedDict
25

36
from sc2reader.objects import Replay
47
from sc2reader.processors import *
@@ -25,3 +28,20 @@ def __init__(self):
2528
ApmProcessor,
2629
ResultsProcessor
2730
]
31+
32+
class NoEventsConfig(object):
33+
def __init__(self):
34+
self.ReplayClass = Replay
35+
self.readers = OrderedDict()
36+
self.readers['replay.initData'] = [ ReplayInitDataReader() ]
37+
self.readers['replay.details'] = [ ReplayDetailsReader() ]
38+
self.readers['replay.attributes.events'] = [ AttributeEventsReader_17326(), AttributeEventsReader() ]
39+
self.readers['replay.message.events'] = [ MessageEventsReader() ]
40+
41+
self.processors = [
42+
PeopleProcessor,
43+
AttributeProcessor,
44+
TeamsProcessor,
45+
MessageProcessor,
46+
RecorderProcessor,
47+
]

0 commit comments

Comments
 (0)