Do not use index() in 'join' and 'part' commands
authorStefan Hajnoczi <stefanha@gmail.com>
Sun, 28 Jun 2009 07:53:24 +0000 (08:53 +0100)
committerStefan Hajnoczi <stefanha@gmail.com>
Sun, 28 Jun 2009 07:53:24 +0000 (08:53 +0100)
cmds.py

diff --git a/cmds.py b/cmds.py
index cf83eb6..0f77a96 100644 (file)
--- a/cmds.py
+++ b/cmds.py
@@ -159,18 +159,18 @@ op_commands['log'] = log_command
 
 def join_command(who, target, replyto, words):
     '''Join a channel (join #etherboot)'''
 
 def join_command(who, target, replyto, words):
     '''Join a channel (join #etherboot)'''
-    index = words.index('join') + 1
-    if index < len(words):
-        channel = words[index]
-        return 'CMD', ('JOIN %s' % (channel.startswith('#') and channel or '#' + channel),)
+    if len(words) != 2:
+        return
+    channel = words[1]
+    return 'CMD', ('JOIN %s' % (channel.startswith('#') and channel or '#' + channel),)
 op_commands['join'] = join_command
 
 def part_command(who, target, replyto, words):
     '''Leave a channel (part #etherboot)'''
 op_commands['join'] = join_command
 
 def part_command(who, target, replyto, words):
     '''Leave a channel (part #etherboot)'''
-    index = words.index('part') + 1
-    if index < len(words):
-        channel = words[index]
-        return 'CMD', ('PART %s' % (channel.startswith('#') and channel or '#' + channel),)
+    if len(words) != 2:
+        return
+    channel = words[1]
+    return 'CMD', ('PART %s' % (channel.startswith('#') and channel or '#' + channel),)
 op_commands['part'] = part_command
 
 def privmsg_command(who, target, replyto, words):
 op_commands['part'] = part_command
 
 def privmsg_command(who, target, replyto, words):