Context Navigation

source: trunk/amazonbot/amazonbot.py @ 18

Revision 18, 10.8 KB checked in by atzm, 18 years ago (diff)
add count comments

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

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

Download in other formats: