1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
|
#--depends-on commands
#--depends-on permissions
import time
from src import EventManager, ModuleManager, utils
SILENCE_TIME = 60*5 # 5 minutes
class Module(ModuleManager.BaseModule):
def on_load(self):
self.exports.add("is-silenced", self._is_silenced)
def _is_silenced(self, target):
silence_until = target.get_setting("silence-until", None)
if not silence_until == None:
if time.time()<silence_until:
return True
else:
target.del_setting("silence-until")
return False
@utils.hook("received.command.silence")
@utils.kwarg("channel_only", True)
@utils.kwarg("help", "Prevent me saying anything for a period of time "
"(default: 5 minutes)")
@utils.kwarg("usage", "[+time]")
@utils.kwarg("require_mode", "high")
@utils.kwarg("require_access", "low,silence")
@utils.kwarg("permission", "silence")
def silence(self, event):
duration = SILENCE_TIME
if event["args"] and event["args_split"][0].startswith("+"):
duration = utils.datetime.parse.from_pretty_time(
event["args_split"][0][1:])
if duration == None:
raise utils.EventError("Invalid duration provided")
silence_until = time.time()+duration
event["target"].set_setting("silence-until", silence_until)
event["stdout"].write("Ok, I'll be back")
@utils.hook("received.command.unsilence")
@utils.kwarg("help", "Unsilence me")
@utils.kwarg("unsilence", True)
@utils.kwarg("channel_only", True)
@utils.kwarg("require_mode", "high")
@utils.kwarg("require_access", "low,unsilence")
@utils.kwarg("permission", "unsilence")
def unsiltence(self, event):
silence_until = event["target"].get_setting("silence-until", None)
if not silence_until == None:
event["target"].del_setting("silence-until")
event["stdout"].write("Ok. I've been unsilenced")
else:
event["stderr"].write("I am not silenced")
@utils.hook("preprocess.command", priority=EventManager.PRIORITY_HIGH)
def preprocess_command(self, event):
if event["is_channel"] and not event["hook"].get_kwarg(
"unsilence", False):
silence_until = event["target"].get_setting("silence-until", None)
if silence_until:
if self._is_silenced(event["target"]):
return utils.consts.PERMISSION_HARD_FAIL, None
@utils.hook("unknown.command")
@utils.kwarg("priority", EventManager.PRIORITY_HIGH)
def unknown_command(self, event):
if event["is_channel"] and self._is_silenced(event["target"]):
event.eat()
|
