Backup_Repos/calibre
1
0
Fork 0
mirror of git://github.com/kovidgoyal/calibre.git synced 2026-01-06 20:46:19 +01:00
Code Issues Projects Releases Wiki Activity

Make BD file detection more accurate

Browse source
This commit is contained in:
Kovid Goyal 2007-10-10 19:00:34 +00:00
parent d8b4bffb7e
commit 9e1498969d
1 changed files with 11 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
src/libprs500/ebooks/lrf/html/convert_from.py
View file

@ -229,12 +229,19 @@ def is_baen(self, soup):
return bool(soup.find('meta', attrs={'name':'Publisher',
'content':re.compile('Baen', re.IGNORECASE)}))
def is_book_designer(self, soup):
return bool(soup.find('h2', attrs={'id':'BookTitle'}))
def is_book_designer(self, raw):
return bool(re.search('<H2[^><]*id=BookTitle', raw))
def preprocess(self, raw):
nmassage = copy.copy(BeautifulSoup.MARKUP_MASSAGE)
nmassage.extend(HTMLConverter.MARKUP_MASSAGE)
if not self.book_designer and self.is_book_designer(raw):
self.book_designer = True
self.logger.info('\tBook Designer file detected.')
self.logger.info('\tParsing HTML...')
if self.baen:
nmassage.extend(HTMLConverter.BAEN)
@ -255,12 +262,8 @@ def preprocess(self, raw):
if not self.baen and self.is_baen(soup):
self.baen = True
self.logger.info('Baen file detected. Re-parsing...')
self.logger.info('\tBaen file detected. Re-parsing...')
return self.preprocess(raw)
if not self.book_designer and self.is_book_designer(soup):
self.book_designer = True
self.logger.info('Book Designer file detected. Re-parsing...')
return self.preprocess(raw)
if self.book_designer:
t = soup.find(id='BookTitle')
if t:
@ -286,7 +289,7 @@ def start_on_file(self, path, is_root=True, link_level=0):
path = os.path.abspath(path)
os.chdir(os.path.dirname(path))
self.file_name = os.path.basename(path)
self.logger.info('Processing %s\n\tParsing HTML...', self.file_name)
self.logger.info('Processing %s', self.file_name)
sys.stdout.flush()
soup = self.preprocess(open(self.file_name, 'rb').read())
self.logger.info('\tConverting to BBeB...')