add, remove now accept iterables

update README

patch bump
This commit is contained in:
onyx-and-iris
2022-10-06 18:07:41 +01:00
parent 2f82e0b1fc
commit 550df917fb
3 changed files with 26 additions and 7 deletions

View File

@@ -1,4 +1,5 @@
import logging
from typing import Iterable, Union
class Event:
@@ -41,8 +42,14 @@ class Event:
def any(self) -> bool:
return any(self.subs.values())
def add(self, event):
setattr(self, event, True)
def add(self, events: Union[str, Iterable[str]]):
if isinstance(events, str):
events = [events]
for event in events:
setattr(self, event, True)
def remove(self, event):
setattr(self, event, False)
def remove(self, events: Union[str, Iterable[str]]):
if isinstance(events, str):
events = [events]
for event in events:
setattr(self, event, False)