def _get_api_pkgs(self, resolved_model_dir): """ Return api packages for resolved YANG modules. Each module will be represented as an api package. Args: resolved_model_dir (str): Path to resolved YANG modules. Returns: api_pkgs (List[.api_model.Package]): List of api packages. """ pyang_builder = PyangModelBuilder(resolved_model_dir) modules = pyang_builder.parse_and_return_modules() # build api model packages if not self.groupings_as_class: api_pkgs = ApiModelBuilder(self.iskeyword, self.language).generate(modules) else: api_pkgs = GroupingClassApiModelBuilder( self.iskeyword, self.language).generate(modules) api_pkgs.extend(SubModuleBuilder().generate( pyang_builder.get_submodules(), self.iskeyword)) return api_pkgs
def _build_api_model_packages(self, modules, groupings_as_class): packages = [] if not groupings_as_class: packages = ApiModelBuilder().generate(modules) else: packages = GroupingClassApiModelBuilder().generate(modules) return packages