Esempio n. 1
0
def export_notebook(nb, name, templating="test.tplx", debug=False):
    shutil.copyfile(pkg_resources.resource_filename(__name__, templating),
                    "test.tplx")
    pdf_exporter = PDFExporter()
    pdf_exporter.template_file = "test.tplx"
    print("Attempting to compile LaTeX")
    try:
        pdf_output = pdf_exporter.from_notebook_node(nb)
        with open("%s.pdf" % name, "wb") as output_file:
            output_file.write(pdf_output[0])
            print("Finished generating PDF")
    except nbconvert.pdf.LatexFailed as error:
        print("There was an error generating your LaTeX")
        output = error.output
        if debug:
            print("Showing full error message from PDFTex")

        else:
            print("Showing concise error message")
            output = "\n".join(error.output.split("\n")[-15:])
        print("=" * 30)
        print(output)
        print("=" * 30)
        return None
    return "%s.pdf" % name
Esempio n. 2
0
def has_error(nb):
    pdf_exporter = PDFExporter()
    try:
        pdf_exporter.from_notebook_node(nb)
        return None
    except nbconvert.pdf.LatexFailed as error:
        return "\n".join(error.output.split("\n")[-15:])
Esempio n. 3
0
def ipython_to_pdf(raw_executed_ipynb: str, report_title: str) -> AnyStr:
    pdf_exporter = PDFExporter(Config())
    resources = ResourcesDict()
    resources["metadata"] = ResourcesDict()
    resources["metadata"]["name"] = report_title
    pdf, _ = pdf_exporter.from_notebook_node(nbformat.reads(
        raw_executed_ipynb, as_version=nbformat.v4.nbformat),
                                             resources=resources)
    return pdf
Esempio n. 4
0
    def __init__(self, builder):
        self.pdfdir = builder.outdir + "/pdf"  #pdf directory
        self.texdir = builder.outdir + "/executed"  #latex directory
        self.texbookdir = builder.outdir + "/texbook"  # tex files for book pdf

        for path in [self.pdfdir, self.texdir]:
            ensuredir(path)

        self.pdf_exporter = PDFExporter()
        self.tex_exporter = LatexExporter()
        self.index_book = builder.config['tojupyter_pdf_book_index']
Esempio n. 5
0
    def to_pdf(self):
        pdf = PDFExporter(config={
            "NbConvertBase": {
                "display_data_priority": self.DATA_DISPLAY_PRIORITY
            }
        })

        resources = {"metadata": {"path": str(self.nb_home)}}
        pdf_data, _ = pdf.from_notebook_node(self.nb, resources)
        fn = self.path.with_suffix(".pdf")
        with open(fn, mode="wb") as f:
            f.write(pdf_data)
Esempio n. 6
0
    def _create_report(self, report_kernel_name: str, report_export_html: bool,
                       report_export_pdf: bool):
        """Creates report from notebook-template and stores it in different formats all figures.

            - Jupyter Notebook
            - HTML
            - PDF
        """
        assert not self.mode_ipython, 'Create report is only possible when not in ipython mode'

        filepath_template = dirname(
            imageatm.notebooks.__file__) + '/evaluation_template.ipynb'
        filepath_notebook = self.evaluation_dir / 'evaluation_report.ipynb'
        filepath_html = self.evaluation_dir / 'evaluation_report.html'
        filepath_pdf = self.evaluation_dir / 'evaluation_report.pdf'

        pm.execute_notebook(str(filepath_template),
                            str(filepath_notebook),
                            parameters=dict(image_dir=str(self.image_dir),
                                            job_dir=str(self.job_dir)),
                            kernel_name=report_kernel_name)

        with open(filepath_notebook) as f:
            nb = nbformat.read(f, as_version=4)

        if report_export_html:
            self.logger.info('\n****** Create HTML ******\n')
            with open(filepath_notebook) as f:
                nb = nbformat.read(f, as_version=4)

            html_exporter = HTMLExporter()
            html_data, resources = html_exporter.from_notebook_node(nb)

            with open(filepath_html, 'w') as f:
                f.write(html_data)
                f.close()

        if report_export_pdf:
            self.logger.info('\n****** Create PDF ******\n')

            pdf_exporter = PDFExporter()
            pdf_exporter.template_file = dirname(
                imageatm.notebooks.__file__
            ) + '/tex_templates/evaluation_report.tplx'
            pdf_data, resources = pdf_exporter.from_notebook_node(
                nb, resources={'metadata': {
                    'name': 'Evaluation Report'
                }})

            with open(filepath_pdf, 'wb') as f:
                f.write(pdf_data)
                f.close()
    def __init__(self, output):
        assert output in SUPPORTED_FORMATS, f"supported formats are {SUPPORTED_FORMATS}"
        self.read_encoding = "utf-8"
        self.write_encoding = "utf-8"
        self.format = output

        if self.format == "pdf":
            pdf = PDFExporter()
            pdf.exclude_output_prompt = True
            pdf.exclude_input = True
            self.exporter = pdf
        elif self.format == "rst":
            self.exporter = RSTExporter()
        else:
            self.exporter = MarkdownExporter()
Esempio n. 8
0
def pdf_convertor(in_name, out_name, current_dir, new_dir):
    #notebook_filename = "telstra-exploring-data.ipynb"
    with open(in_name + ".ipynb") as f:
        print 'jupyter notebook is taken from dir:', current_dir
        nb = nbformat.read(f, as_version=4)

    ep = ExecutePreprocessor(timeout=600, kernel_name='python2')  #python3
    ep.preprocess(nb, {'metadata': {'path': new_dir}})
    pdf_exporter = PDFExporter()
    pdf_data, resources = pdf_exporter.from_notebook_node(nb)

    with open(out_name + ".pdf", "wb") as f:
        print 'pdf report is saved in dir:', new_dir
        f.write(pdf_data)
        f.close()
Esempio n. 9
0
def ipython_to_pdf(raw_executed_ipynb: str,
                   report_title: str,
                   hide_code: bool = False) -> AnyStr:
    c = Config()
    c.PDFExporter.exclude_input = hide_code
    c.PDFExporter.exclude_output_prompt = hide_code
    c.HTMLExporter.template_file = pkg_resources.resource_filename(
        __name__, "../nbtemplates/notebooker_pdf_output.tplx")
    pdf_exporter = PDFExporter(c)
    resources = ResourcesDict()
    resources["metadata"] = ResourcesDict()
    resources["metadata"]["name"] = report_title
    pdf, _ = pdf_exporter.from_notebook_node(nbformat.reads(
        raw_executed_ipynb, as_version=nbformat.v4.nbformat),
                                             resources=resources)
    return pdf
Esempio n. 10
0
def make_pdf(out, fname):
    """Make a PDF out of an ipynb."""
    # Now output the notebook to pdf
    if report:

        c = Config()
        c.TemplateExporter.exclude_input_prompt = True
        c.TemplateExporter.exclude_output_prompt = True
        c.TemplateExporter.exclude_input = True

        exporter = PDFExporter(config=c)
        body, resources = exporter.from_filename(out / fname)
        with open(out / fname.with_suffix(".pdf"), "wb") as fl:
            fl.write(body)

        console.print(f"Saved PDF to '{out / fname.with_suffix('.pdf')}'")
Esempio n. 11
0
    def to_pdf(self):
        if self.first:
            td = TemporaryDirectory()
            preprocessors = self.get_preprocessors('pdf', td=td)
            self.preprocess(preprocessors)

        pdf = PDFExporter(config={
            'NbConvertBase': {
                'display_data_priority': self.DISPLAY_DATA_PRIORITY
            }
        })

        pdf_data, self.resources = pdf.from_notebook_node(
            self.nb, self.resources)
        fn = self.final_nb_home / (self.document_name + '.pdf')
        with open(fn, mode='wb') as f:
            f.write(pdf_data)
Esempio n. 12
0
def to_pdf(nb: NotebookNode, kernel: str, fpath: Path) -> Tuple[Any, Dict]:
    """
    Compile final notebook into a single PDF while executing it.

    :param nb: The compiled notebook object with all notebook cells.
    :param kernel: String name of the kernel to output.
    :param fpath: The path to write hte notebook to.
    """
    ep = ExecutePreprocessor(timeout=600, kernel_name=kernel)
    ep.preprocess(nb)

    strip_execution_count(nb)
    pdf_exporter = PDFExporter()
    body, resources = pdf_exporter.from_notebook_node(nb)

    with open(fpath, "wb") as f:
        f.write(body)
Esempio n. 13
0
    def to_pdf_latex(self):
        if 'outputs' in self.resources:
            # remove outputs from MarkdownExporter if used
            self.resources.pop('outputs')

        # must download html images for latex to use them
        from ._preprocessors import MarkdownHTTPPreprocessor
        temp_dir = Path(self.td.name)
        self.resources['temp_dir'] = temp_dir
        MarkdownHTTPPreprocessor().preprocess(self.nb, self.resources)

        for filename, image_data in self.resources['image_data_dict'].items():
            fn_pieces = filename.split('_')
            cell_idx = int(fn_pieces[1])
            ext = fn_pieces[-1].split('.')[-1]
            new_filename = str(temp_dir / filename)

            # extract first image from gif and use as png for latex pdf
            if ext == 'gif':
                buffer = io.BytesIO(image_data)
                arr = mimage.imread(buffer, format='gif')
                new_filename = filename.split('.')[0] + '.png'
                new_filename = str(temp_dir / new_filename)
                mimage.imsave(new_filename, arr)
            else:
                with open(new_filename, 'wb') as f:
                    f.write(image_data)

            cell = self.nb.cells[cell_idx]
            cell['source'] = cell['source'].replace(filename, new_filename)

        pdf = PDFExporter(config={
            'NbConvertBase': {
                'display_data_priority': self.DISPLAY_DATA_PRIORITY
            }
        })
        pdf_data, self.resources = pdf.from_notebook_node(
            self.nb, self.resources)
        self.return_data['pdf_data'] = pdf_data
        if not self.web_app:
            fn = self.final_nb_home / (self.document_name + '.pdf')
            with open(fn, mode='wb') as f:
                f.write(pdf_data)
Esempio n. 14
0
    def __init__(self, output_format, destination_mode):
        assert output_format in SUPPORTED_FORMATS, f"supported formats are {SUPPORTED_FORMATS}"
        assert (
            destination_mode in SUPPORTED_DESTINATIONS_MODES
        ), f"supported destination modes are {SUPPORTED_DESTINATIONS_MODES}"

        self.read_encoding = "utf-8"
        self.write_encoding = "utf-8"
        self.format = output_format
        self.destination_mode = destination_mode

        if self.format == "pdf":
            pdf = PDFExporter()
            pdf.exclude_output_prompt = True
            pdf.exclude_input = True
            self.exporter = pdf
        elif self.format == "rst":
            self.exporter = RSTExporter()
        else:
            self.exporter = MarkdownExporter()
Esempio n. 15
0
def convert_notebook(report_in_path, report_out_path, **kwargs):

    curdir = os.path.abspath(os.getcwd())
    indir, _ = os.path.split(report_in_path)
    outdir, _ = os.path.split(report_out_path)
    os.makedirs(outdir, exist_ok=True)

    config = {
        "ExecutePreprocessor": {
            "enabled": True,
            "timeout": -1
        },
        "TemplateExporter": {
            "exclude_output_prompt": True,
            "exclude_input": True,
            "exclude_input_prompt": True
        },
    }

    nb = nbformat.read(open(report_in_path), as_version=4)
    html_exporter = HTMLExporter(config=config)

    # no exectute for PDFs
    config["ExecutePreprocessor"]["enabled"] = False
    pdf_exporter = PDFExporter(config=config)

    # change dir to notebook dir, to execute notebook
    os.chdir(indir)

    body, resources = (html_exporter.from_notebook_node(nb))

    pdf_body, pdf_resources = (pdf_exporter.from_notebook_node(nb))

    # change back to original directory
    os.chdir(curdir)

    with open(report_out_path.replace(".pdf", ".html"), 'w') as fh:
        fh.write(body)

    with open(report_out_path.replace(".html", ".pdf"), 'wb') as fh:
        fh.write(pdf_body)
Esempio n. 16
0
def export_notebook(nb, pdf_path, template="test.tplx", debug=False):
    """Write notebook as PDF to pdf_path. Return True on success or False on error."""
    shutil.copyfile(pkg_resources.resource_filename(__name__, template), "test.tplx")
    pdf_exporter = PDFExporter()
    pdf_exporter.template_file = "test.tplx"
    try:
        pdf_output = pdf_exporter.from_notebook_node(nb)
        with open(pdf_path, "wb") as out:
            out.write(pdf_output[0])
            print("Saved", pdf_path)
        return True
    except nbconvert.pdf.LatexFailed as error:
        print("There was an error generating your LaTeX")
        output = error.output
        if not debug:
            print("To see the full error message, run with debug=True")
            output = "\n".join(error.output.split("\n")[-15:])
        print("=" * 30)
        print(output)
        print("=" * 30)
        return False
Esempio n. 17
0
def save_notebook(notebook_filename):
    now = str(datetime.datetime.now()).replace(" ", "").replace(".",
                                                                "-").replace(
                                                                    ":", "-")
    with open(notebook_filename) as f:
        nb = nbformat.read(f, as_version=4)

    # ep = nbconvert.ExecutePreprocessor(timeout=600, kernel_name='python3')

    # ep.preprocess(nb, {'metadata': {'path': './'}})

    pdf_exporter = PDFExporter()

    pdf_data, resources = pdf_exporter.from_notebook_node(nb)
    file_result_name = "results/" + notebook_filename.replace(".ipynb",
                                                              now) + ".pdf"

    with open(file_result_name, "wb") as f:
        f.write(pdf_data)
        f.close()
    return file_result_name
Esempio n. 18
0
def run_notebook_convert(dir_experiment_run: Path = None) -> Path:
    """
    Run and convert the notebook to html and pdf.
    """
    # Copy the experiment_vis jupyter notebook to the experiment dir
    notebook_path = Path(__file__).parent.parent / 'experiment_vis/experiment_vis.ipynb'
    dest_notebook_path = dir_experiment_run / 'experiment_vis.ipynb'

    # copy notebook to experiment run
    shutil.copyfile(notebook_path, dest_notebook_path)

    log.info(f'Executing experiment vis notebook {dest_notebook_path}.')
    with open(dest_notebook_path) as f:
        nb = nbformat.read(f, as_version=4)
    ep = ExecutePreprocessor(timeout=600, kernel_name='python3')
    ep.preprocess(nb, {'metadata': {'path': str(dest_notebook_path.parent)}})

    nbconvert_path = dest_notebook_path.with_suffix('.nbconvert.ipynb')

    with open(nbconvert_path, 'w', encoding='utf-8') as f:
        nbformat.write(nb, f)

    log.info('Converting notebook to html.')
    html_path = nbconvert_path.with_suffix('.html')
    html_exporter = HTMLExporter()
    html_exporter.template_name = 'classic'
    (body, resources) = html_exporter.from_notebook_node(nb)
    with open(html_path, 'w') as f:
        f.write(body)

    log.info('Converting notebook to pdf.')
    pdf_path = nbconvert_path.with_suffix('.pdf')
    pdf_exporter = PDFExporter()
    pdf_exporter.template_name = 'classic'
    (body, resources) = pdf_exporter.from_notebook_node(nb)
    pdf_path.write_bytes(body)

    return pdf_path
Esempio n. 19
0
def pdf_from_latex(pdf=None, verbose=False):
    '''Export the notebook to PDF via LaTeX.
    This is not fast because you have to install texlive.
    verbose is not used right now.
    '''

    if not shutil.which('xelatex'):
        aptinstall('texlive-xetex')

    fname, fid = current_notebook()
    ipynb = notebook_string(fid)

    exporter = PDFExporter()

    nb = nbformat.reads(ipynb, as_version=4)
    body, resources = exporter.from_notebook_node(nb)

    if pdf is None:
        pdf = fname.replace(".ipynb", ".pdf")

    tmpdirname = tempfile.TemporaryDirectory().name

    if not os.path.isdir(tmpdirname):
        os.mkdir(tmpdirname)

    apdf = os.path.join(tmpdirname, pdf)

    if os.path.exists(apdf):
        os.unlink(apdf)

    with open(apdf, 'wb') as f:
        f.write(body)

    if os.path.exists(apdf):
        files.download(apdf)
    else:
        print(f'{apdf} not found')
Esempio n. 20
0
 def to_pdf(self):
     return PDFExporter().from_notebook_node(self._content)[0]
Esempio n. 21
0
import sys, os
import nbformat
from nbconvert.preprocessors import ExecutePreprocessor
from nbconvert import PDFExporter
dir = os.path.split(os.getcwd())[0]
if dir not in sys.path:
    sys.path.append(dir)

notebook_filename = "C:\Users\Usuario\Desktop\Universidad\9 semestre\Conmutacionpython\Notebook\central.ipynb"

with open(notebook_filename) as f:
    nb = nbformat.read(f, as_version=4)

ep = ExecutePreprocessor(timeout=600, kernel_name='python2')

ep.preprocess(
    nb, {
        'metadata': {
            'path':
            'C:\Users\Usuario\Desktop\Universidad\9 semestre\Conmutacionpython\Notebook/'
        }
    })

pdf_exporter = PDFExporter()

pdf_data, resources = pdf_exporter.from_notebook_node(nb)

with open("notebook.pdf", "wb") as f:
    f.write(pdf_data)
    f.close()
Esempio n. 22
0
    def get(self, format: str, path: str):
        """Handle the GET method call."""
        if format != 'pdf':
            self.log.exception('format must be pdf')
            raise web.HTTPError(500, 'format must be pdf')

        self.config.PDFExporter.preprocessors = [thermohw.ExtractAttachmentsPreprocessor]
        self.config.PDFExporter.template_file = os.path.join(thermohw_dir, 'homework.tpl')
        self.config.PDFExporter.filters = {'convert_div': thermohw.convert_div,
                                           'convert_raw_html': thermohw.convert_raw_html}
        self.config.PDFExporter.latex_count = 1

        exporter = PDFExporter(config=self.config, log=self.log)
        exporter.writer.build_directory = '.'

        pdfs = []

        path = path.strip('/').strip()
        paths = path.split('.ipynb')

        for path in paths:
            if not path:
                continue
            path += '.ipynb'
            # If the notebook relates to a real file (default contents manager),
            # give its path to nbconvert.
            ext_resources_dir: Union[str, None]
            basename: str
            os_path: str
            if hasattr(self.contents_manager, '_get_os_path'):
                os_path = self.contents_manager._get_os_path(path)
                ext_resources_dir, basename = os.path.split(os_path)
            else:
                ext_resources_dir = None

            model: Dict[str, str] = self.contents_manager.get(path=path)
            name: str = model['name']
            if model['type'] != 'notebook':
                # not a notebook, redirect to files
                return FilesRedirectHandler.redirect_to_files(self, path)

            nb = model['content']

            self.set_header('Last-Modified', model['last_modified'])

            # create resources dictionary
            mod_date: str = model['last_modified'].strftime(text.date_format)
            nb_title: str = os.path.splitext(name)[0]

            config_dir: str = self.application.settings['config_dir']

            resource_dict: Dict[str, str] = {
                "metadata": {
                    "name": nb_title,
                    "modified_date": mod_date
                },
                "config_dir": config_dir,
            }

            if ext_resources_dir:
                resource_dict['metadata']['path'] = ext_resources_dir

            output: bytes
            try:
                output, _ = exporter.from_notebook_node(
                    nb,
                    resources=resource_dict
                )
            except Exception as e:
                self.log.exception("nbconvert failed: %s", e)
                raise web.HTTPError(500, "nbconvert failed: %s" % e)

            pdfs.append(io.BytesIO(output))

        writer = PdfWriter()
        for pdf in pdfs:
            writer.addpages(PdfReader(pdf).pages)
        bio = io.BytesIO()
        writer.write(bio)
        bio.seek(0)
        output = bio.read()
        bio.close()

        # Force download if requested
        if self.get_argument('download', 'false').lower() == 'true':
            filename = 'final_output.pdf'
            self.set_header('Content-Disposition',
                            'attachment; filename="{}"'.format(filename))

        # MIME type
        if exporter.output_mimetype:
            self.set_header('Content-Type',
                            '{}; charset=utf-8'.format(exporter.output_mimetype))

        self.set_header('Cache-Control', 'no-store, no-cache, must-revalidate, max-age=0')
        self.finish(output)
from nbconvert.preprocessors import ExecutePreprocessor
import datasets
output_folderpath = Path("reports/datasets_analysis")
output_folderpath.mkdir(parents=True, exist_ok=True)

for dataset_name in datasets.datasets_by_name_all:
    print(f"Dataset {dataset_name}")
    print("     Reading notebook...")
    notebook_filename = "Exploratory analysis.ipynb"
    notebook = nbformat.read(notebook_filename, as_version=4)

    cell = notebook.cells[0]["source"] = f"dataset_name = '{dataset_name}'"

    # # 2. Instantiate the exporter. We use the `classic` template for now; we'll get into more details
    # # later about how to customize the exporter further.
    print("     Preprocessing notebook..")
    ep = ExecutePreprocessor(timeout=600, kernel_name='python3')
    ep.preprocess(notebook)

    print("     Exporting to pdf..")
    exporter = PDFExporter()
    #exporter.template_name = 'classic'
    pdf_data, resources = exporter.from_notebook_node(notebook)

    output_filepath = output_folderpath / f"{dataset_name}.pdf"
    print("     Writing..")
    with open(output_filepath, "wb") as f:
        f.write(pdf_data)
        f.close()
    print("     Done")
Esempio n. 24
0
def package(parser, context, args):
    ld = arg_date(parser, args)
    eld_admin = '_private/' + ld
    eld_solutions = '_private/%s/solutions' % ld
    eld_notebook = '%s/exam-%s.ipynb' % (eld_solutions, ld)
    target_student = get_target_student(ld)
    target_student_pdf = '%s/%s' % (get_target_student(ld),
                                    get_exam_text_filename(ld, 'pdf'))
    # no pdf as hiding cells is too boring, have still
    # to properly review cells filtering https://github.com/DavidLeoni/jupman/issues/4
    # target_student_pdf = target_student + '/' + 'exam-' + ld + '.pdf'
    target_student_zip = "%s/server/%s-%s-exam" % (eld_admin, jm.filename, ld
                                                   )  # without '.zip'
    target_server_zip = "%s/%s-%s-server" % (eld_admin, jm.filename, ld
                                             )  # without '.zip'

    if not os.path.exists(jm.build):
        fatal("%s WAS NOT BUILT !" % jm.build)

    if not os.path.exists(eld_solutions):
        fatal("MISSING SOURCE SOLUTION EXERCISES: " + eld_solutions)

    if os.path.exists(target_student):
        fatal("TARGET STUDENT EXERCISES DIRECTORY ALREADY EXISTS: " +
              target_student)

    try:
        dir_names = os.listdir(jm.build)
    except Exception as e:
        fatal("ERROR WITH DIR %s" % jm.build, ex=e)

    if len(dir_names) == 0:
        fatal("SITE DIRECTORY AT %s WAS NOT BUILT !" % jm.build)

    server_jupman = "%s/server/%s" % (eld_admin, jm.filename)

    if os.path.exists(server_jupman):
        jmt.delete_tree(server_jupman, "_private/%s/server" % ld)

    info("Copying built website ...")
    shutil.copytree(jm.build, server_jupman)

    info("Building pdf ..")
    import nbformat
    import nbconvert
    from nbconvert import PDFExporter

    pdf_exporter = PDFExporter()

    #Dav dic 2019:  couldn't make it work, keeps complaining about missing files
    #pdf_exporter.template_file = '_templates/classic.tplx
    # as a result we have stupid extra date and worse extra numbering in headers

    from nbconvert.preprocessors import ExecutePreprocessor
    with open(eld_notebook) as f:
        nb = nbformat.read(f, as_version=4)
        old_title = jmt._replace_title(nb, eld_notebook, "").strip('#')

        nb.cells = [cell for cell in nb.cells \
                             if not ('nbsphinx' in cell.metadata \
                                     and cell.metadata['nbsphinx'] == 'hidden')]

        (body, resources) = pdf_exporter.from_notebook_node(
            nb, resources={'metadata': {
                'name': old_title
            }})

        if not os.path.exists(target_student):
            os.makedirs(target_student)
        #ep = ExecutePreprocessor(timeout=600, kernel_name='python3')
        #ep.preprocess(nb, {'metadata': {'path': './'}})
        with open(target_student_pdf, 'wb') as pdf_f:
            #print("resources = %s" % resources)
            pdf_f.write(body)

    info("Copying exercises to " + str(target_student))
    jm.copy_code(eld_solutions, target_student, copy_solutions=False)

    info("Creating student exercises zip:  %s.zip" % target_student_zip)

    def mysub(fname):
        if fname.startswith('_private/'):
            return fname[len('_private/YYYY-MM-DD/student-zip/'):]
        else:
            return '/%s/%s' % (jm.get_exam_student_folder(ld), fname)

    jm.zip_paths([target_student] + jm.chapter_files, target_student_zip,
                 mysub)
    #shutil.make_archive(target_student_zip, 'zip', target_student_zip)
    info("Creating server zip: %s.zip" % target_server_zip)
    shutil.make_archive(target_server_zip, 'zip', eld_admin + "/server")
    print("")
    info("You can now browse the website at:  %s" %
         (os.path.abspath(eld_admin + "/server/" + jm.filename +
                          "/html/index.html")))
    print("")
Esempio n. 25
0
def package(parser, context, args):

    parser.add_argument('date', help="date in format 'yyyy-mm-dd'")
    parser.add_argument('-t',
                        '--site',
                        action='store_true',
                        help="zips the site")
    parser.add_argument('-r',
                        '--server',
                        action='store_true',
                        help="zips the server")

    vs = vars(parser.parse_args(args))
    ld = jmt.parse_date_str(vs['date'])

    zip_site = vs['site']
    zip_server = vs['server']

    eld_admin = '_private/' + ld
    eld_solutions = '_private/%s/solutions' % ld
    eld_notebook = '%s/exam-%s.ipynb' % (eld_solutions, ld)
    target_student = get_target_student(ld)
    target_student_pdf = '%s/%s' % (get_target_student(ld),
                                    get_exam_text_filename(ld, 'pdf'))
    target_student_zip = "%s/server/%s-%s-exam" % (eld_admin, jm.filename, ld
                                                   )  # without '.zip'
    target_server_zip = "%s/%s-%s-server" % (eld_admin, jm.filename, ld
                                             )  # without '.zip'

    if not os.path.exists(jm.build):
        fatal("%s WAS NOT BUILT !" % jm.build)

    if not os.path.exists(eld_solutions):
        fatal("MISSING SOURCE SOLUTION EXERCISES: " + eld_solutions)

    if os.path.exists(target_student):
        fatal("TARGET STUDENT EXERCISES DIRECTORY ALREADY EXISTS: " +
              target_student)

    try:
        dir_names = os.listdir(jm.build)
    except Exception as e:
        fatal("ERROR WITH DIR %s" % jm.build, ex=e)

    if len(dir_names) == 0:
        fatal("SITE DIRECTORY AT %s WAS NOT BUILT !" % jm.build)

    server_jupman = "%s/server/%s" % (eld_admin, jm.filename)

    if os.path.exists(server_jupman):
        jmt.delete_tree(server_jupman, "_private/%s/server" % ld)

    if zip_site:
        info("Copying built website ...")
        shutil.copytree(jm.build, server_jupman)

    if not os.path.exists(target_student):
        os.makedirs(target_student)

    info("Copying exercises to " + str(target_student))
    jm.copy_code(eld_solutions, target_student, copy_solutions=False)

    info("Building pdf ..")
    import nbformat
    import nbconvert
    from nbconvert import PDFExporter
    pdf_exporter = PDFExporter()

    #Dav dic 2019:  couldn't make it work, keeps complaining about missing files
    #pdf_exporter.template_file = '_templates/classic.tplx
    # as a result we have stupid extra date and worse extra numbering in headers

    with open('%s/exam-%s.ipynb' % (target_student, ld)) as f:
        ex_nb = nbformat.read(f, as_version=4)
        old_title = jmt._replace_title(ex_nb, eld_notebook,
                                       "").strip('#').strip(jm.ipynb_exercises)

        ex_nb.cells = [cell for cell in ex_nb.cells \
                             if not ('nbsphinx' in cell.metadata \
                                     and cell.metadata['nbsphinx'] == 'hidden')]

        (body, resources) = pdf_exporter.from_notebook_node(
            ex_nb, resources={'metadata': {
                'name': old_title
            }})

        with open(target_student_pdf, 'wb') as pdf_f:
            pdf_f.write(body)

    info("Creating student exercises zip:  %s.zip" % target_student_zip)

    def remap(fname):
        if fname.startswith('_private/'):
            return fname[len('_private/YYYY-MM-DD/student-zip/'):]
        else:
            return '/%s/%s' % (jm.get_exam_student_folder(ld), fname)

    deglobbed_common_files, deglobbed_common_files_patterns = jm._common_files_maps(
        target_student_zip)

    jm.zip_paths([target_student] + deglobbed_common_files,
                 target_student_zip,
                 patterns=deglobbed_common_files_patterns,
                 remap=remap)

    if zip_server:
        info("Creating server zip: %s.zip" % target_server_zip)
        shutil.make_archive(target_server_zip, 'zip', eld_admin + "/server")

    if zip_server and zip_site:
        print("")
        info("You can now browse the website at:  %s" %
             (os.path.abspath(eld_admin + "/server/" + jm.filename +
                              "/html/index.html")))
        print("")
Esempio n. 26
0
def generate_report_notebook(config, job_type, data_dir, processed_data_dir,
                             genome_dirpath, filenames, report_filename):
    file_loader = FileSystemLoader(os.path.join(ROOT_DIR, 'IDSort/src'))
    env = Environment(loader=file_loader)
    report_template = env.get_template('genome_report_template.ipynb')

    if job_type == 'sort':
        # convert given genomes to h5 files, and convert given inp files to
        # genomes and then to h5 files

        inp_to_genome_config = {
            'analysis': False,
            'readable': False,
            'create_genome': True,
            'id_filename': config['process_genome']['id_filename'],
            'magnets_filename': config['process_genome']['magnets_filename'],
            'id_template': config['process_genome']['id_template']
        }
        genome_to_h5_config = {
            'analysis': True,
            'readable': True,
            'create_genome': False,
            'id_filename': config['process_genome']['id_filename'],
            'magnets_filename': config['process_genome']['magnets_filename'],
            'id_template': config['process_genome']['id_template']
        }

        genome_h5_filepaths = []
        for filename in filenames:
            if filename.endswith('.genome'):
                filepath = os.path.join(genome_dirpath, filename)
                run_process_genome(genome_to_h5_config, filepath,
                                   processed_data_dir)
                genome_h5_filepaths.append(
                    os.path.join(processed_data_dir, filename + '.h5'))
            elif filename.endswith('.inp'):
                filepath = os.path.join(processed_data_dir, filename)
                run_process_genome(inp_to_genome_config, filepath,
                                   genome_dirpath)
                new_genome_filename = filename + '.genome'
                new_genome_filepath = os.path.join(genome_dirpath,
                                                   new_genome_filename)
                run_process_genome(genome_to_h5_config, new_genome_filepath,
                                   processed_data_dir)
                genome_h5_filepaths.append(
                    os.path.join(processed_data_dir,
                                 new_genome_filename + '.h5'))
    elif job_type == 'shim':
        genome_h5_filepaths = [
            os.path.join(genome_dirpath, filename) for filename in filenames
        ]

    report_output = report_template.render(
        job_type=job_type, genome_h5_filepaths=genome_h5_filepaths)

    notebook_name = 'genome_report.ipynb'
    notebook_path = os.path.join(data_dir, notebook_name)

    with open(notebook_path, 'w') as notebook:
        notebook.write(report_output)

    # create dir for genome reports
    genome_report_dir = 'genome_reports/'
    genome_report_dirpath = os.path.join(data_dir, genome_report_dir)
    if not os.path.exists(genome_report_dirpath):
        os.makedirs(genome_report_dirpath)

    # execute notebook
    with open(notebook_path, 'r') as notebook:
        nb = nbformat.read(notebook, as_version=4)

    ep = ExecutePreprocessor()
    ep.preprocess(nb, {'metadata': {'path': data_dir}})
    pdf_exporter = PDFExporter()
    pdf_exporter.exclude_output_prompt = True
    pdf_exporter.exclude_input = True
    pdf_data, resources = pdf_exporter.from_notebook_node(nb)

    # check the name of the report to see if a specific name has been given or
    # not
    if report_filename == 'genome_report.pdf':
        # concatenate all genome/inp files using "_" as a separator to form the
        # filename
        report_filename = '_'.join(filenames) + '.pdf'

    report_filepath = os.path.join(genome_report_dirpath, report_filename)
    with open(report_filepath, 'wb') as report:
        report.write(pdf_data)