add probability field to rules

README.md

@@ -49,6 +49,8 @@ Rules have five fields. Only `matches` and `template` are required.
 * `rooms` - The list of rooms where the rule should apply.
             If empty, the rule will apply to all rooms the bot is in.
 * `matches` - The regex or list of regexes to match.
+* `probability` - The probability the rule will be executed expressed as a float
+                  between 0 and 1.0. Defaults to 1.0.
 * `template` - The name of the template to use.
 * `variables` - A key-value map of variables to extend or override template variables.
                 Like with template variables, the values are parsed as Jinja2 templates.

reactbot/config.py

@@ -59,6 +59,7 @@ def _make_rule(self, name: str, rule: Dict[str, Any]) -> Rule:
                         not_rooms=set(rule.get("not_rooms", [])),
                         matches=self._compile_all(rule["matches"]),
                         not_matches=self._compile_all(rule.get("not_matches", [])),
+                        probability=rule.get("probability", 1.0),
                         type=EventType.find(rule["type"]) if "type" in rule else None,
                         template=self.templates[rule["template"]],
                         variables=self._parse_variables(rule))

reactbot/rule.py

@@ -18,6 +18,8 @@
 from attr import dataclass
 from jinja2 import Template as JinjaTemplate
 
+from random import random
+
 from mautrix.types import RoomID, EventType
 
 from maubot import MessageEvent
@@ -34,6 +36,7 @@ class Rule:
     not_rooms: Set[RoomID]
     matches: List[RPattern]
     not_matches: List[RPattern]
+    probability: float
     template: Template
     type: Optional[EventType]
     variables: Dict[str, Any]
@@ -49,6 +52,8 @@ def match(self, evt: MessageEvent) -> Optional[Match]:
             return None
         elif evt.room_id in self.not_rooms:
             return None
+        elif self.probability < random():
+            return None
         for pattern in self.matches:
             match = pattern.search(evt.content.body)
             if match: