From 089823d8843efddc6ac65ddf7030378e974f8bf3 Mon Sep 17 00:00:00 2001
From: nemunaire <nemunaire@nemunai.re>
Date: Wed, 14 Oct 2015 06:47:01 +0200
Subject: [PATCH] [rnd] New command to choice between cmd

---
 modules/rnd.py | 27 ++++++++++++++++++++++-----
 1 file changed, 22 insertions(+), 5 deletions(-)

diff --git a/modules/rnd.py b/modules/rnd.py
index 6ecf9eb..84c5693 100644
--- a/modules/rnd.py
+++ b/modules/rnd.py
@@ -1,17 +1,20 @@
-# coding=utf-8
-
 """Help to make choice"""
 
-import random
+# PYTHON STUFFS #######################################################
 
+import random
+import shlex
+
+from nemubot import context
 from nemubot.exception import IRCException
 from nemubot.hooks import hook
-
-nemubotversion = 3.4
+from nemubot.message import Command
 
 from more import Response
 
 
+# MODULE INTERFACE ####################################################
+
 @hook("cmd_hook", "choice")
 def cmd_choice(msg):
     if not len(msg.args):
@@ -20,3 +23,17 @@ def cmd_choice(msg):
     return Response(random.choice(msg.args),
                     channel=msg.channel,
                     nick=msg.nick)
+
+
+@hook("cmd_hook", "choicecmd")
+def cmd_choice(msg):
+    if not len(msg.args):
+        raise IRCException("indicate some command to pick!")
+
+    choice = shlex.split(random.choice(msg.args))
+
+    return [x for x in context.subtreat(Command(choice[0][1:],
+                                                choice[1:],
+                                                to_response=msg.to_response,
+                                                frm=msg.frm,
+                                                server=msg.server))]