### # Copyright (c) 2002-2004, Jeremiah Fincher # All rights reserved. # # Redistribution and use in source and binary forms, with or without # modification, are permitted provided that the following conditions are met: # # * Redistributions of source code must retain the above copyright notice, # this list of conditions, and the following disclaimer. # * Redistributions in binary form must reproduce the above copyright notice, # this list of conditions, and the following disclaimer in the # documentation and/or other materials provided with the distribution. # * Neither the name of the author of this software nor the name of # contributors to this software may be used to endorse or promote products # derived from this software without specific prior written consent. # # THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" # AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE # IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE # ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE # LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR # CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF # SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS # INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN # CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) # ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE # POSSIBILITY OF SUCH DAMAGE. ### """ Amazon module, to use Amazon's Web Services. """ import supybot __revision__ = "$Id$" __author__ = supybot.authors.jamessan import getopt import supybot.plugins as plugins import amazon import supybot.registry as registry import supybot.conf as conf import supybot.utils as utils import supybot.ircutils as ircutils import supybot.privmsgs as privmsgs import supybot.callbacks as callbacks def configure(advanced): from supybot.questions import output, expect, anything, something, yn output('To use Amazon\'s Web Services, you must have a license key.') if yn('Do you have a license key?'): key = anything('What is it?') conf.registerPlugin('Amazon', True) conf.supybot.plugins.Amazon.licenseKey.set(key) else: output("""You'll need to get a key before you can use this plugin. You can apply for a key at http://www.amazon.com/webservices/""") class Region(registry.OnlySomeStrings): validStrings = ('us', 'uk', 'de', 'jp') class LicenseKey(registry.String): def set(self, s): # In case we decide we need to recover original = getattr(self, 'value', self._default) registry.String.set(self, s) if self.value: amazon.setLicense(self.value) conf.registerPlugin('Amazon') conf.registerChannelValue(conf.supybot.plugins.Amazon, 'bold', registry.Boolean(True, """Determines whether the results are bolded.""")) conf.registerGlobalValue(conf.supybot.plugins.Amazon, 'licenseKey', LicenseKey('', """Sets the license key for using Amazon Web Services. Must be set before any other commands in the plugin are used.""", private=True)) conf.registerChannelValue(conf.supybot.plugins.Amazon, 'linkSnarfer', registry.Boolean(False, """Determines whether the bot will reply to Amazon.com URLs in the channel with a description of the item at the URL.""")) conf.registerChannelValue(conf.supybot.plugins.Amazon, 'region', Region('us', """Determines the region that will be used when performing searches.""")) class Amazon(callbacks.PrivmsgCommandAndRegexp): threaded = True callBefore = ['URL'] regexps = ['amzSnarfer'] def __init__(self): self.__parent = super(Amazon, self) self.__parent.__init__() def callCommand(self, name, irc, msg, *L, **kwargs): try: self.__parent.callCommand(name, irc, msg, *L, **kwargs) except amazon.NoLicenseKey, e: irc.error('You must have a free Amazon web services license key ' 'in order to use this command. You can get one at ' '<http://www.amazon.com/webservices>. Once you have ' 'one, you can set it with the command ' '"config supybot.plugins.Amazon.licensekey <key>".') def _genResults(self, reply, attribs, items, url, bold, bold_item): results = {} res = [] if isinstance(items, amazon.Bag): items = [items] for item in items: try: for k,v in attribs.iteritems(): results[v] = getattr(item, k, 'unknown') if isinstance(results[v], amazon.Bag): results[v] = getattr(results[v], k[:-1], 'unknown') if not isinstance(results[v], basestring): results[v] = utils.commaAndify(results[v]) if bold_item in results: if bold: results[bold_item] = ircutils.bold(results[bold_item]) else: results[bold_item] = '"%s"' % results[bold_item] if not url: results['url'] = '' else: results['url'] = ' <%s>' % results['url'] s = reply % results if isinstance(s, unicode): s = s.encode('utf-8') res.append(str(s)) except amazon.AmazonError, e: self.log.warning(str(e)) except UnicodeEncodeError, e: self.log.warning(str(e)) return res def isbn(self, irc, msg, args): """[--url] <isbn> Returns the book matching the given ISBN number. If --url is specified, a link to amazon.com's page for the book will also be returned. """ opts = ['url'] (optlist, rest) = getopt.getopt(args, '', opts) url = False for (option, argument) in optlist: option = option.lstrip('-') if option == 'url': url = True isbn = privmsgs.getArgs(rest) isbn = isbn.replace('-', '').replace(' ', '') attribs = {'ProductName' : 'title', 'Manufacturer' : 'publisher', 'Authors' : 'author', 'OurPrice' : 'price', 'URL' : 'url' } s = '%(title)s, written by %(author)s; published by ' \ '%(publisher)s; price: %(price)s%(url)s' chan = msg.args[0] bold = self.registryValue('bold', chan) region = self.registryValue('region', chan) try: book = amazon.searchByKeyword(isbn, locale=region) res = self._genResults(s, attribs, book, url, bold, 'title') if res: irc.reply(utils.commaAndify(res)) return except amazon.AmazonError, e: pass irc.error('No book was found with that ISBN.') def books(self, irc, msg, args): """[--url] <keywords> Returns the books matching the given <keywords> search. If --url is specified, a link to amazon.com's page for the book will also be returned. """ opts = ['url'] (optlist, rest) = getopt.getopt(args, '', opts) url = False for (option, argument) in optlist: option = option.lstrip('-') if option == 'url': url = True keyword = privmsgs.getArgs(rest) attribs = {'ProductName' : 'title', 'Manufacturer' : 'publisher', 'Authors' : 'author', 'OurPrice' : 'price', 'URL' : 'url' } s = '%(title)s, written by %(author)s; published by ' \ '%(publisher)s; price: %(price)s%(url)s' chan = msg.args[0] region = self.registryValue('region', chan) bold = self.registryValue('bold', chan) try: books = amazon.searchByKeyword(keyword, locale=region) res = self._genResults(s, attribs, books, url, bold, 'title') if res: irc.reply(utils.commaAndify(res)) return except amazon.AmazonError, e: pass irc.error('No books were found with that keyword search.') def videos(self, irc, msg, args): """[--url] [--{dvd,vhs}] <keywords> Returns the videos matching the given <keyword> search. If --url is specified, a link to amazon.com's page for the video will also be returned. Search defaults to using --dvd. """ opts = ['url'] products = ['dvd', 'vhs'] (optlist, rest) = getopt.getopt(args, '', opts + products) url = False product = 'dvd' for (option, argument) in optlist: option = option.lstrip('-') if option == 'url': url = True if option in products: product = option keyword = privmsgs.getArgs(rest) attribs = {'ProductName' : 'title', 'Manufacturer' : 'publisher', 'MpaaRating' : 'mpaa', 'Media' : 'media', 'ReleaseDate' : 'date', 'OurPrice' : 'price', 'URL' : 'url' } s = '%(title)s (%(media)s), rated %(mpaa)s; released ' \ '%(date)s; published by %(publisher)s; price: %(price)s%(url)s' chan = msg.args[0] region = self.registryValue('region', chan) bold = self.registryValue('bold', chan) try: videos = amazon.searchByKeyword(keyword, product_line=product, locale=region) res = self._genResults(s, attribs, videos, url, bold, 'title') if res: irc.reply(utils.commaAndify(res)) return except amazon.AmazonError, e: pass irc.error('No videos were found with that keyword search.') def asin(self, irc, msg, args): """[--url] <asin> Returns the item matching the given ASIN number. If --url is specified, a link to amazon.com's page for the item will also be returned. """ opts = ['url'] (optlist, rest) = getopt.getopt(args, '', opts) url = False for (option, argument) in optlist: option = option.lstrip('-') if option == 'url': url = True asin = privmsgs.getArgs(rest) asin = asin.replace('-', '').replace(' ', '') attribs = {'ProductName' : 'title', 'OurPrice' : 'price', 'URL' : 'url' } s = '%(title)s; price: %(price)s%(url)s' chan = msg.args[0] region = self.registryValue('region', chan) bold = self.registryValue('bold', chan) try: item = amazon.searchByASIN(asin, locale=region) res = self._genResults(s, attribs, item, url, bold, 'title') if res: irc.reply(utils.commaAndify(res)) return except amazon.AmazonError, e: pass irc.error('No item was found with that ASIN.') def upc(self, irc, msg, args): """[--url] <upc> Returns the item matching the given UPC number. If --url is specified, a link to amazon.com's page for the item will also be returned. Only items in the following categories may be found via upc search: music, classical, software, dvd, video, vhs, electronics, pc-hardware, and photo. """ opts = ['url'] (optlist, rest) = getopt.getopt(args, '', opts) url = False for (option, argument) in optlist: option = option.lstrip('-') if option == 'url': url = True upc = privmsgs.getArgs(rest) upc = upc.replace('-', '').replace(' ', '') attribs = {'ProductName' : 'title', 'Manufacturer' : 'manufacturer', 'OurPrice' : 'price', 'URL' : 'url' } s = '%(title)s %(manufacturer)s; price: %(price)s%(url)s' chan = msg.args[0] region = self.registryValue('region', chan) bold = self.registryValue('bold', chan) try: item = amazon.searchByUPC(upc, locale=region) res = self._genResults(s, attribs, item, url, bold, 'title') if res: irc.reply(utils.commaAndify(res)) return except amazon.AmazonError, e: pass irc.error('No item was found with that UPC.') def author(self, irc, msg, args): """[--url] <author> Returns a list of books written by the given author. If --url is specified, a link to amazon.com's page for the book will also be returned. """ opts = ['url'] (optlist, rest) = getopt.getopt(args, '', opts) url = False for (option, argument) in optlist: option = option.lstrip('-') if option == 'url': url = True author = privmsgs.getArgs(rest) attribs = {'ProductName' : 'title', 'Manufacturer' : 'publisher', 'Authors' : 'author', 'OurPrice' : 'price', 'URL' : 'url' } s = '%(title)s, written by %(author)s; published by ' \ '%(publisher)s; price: %(price)s%(url)s' chan = msg.args[0] region = self.registryValue('region', chan) bold = self.registryValue('bold', chan) try: books = amazon.searchByAuthor(author, locale=region) res = self._genResults(s, attribs, books, url, bold, 'title') if res: irc.reply(utils.commaAndify(res)) return except amazon.AmazonError, e: pass irc.error('No books were found by that author.') # FIXME: Until I get a *good* list of categories (ones that actually work), # these commands will remain unavailable ''' _textToNode = {'dvds':'130', 'magazines':'599872', 'music':'301668', 'software':'491286', 'vhs':'404272', 'kitchen':'491864', 'video games':'471280', 'toys':'491290', 'camera':'502394', 'outdoor':'468250', 'computers':'565118', 'tools':'468240', 'electronics':'172282' } def categories(self, irc, msg, args): """takes no arguments Returns a list of valid categories to use with the bestsellers commands. """ cats = self._textToNode.keys() cats.sort() irc.reply(utils.commaAndify(cats)) def bestsellers(self, irc, msg, args): """[--url] <category> Returns a list of best selling items in <category>. The 'categories' command will return a list of the available categories. If --url is specified, a link to amazon.com's page for the item will also be returned. """ opts = ['url'] (optlist, rest) = getopt.getopt(args, '', opts) url = False for (option, argument) in optlist: option = option.lstrip('-') if option == 'url': url = True category = privmsgs.getArgs(rest).lower() if category not in self._textToNode: irc.error('An invalid category was specified. The categories' ' command will return a list of valid categories') return category = self._textToNode[category] attribs = {'ProductName' : 'title', 'Manufacturer' : 'publisher', 'URL' : 'url' } s = '"%(title)s", from %(publisher)s.%(url)s' try: #self.log.warning(category) items = amazon.browseBestSellers(category) #self.log.warning(items) res = self._genResults(s, attribs, items, url) if res: irc.reply(utils.commaAndify(res)) return except amazon.AmazonError, e: pass irc.error('No items were found on that best seller list.') ''' def artist(self, irc, msg, args): """[--url] [--{music,classical}] <artist> Returns a list of items by the given artist. If --url is specified, a link to amazon.com's page for the match will also be returned. The search defaults to using --music. """ products = ['music', 'classical'] opts = ['url'] (optlist, rest) = getopt.getopt(args, '', products + opts) url = False product = '' for (option, argument) in optlist: option = option.lstrip('-') if option == 'url': url = True if option in products: product = option product = product or 'music' artist = privmsgs.getArgs(rest) attribs = {'ProductName' : 'title', 'Manufacturer' : 'publisher', 'Artists' : 'artist', 'Media' : 'media', 'OurPrice' : 'price', 'URL' : 'url' } s = '%(title)s (%(media)s), by %(artist)s; published by ' \ '%(publisher)s; price: %(price)s%(url)s' chan = msg.args[0] region = self.registryValue('region', chan) bold = self.registryValue('bold', chan) try: items = amazon.searchByArtist(artist, product_line=product, locale=region) res = self._genResults(s, attribs, items, url, bold, 'title') if res: irc.reply(utils.commaAndify(res)) return except amazon.AmazonError, e: pass irc.error('No items were found by that artist.') def actor(self, irc, msg, args): """[--url] [--{dvd,vhs,video}] <actor> Returns a list of items starring the given actor. If --url is specified, a link to amazon.com's page for the match will also be returned. The search defaults to using --dvd. """ products = ['dvd', 'video', 'vhs'] opts = ['url'] (optlist, rest) = getopt.getopt(args, '', products + opts) url = False product = '' for (option, argument) in optlist: option = option.lstrip('-') if option == 'url': url = True if option in products: product = option product = product or 'dvd' actor = privmsgs.getArgs(rest) attribs = {'ProductName' : 'title', 'Manufacturer' : 'publisher', 'MpaaRating' : 'mpaa', 'Media' : 'media', 'ReleaseDate' : 'date', 'OurPrice' : 'price', 'URL' : 'url' } s = '%(title)s (%(media)s), rated %(mpaa)s; released ' \ '%(date)s; published by %(publisher)s; price: %(price)s%(url)s' chan = msg.args[0] region = self.registryValue('region', chan) bold = self.registryValue('bold', chan) try: items = amazon.searchByActor(actor, product_line=product, locale=region) res = self._genResults(s, attribs, items, url, bold, 'title') if res: irc.reply(utils.commaAndify(res)) return except amazon.AmazonError, e: pass irc.error('No items were found starring that actor.') def director(self, irc, msg, args): """[--url] [--{dvd,vhs,video}] <director> Returns a list of items by the given director. If --url is specified, a link to amazon.com's page for the match will also be returned. The search defaults to using --dvd. """ products = ['dvd', 'video', 'vhs'] opts = ['url'] (optlist, rest) = getopt.getopt(args, '', products + opts) url = False product = '' for (option, argument) in optlist: option = option.lstrip('-') if option == 'url': url = True if option in products: product = option product = product or 'dvd' director = privmsgs.getArgs(rest) attribs = {'ProductName' : 'title', 'Manufacturer' : 'publisher', 'MpaaRating' : 'mpaa', 'Media' : 'media', 'ReleaseDate' : 'date', 'OurPrice' : 'price', 'URL' : 'url' } s = '%(title)s (%(media)s), rated %(mpaa)s; released ' \ '%(date)s; published by %(publisher)s; price: %(price)s%(url)s' chan = msg.args[0] region = self.registryValue('region', chan) bold = self.registryValue('bold', chan) try: items = amazon.searchByDirector(director, product_line=product, locale=region) res = self._genResults(s, attribs, items, url, bold, 'title') if res: irc.reply(utils.commaAndify(res)) return except amazon.AmazonError, e: pass irc.error('No items were found by that director.') def manufacturer(self, irc, msg, args): """ [--url] \ [--{pc-hardware,kitchen,electronics,videogames,software,photo}] \ <manufacturer> Returns a list of items by the given manufacturer. If --url is specified, a link to amazon.com's page for the match will also be returned. The search defaults to using --pc-hardware. """ products = ['electronics', 'kitchen', 'videogames', 'software', 'photo', 'pc-hardware'] opts = ['url'] (optlist, rest) = getopt.getopt(args, '', products + opts) url = False product = '' for (option, argument) in optlist: option = option.lstrip('-') if option == 'url': url = True if option in products: product = option product = product or 'pc-hardware' manufacturer = privmsgs.getArgs(rest) attribs = {'ProductName' : 'title', 'OurPrice' : 'price', 'URL' : 'url' } s = '%(title)s; price: %(price)s%(url)s' chan = msg.args[0] region = self.registryValue('region', chan) bold = self.registryValue('bold', chan) try: items = amazon.searchByManufacturer(manufacturer, product_line=product, locale=region) res = self._genResults(s, attribs, items, url, bold, 'title') if res: irc.reply(utils.commaAndify(res)) return except amazon.AmazonError, e: pass irc.error('No items were found by that manufacturer.') def amzSnarfer(self, irc, msg, match): r"http://www.amazon.com/exec/obidos/(?:tg/detail/-/|ASIN/)([^/]+)" if not self.registryValue('linkSnarfer', msg.args[0]): return match = match.group(1) attribs = {'ProductName' : 'title', 'Manufacturer' : 'publisher', 'Authors' : 'author', 'MpaaRating' : 'mpaa', 'Media' : 'media', 'ReleaseDate' : 'date', 'OurPrice' : 'price', 'Artists' : 'artist', } s = '%(title)s; %(artist)s; %(author)s; %(mpaa)s; %(media)s; '\ '%(date)s; %(publisher)s; price: %(price)s' chan = msg.args[0] region = self.registryValue('region', chan) bold = self.registryValue('bold', chan) try: item = amazon.searchByASIN(match, locale=region) res = self._genResults(s, attribs, item, False, bold, 'title') if res: res = utils.commaAndify(res) res = res.replace('; unknown', '') res = res.replace('; price: unknown', '') irc.reply(res, prefixName=False) return except amazon.AmazonError, e: pass self.log.warning('No item was found with that ASIN.') Class = Amazon # vim:set shiftwidth=4 tabstop=8 expandtab textwidth=78: