first draft of vote.py

1 files changed, 90 insertions, 0 deletions

diff --git a/modules/vote.py b/modules/vote.py
new file mode 100644
index 00000000..5713f889
--- /dev/null
+++ b/modules/vote.py
@@ -0,0 +1,90 @@
+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
+        """
+        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
+        """
+        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)