def __init__(self, trans_id=None, nb_path=None): if not trans_id: trans_id = random_string(size=10) self.log = create_adapter(logging.getLogger(__name__), trans_id, nb_path) self.trans_id = trans_id self.nb_path = nb_path
import nbformat from tabulate import tabulate from kale.core import Kale from kale.nbparser import parser from kale.static_analysis import ast from kale.utils import pod_utils, kfp_utils from kale.marshal import resource_load from kale.rpc.log import create_adapter from kale.rpc.errors import RPCInternalError KALE_MARSHAL_DIR_POSTFIX = ".kale.marshal.dir" KALE_PIPELINE_STEP_ENV = "KALE_PIPELINE_STEP" logger = create_adapter(logging.getLogger(__name__)) def resume_notebook_path(request): p = os.environ.get("KALE_NOTEBOOK_PATH") if p and not os.path.isfile(p): raise RuntimeError("env path KALE_NOTEBOOK_PATH=%s is not a file" % p) if not p: return None home = os.environ.get("HOME") if not home.endswith('/'): home = home + '/' # JupyterLab needs a relative path to open a file # JP should always run form the HOME directory, so we strip the