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
|
import binascii, os, uuid
from src import ModuleManager, utils
STR_NOVOTE = "There is currently no vote running."
class Module(ModuleManager.BaseModule):
def _get_vote(self, channel):
return channel.get_setting("vote", None)
def _set_vote(self, channel, vote):
channel.set_setting("vote", vote)
def _del_vote(self, channel):
channel.del_setting("vote")
def _random_id(self):
return binascii.hexlify(os.urandom(4)).decode("ascii")
def _archive_vote(self, channel):
vote = self._get_vote(channel)
vote_id = self._random_id()
channel.set_setting("vote-%s" % vote_id, vote)
self._del_vote(channel)
return vote_id
def _cast_vote(self, channel, user, yes):
vote = self._get_vote(channel)
key = "yes" if yes else "no"
voters = vote["yes"]+vote["no"]
if user.name in voters:
return False
vote[key].append(user.name)
self._set_vote(channel, vote)
return True
@utils.hook("received.command.startvote", channel_only=True, min_args=1)
def start_vote(self, event):
"""
:help: Start a yes/no vote
:usage: <description>
:require_mode: o
:permission: vote
"""
current_vote = self._get_vote(event["target"])
if not current_vote == None:
raise utils.EventError("There's already a vote running")
self._set_vote(event["target"], {"description": event["args"],
"yes": [], "no": [], "electorate": []})
event["stdout"].write(
"Vote started. use '%svote yes' or '%svote no' to vote." % (
event["command_prefix"], event["command_prefix"]))
@utils.hook("received.command.endvote", channel_only=True)
def end_vote(self, event):
"""
:help: End the current yes/no vote
:require_mode: o
:permission: vote
"""
vote = self._get_vote(event["target"])
if vote == None:
event["stderr"].write(STR_NOVOTE)
else:
vote_id = self._archive_vote(event["target"])
event["stdout"].write("Vote %s ended: %s (%s yes, %s no)" %
(vote_id, vote["description"], len(vote["yes"]),
len(vote["no"])))
@utils.hook("received.command.vote", channel_only=True)
def vote(self, event):
"""
:help: Vote in the channel's current vote
:usage: yes|no
"""
vote = self._get_vote(event["target"])
if vote == None:
raise utils.EventError(STR_NOVOTE)
if not event["args"]:
event["stdout"].write("Current vote: %s (%s yes, %s no)" %
(vote["description"], len(vote["yes"]), len(vote["no"])))
else:
choice = event["args_split"][0].lower()
if not choice in ["yes", "no"]:
raise utils.EventError("Please vote 'yes' or 'no'")
if self._cast_vote(event["target"], event["user"], choice=="yes"):
event["stdout"].write("%s: your vote has been cast." %
event["user"].nickname)
else:
event["stderr"].write("%s: you have already voted." %
event["user"].nickname)
|
