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
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
|
#--depends-on commands
#--depends-on config
#--depends-on permissions
import re, time
from src import EventManager, ModuleManager, utils
WORD_STOP = [",", ":"]
KARMA_DELAY_SECONDS = 3
REGEX_KARMA = re.compile(r"^(.*)(\+{2}|\-{2})$")
@utils.export("channelset", {"setting": "karma-verbose",
"help": "Enable/disable automatically responding to karma changes",
"validate": utils.bool_or_none, "example": "on"})
@utils.export("serverset", {"setting": "karma-nickname-only",
"help": "Enable/disable karma being for nicknames only",
"validate": utils.bool_or_none, "example": "on"})
class Module(ModuleManager.BaseModule):
def _karma_str(self, karma):
karma_str = str(karma)
if karma < 0:
return utils.irc.color(str(karma), utils.consts.RED)
elif karma > 0:
return utils.irc.color(str(karma), utils.consts.LIGHTGREEN)
return str(karma)
@utils.hook("new.user")
def new_user(self, event):
event["user"].last_karma = None
@utils.hook("command.regex")
@utils.kwarg("command", "karma")
@utils.kwarg("pattern", REGEX_KARMA)
def channel_message(self, event):
verbose = event["target"].get_setting("karma-verbose", False)
nickname_only = event["server"].get_setting("karma-nickname-only",
False)
if not event["user"].last_karma or (time.time()-event["user"
].last_karma) >= KARMA_DELAY_SECONDS:
target = event["match"].group(1).strip().rstrip("".join(WORD_STOP))
if not target:
return
if event["server"].irc_lower(target) == event["user"].name:
if verbose:
event["stderr"].write("You cannot change your own karma")
return
setting = "karma-%s" % target
setting_target = event["server"]
if nickname_only:
user = event["server"].get_user(target)
setting = "karma"
setting_target = user
if not event["target"].has_user(user):
return
positive = event["match"].group(2)[0] == "+"
karma = setting_target.get_setting(setting, 0)
karma += 1 if positive else -1
if not karma == 0:
setting_target.set_setting(setting, karma)
else:
setting_target.del_setting(setting)
karma_str = self._karma_str(karma)
if verbose:
event["stdout"].write(
"%s now has %s karma" % (target, karma_str))
event["user"].last_karma = time.time()
elif verbose:
event["stderr"].write("Try again in a couple of seconds")
@utils.hook("received.command.karma")
def karma(self, event):
"""
:help: Get your or someone else's karma
:usage: [target]
"""
if event["args"]:
target = event["args"]
else:
target = event["user"].nickname
target = target.strip()
if event["server"].get_setting("karma-nickname-only", False):
karma = event["server"].get_user(target).get_setting("karma", 0)
else:
karma = event["server"].get_setting("karma-%s" % target, 0)
karma_str = self._karma_str(karma)
event["stdout"].write("%s has %s karma" % (target, karma_str))
@utils.hook("received.command.resetkarma", min_args=1)
def reset_karma(self, event):
"""
:help: Reset a specified karma to 0
:usage: <target>
:permission: resetkarme
"""
setting = "karma-%s" % event["args_split"][0]
karma = event["server"].get_setting(setting, 0)
if karma == 0:
event["stderr"].write("%s already has 0 karma" % event[
"args_split"][0])
else:
event["server"].del_setting(setting)
event["stdout"].write("Reset karma for %s" % event[
"args_split"][0])
|
