def open_notebook(arg, _user_info_db=_user_info_db, scene_root=_scene_root, history_db=_history_db, library_db=_library_db, model_db=_model_db, azure_data_studio_location=_azure_data_studio_location, current_scene_db=None, test=False): """handles opening a notebook""" from src.praxxis.notebook import open_notebook from src.praxxis.util import roots from src.praxxis.sqlite import sqlite_user default_editor = sqlite_user.get_user_info(_user_info_db)[6][1] editor = input_editor = arg.viewer if input_editor == None: editor = default_editor if current_scene_db is None: current_scene_db = roots.get_current_scene_db(scene_root, history_db) open_notebook.open_notebook(arg, current_scene_db, library_db, azure_data_studio_location, editor, True)
def set_param(arg, scene_root=_scene_root, history_db=_history_db, current_scene_db=None): """handles setting a parameter""" from src.praxxis.parameter import set_param from src.praxxis.util import roots if current_scene_db is None: current_scene_db = roots.get_current_scene_db(scene_root, history_db) set_param.set_param(arg, scene_root, history_db, current_scene_db)
def end_scene(arg, scene_root=_scene_root, history_db=_history_db, current_scene_db=None): """handles ending a scene""" from src.praxxis.scene import end_scene from src.praxxis.util import roots if current_scene_db is None: current_scene_db = roots.get_current_scene_db(scene_root, history_db) end_scene.end_scene(arg, scene_root, history_db, current_scene_db)
def view_notebook_param(arg, scene_root=_scene_root, library_db=_library_db, history_db=_history_db, current_scene_db=None): """handles viewing notebook parameters""" from src.praxxis.parameter import list_param from src.praxxis.util import roots if current_scene_db is None: current_scene_db = roots.get_current_scene_db(scene_root, history_db) list_param.list_notebook_param(arg, library_db, current_scene_db)
def remove_notebook(arg, scene_root=_scene_root, history_db=_history_db, library_db=_library_db, current_scene_db=None): """handles removing a notebook from the CLI""" from src.praxxis.notebook import remove_notebook from src.praxxis.util import roots if current_scene_db is None: current_scene_db = roots.get_current_scene_db(scene_root, history_db) remove_notebook.remove_notebook(arg, library_db, current_scene_db)
def add_output(arg, output_root=_output_root, scene_root=_scene_root, history_db=_history_db, current_scene_db=None, user_info_db=_user_info_db): """handles adding output to a notebook""" from src.praxxis.notebook import add_output from src.praxxis.util import roots if current_scene_db is None: current_scene_db = roots.get_current_scene_db(scene_root, history_db) add_output.add_output(arg, output_root, current_scene_db, user_info_db)
def pull_notebook_param(arg, library_db=_library_db, scene_root=_scene_root, history_db=_history_db, current_scene_db=None): """handles pulling parameters out of a notebook""" from src.praxxis.parameter import pull_param from src.praxxis.util import roots if current_scene_db is None: current_scene_db = roots.get_current_scene_db(scene_root, history_db) pull_param.pull_notebook_parameter(arg, library_db, current_scene_db)
def history(arg, scene_root=_scene_root, history_db=_history_db, library_db=_library_db, current_scene_db=None): """handles displaying scene history""" from src.praxxis.scene import history from src.praxxis.util import roots if current_scene_db is None: current_scene_db = roots.get_current_scene_db(scene_root, history_db) history.history(history_db, library_db, current_scene_db)
def list_param(arg, scene_root=_scene_root, history_db=_history_db, query_start=_query_start, query_end=_query_end, current_scene_db=None): """handles listing parameters in current scene""" from src.praxxis.parameter import list_param from src.praxxis.util import roots if current_scene_db is None: current_scene_db = roots.get_current_scene_db(scene_root, history_db) list_param.list_param(current_scene_db, query_start, query_end)
def search_notebook(arg, scene_root=_scene_root, history_db=_history_db, library_db=_library_db, query_start=_query_start, query_end=_query_end, current_scene_db=None): """handles searching for a notebook""" from src.praxxis.notebook import search_notebook from src.praxxis.util import roots if current_scene_db is None: current_scene_db = roots.get_current_scene_db(scene_root, history_db) search_notebook.search_notebook(arg, library_db, current_scene_db, query_start, query_end)
def view_library_param(arg, scene_root=_scene_root, history_db=_history_db, library_db=_library_db, query_start=_query_start, query_end=_query_end, current_scene_db=None): """handles viewing library parameters""" from src.praxxis.parameter import list_param from src.praxxis.util import roots if current_scene_db is None: current_scene_db = roots.get_current_scene_db(scene_root, history_db) list_param.list_library_param(arg, library_db, current_scene_db, query_start, query_end)
def list_notebook(arg, scene_root=_scene_root, history_db=_history_db, library_root=_library_root, library_db=_library_db, query_start=_query_start, query_end=_query_end, current_scene_db=None): """calls the function to list notebooks""" from src.praxxis.notebook import list_notebook from src.praxxis.util import roots if current_scene_db is None: current_scene_db = roots.get_current_scene_db(scene_root, history_db) list_notebook.list_notebook(library_db, current_scene_db, query_start, query_end)
def pull_library_param(arg, library_db=_library_db, scene_root=_scene_root, history_db=_history_db, current_scene_db=None, query_start=_query_start, query_end=_query_end): """handles pulling parameters out of a library""" from src.praxxis.parameter import pull_param from src.praxxis.util import roots if current_scene_db is None: current_scene_db = roots.get_current_scene_db(scene_root, history_db) pull_param.pull_library_parameter(arg, library_db, current_scene_db, query_start, query_end)
def add_rule_to_ruleset(arg, rulesengine_db=_rulesengine_db, library_db=_library_db, current_scene_db=None, scene_root=_scene_root, history_db=_history_db, query_start=_query_start, query_end=_query_end): """handles adding a rule to a ruleset""" from src.praxxis.util import roots if current_scene_db is None: current_scene_db = roots.get_current_scene_db(scene_root, history_db) from src.praxxis.rulesengine import add_rule_to_ruleset add_rule_to_ruleset.add_rule_to_ruleset(arg, rulesengine_db, library_db, current_scene_db, query_start, query_end)
def next_notebook(arg, user_info_db=_user_info_db, scene_root=_scene_root, history_db=_history_db, current_scene_db=None, library_db=_library_db, rulesengine_db=_rulesengine_db, query_start=_query_start, query_end=_query_end): """handles getting the predicted next notebook""" from src.praxxis.notebook import what_next from src.praxxis.util import roots if current_scene_db is None: current_scene_db = roots.get_current_scene_db(scene_root, history_db) what_next.what_next(arg, user_info_db, current_scene_db, library_db, rulesengine_db, query_start, query_end)
def open_notebook(arg, scene_root=_scene_root, history_db=_history_db, library_db=_library_db, azure_data_studio_location=_azure_data_studio_location, current_scene_db=None, test=False): """handles opening a notebook""" from src.praxxis.notebook import open_notebook from src.praxxis.util import roots # TODO: allow for selecting your own editor on first run editor = "vim" if current_scene_db is None: current_scene_db = roots.get_current_scene_db(scene_root, history_db) open_notebook.open_notebook(arg, current_scene_db, library_db, azure_data_studio_location, editor, True)
def run_notebook(arg, user_info_db=_user_info_db, output_root=_output_root, library_root=_library_root, library_db=_library_db, scene_root=_scene_root, history_db=_history_db, current_scene_db=None, query_start=_query_start, query_end=_query_end): """handles running a notebook""" from src.praxxis.notebook import run_notebook from src.praxxis.util import roots if current_scene_db is None: current_scene_db = roots.get_current_scene_db(scene_root, history_db) run_notebook.run_notebook(arg, user_info_db, output_root, current_scene_db, library_root, library_db, query_start, query_end)
def edit_ruleset(arg, rulesengine_db=_rulesengine_db, library_db=_library_db, current_scene_db=None, scene_root=_scene_root, history_db=_history_db, query_start=_query_start, query_end=_query_end): """handles editing a ruleset""" from src.praxxis.util import roots if current_scene_db is None: current_scene_db = roots.get_current_scene_db(scene_root, history_db) if arg.action == 'a': add_rule_to_ruleset(arg, rulesengine_db) elif arg.action == 'd': delete_rule_from_ruleset(arg, rulesengine_db) elif arg.action == 'm': modify_rule_in_ruleset(arg, rulesengine_db) return
def setup(init_root, library_root, telemetry_db, library_db, output_root, scene_root, history_db, default_scene_name, query_start, query_end, rulesengine_root, rulesengine_db): """ sets up directories in the temp dir """ from src.praxxis.sqlite import sqlite_init from src.praxxis.sqlite import sqlite_scene from src.praxxis.sqlite import sqlite_library from src.praxxis.sqlite import sqlite_rulesengine from src.praxxis.util import roots from src.praxxis.scene import new_scene from src.praxxis.scene import list_scene from src.praxxis.parameter import list_param from src.praxxis.library import list_library from src.praxxis.notebook import list_notebook from src.praxxis.sqlite import sqlite_telemetry if not os.path.exists(init_root): os.mkdir(init_root) assert os.path.exists(init_root) if not os.path.exists(library_root): os.mkdir(library_root) assert os.path.exists(library_root) if not os.path.exists(library_db): sqlite_init.init_library_db(library_db) assert os.path.exists(library_db) if not os.path.exists(output_root): os.mkdir(output_root) assert os.path.exists(output_root) if not os.path.exists(scene_root): os.mkdir(scene_root) assert os.path.exists(scene_root) if not os.path.exists(history_db): sqlite_init.init_history(history_db, default_scene_name) assert os.path.exists(history_db) if not os.path.exists(telemetry_db): sqlite_init.init_user_info(telemetry_db, 0) if not os.path.exists(rulesengine_root): os.mkdir(rulesengine_root) assert os.path.exists(rulesengine_root) if not os.path.exists(rulesengine_db): sqlite_init.init_rulesengine_db(rulesengine_db) new_scene.new_scene(default_scene_name, scene_root, history_db) yield current_scene_db = roots.get_current_scene_db(scene_root, history_db) assert len(list_scene.list_scene(init_root, history_db)) == 1 assert len(list_param.list_param(current_scene_db, query_start, query_end)) == 0 assert len(list_library.list_library(library_db, query_start, query_end)) == 0 assert len( list_notebook.list_notebook(library_db, current_scene_db, query_start, query_end)) == 0
def current_scene_db(init_root, scene_root, history_db): from src.praxxis.util import roots return roots.get_current_scene_db(scene_root, history_db)
def test_get_current_scene_db(setup, scene_root, history_db): import os current_scene_db = roots.get_current_scene_db(scene_root, history_db) assert os.path.basename(current_scene_db) == "scene.db"