class Engine: def __init__(self, pdf_path, debug=False): self.pdf_path = pdf_path if False is debug: self.pdf = KindlePDF() else: self.pdf = TextOutput() def parse(self, path): files = self.get_files(path) for fpath in files: plain = Plain(fpath) plain.parse(self.pdf) self.create_pdf() def set_cover(self, cover_path): self.pdf.set_cover(cover_path) def enable_bookmark(self): self.pdf.create_table_of_contents() def get_files(self, path): if os.path.isdir(path): files = self.parse_dir_files(path) elif os.path.isfile(path): files = [path] return files def parse_dir_files(self, path): fdir = Directory() return fdir.get_files(path) def create_pdf(self): self.pdf.build(self.pdf_path)
def __init__(self, pdf_path, debug=False): self.pdf_path = pdf_path if False is debug: self.pdf = KindlePDF() else: self.pdf = TextOutput()