+from setuptools.command.install import install
+
+
+class Install_osm_im(install):
+ """Generation of .py files from yang models"""
+ model_dir = "models/yang"
+ im_dir = "osm_im"
+
+ def pipinstall(self, package):
+ """pip install for executable dependencies"""
+ subprocess.call([sys.executable, "-m", "pip", "install", package])
+
+ def run(self):
+ self.pipinstall('pyang')
+ self.pipinstall('pyangbind')
+ import pyangbind
+ print("Creating dir {}/{} for python artifacts".format(os.getcwd(), self.im_dir))
+ path = "{}/{}".format(os.getcwd(), self.im_dir)
+ if not os.path.exists(path):
+ os.makedirs(path)
+ open("{}/{}/__init__.py".format(os.getcwd(), self.im_dir), 'a').close()
+ for files_item in ['vnfd', 'nsd', 'nst']:
+ protoc_command = ["pyang",
+ "-Werror",
+ "--plugindir",
+ "{}/plugin".format(os.path.dirname(pyangbind.__file__)),
+ "--path",
+ self.model_dir,
+ "-f", "pybind",
+ "-o",
+ "{}/{}.py".format(self.im_dir, files_item),
+ "{}/{}.yang".format(self.model_dir, files_item)]
+ print("Generating {}.py from {}.yang".format(files_item, files_item))
+ if subprocess.call(protoc_command) != 0:
+ sys.exit(-1)
+ # To ensure generated files are copied to the python installation folder
+ self.copy_tree(self.im_dir, "{}{}".format(self.install_lib, self.im_dir))
+ install.run(self)
+