analyse_nb.py

import argparse
import sys, os

import Osiris

parser = argparse.ArgumentParser()
parser.add_argument('-n', '--notebook-name', type=str, required=True)
parser.add_argument('-e', '--execute', type=str, required=True)
parser.add_argument('-v', '--verbose', action='store_true', default=True)
parser.add_argument('-m', '--match-pattern', type=str, default=None)
parser.add_argument('-s', '--self-reproduce', action='store_true', default=False)
parser.add_argument('-a', '--all', action='store_true', default=False)
parser.add_argument('-d', '--debug', type=int, default=None)
args = parser.parse_args()

# Parameters (required)
path = args.notebook_name
execute = args.execute
assert execute in ['normal', 'OEC', 'dependency']

# Parameters (optional)
verbose = args.verbose
match_pattern = args.match_pattern
self_reproduce = args.self_reproduce
debug = args.debug
analyse_all_dependency = args.all
if match_pattern is not None:
    match_pattern = match_pattern.lstrip()
    assert match_pattern in ['strong', 'weak', 'best_effort']

root_path = os.getcwd()

def analyse_nb(path, execute, verbose, match_pattern, self_reproduce, debug, analyse_all_dependency):
    interface = Osiris.UserInterface(path, execute, verbose, analyse_all_dependency)
    is_executable = interface.analyse_executability()

    if is_executable:
        # reproducibility 
        if match_pattern is not None:
            os.chdir(root_path)
            interface.analyse_reproducibility(match_pattern)

        # self-reproducibility 
        if self_reproduce:
            os.chdir(root_path)
            interface.analyse_self_reproducibility()

        # debug 
        if debug is not None:
            os.chdir(root_path)
            interface.analyse_status_difference_for_a_cell(debug)

analyse_nb(path, execute, verbose, match_pattern, self_reproduce, debug, analyse_all_dependency)