@autorest/python 6.1.10 → 6.2.0

package/autorest/codegen/serializers/__init__.py

@@ -3,18 +3,22 @@
 # Licensed under the MIT License. See License.txt in the project root for
 # license information.
 # --------------------------------------------------------------------------
-from typing import Dict, List, Optional, Any, Union, cast
+import logging
+from typing import List, Optional, Any, Union, cast
 from pathlib import Path
 from jinja2 import PackageLoader, Environment, FileSystemLoader, StrictUndefined
-from autorest.codegen.models.operation_group import OperationGroup
-from autorest.codegen.models.request_builder import OverloadedRequestBuilder
 
 from ... import ReaderAndWriter, ReaderAndWriterAutorest
 from ...jsonrpc import AutorestAPI
-from ..models import CodeModel, OperationGroup, RequestBuilder
-from ..models import TokenCredentialType
+from ..models import (
+    OperationGroup,
+    RequestBuilder,
+    OverloadedRequestBuilder,
+    NamespaceModel,
+    CodeModel,
+)
 from .enum_serializer import EnumSerializer
-from .general_serializer import GeneralSerializer
+from .general_serializer import GeneralNamespaceSerializer, GeneralSerializer
 from .model_init_serializer import ModelInitSerializer
 from .model_serializer import DpgModelSerializer, MsrestModelSerializer
 from .operations_init_serializer import OperationsInitSerializer
@@ -22,6 +26,10 @@ from .operation_groups_serializer import OperationGroupsSerializer
 from .metadata_serializer import MetadataSerializer
 from .request_builders_serializer import RequestBuildersSerializer
 from .patch_serializer import PatchSerializer
+from .sample_serializer import SampleSerializer
+from ..._utils import to_snake_case
+
+_LOGGER = logging.getLogger(__name__)
 
 __all__ = [
     "JinjaSerializer",
@@ -41,15 +49,27 @@ _REGENERATE_FILES = {"setup.py", "MANIFEST.in"}
 
 class JinjaSerializer(ReaderAndWriter):  # pylint: disable=abstract-method
     def __init__(
-        self, code_model: CodeModel, *, output_folder: Union[str, Path], **kwargs: Any
+        self,
+        code_model: CodeModel,
+        namespace_model: NamespaceModel,
+        *,
+        output_folder: Union[str, Path],
+        **kwargs: Any,
     ) -> None:
         super().__init__(output_folder=output_folder, **kwargs)
         self.code_model = code_model
+        self.namespace_model = namespace_model
 
     @property
     def has_aio_folder(self) -> bool:
-        return not self.code_model.options["no_async"] and bool(
-            self.code_model.request_builders
+        return not self.namespace_model.options["no_async"] and bool(
+            self.namespace_model.operation_groups
+        )
+
+    @property
+    def has_operations_folder(self) -> bool:
+        return self.namespace_model.options["show_operations"] and bool(
+            self.namespace_model.operation_groups
         )
 
     def serialize(self) -> None:
@@ -64,27 +84,24 @@ class JinjaSerializer(ReaderAndWriter):  # pylint: disable=abstract-method
 
         namespace_path = (
             Path(".")
-            if self.code_model.options["no_namespace_folders"]
-            else Path(*(self.code_model.namespace.split(".")))
+            if self.namespace_model.options["no_namespace_folders"]
+            else Path(*(self.namespace_model.namespace.split(".")))
         )
         # if there was a patch file before, we keep it
         self._keep_patch_file(namespace_path / Path("_patch.py"), env)
         if self.has_aio_folder:
             self._keep_patch_file(namespace_path / Path("aio") / Path("_patch.py"), env)
 
-        if self.code_model.options["models_mode"] and (
-            self.code_model.model_types or self.code_model.enums
+        if self.namespace_model.options["models_mode"] and (
+            self.namespace_model.model_types or self.namespace_model.enums
         ):
             self._keep_patch_file(
                 namespace_path / Path("models") / Path("_patch.py"), env
             )
-        if (
-            self.code_model.options["show_operations"]
-            and self.code_model.operation_groups
-        ):
+        if self.has_operations_folder:
             self._keep_patch_file(
                 namespace_path
-                / Path(self.code_model.operations_folder_name)
+                / Path(self.namespace_model.operations_folder_name)
                 / Path("_patch.py"),
                 env,
             )
@@ -92,7 +109,7 @@ class JinjaSerializer(ReaderAndWriter):  # pylint: disable=abstract-method
                 self._keep_patch_file(
                     namespace_path
                     / Path("aio")
-                    / Path(self.code_model.operations_folder_name)
+                    / Path(self.namespace_model.operations_folder_name)
                     / Path("_patch.py"),
                     env,
                 )
@@ -101,8 +118,8 @@ class JinjaSerializer(ReaderAndWriter):  # pylint: disable=abstract-method
             env=env, namespace_path=namespace_path
         )
 
-        if self.code_model.request_builders:
-            if self.code_model.options["builders_visibility"] != "embedded":
+        if self.namespace_model.operation_groups:
+            if self.namespace_model.options["builders_visibility"] != "embedded":
                 self._serialize_and_write_rest_layer(
                     env=env, namespace_path=namespace_path
                 )
@@ -112,25 +129,22 @@ class JinjaSerializer(ReaderAndWriter):  # pylint: disable=abstract-method
                     namespace_path=namespace_path,
                 )
 
-        if (
-            self.code_model.options["show_operations"]
-            and self.code_model.operation_groups
-        ):
+        if self.has_operations_folder:
             self._serialize_and_write_operations_folder(
                 env=env, namespace_path=namespace_path
             )
-            if self.code_model.options["multiapi"]:
+            if self.namespace_model.options["multiapi"]:
                 self._serialize_and_write_metadata(
                     env=env, namespace_path=namespace_path
                 )
 
-        if self.code_model.options["models_mode"] and (
-            self.code_model.model_types or self.code_model.enums
+        if self.namespace_model.options["models_mode"] and (
+            self.namespace_model.model_types or self.namespace_model.enums
         ):
             self._serialize_and_write_models_folder(
                 env=env, namespace_path=namespace_path
             )
-        if not self.code_model.options["models_mode"]:
+        if not self.namespace_model.options["models_mode"]:
             # keep models file if users ended up just writing a models file
             if self.read_file(namespace_path / Path("models.py")):
                 self.write_file(
@@ -138,68 +152,49 @@ class JinjaSerializer(ReaderAndWriter):  # pylint: disable=abstract-method
                     cast(str, self.read_file(namespace_path / Path("models.py"))),
                 )
 
-        if self.code_model.options["package_mode"]:
-            self._serialize_and_write_package_files(out_path=namespace_path)
-
-    def _serialize_and_write_package_files(self, out_path: Path) -> None:
-        def _serialize_and_write_package_files_proc(**kwargs: Any):
-            for template_name in package_files:
-                file = template_name.replace(".jinja2", "")
-                output_name = out_path / file
-                if not self.read_file(output_name) or file in _REGENERATE_FILES:
-                    template = env.get_template(template_name)
-                    render_result = template.render(**kwargs)
-                    self.write_file(output_name, render_result)
-
-        def _prepare_params() -> Dict[Any, Any]:
-            package_parts = (self.code_model.options["package_name"] or "").split("-")[
-                :-1
-            ]
-            token_cred = isinstance(
-                getattr(self.code_model.credential, "type", None), TokenCredentialType
-            )
-            version = self.code_model.options["package_version"]
-            if any(x in version for x in ["a", "b", "rc"]) or version[0] == "0":
-                dev_status = "4 - Beta"
-            else:
-                dev_status = "5 - Production/Stable"
-            params = {
-                "token_credential": token_cred,
-                "pkgutil_names": [
-                    ".".join(package_parts[: i + 1]) for i in range(len(package_parts))
-                ],
-                "init_names": [
-                    "/".join(package_parts[: i + 1]) + "/__init__.py"
-                    for i in range(len(package_parts))
-                ],
-                "dev_status": dev_status,
-                "client_name": self.code_model.client.name,
-                "namespace": self.code_model.namespace,
-                "code_model": self.code_model,
-            }
-            params.update(self.code_model.options)
-            params.update(self.code_model.package_dependency)
-            return params
-
-        out_path = out_path / Path("../" * (self.code_model.namespace.count(".") + 1))
-        if self.code_model.options["package_mode"] in ("dataplane", "mgmtplane"):
+        if self.namespace_model.options["package_mode"]:
+            self._serialize_and_write_package_files(namespace_path=namespace_path)
+
+        if (
+            self.namespace_model.options["show_operations"]
+            and self.namespace_model.operation_groups
+            and self.namespace_model.options["generate_sample"]
+            and not self.namespace_model.options["multiapi"]
+        ):
+            self._serialize_and_write_sample(env, namespace_path)
+
+    def _serialize_and_write_package_files(self, namespace_path: Path) -> None:
+        root_of_sdk = self._package_root_folder(namespace_path)
+        if self.namespace_model.options["package_mode"] in ("dataplane", "mgmtplane"):
             env = Environment(
-                loader=PackageLoader("autorest.codegen", "templates"),
+                loader=PackageLoader(
+                    "autorest.codegen", "templates/packaging_templates"
+                ),
                 undefined=StrictUndefined,
             )
+
             package_files = _PACKAGE_FILES
-            _serialize_and_write_package_files_proc(**_prepare_params())
-        elif Path(self.code_model.options["package_mode"]).exists():
+        elif Path(self.namespace_model.options["package_mode"]).exists():
             env = Environment(
                 loader=FileSystemLoader(
-                    str(Path(self.code_model.options["package_mode"]))
+                    str(Path(self.namespace_model.options["package_mode"]))
                 ),
                 keep_trailing_newline=True,
                 undefined=StrictUndefined,
             )
             package_files = env.list_templates()
-            params = self.code_model.options["package_configuration"] or {}
-            _serialize_and_write_package_files_proc(**params)
+        else:
+            return
+        serializer = GeneralSerializer(self.code_model, env)
+        params = self.namespace_model.options["package_configuration"] or {}
+        for template_name in package_files:
+            file = template_name.replace(".jinja2", "")
+            output_name = root_of_sdk / file
+            if not self.read_file(output_name) or file in _REGENERATE_FILES:
+                self.write_file(
+                    output_name,
+                    serializer.serialize_package_file(template_name, **params),
+                )
 
     def _keep_patch_file(self, path_file: Path, env: Environment):
         if self.read_file(path_file):
@@ -207,7 +202,9 @@ class JinjaSerializer(ReaderAndWriter):  # pylint: disable=abstract-method
         else:
             self.write_file(
                 path_file,
-                PatchSerializer(env=env, code_model=self.code_model).serialize(),
+                PatchSerializer(
+                    env=env, namespace_model=self.namespace_model
+                ).serialize(),
             )
 
     def _serialize_and_write_models_folder(
@@ -217,34 +214,43 @@ class JinjaSerializer(ReaderAndWriter):  # pylint: disable=abstract-method
         models_path = namespace_path / Path("models")
         serializer = (
             DpgModelSerializer
-            if self.code_model.options["models_mode"] == "dpg"
+            if self.namespace_model.options["models_mode"] == "dpg"
             else MsrestModelSerializer
         )
-        if self.code_model.model_types:
+        if self.namespace_model.model_types:
             self.write_file(
-                models_path / Path(f"{self.code_model.models_filename}.py"),
-                serializer(code_model=self.code_model, env=env).serialize(),
+                models_path / Path(f"{self.namespace_model.models_filename}.py"),
+                serializer(namespace_model=self.namespace_model, env=env).serialize(),
             )
-        if self.code_model.enums:
+        if self.namespace_model.enums:
             self.write_file(
-                models_path / Path(f"{self.code_model.enums_filename}.py"),
-                EnumSerializer(code_model=self.code_model, env=env).serialize(),
+                models_path / Path(f"{self.namespace_model.enums_filename}.py"),
+                EnumSerializer(
+                    namespace_model=self.namespace_model, env=env
+                ).serialize(),
             )
         self.write_file(
             models_path / Path("__init__.py"),
-            ModelInitSerializer(code_model=self.code_model, env=env).serialize(),
+            ModelInitSerializer(
+                namespace_model=self.namespace_model, env=env
+            ).serialize(),
         )
 
     def _serialize_and_write_rest_layer(
         self, env: Environment, namespace_path: Path
     ) -> None:
-        rest_path = namespace_path / Path(self.code_model.rest_layer_name)
-        group_names = {rb.group_name for rb in self.code_model.request_builders}
+        rest_path = namespace_path / Path(self.namespace_model.rest_layer_name)
+        group_names = {
+            rb.group_name
+            for c in self.namespace_model.clients
+            for rb in c.request_builders
+        }
 
         for group_name in group_names:
             request_builders = [
                 r
-                for r in self.code_model.request_builders
+                for c in self.namespace_model.clients
+                for r in c.request_builders
                 if r.group_name == group_name
             ]
             self._serialize_and_write_single_rest_layer(
@@ -253,7 +259,7 @@ class JinjaSerializer(ReaderAndWriter):  # pylint: disable=abstract-method
         if not "" in group_names:
             self.write_file(
                 rest_path / Path("__init__.py"),
-                self.code_model.options["license_header"],
+                self.namespace_model.options["license_header"],
             )
 
     def _serialize_and_write_single_rest_layer(
@@ -268,7 +274,7 @@ class JinjaSerializer(ReaderAndWriter):  # pylint: disable=abstract-method
         self.write_file(
             output_path / Path("_request_builders.py"),
             RequestBuildersSerializer(
-                code_model=self.code_model,
+                namespace_model=self.namespace_model,
                 env=env,
                 request_builders=request_builders,
             ).serialize_request_builders(),
@@ -278,7 +284,7 @@ class JinjaSerializer(ReaderAndWriter):  # pylint: disable=abstract-method
         self.write_file(
             output_path / Path("__init__.py"),
             RequestBuildersSerializer(
-                code_model=self.code_model,
+                namespace_model=self.namespace_model,
                 env=env,
                 request_builders=request_builders,
             ).serialize_init(),
@@ -293,14 +299,14 @@ class JinjaSerializer(ReaderAndWriter):  # pylint: disable=abstract-method
         filename = operation_group.filename if operation_group else "_operations"
         # write first sync file
         operation_group_serializer = OperationGroupsSerializer(
-            code_model=self.code_model,
+            namespace_model=self.namespace_model,
             env=env,
             async_mode=False,
             operation_group=operation_group,
         )
         self.write_file(
             namespace_path
-            / Path(self.code_model.operations_folder_name)
+            / Path(self.namespace_model.operations_folder_name)
             / Path(f"{filename}.py"),
             operation_group_serializer.serialize(),
         )
@@ -308,7 +314,7 @@ class JinjaSerializer(ReaderAndWriter):  # pylint: disable=abstract-method
         if self.has_aio_folder:
             # write async operation group and operation files
             operation_group_async_serializer = OperationGroupsSerializer(
-                code_model=self.code_model,
+                namespace_model=self.namespace_model,
                 env=env,
                 async_mode=True,
                 operation_group=operation_group,
@@ -317,7 +323,7 @@ class JinjaSerializer(ReaderAndWriter):  # pylint: disable=abstract-method
                 (
                     namespace_path
                     / Path("aio")
-                    / Path(self.code_model.operations_folder_name)
+                    / Path(self.namespace_model.operations_folder_name)
                     / Path(f"{filename}.py")
                 ),
                 operation_group_async_serializer.serialize(),
@@ -328,11 +334,11 @@ class JinjaSerializer(ReaderAndWriter):  # pylint: disable=abstract-method
     ) -> None:
         # write sync operations init file
         operations_init_serializer = OperationsInitSerializer(
-            code_model=self.code_model, env=env, async_mode=False
+            namespace_model=self.namespace_model, env=env, async_mode=False
         )
         self.write_file(
             namespace_path
-            / Path(self.code_model.operations_folder_name)
+            / Path(self.namespace_model.operations_folder_name)
             / Path("__init__.py"),
             operations_init_serializer.serialize(),
         )
@@ -340,23 +346,23 @@ class JinjaSerializer(ReaderAndWriter):  # pylint: disable=abstract-method
         # write async operations init file
         if self.has_aio_folder:
             operations_async_init_serializer = OperationsInitSerializer(
-                code_model=self.code_model, env=env, async_mode=True
+                namespace_model=self.namespace_model, env=env, async_mode=True
             )
             self.write_file(
                 namespace_path
                 / Path("aio")
-                / Path(self.code_model.operations_folder_name)
+                / Path(self.namespace_model.operations_folder_name)
                 / Path("__init__.py"),
                 operations_async_init_serializer.serialize(),
             )
 
-        if self.code_model.options["combine_operation_files"]:
+        if self.namespace_model.options["combine_operation_files"]:
             self._serialize_and_write_operations_file(
                 env=env,
                 namespace_path=namespace_path,
             )
         else:
-            for operation_group in self.code_model.operation_groups:
+            for operation_group in self.namespace_model.operation_groups:
                 self._serialize_and_write_operations_file(
                     env=env,
                     namespace_path=namespace_path,
@@ -364,7 +370,9 @@ class JinjaSerializer(ReaderAndWriter):  # pylint: disable=abstract-method
                 )
 
     def _serialize_and_write_version_file(
-        self, namespace_path: Path, general_serializer: GeneralSerializer
+        self,
+        namespace_path: Path,
+        general_namespace_serializer: GeneralNamespaceSerializer,
     ):
         def _read_version_file(original_version_file_name: str) -> str:
             return self.read_file(namespace_path / original_version_file_name)
@@ -375,135 +383,184 @@ class JinjaSerializer(ReaderAndWriter):  # pylint: disable=abstract-method
                 _read_version_file(original_version_file_name),
             )
 
-        keep_version_file = self.code_model.options["keep_version_file"]
+        keep_version_file = self.namespace_model.options["keep_version_file"]
         if keep_version_file and _read_version_file("_version.py"):
             _write_version_file(original_version_file_name="_version.py")
         elif keep_version_file and _read_version_file("version.py"):
             _write_version_file(original_version_file_name="version.py")
-        elif self.code_model.options["package_version"]:
+        elif self.namespace_model.options["package_version"]:
             self.write_file(
                 namespace_path / Path("_version.py"),
-                general_serializer.serialize_version_file(),
+                general_namespace_serializer.serialize_version_file(),
             )
 
     def _serialize_and_write_top_level_folder(
         self, env: Environment, namespace_path: Path
     ) -> None:
+        general_namespace_serializer = GeneralNamespaceSerializer(
+            namespace_model=self.namespace_model, env=env, async_mode=False
+        )
         general_serializer = GeneralSerializer(
-            code_model=self.code_model, env=env, async_mode=False
+            code_model=self.code_model,
+            env=env,
         )
 
         self.write_file(
             namespace_path / Path("__init__.py"),
-            general_serializer.serialize_init_file(),
+            general_namespace_serializer.serialize_init_file(),
         )
         p = namespace_path.parent
         while p != Path("."):
             # write pkgutil init file
             self.write_file(
                 p / Path("__init__.py"),
-                general_serializer.serialize_pkgutil_init_file(),
+                general_namespace_serializer.serialize_pkgutil_init_file(),
             )
             p = p.parent
 
         # Write the service client
-        if self.code_model.request_builders:
+        if self.namespace_model.operation_groups:
             self.write_file(
-                namespace_path / Path(f"{self.code_model.client.filename}.py"),
-                general_serializer.serialize_service_client_file(),
+                namespace_path / Path(f"{self.namespace_model.client_filename}.py"),
+                general_namespace_serializer.serialize_service_client_file(),
             )
 
-        if self.code_model.need_vendored_code(async_mode=False):
+        if self.namespace_model.need_vendored_code(async_mode=False):
             self.write_file(
                 namespace_path / Path("_vendor.py"),
-                general_serializer.serialize_vendor_file(),
+                general_namespace_serializer.serialize_vendor_file(),
             )
 
-        self._serialize_and_write_version_file(namespace_path, general_serializer)
+        self._serialize_and_write_version_file(
+            namespace_path, general_namespace_serializer
+        )
 
         # write the empty py.typed file
         self.write_file(namespace_path / Path("py.typed"), "# Marker file for PEP 561.")
 
         if (
-            not self.code_model.options["client_side_validation"]
-            and not self.code_model.options["multiapi"]
+            not self.namespace_model.options["client_side_validation"]
+            and not self.namespace_model.options["multiapi"]
         ):
             self.write_file(
                 namespace_path / Path("_serialization.py"),
-                general_serializer.serialize_serialization_file(),
+                general_namespace_serializer.serialize_serialization_file(),
             )
-        if self.code_model.options["models_mode"] == "dpg":
+        if self.namespace_model.options["models_mode"] == "dpg":
             self.write_file(
                 namespace_path / Path("_model_base.py"),
-                general_serializer.serialize_model_base_file(),
+                general_namespace_serializer.serialize_model_base_file(),
             )
 
-        if any(og for og in self.code_model.operation_groups if og.need_validation):
+        if any(
+            og for og in self.namespace_model.operation_groups if og.need_validation
+        ):
             self.write_file(
                 namespace_path / Path("_validation.py"),
-                general_serializer.serialize_validation_file(),
+                general_namespace_serializer.serialize_validation_file(),
             )
 
         # Write the config file
-        if self.code_model.request_builders:
+        if self.namespace_model.request_builders:
             self.write_file(
                 namespace_path / Path("_configuration.py"),
-                general_serializer.serialize_config_file(),
+                general_namespace_serializer.serialize_config_file(),
             )
 
         # Write the setup file
-        if self.code_model.options["basic_setup_py"]:
+        if self.namespace_model.options["basic_setup_py"]:
             self.write_file(Path("setup.py"), general_serializer.serialize_setup_file())
 
     def _serialize_and_write_aio_top_level_folder(
         self, env: Environment, namespace_path: Path
     ) -> None:
-        aio_general_serializer = GeneralSerializer(
-            code_model=self.code_model, env=env, async_mode=True
+        aio_general_namespace_serializer = GeneralNamespaceSerializer(
+            namespace_model=self.namespace_model, env=env, async_mode=True
         )
 
         aio_path = namespace_path / Path("aio")
 
         # Write the __init__ file
         self.write_file(
-            aio_path / Path("__init__.py"), aio_general_serializer.serialize_init_file()
+            aio_path / Path("__init__.py"),
+            aio_general_namespace_serializer.serialize_init_file(),
         )
 
         # Write the service client
-        if self.code_model.request_builders:
+        if self.namespace_model.request_builders:
             self.write_file(
-                aio_path / Path(f"{self.code_model.client.filename}.py"),
-                aio_general_serializer.serialize_service_client_file(),
+                aio_path / Path(f"{self.namespace_model.client_filename}.py"),
+                aio_general_namespace_serializer.serialize_service_client_file(),
             )
 
         # Write the config file
         self.write_file(
             aio_path / Path("_configuration.py"),
-            aio_general_serializer.serialize_config_file(),
+            aio_general_namespace_serializer.serialize_config_file(),
         )
-        if self.code_model.need_vendored_code(async_mode=True):
+        if self.namespace_model.need_vendored_code(async_mode=True):
             self.write_file(
                 aio_path / Path("_vendor.py"),
-                aio_general_serializer.serialize_vendor_file(),
+                aio_general_namespace_serializer.serialize_vendor_file(),
             )
 
     def _serialize_and_write_metadata(
         self, env: Environment, namespace_path: Path
     ) -> None:
-        metadata_serializer = MetadataSerializer(self.code_model, env)
+        metadata_serializer = MetadataSerializer(self.namespace_model, env)
         self.write_file(
             namespace_path / Path("_metadata.json"), metadata_serializer.serialize()
         )
 
+    # find root folder where "setup.py" is
+    def _package_root_folder(self, namespace_path: Path) -> Path:
+        return namespace_path / Path(
+            "../" * (self.namespace_model.namespace.count(".") + 1)
+        )
+
+    def _serialize_and_write_sample(self, env: Environment, namespace_path: Path):
+        out_path = self._package_root_folder(namespace_path) / Path("generated_samples")
+        for op_group in self.namespace_model.operation_groups:
+            for operation in op_group.operations:
+                samples = operation.yaml_data["samples"]
+                if not samples:
+                    continue
+                for key, value in samples.items():
+                    file_name = to_snake_case(key) + ".py"
+                    try:
+                        self.write_file(
+                            out_path / file_name,
+                            SampleSerializer(
+                                namespace_model=self.namespace_model,
+                                env=env,
+                                operation_group=op_group,
+                                operation=operation,
+                                sample=value,
+                                file_name=file_name,
+                                sample_origin_name=key,
+                            ).serialize(),
+                        )
+                    except Exception as e:  # pylint: disable=broad-except
+                        # sample generation shall not block code generation, so just log error
+                        _LOGGER.error(
+                            "error happens when generate sample with {%s}: {%s}", key, e
+                        )
+
 
 class JinjaSerializerAutorest(JinjaSerializer, ReaderAndWriterAutorest):
     def __init__(
         self,
         autorestapi: AutorestAPI,
         code_model: CodeModel,
+        namespace_model: NamespaceModel,
         *,
         output_folder: Union[str, Path],
+        **kwargs: Any,
     ) -> None:
         super().__init__(
-            autorestapi=autorestapi, code_model=code_model, output_folder=output_folder
+            autorestapi=autorestapi,
+            code_model=code_model,
+            namespace_model=namespace_model,
+            output_folder=output_folder,
+            **kwargs,
         )