mirror of
https://github.com/gryf/ebook-converter.git
synced 2026-01-22 21:05:45 +01:00
35 lines
1.1 KiB
Python
35 lines
1.1 KiB
Python
import os
|
|
|
|
from ebook_converter.customize.conversion import InputFormatPlugin
|
|
from ebook_converter.ebooks.pdb.header import PdbHeaderReader
|
|
from ebook_converter.ebooks.pdb import PDBError, IDENTITY_TO_NAME, get_reader
|
|
|
|
|
|
class PDBInput(InputFormatPlugin):
|
|
|
|
name = 'PDB Input'
|
|
author = 'John Schember'
|
|
description = 'Convert PDB to HTML'
|
|
file_types = {'pdb', 'updb'}
|
|
commit_name = 'pdb_input'
|
|
|
|
def convert(self, stream, options, file_ext, log,
|
|
accelerators):
|
|
|
|
header = PdbHeaderReader(stream)
|
|
Reader = get_reader(header.ident)
|
|
|
|
if Reader is None:
|
|
raise PDBError('No reader available for format within container.'
|
|
'\n Identity is %s. Book type is %s' %
|
|
(header.ident,
|
|
IDENTITY_TO_NAME.get(header.ident, 'Unknown')))
|
|
|
|
log.debug('Detected ebook format as: %s with identity: %s' %
|
|
(IDENTITY_TO_NAME[header.ident], header.ident))
|
|
|
|
reader = Reader(header, stream, log, options)
|
|
opf = reader.extract_content(os.getcwd())
|
|
|
|
return opf
|