Changeset 12 for trunk/amazonbot

Timestamp:

06/20/06 18:41:29 (18 years ago)

Author:

atzm

Message:

command implemented

File:

• trunk/amazonbot/amazonbot.py (modified) (7 diffs)

trunk/amazonbot/amazonbot.py

@@ -11 +11 @@
 import sys
 import time
+import shlex
 import random
+import getopt
+
 import MeCab
 import nkf
@@ -75 +78 @@
 class AmazonBotBase(SingleServerIRCBot):
         """アマゟンボットのベヌスクラス
-        こい぀単䜓では受け取ったメッセヌゞの圢æ
+        単䜓では受け取ったメッセヌゞの圢æ
 ‹çŽ 解析ず名詞抜出たでしかやらない
         サブクラスで process_keyword を実è£
 しお Amazon ぞク゚リを投げるべし
+
+        サブクラスには onmsg_HOGEHOGE(self, conn, ev, args) ãƒ¡ã‚œãƒƒãƒ‰ã‚’äœœã‚‹ã“ãšã§ã‚³ãƒžãƒ³ãƒ‰è¿œåŠ å¯èƒœ
+        コマンド曞匏は !HOGEHOGE arg [, arg2, ...] ずなる
         """
         def __init__(self):
@@ -106 +112 @@
 
         def on_pubmsg(self, c, e):
+                msg = unicoding(e.arguments()[0])
+                if msg[0] == '!':
+                        words = shlex.split(ununicoding(msg, 'utf-8')[1:])
+                        method = getattr(self, 'onmsg_%s' % words[0], lambda *arg: False)
+                        return method(c, e, words[1:]) # words[0] == command name
+
                 _current_time = time.time()
                 if _current_time < self._prev_time + config.get('bot', 'freq', 'int'):
@@ -112 +124 @@
                                 print >> sys.stderr, 'DEBUG> Not expired: prev time is %s' % prev
                         return False
-
                 self._prev_time = _current_time
-                msg = unicoding(e.arguments()[0])
 
                 self.silence(msg, c, e)
@@ -132 +142 @@
                         try:
                                 message = ununicoding(': '.join([content, title, url]))
-                        except UnicodeError, e:
+                        except UnicodeError, err:
                                 # なぜかたたに unicode オブゞェクトを iso-2022-jp で゚ンコヌドできない
                                 if __debug__:
-                                        print >> sys.stderr, 'DEBUG> %s' % str(e)
+                                        print >> sys.stderr, 'DEBUG> %s' % str(err)
                                 return False
 
@@ -172 +182 @@
                 return 'AmazonBot by %s, based on python-irclib' % __author__
 
+        def onmsg_isbn(self, c, e, args):
+                """Syntax: !isbn <ISBN number>
+                """
+                return self.onmsg_asin(c, e, args)
+        def onmsg_asin(self, c, e, args):
+                """Syntax: !asin <ASIN number>
+                """
+                if __debug__:
+                        print >> sys.stderr, 'DEBUG> in asin command: %s' % str(args)
+
+                try:
+                        data = my_amazon.searchByASIN(args[0])
+                except my_amazon.AmazonError, err:
+                        ch = e.target()
+                        c.privmsg(ch, ununicoding(config.get('bot', 'no_products')))
+                        if __debug__:
+                                print >> sys.stderr, 'DEBUG> Caught AmazonError in onmsg_asin: %s' % str(err)
+                        return False
+
+                return self._process_onmsg(c, e, data)
+
+        def onmsg_a(self, c, e, args):
+                """Syntax: !a [-h] [-t type] keyword1 [, keyword2, ...]
+                """
+                return self.onmsg_amazon(c, e, args)
+        def onmsg_amazon(self, c, e, args):
+                """Syntax: !amazon [-h] [-t type] keyword1 [, keyword2, ...]
+                """
+                if __debug__:
+                        print >> sys.stderr, 'DEBUG> in amazon command: %s' % str(args)
+
+                try:
+                        options, rest = getopt.getopt(args, 't:h', ['type=', 'help'])
+                except getopt.GetoptError, err:
+                        if __debug__:
+                                print >> sys.stderr, 'DEBUG> Caught GetoptError in onmsg_amazon: %s' % str(err)
+                        return False
+
+                keyword = ' '.join(rest)
+                product_line = 'books-jp'
+                for opt, val in options:
+                        if opt in ['-t', '--type']:
+                                product_line = val
+                                break
+                        elif opt in ['-h', '--help']:
+                                available = [
+                                        'books-jp (和曞, default)', 'books-us (掋曞)',
+                                        'music-jp (ポピュラヌ音楜)', 'classical-jp (クラシック音楜)',
+                                        'dvd-jp (DVD)', 'vhs-jp (ビデオ)',
+                                        'electronics-jp (゚レクトロニクス)', 'kitchen-jp (ホヌムキッチン)',
+                                        'software-jp (゜フトりェア)', 'videogames-jp (ゲヌム)',
+                                        'magazines-jp (雑誌)', 'toys-jp (おもちゃホビヌ)',
+                                        ]
+                                ch = e.target()
+                                c.privmsg(ch, ununicoding('Available types: ' + ', '.join(available)))
+                                return True
+
+                if __debug__:
+                        fmt = 'DEBUG> keyword="%s", product_line=%s'
+                        print >> sys.stderr, fmt % (ununicoding(keyword, 'euc-jp'), product_line)
+
+                try:
+                        data = my_amazon.searchByKeyword(keyword, product_line=product_line)
+                except my_amazon.AmazonError, err:
+                        ch = e.target()
+                        c.privmsg(ch, ununicoding(config.get('bot', 'no_products')))
+                        if __debug__:
+                                print >> sys.stderr, 'DEBUG> Caught AmazonError in onmsg_amazon: %s' % str(err)
+                        return False
+
+                return self._process_onmsg(c, e, data)
+
+        def onmsg_help(self, c, e, data):
+                pass
+
+        def _process_onmsg(self, c, e, data):
+                if type(data.Details) is not list:
+                        data.Details = [data.Details]
+
+                detail = random.choice(data.Details)
+                title = ununicoding(detail.ProductName)
+                url = ununicoding(detail.URL)
+                channel = config.get('irc', 'channel')
+                c.privmsg(channel, '%(title)s: %(url)s' % locals())
+
+                return True
+
         def process_keyword(self, keyword):
                 keyword = ununicoding(keyword, 'utf-8')
@@ -181 +278 @@
                         if type(data.ProductLine) is not type([]):
                                 data.ProductLine = [data.ProductLine]
-                except my_amazon.AmazonError, e:
-                        if __debug__:
-                                print >> sys.stderr, 'DEBUG> Caught AmazonError: %s' % str(e)
+                except my_amazon.AmazonError, err:
+                        if __debug__:
+                                print >> sys.stderr, 'DEBUG> Caught AmazonError: %s' % str(err)
                         return [None, None]