vbatts/maubot_reactbot
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Add option to exclude specific rooms from a rule

Browse source
This commit is contained in:
Tulir Asokan 2019-07-31 19:28:15 +03:00
parent 50141c8a92
commit 217351d141
2 changed files with 4 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
reactbot/config.py
View file

@ -53,6 +53,7 @@ class Config(BaseProxyConfig):
def _make_rule(self, name: str, rule: Dict[str, Any]) -> Rule: def _make_rule(self, name: str, rule: Dict[str, Any]) -> Rule:
try: try:
return Rule(rooms=set(rule.get("rooms", [])), return Rule(rooms=set(rule.get("rooms", [])),
not_rooms=set(rule.get("not_rooms", [])),
matches=self._compile_all(rule["matches"]), matches=self._compile_all(rule["matches"]),
not_matches=self._compile_all(rule.get("not_matches", [])), not_matches=self._compile_all(rule.get("not_matches", [])),
type=EventType.find(rule["type"]) if "type" in rule else None, type=EventType.find(rule["type"]) if "type" in rule else None,

3
reactbot/rule.py
View file

@ -31,6 +31,7 @@ RPattern = Union[Pattern, SimplePattern]
@dataclass @dataclass
class Rule: class Rule:
rooms: Set[RoomID] rooms: Set[RoomID]
not_rooms: Set[RoomID]
matches: List[RPattern] matches: List[RPattern]
not_matches: List[RPattern] not_matches: List[RPattern]
template: Template template: Template
@ -46,6 +47,8 @@ class Rule:
def match(self, evt: MessageEvent) -> Optional[Match]: def match(self, evt: MessageEvent) -> Optional[Match]:
if len(self.rooms) > 0 and evt.room_id not in self.rooms: if len(self.rooms) > 0 and evt.room_id not in self.rooms:
return None return None
elif evt.room_id in self.not_rooms:
return None
for pattern in self.matches: for pattern in self.matches:
match = pattern.search(evt.content.body) match = pattern.search(evt.content.body)
if match: if match: