@autorest/python 5.15.0 → 5.18.0

package/autorest/codegen/templates/request_builder.py.jinja2

@@ -4,23 +4,32 @@
 {% if code_model.options["builders_visibility"] == "public" %}
     {{ op_tools.description(request_builder, request_builder_serializer) | indent }}
 {% endif %}
-{% if request_builder_serializer.pop_kwargs_from_signature(request_builder) %}
+{% if not request_builder.is_overload %}
+    {% if request_builder.abstract %}
+    raise NotImplementedError(
+        "You need to write a custom operation for '{{ request_builder.name }}'. "
+        "Please refer to https://aka.ms/azsdk/python/dpcodegen/python/customize to learn how to customize."
+    )
+    {% else %}
+        {% if request_builder_serializer.pop_kwargs_from_signature(request_builder) %}
     {{ op_tools.serialize(request_builder_serializer.pop_kwargs_from_signature(request_builder)) | indent }}
-{%- endif -%}
-{% if request_builder_serializer.declare_non_inputtable_constants(request_builder) %}
+        {%- endif -%}
+    {% if request_builder_serializer.declare_non_inputtable_constants(request_builder) %}
     {{ op_tools.serialize(request_builder_serializer.declare_non_inputtable_constants(request_builder)) | indent }}
-{% endif %}
+        {% endif %}
     # Construct URL
     {{ request_builder_serializer.construct_url(request_builder) }}
-{% if request_builder.parameters.path %}
+    {% if request_builder.parameters.path %}
     {{ op_tools.serialize(request_builder_serializer.serialize_path(request_builder)) | indent }}
     _url = _format_url_section(_url, **path_format_arguments)
-{% endif %}
+        {% endif %}
 
-{% if request_builder.parameters.query %}
+        {% if request_builder.parameters.query %}
     {{ op_tools.serialize(request_builder_serializer.serialize_query(request_builder)) | indent }}
-{% endif %}
-{% if request_builder.parameters.headers %}
+        {% endif %}
+        {% if request_builder.parameters.headers %}
     {{ op_tools.serialize(request_builder_serializer.serialize_headers(request_builder)) | indent }}
-{% endif %}
+        {% endif %}
     {{ op_tools.serialize(request_builder_serializer.create_http_request(request_builder)) | indent }}
+        {% endif %}
+{% endif %}

package/autorest/codegen/templates/setup.py.jinja2

@@ -7,7 +7,7 @@ import re
 {% endif -%}
 from setuptools import setup, find_packages
 
-{% set package_name_render = package_name or code_model.class_name %}
+{% set package_name_render = package_name or code_model.client.name %}
 
 PACKAGE_NAME = "{{ package_name_render|lower }}"
 {% if package_mode -%}
@@ -30,7 +30,7 @@ if not version:
 {% else %}
 version = "{{ code_model.options['package_version'] }}"
 {% set description = "%s"|format(package_name_render) %}
-{% set long_description = code_model.description %}
+{% set long_description = code_model.client.description %}
 {% set author_email = "" %}
 {% set url = "" %}
 {% endif -%}
@@ -65,12 +65,18 @@ setup(
     packages=find_packages(
         exclude=[
             "tests",
+            {% if pkgutil_names %}
             # Exclude packages that will be covered by PEP420 or nspkg
+            {% endif %}
             {%- for pkgutil_name in pkgutil_names %}
             "{{ pkgutil_name }}",
             {%- endfor %}
         ]
     ),
+    include_package_data=True,
+    package_data={
+        'pytyped': ['py.typed'],
+    },
     {% else %}
     packages=find_packages(),
     include_package_data=True,
@@ -87,7 +93,7 @@ setup(
     python_requires=">=3.6",
     {% else %}
     long_description="""\
-    {{ code_model.description }}
+    {{ code_model.client.description }}
     """
     {% endif %}
 )

package/autorest/codegen/templates/vendor.py.jinja2

@@ -30,7 +30,7 @@ def _format_url_section(template, **kwargs):
 class MixinABC(ABC):
     """DO NOT use this class. It is for internal typing use only."""
     _client: "{{ keywords.async_class }}PipelineClient"
-    _config: {{ code_model.class_name }}Configuration
+    _config: {{ code_model.client.name }}Configuration
     _serialize: "Serializer"
     _deserialize: "Deserializer"
 {% endif %}

package/autorest/jsonrpc/__init__.py

@@ -7,7 +7,7 @@ from abc import ABC, abstractmethod
 from enum import Enum
 from typing import Any, List, Optional, Union
 import logging
-import logging.config # need to include this extra import so mypy doesn't throw logging module has no config
+import logging.config  # need to include this extra import so mypy doesn't throw logging module has no config
 from pathlib import Path
 
 
@@ -45,13 +45,12 @@ class AutorestHandler(logging.Handler):
         # Initialize this handler with the max loglevel, since
         # autorest is deciding what to show, not us
         # so we want to log everything and let autorest filters.
-        super(AutorestHandler, self).__init__(logging.DEBUG)
+        super().__init__(logging.DEBUG)
         self._autorest_api = autorest_api
 
     @staticmethod
     def _get_log_level(level: int) -> Channel:
-        """Convert Python log levels to Autorest Channel.
-        """
+        """Convert Python log levels to Autorest Channel."""
         return _LEVEL_MAPPING.get(level, Channel.Warning)
 
     def emit(self, record: logging.LogRecord) -> None:
@@ -65,8 +64,7 @@ class AutorestHandler(logging.Handler):
 
 
 class AutorestAPI(ABC):
-    """Defines the base interface of communication to Autorest from the plugin.
-    """
+    """Defines the base interface of communication to Autorest from the plugin."""
 
     def __init__(self) -> None:
         if Path("logging.conf").exists():
@@ -106,18 +104,15 @@ class AutorestAPI(ABC):
 
     @abstractmethod
     def list_inputs(self) -> List[str]:
-        """List possible inputs for this plugin.
-        """
+        """List possible inputs for this plugin."""
 
     @abstractmethod
     def get_value(self, key: str) -> Any:
-        """Get a value from configuration.
-        """
+        """Get a value from configuration."""
 
     @abstractmethod
     def message(self, channel: Channel, text: str) -> None:
-        """Send a log message to autorest.
-        """
+        """Send a log message to autorest."""
 
     def get_boolean_value(self, key: str, default: bool = None) -> Optional[bool]:
         """Check if value is present on the line, and interpret it as bool if it was.

package/autorest/jsonrpc/localapi.py

@@ -14,10 +14,11 @@ _LOGGER = logging.getLogger(__name__)
 
 
 class LocalAutorestAPI(AutorestAPI):
-    """A local API that will write on local disk.
-    """
+    """A local API that will write on local disk."""
 
-    def __init__(self, reachable_files: List[str] = None, output_folder: str = "generated") -> None:
+    def __init__(
+        self, reachable_files: List[str] = None, output_folder: str = "generated"
+    ) -> None:
         super().__init__()
         if reachable_files is None:
             reachable_files = []

package/autorest/jsonrpc/server.py

@@ -18,27 +18,42 @@ _LOGGER = logging.getLogger(__name__)
 
 @dispatcher.add_method
 def GetPluginNames():
-    return ["codegen", "m2r", "namer", "black", "multiapiscript"]
+    return [
+        "codegen",
+        "m2r",
+        "preprocess",
+        "m4reformatter",
+        "black",
+        "multiapiscript",
+        "postprocess",
+    ]
 
 
 @dispatcher.add_method
 def Process(plugin_name: str, session_id: str) -> bool:
     # pylint: disable=import-outside-toplevel
-    """JSON-RPC process call.
-    """
+    """JSON-RPC process call."""
     from .stdstream import StdStreamAutorestAPI
 
     with contextlib.closing(StdStreamAutorestAPI(session_id)) as stdstream_connection:
 
-        _LOGGER.debug("Autorest called process with plugin_name '%s' and session_id: '%s'", plugin_name, session_id)
+        _LOGGER.debug(
+            "Autorest called process with plugin_name '%s' and session_id: '%s'",
+            plugin_name,
+            session_id,
+        )
         if plugin_name == "m2r":
             from ..m2r import M2R as PluginToLoad
-        elif plugin_name == "namer":
-            from ..namer import Namer as PluginToLoad  # type: ignore
+        elif plugin_name == "preprocess":
+            from ..preprocess import PreProcessPlugin as PluginToLoad  # type: ignore
+        elif plugin_name == "m4reformatter":
+            from ..m4reformatter import M4Reformatter as PluginToLoad  # type: ignore
         elif plugin_name == "codegen":
             from ..codegen import CodeGenerator as PluginToLoad  # type: ignore
+        elif plugin_name == "postprocess":
+            from ..postprocess import PostProcessPlugin as PluginToLoad  # type: ignore
         elif plugin_name == "black":
-            from ..black import BlackScriptPlugin  as PluginToLoad  # type: ignore
+            from ..black import BlackScriptPlugin as PluginToLoad  # type: ignore
         elif plugin_name == "multiapiscript":
             from ..multiapi import MultiApiScriptPlugin as PluginToLoad  # type: ignore
         else:
@@ -58,11 +73,15 @@ def Process(plugin_name: str, session_id: str) -> bool:
 def main() -> None:
     # If --python.debugger is specified on the command line, we call the server.py file internally
     # with flag --debug.
-    if '--debug' in sys.argv or os.environ.get("AUTOREST_PYTHON_ATTACH_VSCODE_DEBUG", False):
+    if "--debug" in sys.argv or os.environ.get(
+        "AUTOREST_PYTHON_ATTACH_VSCODE_DEBUG", False
+    ):
         try:
             import ptvsd  # pylint: disable=import-outside-toplevel
         except ImportError:
-            raise SystemExit("Please pip install ptvsd in order to use VSCode debugging")
+            raise SystemExit(
+                "Please pip install ptvsd in order to use VSCode debugging"
+            )
 
         # 5678 is the default attach port in the VS Code debug configurations
         ptvsd.enable_attach(address=("localhost", 5678), redirect_output=True)

package/autorest/jsonrpc/stdstream.py

@@ -53,8 +53,7 @@ def write_message(message: str, stream: BinaryIO = sys.stdout.buffer) -> None:
 
 
 class StdStreamAutorestAPI(AutorestAPI):
-    """The stream API with Autorest
-    """
+    """The stream API with Autorest"""
 
     def __init__(self, session_id: str) -> None:
         super().__init__()
@@ -73,19 +72,25 @@ class StdStreamAutorestAPI(AutorestAPI):
     def read_file(self, filename: Union[str, Path]) -> str:
         _LOGGER.debug("Asking content for file %s", filename)
         filename = os.fspath(filename)
-        request = JSONRPC20Request(method="ReadFile", params=[self.session_id, filename], _id=42)
+        request = JSONRPC20Request(
+            method="ReadFile", params=[self.session_id, filename], _id=42
+        )
         write_message(request.json)
         return json.loads(read_message())["result"]
 
     def list_inputs(self) -> List[str]:
         _LOGGER.debug("Calling list inputs to Autorest")
-        request = JSONRPC20Request(method="ListInputs", params=[self.session_id, None], _id=42)
+        request = JSONRPC20Request(
+            method="ListInputs", params=[self.session_id, None], _id=42
+        )
         write_message(request.json)
         return json.loads(read_message())["result"]
 
     def get_value(self, key: str) -> Any:
         _LOGGER.debug("Calling get value to Autorest: %s", key)
-        request = JSONRPC20Request(method="GetValue", params=[self.session_id, key], _id=42)
+        request = JSONRPC20Request(
+            method="GetValue", params=[self.session_id, key], _id=42
+        )
         write_message(request.json)
         return json.loads(read_message())["result"]
 
@@ -96,5 +101,7 @@ class StdStreamAutorestAPI(AutorestAPI):
             "Channel": channel.value,
             "Text": text,
         }
-        request = JSONRPC20Request(method="Message", params=[self.session_id, message], is_notification=True)
+        request = JSONRPC20Request(
+            method="Message", params=[self.session_id, message], is_notification=True
+        )
         write_message(request.json)

package/autorest/m2r/__init__.py

@@ -27,18 +27,16 @@ class AutorestRender(m2r.RestRenderer):
 
 
 class M2R(YamlUpdatePlugin):
-    """A plugin to convert any description and summary from MD to RST.
-    """
+    """A plugin to convert any description and summary from MD to RST."""
+
     def update_yaml(self, yaml_data: Dict[str, Any]) -> None:
-        """Convert in place the YAML str.
-        """
+        """Convert in place the YAML str."""
         self._convert_docstring_no_cycles(yaml_data, set())
 
     def _convert_docstring_no_cycles(
         self, yaml_data: Dict[str, Any], node_list: Set[int]
     ) -> None:
-        """Walk the YAML tree to convert MD to RST.
-        """
+        """Walk the YAML tree to convert MD to RST."""
         if id(yaml_data) in node_list:
             return
         node_list.add(id(yaml_data))
@@ -55,8 +53,7 @@ class M2R(YamlUpdatePlugin):
 
     @staticmethod
     def convert_to_rst(string_to_convert: str) -> str:
-        """Convert that string from MD to RST.
-        """
+        """Convert that string from MD to RST."""
         try:
             return m2r.convert(string_to_convert, renderer=AutorestRender()).strip()
         except Exception:  # pylint: disable=broad-except