Skip to content

Commit fb990a0

Browse files
committed
Allow listener registration to one or more events at once.
1 parent 0e309c4 commit fb990a0

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

sc2reader/__init__.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -126,8 +126,12 @@ def get_listeners(self, replay):
126126
return listeners
127127

128128

129-
def register_listener(self, event_class, listener, filterfunc=lambda r: True):
130-
self.registered_listeners[event_class].append((filterfunc, listener))
129+
def register_listener(self, events, listener, filterfunc=lambda r: True):
130+
try:
131+
for event in events:
132+
self.registered_listeners[event].append((filterfunc, listener))
133+
except TypeError:
134+
self.registered_listeners[event].append((filterfunc, listener))
131135

132136
def register_reader(self, data_file, reader, filterfunc=lambda r: True):
133137
self.registered_readers[data_file].insert(0,(filterfunc, reader))

0 commit comments

Comments
 (0)