Maximum CD 2011 January/maximum-cd-2011-01.iso/DiscContents/calibre-0.7.26.msi/file_956

Python Compiled Bytecode | 2010-10-31 | 6.4 KB | 166 lines

# Source Generated with Decompyle++
# File: in.pyc (Python 2.6)

__license__ = 'GPL v3'
__copyright__ = '2008, Kovid Goyal <kovid at kovidgoyal.net>'
import sys
import re
from urllib import quote
from calibre.utils.config import OptionParser
from calibre.ebooks.metadata.book.base import Metadata
from calibre.ebooks.BeautifulSoup import BeautifulStoneSoup
from calibre import browser
BASE_URL = 'http://isbndb.com/api/books.xml?access_key=%(key)s&page_number=1&results=subjects,authors,texts&'

class ISBNDBError(Exception):
    pass


def fetch_metadata(url, max = 100, timeout = 5):
    books = []
    page_number = 1
    total_results = sys.maxint
    br = browser()
    while len(books) < total_results and max > 0:

        try:
            raw = br.open(url, timeout = timeout).read()
        except Exception:
            err = None
            raise ISBNDBError('Could not fetch ISBNDB metadata. Error: ' + str(err))

        soup = BeautifulStoneSoup(raw, convertEntities = BeautifulStoneSoup.XML_ENTITIES)
        book_list = soup.find('booklist')
        if book_list is None:
            errmsg = soup.find('errormessage').string
            raise ISBNDBError('Error fetching metadata: ' + errmsg)
        book_list is None
        total_results = int(book_list['total_results'])
        page_number += 1
        np = '&page_number=%s&' % page_number
        url = re.sub('\\&page_number=\\d+\\&', np, url)
        books.extend(book_list.findAll('bookdata'))
        max -= 1
    return books


class ISBNDBMetadata(Metadata):

    def __init__(self, book):
        Metadata.__init__(self, None)

        def tostring(e):
            if not hasattr(e, 'string'):
                return None
            ans = e.string
            if ans is not None:
                ans = unicode(ans).strip()

            if not ans:
                ans = None

            return ans

        self.isbn = unicode(book.get('isbn13', book.get('isbn')))
        title = tostring(book.find('titlelong'))
        if not title:
            title = tostring(book.find('title'))

        self.title = title
        self.title = unicode(self.title).strip()
        authors = []
        au = tostring(book.find('authorstext'))
        if authors:
            self.authors = authors


        try:
            self.author_sort = tostring(book.find('authors').find('person'))
            if self.authors and self.author_sort == self.authors[0]:
                self.author_sort = None
        except:
            pass

        self.publisher = tostring(book.find('publishertext'))
        summ = tostring(book.find('summary'))
        if summ:
            self.comments = 'SUMMARY:\n' + summ




def build_isbn(base_url, opts):
    return base_url + 'index1=isbn&value1=' + opts.isbn


def build_combined(base_url, opts):
    query = ''
    for e in (opts.title, opts.author, opts.publisher):
        if e is not None:
            query += ' ' + e
            continue

    query = query.strip()
    if len(query) == 0:
        raise ISBNDBError('You must specify at least one of --author, --title or --publisher')
    len(query) == 0
    query = re.sub('\\s+', '+', query)
    if isinstance(query, unicode):
        query = query.encode('utf-8')

    return base_url + 'index1=combined&value1=' + quote(query, '+')


def option_parser():
    parser = OptionParser(usage = _('\n%prog [options] key\n\nFetch metadata for books from isndb.com. You can specify either the\nbooks ISBN ID or its title and author. If you specify the title and author,\nthen more than one book may be returned.\n\nkey is the account key you generate after signing up for a free account from isbndb.com.\n\n'))
    parser.add_option('-i', '--isbn', default = None, dest = 'isbn', help = _('The ISBN ID of the book you want metadata for.'))
    parser.add_option('-a', '--author', dest = 'author', default = None, help = _('The author whose book to search for.'))
    parser.add_option('-t', '--title', dest = 'title', default = None, help = _('The title of the book to search for.'))
    parser.add_option('-p', '--publisher', default = None, dest = 'publisher', help = _('The publisher of the book to search for.'))
    parser.add_option('-v', '--verbose', default = False, action = 'store_true', help = _('Verbose processing'))
    return parser


def create_books(opts, args, timeout = 5):
    base_url = BASE_URL % dict(key = args[1])
    if opts.isbn is not None:
        url = build_isbn(base_url, opts)
    else:
        url = build_combined(base_url, opts)
    if opts.verbose:
        print 'ISBNDB query: ' + url

    tans = [ ISBNDBMetadata(book) for book in fetch_metadata(url, timeout = timeout) ]
    ans = []
    for x in tans:
        add = True
        for y in ans:
            if y.isbn == x.isbn:
                add = False
                continue
            []

        if add:
            ans.append(x)
            continue
        []

    return ans


def main(args = sys.argv):
    parser = option_parser()
    (opts, args) = parser.parse_args(args)
    if len(args) != 2:
        parser.print_help()
        print 'You must supply the isbndb.com key'
        return 1
    for book in create_books(opts, args):
        print unicode(book).encode('utf-8')

    return 0

if __name__ == '__main__':
    sys.exit(main())