Context Navigation

source: trunk/amazonbot/amazonbot.py @ 17

Revision 17, 10.3 KB checked in by atzm, 18 years ago (diff)
debug

Rev	Line
[8]	1	#!/usr/bin/env python
	2	# -- coding: utf-8 --
	3
	4	__version__ = '$Revision$'
	5	__author__ = 'Atzm WATANABE <sitosito@p.chan.ne.jp>'
	6	__date__ = '$Date$'
	7	__copyright__ = 'Copyright(C) 2006 Atzm WATANABE, all rights reserved.'
	8	__license__ = 'Python'
	9
	10	import re
[9]	11	import sys
[10]	12	import time
[12]	13	import shlex
[8]	14	import random
[12]	15	import getopt
	16
[8]	17	import MeCab
[10]	18	import nkf
[8]	19
	20	from ircbot import SingleServerIRCBot
	21	from irclib import nm_to_n
	22
	23	import config
	24	config.init()
	25
	26	import my_amazon
	27	my_amazon.setLocale(config.get('amazon', 'locale'))
	28	my_amazon.setLicense(config.get('amazon', 'access_key'))
	29
	30	try:
	31	set, frozenset
	32	except NameError:
	33	from sets import Set as set, ImmutableSet as frozenset
	34
	35	def uniq(sequence):
[10]	36	"""ãªã¹ãããéè€ãåãé€ã (é çªãçãã®ã§æ³šæ)
	37	"""
[8]	38	return list(set(sequence))
	39
[10]	40	def unicoding(text):
	41	"""text ãåŒ·å¶çã« unicode ãªããžã§ã¯ãã«å€æ
	42	"""
	43	if type(text) is unicode:
	44	return text
	45	return unicode(nkf.nkf('-w', text), 'utf-8')
	46
	47	def ununicoding(text, encoding='iso-2022-jp'):
	48	"""text ãæå®ããã encoding ã§ãšã³ã³ãŒããïŒraw str ã«åŒ·å¶å€æ
	49	"""
	50	if type(text) is not unicode:
	51	return unicoding(text).encode(encoding)
	52	return text.encode(encoding)
	53
[8]	54	def mecab_parse(text):
[10]	55	"""MeCab ãäœ¿ã£ãŠåœ¢æçŽ è§£æãïŒåºæåè©ãšäžè¬åè©ã ããæœåºãã
	56	"""
[8]	57	def choice_nominal(wlist):
	58	res = []
	59	for word, wtype in wlist:
	60	wtypes = wtype.split('-')
	61	if 'åºæåè©' in wtypes or ('åè©' in wtypes and 'äžè¬' in wtypes):
[10]	62	res.append(unicoding(word))
[8]	63	return res
	64
[10]	65	text = ununicoding(text, 'utf-8')
[8]	66	result = []
	67	tag = MeCab.Tagger('-Ochasen')
	68	for line in tag.parse(text).split('\n'):
	69	if not line or line == 'EOS':
	70	break
	71	words = line.split()
	72	result.append((words[0], words[-1])) # word, word-type
	73
	74	result = uniq(choice_nominal(result))
	75	return result
	76
	77	class AmazonBotBase(SingleServerIRCBot):
[10]	78	"""ã¢ããŸã³ãããã®ããŒã¹ã¯ã©ã¹
[12]	79	åäœã§ã¯ïŒåãåã£ãã¡ãã»ãŒãžã®åœ¢æçŽ è§£æãšåè©æœåºãŸã§ãããããªã
[10]	80	ãµãã¯ã©ã¹ã§ process_keyword ãå®è£ããŠ Amazon ãžã¯ãšãªãæããã¹ã
[12]	81
[14]	82	ãµãã¯ã©ã¹ã«ã¯ onmsg_HOGEHOGE(self, conn, ev, to, args) ã¡ãœãããäœãããšã§ã³ãã³ãè¿œå å¯èœ
[12]	83	ã³ãã³ãæžåŒã¯ !HOGEHOGE arg [, arg2, ...] ãšãªã
[14]	84	ãã«ãã¯ã¡ãœããã« docstring ãæžãã° OK
[10]	85	"""
[8]	86	def __init__(self):
	87	_server = [(config.get('irc', 'server'), config.get('irc', 'port', 'int'))]
	88	_nick = config.get('bot', 'nick')
	89
[15]	90	self._prev_time = time.time() - config.get('bot', 'freq', 'int')
[8]	91	self._silent = False
	92	SingleServerIRCBot.__init__(self, _server, _nick, _nick)
	93
[10]	94	def start(self):
	95	try:
	96	SingleServerIRCBot.start(self)
	97	except KeyboardInterrupt:
	98	self.die(ununicoding(config.get('bot', 'bye')))
[8]	99
	100	def on_welcome(self, c, e):
	101	c.join(config.get('irc', 'channel'))
	102	if __debug__:
[10]	103	print >> sys.stderr, 'DEBUG> Joined %s' % config.get('irc', 'channel')
[8]	104
	105	def on_nicknameinuse(self, c, e):
	106	c.nick(c.get_nickname() + '_')
	107
	108	def on_privmsg(self, c, e):
[14]	109	return self.on_pubmsg(c, e, to=nm_to_n(e.source()))
[8]	110
[14]	111	def on_pubmsg(self, c, e, to=config.get('irc', 'channel')):
[12]	112	msg = unicoding(e.arguments()[0])
[14]	113
	114	if __debug__:
[17]	115	try:
	116	print >> sys.stderr, 'DEBUG> pubmsg incoming "%s", should be reply to %s' % (ununicoding(msg, 'euc-jp'), to)
	117	except:
	118	print >> sys.stderr, 'DEBUG> pubmsg incoming, but could not en/decode message. should be reply to %s' % to
[14]	119
[12]	120	if msg[0] == '!':
[17]	121	try:
	122	words = shlex.split(ununicoding(msg, 'utf-8')[1:])
	123	except:
	124	return False
[16]	125	if not words:
	126	return False
[12]	127	method = getattr(self, 'onmsg_%s' % words[0], lambda *arg: False)
[14]	128	return method(c, e, to, words[1:]) # words[0] == command name
[12]	129
[11]	130	_current_time = time.time()
	131	if _current_time < self._prev_time + config.get('bot', 'freq', 'int'):
[9]	132	if __debug__:
[15]	133	cur = time.strftime('%H:%M:%S', time.localtime(_current_time))
[14]	134	go = time.strftime('%H:%M:%S', time.localtime(self._prev_time + config.get('bot', 'freq', 'int')))
[15]	135	print >> sys.stderr, 'DEBUG> Not expired: now %s, be expired at: %s' % (cur, go)
[9]	136	return False
[11]	137	self._prev_time = _current_time
[10]	138
[14]	139	self.silence(msg, c, e, to)
[8]	140	if self._silent:
	141	return False
	142
[10]	143	nominals = mecab_parse(msg)
[8]	144	if not nominals:
[10]	145	if __debug__:
	146	print >> sys.stderr, "DEBUG> Couldn't find nominal words"
[8]	147	return False
	148
	149	title, url = self.process_keyword(' '.join(nominals))
	150	if title and url:
[10]	151	content = unicoding(config.get('bot', 'content'))
[9]	152	try:
[10]	153	message = ununicoding(': '.join([content, title, url]))
[12]	154	except UnicodeError, err:
[11]	155	# ãªããããŸã« unicode ãªããžã§ã¯ãã iso-2022-jp ã§ãšã³ã³ãŒãã§ããªã
	156	if __debug__:
[12]	157	print >> sys.stderr, 'DEBUG> %s' % str(err)
[9]	158	return False
	159
[14]	160	c.notice(to, message)
[8]	161	return True
	162	return False
	163
[10]	164	ACTIVE_PATTERN = re.compile(unicoding(config.get('bot', 'active_pattern')))
	165	SILENT_PATTERN = re.compile(unicoding(config.get('bot', 'silent_pattern')))
[14]	166	def silence(self, msg, c, e, to):
[10]	167	active = self.ACTIVE_PATTERN.search(msg)
	168	silent = self.SILENT_PATTERN.search(msg)
[9]	169	if __debug__:
[10]	170	print >> sys.stderr, 'DEBUG> ACT_PATT: %s, SIL_PATT: %s' % (str(active), str(silent))
[9]	171
	172	if active:
[8]	173	self._silent = False
[14]	174	c.notice(to, ununicoding(config.get('bot', 'thanks')))
[9]	175	elif silent:
[8]	176	self._silent = True
[14]	177	c.notice(to, ununicoding(config.get('bot', 'sorry')))
[8]	178
	179	def process_keyword(self, keyword):
	180	return [None, None]
	181
	182	class AmazonBot(AmazonBotBase):
[10]	183	"""ã¢ããŸã³ãããã®å®è£ã¯ã©ã¹
	184	process_keyword ã¡ãœããã§ Amazon ãžã¯ãšãªãæããŠçµæãè¿ã
	185	"""
[14]	186	_AVAIL_PRODUCT_LINES = {
	187	'books-jp': '(åæž, default)',
	188	'books-us': '(æŽæž)',
	189	'music-jp': '(ããã¥ã©ãŒé³æ¥œ)',
	190	'classical-jp': '(ã¯ã©ã·ãã¯é³æ¥œ)',
	191	'dvd-jp': '(DVD)',
	192	'vhs-jp': '(ãããª)',
	193	'electronics-jp': '(ãšã¬ã¯ãããã¯ã¹)',
	194	'kitchen-jp': '(ããŒã ïŒãããã³)',
	195	'software-jp': '(ãœãããŠã§ã¢)',
	196	'videogames-jp': '(ã²ãŒã )',
	197	'magazines-jp': '(éèª)',
	198	'toys-jp': '(ããã¡ãïŒãããŒ)',
	199	}
	200
[8]	201	def __init__(self):
	202	AmazonBotBase.__init__(self)
	203
[10]	204	def get_version(self):
	205	return 'AmazonBot by %s, based on python-irclib' % __author__
	206
[14]	207	def onmsg_isbn(self, c, e, to, args):
[12]	208	"""Syntax: !isbn <ISBN number>
	209	"""
[14]	210	return self.onmsg_asin(c, e, to, args)
	211	def onmsg_asin(self, c, e, to, args):
[12]	212	"""Syntax: !asin <ASIN number>
	213	"""
	214	if __debug__:
	215	print >> sys.stderr, 'DEBUG> in asin command: %s' % str(args)
	216
	217	try:
	218	data = my_amazon.searchByASIN(args[0])
	219	except my_amazon.AmazonError, err:
[14]	220	c.notice(to, ununicoding(config.get('bot', 'no_products')))
[12]	221	if __debug__:
	222	print >> sys.stderr, 'DEBUG> Caught AmazonError in onmsg_asin: %s' % str(err)
	223	return False
[14]	224	except IndexError, err:
	225	c.notice(to, 'Please specify an argument.')
	226	return False
[12]	227
[14]	228	return self._process_onmsg(c, e, to, data)
[12]	229
[14]	230	def onmsg_k(self, c, e, to, args): return self.onmsg_keyword(c, e, to, args)
	231	def onmsg_keyword(self, c, e, to, args):
	232	"""Syntax: !keyword [-h] [-t type] <keyword1> [, keyword2, ...]
[12]	233	"""
	234	if __debug__:
[14]	235	print >> sys.stderr, 'DEBUG> in keyword command: %s' % str(args)
[12]	236
	237	try:
	238	options, rest = getopt.getopt(args, 't:h', ['type=', 'help'])
	239	except getopt.GetoptError, err:
	240	if __debug__:
[14]	241	print >> sys.stderr, 'DEBUG> Caught GetoptError in onmsg_keyword: %s' % str(err)
[12]	242	return False
	243
[14]	244	keyword = ' '.join(rest).strip()
[12]	245	product_line = 'books-jp'
	246	for opt, val in options:
	247	if opt in ['-t', '--type']:
[14]	248	if val not in self._AVAIL_PRODUCT_LINES.keys():
	249	c.notice(to, 'Type "%s" is not available.' % val)
	250	return False
	251
[12]	252	product_line = val
	253	break
[14]	254
[12]	255	elif opt in ['-h', '--help']:
[14]	256	_from = nm_to_n(e.source()) # ãã°ãæµããŠããŸãã®ã§ãã«ãã¯çŽæ¥éä¿¡åãž
	257	c.notice(_from, ununicoding('Available types:'))
	258
	259	for key, val in self._AVAIL_PRODUCT_LINES.iteritems():
	260	time.sleep(1) # XXX: é£ç¶æçš¿ãããšåŒŸãããããšãããã®ã§æ«å®å¯ŸåŠ
	261	c.notice(_from, ununicoding(' * %s: %s' % (key, val)))
	262
[12]	263	return True
	264
[14]	265	if not keyword:
	266	c.notice(to, 'Please specify keywords.')
	267	return False
	268
[12]	269	if __debug__:
	270	fmt = 'DEBUG> keyword="%s", product_line=%s'
	271	print >> sys.stderr, fmt % (ununicoding(keyword, 'euc-jp'), product_line)
	272
	273	try:
	274	data = my_amazon.searchByKeyword(keyword, product_line=product_line)
	275	except my_amazon.AmazonError, err:
[14]	276	c.notice(to, ununicoding(config.get('bot', 'no_products')))
[12]	277	if __debug__:
	278	print >> sys.stderr, 'DEBUG> Caught AmazonError in onmsg_amazon: %s' % str(err)
	279	return False
	280
[14]	281	return self._process_onmsg(c, e, to, data)
[12]	282
[14]	283	def onmsg_h(self, c, e, to, args): return self.onmsg_help(c, e, to, args)
	284	def onmsg_help(self, c, e, to, args):
	285	"""Syntax: !help
	286	"""
	287	if __debug__:
	288	print >> sys.stderr, 'DEBUG> in help command: %s' % str(args)
[12]	289
[14]	290	_from = nm_to_n(e.source()) # ãã°ãæµããŠããŸãã®ã§ãã«ãã¯çŽæ¥éä¿¡åãž
	291	docs = []
	292	for key in dir(self):
	293	val = getattr(self, key, '')
	294	if __debug__:
	295	print >> sys.stderr, 'DEBUG> key=%s, val=%s' % (key, ununicoding(str(val), 'euc-jp'))
	296
	297	if key[:6] != 'onmsg_':
	298	continue
	299
	300	doc = val.__doc__
	301	if doc:
	302	doc = doc.strip()
	303	if not doc:
	304	continue
	305	time.sleep(1) # XXX: é£ç¶æçš¿ãããšåŒŸãããã£ãœãã®ã§æ«å®å¯ŸåŠ
	306	c.notice(_from, doc)
	307
	308	return True
	309
	310	def _process_onmsg(self, c, e, to, data):
[12]	311	if type(data.Details) is not list:
	312	data.Details = [data.Details]
	313
	314	detail = random.choice(data.Details)
	315	title = ununicoding(detail.ProductName)
	316	url = ununicoding(detail.URL)
[14]	317	c.notice(to, '%(title)s: %(url)s' % locals())
[12]	318
	319	return True
	320
[8]	321	def process_keyword(self, keyword):
[10]	322	keyword = ununicoding(keyword, 'utf-8')
	323	if __debug__:
	324	print >> sys.stderr, 'DEBUG> KEYWORD: %s' % ununicoding(keyword, 'euc-jp')
	325
[8]	326	try:
	327	data = my_amazon.searchByBlended(keyword)
	328	if type(data.ProductLine) is not type([]):
	329	data.ProductLine = [data.ProductLine]
[12]	330	except my_amazon.AmazonError, err:
[10]	331	if __debug__:
[12]	332	print >> sys.stderr, 'DEBUG> Caught AmazonError: %s' % str(err)
[8]	333	return [None, None]
	334
	335	product_line = random.choice(data.ProductLine)
	336	detail = random.choice(product_line.ProductInfo.Details)
	337
[10]	338	url = unicoding(getattr(detail, 'URL', None))
	339	product_name = unicoding(getattr(detail, 'ProductName', None))
[8]	340
	341	return [product_name, url]
	342
	343	if __name__ == '__main__':
	344	bot = AmazonBot()
	345	bot.start()
[14]	346	print '> Bye ;)'

Note: See TracBrowser for help on using the repository browser.

Download in other formats: