def parse_code(model_code: str):
root_ns = Namespace("__root__")
main_ns = Namespace("__config__")
main_ns.parent = root_ns
statements = parse(main_ns, "test", model_code)
return statements
def _load_plugins(self, plugin_dir, namespace):
"""
Load all modules in plugin_dir
"""
if not os.path.exists(os.path.join(plugin_dir, "__init__.py")):
raise Exception(
"The plugin directory %s should be a valid python package with a __init__.py file"
% plugin_dir)
mod_name = ".".join(namespace.to_path())
imp.load_package(mod_name, plugin_dir)
for py_file in glob.glob(os.path.join(plugin_dir, "*.py")):
if not py_file.endswith("__init__.py"):
# name of the python module
sub_mod = mod_name + "." + os.path.basename(py_file).split(
".")[0]
# create a namespace for the submodule
new_ns = Namespace(sub_mod.split(".")[-1])
new_ns.parent = namespace
self.graph.add_namespace(new_ns, namespace)
# load the python file
imp.load_source(sub_mod, py_file)