Changeset 22 for trunk/amazonbot

Timestamp:

02/09/07 14:19:24 (18 years ago)

Author:

atzm

Message:

• Add action and localtime command

Location:

trunk/amazonbot

Files:

• amazonbot.ini.sample (modified) (1 diff)
• amazonbot.py (modified) (5 diffs)
• config.py (modified) (2 diffs)

trunk/amazonbot/amazonbot.ini.sample

@@ -1 +1 @@
 [amazon]
-locale = jp
+locale     = jp
 access_key = your access key
 
 [irc]
-server = irc.freenode.net
-port = 6667
+server  = irc.freenode.net
+port    = 6667
 channel = #amazonbot
 
 [bot]
-nick = amazonbot
-content = そんなアナタにこれがオススメ
-thanks = ありがずう
-sorry = 正盎すたんかった
-bye = さようならたた䌚う日たで
-active_pattern = ごめん|ゎメン|すたん|スマン|すいたせん
-silent_pattern = 邪魔|じゃた|ゞャマ|うる(さい|せヌ)|だたれ
+nick           = amazonbot
+content        = そんなアナタにこれがオススメ
+thanks         = ありがずう
+sorry          = 正盎すたんかった
+bye            = さようならたた䌚う日たで
+no_products    = 芋぀かりたせんでした
+active_pattern = ((ごめん|ゎメン)(なさい)?|(すたん|スマン)|(す(い|み)たせん))[なよねヌ・ .。、!]*$
+silent_pattern = (邪魔|じゃた|ゞャマ|うる(さい|せヌ)|だたれ)[だよなヌ-!]*$
+
+[action0]
+input_pattern = お(早|はよ)うございたす
+message       = おはようございたす %s さん
+action        = oper
+start_time    = 06:00
+end_time      = 12:00
+
+[action1]
+input_pattern = もうひずいきじゃ.*パワヌをメテオに
+message       = いいですずも
 
 [freq]

trunk/amazonbot/amazonbot.py

@@ -5 +5 @@
 __author__ = 'Atzm WATANABE <sitosito@p.chan.ne.jp>'
 __date__ = '$Date$'
-__copyright__ = 'Copyright(C) 2006 Atzm WATANABE, all rights reserved.'
+__copyright__ = 'Copyright(C) 2006-2007 Atzm WATANABE, all rights reserved.'
 __license__ = 'Python'
 
@@ -147 +147 @@
             return method(c, e, to, words[1:]) # words[0] == command name
 
+        self.message_action(msg, c, e, to)
+
         # silence
         self.silence(msg, c, e, to)
@@ -217 +219 @@
     def get_prev_time(self):
         return self._prev_time
+
+    def message_action(self, msg, c, e, to):
+        for i in xrange(100):
+            action = 'action%d' % i
+            if not config.has_section(action):
+                break
+
+            c_stime = config.get(action, 'start_time')
+            c_etime = config.get(action, 'end_time')
+
+            try:
+                if c_stime and c_etime:
+                    now = time.time()
+                    [now_y, now_m, now_d] = time.localtime(now)[:3]
+
+                    stime = '%04d/%02d/%02d %s' % (now_y, now_m, now_d, c_stime)
+                    etime = '%04d/%02d/%02d %s' % (now_y, now_m, now_d, c_etime)
+                    stime = time.mktime(time.strptime(stime, '%Y/%m/%d %H:%M'))
+                    etime = time.mktime(time.strptime(etime, '%Y/%m/%d %H:%M'))
+
+                    if not ((stime <= now) and (now <= etime)):
+                        _debug('Out of time: %s - %s' % (c_stime, c_etime))
+                        continue
+            except:
+                _debug('Invalid time: %s - %s' % (str(c_stime), str(c_etime)))
+                continue
+
+            match = re.search(unicoding(config.get(action, 'input_pattern')), msg)
+            if not match:
+                continue
+
+            act = config.get(action, 'action')
+            fmt = config.get(action, 'message')
+            try:
+                _from   = nm_to_n(e.source())
+                message = ununicoding(fmt % _from)
+
+                if not message:
+                    _debug('No message specified')
+                    continue
+
+                if not act:
+                    c.notice(to, message)
+                    continue
+
+                method = getattr(self, 'onact_%s' % act, lambda *arg: False)
+                method(message, c, e, to)
+
+            except KeyboardInterrupt:
+                _debug('Action failed: %s (%s)' % (str(act), str(fmt)))
+
+        return True
 
 class AmazonBot(AmazonBotBase):
@@ -244 +298 @@
         return 'AmazonBot by %s, based on python-irclib' % __author__
 
+    def onmsg_lt(self, c, e, to, args): return self.onmsg_localtime(c, e, to, args)
+    def onmsg_localtime(self, c, e, to, args):
+        """Syntax: !localtime <unix time>
+        """
+        _debug('in localtime command: %s', str(args))
+
+        _from = nm_to_n(e.source())
+        try:
+            sec = float(args[0])
+            c.notice(_from, time.strftime('%b %d %T', time.localtime(sec)))
+        except ValueError, e:
+            c.notice(_from, 'Invalid argument: %s' % args[0])
+            return False
+
+        return True
+
     def onmsg_s(self, c, e, to, args): return self.onmsg_status(c, e, to, args)
     def onmsg_status(self, c, e, to, args):
@@ -385 +455 @@
         return [product_name, url]
 
+    def onact_oper(self, msg, c, e, to):
+        nick = nm_to_n(e.source())
+        _debug('in oper action: %s to %s in %s' % (msg, nick, to))
+        c.notice(to, msg)
+        c.mode(to, '+o %s' % nick)
+        return True
+
+    def onact_nooper(self, msg, c, e, to):
+        nick = nm_to_n(e.source())
+        _debug('in nooper action: %s to %s in %s' % (msg, nick, to))
+        c.notice(to, msg)
+        c.mode(to, '-o %s' % nick)
+        return True
+
+    def onact_kick(self, msg, c, e, to):
+        nick = nm_to_n(e.source())
+        _debug('in kick action: %s to %s in %s' % (msg, nick, to))
+        c.kick(to, nick, msg)
+        return True
+
+    def onact_kick_and_invite(self, msg, c, e, to):
+        nick = nm_to_n(e.source())
+        _debug('in kick_and_invite action: %s to %s in %s' % (msg, nick, to))
+        c.kick(to, nick, msg)
+        c.invite(nick, to)
+        return True
+
+    def onact_nick(self, msg, c, e, to):
+        nick = nm_to_n(e.source())
+        _debug('in nick action: %s to %s in %s' % (msg, nick, to))
+        c.notice(to, msg)
+        c.nick('%s_' % nick)
+        return True
+
+    def onact_topic(self, msg, c, e, to):
+        nick = nm_to_n(e.source())
+        _debug('in topic action: %s to %s in %s' % (msg, nick, to))
+        c.topic(to, msg)
+        return True
+
 if __name__ == '__main__':
     bot = AmazonBot()

trunk/amazonbot/config.py

@@ -7 +7 @@
 
 _userconfig = 'amazonbot.ini'
-_parser = ConfigParser.SafeConfigParser()
+_parser = ConfigParser.RawConfigParser()
 
 def set(sect, key, val):
@@ -30 +30 @@
                         return ''
 
+def has_section(sect):
+    return _parser.has_section(sect)
+
 def write(filename=''):
         if not filename: