def get_tmp_dir(name): if 'REAL_TMP' in os.environ: # Set in tests/__init__.py dir = LocalFolder(os.environ['REAL_TMP']) else: dir = LocalFolder(tempfile.gettempdir()) #~ print("TMPDIR:", dir) dir = dir.folder('test_versioncontrol').folder(name) if dir.exists(): dir.remove_children() dir.remove() assert not dir.exists() dir.touch() return dir
def export_iter(self, pages): basename = encode_filename(pages.name) folder = LocalFolder(get_tmpdir().subdir('mhtml_export_tmp_dir').path) # XXX if folder.exists(): folder.remove_children() else: folder.touch() file = folder.file(basename + '.html') layout = SingleFileLayout(file, pages.prefix) exporter = SingleFileExporter(layout, self.template, 'html', document_root_url=self.document_root_url) for p in exporter.export_iter(pages): yield p encoder = MHTMLEncoder() linker = ExportLinker(pages.notebook, layout, output=file, usebase=True) self.file.write(encoder(layout, linker))