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))