Context Navigation

source: trunk/amazonbot/amazonbot.py @ 20

Revision 20, 12.8 KB checked in by atzm, 17 years ago (diff)
Add !status command and small bug fix

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

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

Download in other formats: