Changeset 10

Timestamp:

06/20/06 14:03:50 (18 years ago)

Author:

atzm

Message:

freq

Location:

trunk/amazonbot

Files:

• amazonbot.ini.sample (modified) (1 diff)
• amazonbot.py (modified) (9 diffs)

trunk/amazonbot/amazonbot.ini.sample

@@ -9 +9 @@
 
 [bot]
+freq = 120
 nick = amazonbot
 content = そんなアナタにこれがオススメ
 thanks = ありがずう
 sorry = 正盎すたんかった
+bye = さようならたた䌚う日たで
 active_pattern = ごめん|ゎメン|すたん|スマン|すいたせん
 silent_pattern = 邪魔|じゃた|ゞャマ|うる(さい|せヌ)|だたれ

trunk/amazonbot/amazonbot.py

@@ -10 +10 @@
 import re
 import sys
+import time
 import random
 import MeCab
+import nkf
 
 from ircbot import SingleServerIRCBot
@@ -29 +31 @@
 
 def uniq(sequence):
+        """リストから重耇を取り陀く (é †ç•ªãŒç‹‚ã†ã®ã§æ³šæ„)
+        """
         return list(set(sequence))
 
+def unicoding(text):
+        """text を匷制的に unicode オブゞェクトに倉換
+        """
+        if type(text) is unicode:
+                return text
+        return unicode(nkf.nkf('-w', text), 'utf-8')
+
+def ununicoding(text, encoding='iso-2022-jp'):
+        """text を指定された encoding で゚ンコヌドしraw str に匷制倉換
+        """
+        if type(text) is not unicode:
+                return unicoding(text).encode(encoding)
+        return text.encode(encoding)
+
 def mecab_parse(text):
+        """MeCab を䜿っお圢æ
+‹çŽ 解析し固有名詞ず䞀般名詞だけを抜出する
+        """
         def choice_nominal(wlist):
                 res = []
@@ -37 +58 @@
                         wtypes = wtype.split('-')
                         if '固有名詞' in wtypes or ('名詞' in wtypes and '䞀般' in wtypes):
-                                res.append(word)
+                                res.append(unicoding(word))
                 return res
 
+        text = ununicoding(text, 'utf-8')
         result = []
         tag = MeCab.Tagger('-Ochasen')
@@ -52 +74 @@
 
 class AmazonBotBase(SingleServerIRCBot):
+        """アマゟンボットのベヌスクラス
+        こい぀単䜓では受け取ったメッセヌゞの圢æ
+‹çŽ 解析ず名詞抜出たでしかやらない
+        サブクラスで process_keyword を実è£
+しお Amazon ぞク゚リを投げるべし
+        """
         def __init__(self):
                 _server = [(config.get('irc', 'server'), config.get('irc', 'port', 'int'))]
                 _nick = config.get('bot', 'nick')
 
+                self._prev_time = time.time()
                 self._silent = False
                 SingleServerIRCBot.__init__(self, _server, _nick, _nick)
 
-        def get_version(self):
-                return 'AmazonBot by %s, based on python-irclib' % __author__
+        def start(self):
+                try:
+                        SingleServerIRCBot.start(self)
+                except KeyboardInterrupt:
+                        self.die(ununicoding(config.get('bot', 'bye')))
 
         def on_welcome(self, c, e):
                 c.join(config.get('irc', 'channel'))
                 if __debug__:
-                        print >> sys.stderr, '* Joined %s' % config.get('irc', 'channel')
+                        print >> sys.stderr, 'DEBUG> Joined %s' % config.get('irc', 'channel')
 
         def on_nicknameinuse(self, c, e):
@@ -74 +106 @@
 
         def on_pubmsg(self, c, e):
-                try:
-                        msg = unicode(e.arguments()[0], 'iso-2022-jp')
-                except UnicodeError:
+                if time.time() > self._prev_time + config.get('bot', 'freq', 'int'):
                         if __debug__:
-                                print >> sys.stderr, 'ERR: incoming message encoding failed: %s' % e.arguments()
+                                prev = time.strftime('%y/%m/%d %H:%M:%S', time.localtime(self._prev_time))
+                                print >> sys.stderr, 'DEBUG> Not expired: prev time is %s' % prev
                         return False
+
+                msg = unicoding(e.arguments()[0])
 
                 self.silence(msg, c, e)
@@ -85 +118 @@
                         return False
 
-                nominals = mecab_parse(msg.encode('utf-8'))
+                nominals = mecab_parse(msg)
                 if not nominals:
+                        if __debug__:
+                                print >> sys.stderr, "DEBUG> Couldn't find nominal words"
                         return False
 
@@ -92 +127 @@
                 if title and url:
                         channel = e.target()
-
-                        content = unicode(config.get('bot', 'content'), 'utf-8')
-                        title = title
-
+                        content = unicoding(config.get('bot', 'content'))
                         try:
-                                message = ('%(content)s: %(title)s %(url)s' % locals()).encode('iso-2022-jp')
+                                message = ununicoding(': '.join([content, title, url]))
                         except UnicodeError:
-                                if __debug__:
-                                        print >> sys.stderr, 'ERR: my message encoding failed: '
-                                        print >> sys.stderr, ' * content> %s' % content
-                                        print >> sys.stderr, ' * title> %s' % title
-                                        print >> sys.stderr, ' * url> %s' % url
                                 return False
+                                # なぜかたたに unicode オブゞェクトを iso-2022-jp で゚ンコヌドできない
 
                         c.privmsg(channel, message)
-
                         return True
-
                 return False
 
-        ACTIVE_PATTERN = re.compile(config.get('bot', 'active_pattern'))
-        SILENT_PATTERN = re.compile(config.get('bot', 'silent_pattern'))
+        ACTIVE_PATTERN = re.compile(unicoding(config.get('bot', 'active_pattern')))
+        SILENT_PATTERN = re.compile(unicoding(config.get('bot', 'silent_pattern')))
         def silence(self, msg, c, e):
                 ch = e.target()
-                active = self.ACTIVE_PATTERN.search(msg.encode('utf-8'))
-                silent = self.SILENT_PATTERN.search(msg.encode('utf-8'))
+                active = self.ACTIVE_PATTERN.search(msg)
+                silent = self.SILENT_PATTERN.search(msg)
                 if __debug__:
-                        print >> sys.stderr, 'ACTIVE_PATTERN: %s, SILENT_PATTERN: %s' % (str(active), str(silent))
+                        print >> sys.stderr, 'DEBUG> ACT_PATT: %s, SIL_PATT: %s' % (str(active), str(silent))
 
                 if active:
                         self._silent = False
-                        c.privmsg(ch, unicode(config.get('bot', 'thanks'), 'utf-8').encode('iso-2022-jp'))
+                        c.privmsg(ch, ununicoding(config.get('bot', 'thanks')))
                 elif silent:
                         self._silent = True
-                        c.privmsg(ch, unicode(config.get('bot', 'sorry'), 'utf-8').encode('iso-2022-jp'))
+                        c.privmsg(ch, ununicoding(config.get('bot', 'sorry')))
 
         def process_keyword(self, keyword):
@@ -132 +158 @@
 
 class AmazonBot(AmazonBotBase):
+        """アマゟンボットの実è£
+クラス
+        process_keyword メ゜ッドで Amazon ぞク゚リを投げお結果を返す
+        """
         def __init__(self):
                 AmazonBotBase.__init__(self)
 
+        def get_version(self):
+                return 'AmazonBot by %s, based on python-irclib' % __author__
+
         def process_keyword(self, keyword):
+                keyword = ununicoding(keyword, 'utf-8')
+                if __debug__:
+                        print >> sys.stderr, 'DEBUG> KEYWORD: %s' % ununicoding(keyword, 'euc-jp')
+
                 try:
                         data = my_amazon.searchByBlended(keyword)
                         if type(data.ProductLine) is not type([]):
                                 data.ProductLine = [data.ProductLine]
-                except my_amazon.AmazonError:
+                except my_amazon.AmazonError, e:
+                        if __debug__:
+                                print >> sys.stderr, 'DEBUG> Caught AmazonError: %s' % str(e)
                         return [None, None]
 
@@ -146 +185 @@
                 detail = random.choice(product_line.ProductInfo.Details)
 
-                url = getattr(detail, 'URL', None)
-                product_name = getattr(detail, 'ProductName', None)
+                url = unicoding(getattr(detail, 'URL', None))
+                product_name = unicoding(getattr(detail, 'ProductName', None))
 
                 return [product_name, url]