@azure-tools/typespec-python 0.24.2 → 0.25.0

package/generator/pygen/codegen/models/primitive_types.py

@@ -0,0 +1,626 @@
+# -------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for
+# license information.
+# --------------------------------------------------------------------------
+import datetime
+import decimal
+from typing import Any, Dict, List, Optional, Union, TYPE_CHECKING
+
+from .base import BaseType
+from .imports import FileImport, ImportType, TypingSection
+
+if TYPE_CHECKING:
+    from .code_model import CodeModel
+
+
+class RawString(object):
+    def __init__(self, string: str) -> None:
+        self.string = string
+
+    def __repr__(self) -> str:
+        return "r'{}'".format(self.string.replace("'", "\\'"))
+
+
+class PrimitiveType(BaseType):  # pylint: disable=abstract-method
+    def description(self, *, is_operation_file: bool) -> str:  # pylint: disable=unused-argument
+        return ""
+
+    def type_annotation(self, **kwargs: Any) -> str:
+        return self.docstring_type(**kwargs)
+
+    def docstring_text(self, **kwargs: Any) -> str:
+        return self.docstring_type(**kwargs)
+
+    def get_json_template_representation(
+        self,
+        *,
+        client_default_value_declaration: Optional[str] = None,
+    ) -> Any:
+        if self.client_default_value is not None:
+            client_default_value_declaration = client_default_value_declaration or self.get_declaration(
+                self.client_default_value
+            )
+        return client_default_value_declaration or self.default_template_representation_declaration
+
+    @property
+    def default_template_representation_declaration(self) -> str:
+        return self.get_declaration(self.docstring_type())
+
+
+class BooleanType(PrimitiveType):
+    @property
+    def serialization_type(self) -> str:
+        return "bool"
+
+    def docstring_type(self, **kwargs: Any) -> str:
+        return "bool"
+
+    @property
+    def instance_check_template(self) -> str:
+        return "isinstance({}, bool)"
+
+
+class BinaryType(PrimitiveType):
+    def __init__(self, yaml_data: Dict[str, Any], code_model: "CodeModel") -> None:
+        super().__init__(yaml_data=yaml_data, code_model=code_model)
+        self.type = "IO"
+
+    @property
+    def serialization_type(self) -> str:
+        return self.type
+
+    def docstring_type(self, **kwargs: Any) -> str:
+        return f"{self.type}[bytes]"
+
+    def type_annotation(self, **kwargs: Any) -> str:
+        return f"{self.type}[bytes]"
+
+    def docstring_text(self, **kwargs: Any) -> str:
+        return f"{self.type}[bytes]"
+
+    @property
+    def default_template_representation_declaration(self) -> str:
+        return self.get_declaration(b"bytes")
+
+    def imports(self, **kwargs: Any) -> FileImport:
+        from .combined_type import CombinedType
+        from .operation import OperationBase
+
+        file_import = FileImport(self.code_model)
+        file_import.add_submodule_import("typing", "IO", ImportType.STDLIB)
+        operation = kwargs.get("operation")
+        if (
+            isinstance(operation, OperationBase)
+            and operation.parameters.has_body
+            and isinstance(operation.parameters.body_parameter.type, CombinedType)
+        ):
+            file_import.add_submodule_import("io", "IOBase", ImportType.STDLIB)
+        return file_import
+
+    @property
+    def instance_check_template(self) -> str:
+        return "isinstance({}, (IOBase, bytes))"
+
+
+class BinaryIteratorType(PrimitiveType):
+    def _iterator_name(self, **kwargs: Any) -> str:
+        return "AsyncIterator" if kwargs.pop("async_mode") else "Iterator"
+
+    @property
+    def serialization_type(self) -> str:
+        return "IO"
+
+    def docstring_type(self, **kwargs: Any) -> str:
+        return f"{self._iterator_name(**kwargs)}[bytes]"
+
+    def type_annotation(self, **kwargs: Any) -> str:
+        return f"{self._iterator_name(**kwargs)}[bytes]"
+
+    def docstring_text(self, **kwargs: Any) -> str:
+        return f"{self._iterator_name(**kwargs)}[bytes]"
+
+    @property
+    def default_template_representation_declaration(self) -> str:
+        return self.get_declaration(b"bytes")
+
+    def imports(self, **kwargs: Any) -> FileImport:
+        file_import = FileImport(self.code_model)
+        file_import.add_submodule_import("typing", self._iterator_name(**kwargs), ImportType.STDLIB)
+        return file_import
+
+    @property
+    def instance_check_template(self) -> str:
+        return "getattr({}, '__aiter__', None) is not None or getattr({}, '__iter__', None) is not None"
+
+
+class AnyType(PrimitiveType):
+    @property
+    def serialization_type(self) -> str:
+        return "object"
+
+    def docstring_type(self, **kwargs: Any) -> str:
+        return "any"
+
+    def type_annotation(self, **kwargs: Any) -> str:
+        return "Any"
+
+    @property
+    def default_template_representation_declaration(self) -> str:
+        return self.get_declaration({})
+
+    def imports(self, **kwargs: Any) -> FileImport:
+        file_import = FileImport(self.code_model)
+        file_import.add_submodule_import("typing", "Any", ImportType.STDLIB, TypingSection.CONDITIONAL)
+        return file_import
+
+    @property
+    def instance_check_template(self) -> str:
+        raise ValueError("Shouldn't do instance check on an anytype, it can be anything")
+
+
+class AnyObjectType(PrimitiveType):
+    @property
+    def serialization_type(self) -> str:
+        return "object"
+
+    def docstring_type(self, **kwargs: Any) -> str:
+        return "JSON"
+
+    def type_annotation(self, **kwargs: Any) -> str:
+        return "JSON"
+
+    @property
+    def default_template_representation_declaration(self) -> str:
+        return self.get_declaration({})
+
+    @property
+    def instance_check_template(self) -> str:
+        return "isinstance({}, MutableMapping)"
+
+    def imports(self, **kwargs: Any) -> FileImport:
+        file_import = FileImport(self.code_model)
+        file_import.define_mutable_mapping_type()
+        return file_import
+
+    @property
+    def type_description(self) -> str:
+        return "JSON"
+
+
+class NumberType(PrimitiveType):  # pylint: disable=abstract-method
+    def __init__(self, yaml_data: Dict[str, Any], code_model: "CodeModel") -> None:
+        super().__init__(yaml_data=yaml_data, code_model=code_model)
+        self.precision: Optional[int] = yaml_data.get("precision")
+        self.multiple: Optional[int] = yaml_data.get("multipleOf")
+        self.maximum: Optional[int] = yaml_data.get("maximum")
+        self.minimum: Optional[int] = yaml_data.get("minimum")
+        self.exclusive_maximum: Optional[int] = yaml_data.get("exclusiveMaximum")
+        self.exclusive_minimum: Optional[int] = yaml_data.get("exclusiveMinimum")
+
+    @property
+    def serialization_constraints(self) -> List[str]:
+        validation_constraints = [
+            (f"maximum_ex={self.maximum}" if self.maximum is not None and self.exclusive_maximum else None),
+            (f"maximum={self.maximum}" if self.maximum is not None and not self.exclusive_maximum else None),
+            (f"minimum_ex={self.minimum}" if self.minimum is not None and self.exclusive_minimum else None),
+            (f"minimum={self.minimum}" if self.minimum is not None and not self.exclusive_minimum else None),
+            f"multiple={self.multiple}" if self.multiple else None,
+        ]
+        return [x for x in validation_constraints if x is not None]
+
+    @property
+    def validation(self) -> Optional[Dict[str, Union[bool, int, str]]]:
+        validation: Dict[str, Union[bool, int, str]] = {}
+        if self.maximum is not None:
+            if self.exclusive_maximum:
+                validation["maximum_ex"] = self.maximum
+            else:
+                validation["maximum"] = self.maximum
+        if self.minimum is not None:
+            if self.exclusive_minimum:
+                validation["minimum_ex"] = self.minimum
+            else:
+                validation["minimum"] = self.minimum
+        if self.multiple:
+            validation["multiple"] = self.multiple
+        return validation or None
+
+    @property
+    def default_template_representation_declaration(self) -> str:
+        default_value = 0 if self.docstring_type() == "int" else 0.0
+        return self.get_declaration(default_value)
+
+
+class IntegerType(NumberType):
+    @property
+    def serialization_type(self) -> str:
+        return "int"
+
+    def docstring_type(self, **kwargs: Any) -> str:
+        return "int"
+
+    def type_annotation(self, **kwargs: Any) -> str:
+        return "int"
+
+    @property
+    def default_template_representation_declaration(self) -> str:
+        return self.get_declaration(0)
+
+    @property
+    def instance_check_template(self) -> str:
+        return "isinstance({}, int)"
+
+
+class FloatType(NumberType):
+    @property
+    def serialization_type(self) -> str:
+        return "float"
+
+    def docstring_type(self, **kwargs: Any) -> str:
+        return "float"
+
+    def type_annotation(self, **kwargs: Any) -> str:
+        return "float"
+
+    @property
+    def default_template_representation_declaration(self) -> str:
+        return self.get_declaration(0.0)
+
+    @property
+    def instance_check_template(self) -> str:
+        return "isinstance({}, float)"
+
+
+class DecimalType(NumberType):
+    @property
+    def serialization_type(self) -> str:
+        return "decimal"
+
+    def docstring_type(self, **kwargs: Any) -> str:
+        return "~" + self.type_annotation()
+
+    def type_annotation(self, **kwargs: Any) -> str:
+        return "decimal.Decimal"
+
+    def docstring_text(self, **kwargs: Any) -> str:
+        return self.type_annotation()
+
+    def get_declaration(self, value: decimal.Decimal) -> str:
+        return str(value)
+
+    def imports(self, **kwargs: Any) -> FileImport:
+        file_import = FileImport(self.code_model)
+        file_import.add_import("decimal", ImportType.STDLIB)
+        return file_import
+
+    @property
+    def default_template_representation_declaration(self) -> str:
+        return self.get_declaration(decimal.Decimal("0.0"))
+
+    @property
+    def instance_check_template(self) -> str:
+        return "isinstance({}, decimal.Decimal)"
+
+
+class StringType(PrimitiveType):
+    def __init__(self, yaml_data: Dict[str, Any], code_model: "CodeModel") -> None:
+        super().__init__(yaml_data=yaml_data, code_model=code_model)
+        self.max_length: Optional[int] = yaml_data.get("maxLength")
+        self.min_length: Optional[int] = (
+            yaml_data.get("minLength", 0) if yaml_data.get("maxLength") else yaml_data.get("minLength")
+        )
+        self.pattern: Optional[str] = yaml_data.get("pattern")
+
+    @property
+    def serialization_constraints(self) -> List[str]:
+        validation_constraints = [
+            f"max_length={self.max_length}" if self.max_length is not None else None,
+            f"min_length={self.min_length}" if self.min_length is not None else None,
+            f"pattern={RawString(self.pattern)}" if self.pattern else None,
+        ]
+        return [x for x in validation_constraints if x is not None]
+
+    @property
+    def validation(self) -> Optional[Dict[str, Union[bool, int, str]]]:
+        validation: Dict[str, Union[bool, int, str]] = {}
+        if self.max_length is not None:
+            validation["max_length"] = self.max_length
+        if self.min_length is not None:
+            validation["min_length"] = self.min_length
+        if self.pattern:
+            # https://github.com/Azure/autorest.python/issues/407
+            validation["pattern"] = RawString(self.pattern)  # type: ignore
+        return validation or None
+
+    def get_declaration(self, value) -> str:
+        return f'"{value}"'
+
+    @property
+    def serialization_type(self) -> str:
+        return "str"
+
+    def docstring_type(self, **kwargs: Any) -> str:
+        return "str"
+
+    @property
+    def instance_check_template(self) -> str:
+        return "isinstance({}, str)"
+
+
+class DatetimeType(PrimitiveType):
+    def __init__(self, yaml_data: Dict[str, Any], code_model: "CodeModel") -> None:
+        super().__init__(yaml_data=yaml_data, code_model=code_model)
+        self.encode = (
+            "rfc3339"
+            if yaml_data.get("encode", "date-time") == "date-time" or yaml_data.get("encode", "date-time") == "rfc3339"
+            else "rfc7231"
+        )
+
+    @property
+    def serialization_type(self) -> str:
+        formats_to_attribute_type = {
+            "rfc3339": "iso-8601",
+            "rfc7231": "rfc-1123",
+        }
+        return formats_to_attribute_type[self.encode]
+
+    def docstring_type(self, **kwargs: Any) -> str:
+        return "~" + self.type_annotation()
+
+    def type_annotation(self, **kwargs: Any) -> str:
+        return "datetime.datetime"
+
+    def docstring_text(self, **kwargs: Any) -> str:
+        return "datetime"
+
+    def get_declaration(self, value: datetime.datetime) -> str:
+        """Could be discussed, since technically I should return a datetime object,
+        but msrest will do fine.
+        """
+        return f'"{value}"'
+
+    def imports(self, **kwargs: Any) -> FileImport:
+        file_import = FileImport(self.code_model)
+        file_import.add_import("datetime", ImportType.STDLIB)
+        return file_import
+
+    @property
+    def default_template_representation_declaration(self):
+        return self.get_declaration(datetime.datetime(2020, 2, 20))
+
+    @property
+    def instance_check_template(self) -> str:
+        return "isinstance({}, datetime.datetime)"
+
+    def imports_for_sample(self) -> FileImport:
+        file_import = super().imports_for_sample()
+        file_import.add_import("isodate", ImportType.STDLIB)
+        return file_import
+
+    @staticmethod
+    def serialize_sample_value(value: Any) -> str:
+        return f"isodate.parse_datetime({repr(value)})"
+
+
+class TimeType(PrimitiveType):
+    @property
+    def serialization_type(self) -> str:
+        return "time"
+
+    def docstring_type(self, **kwargs: Any) -> str:
+        return "~" + self.type_annotation()
+
+    def type_annotation(self, **kwargs: Any) -> str:
+        return "datetime.time"
+
+    def docstring_text(self, **kwargs: Any) -> str:
+        return "time"
+
+    def get_declaration(self, value: datetime.time) -> str:
+        """Could be discussed, since technically I should return a time object,
+        but msrest will do fine.
+        """
+        return f'"{value}"'
+
+    def imports(self, **kwargs: Any) -> FileImport:
+        file_import = FileImport(self.code_model)
+        file_import.add_import("datetime", ImportType.STDLIB)
+        return file_import
+
+    @property
+    def default_template_representation_declaration(self) -> str:
+        return self.get_declaration(datetime.time(12, 30, 0))
+
+    @property
+    def instance_check_template(self) -> str:
+        return "isinstance({}, datetime.time)"
+
+    def imports_for_sample(self) -> FileImport:
+        file_import = super().imports_for_sample()
+        file_import.add_import("isodate", ImportType.STDLIB)
+        return file_import
+
+    @staticmethod
+    def serialize_sample_value(value: Any) -> str:
+        return f"isodate.parse_time({repr(value)})"
+
+
+class UnixTimeType(PrimitiveType):
+    @property
+    def encode(self) -> str:
+        return "unix-timestamp"
+
+    @property
+    def serialization_type(self) -> str:
+        return "unix-time"
+
+    def docstring_type(self, **kwargs: Any) -> str:
+        return "~" + self.type_annotation()
+
+    def type_annotation(self, **kwargs: Any) -> str:
+        return "datetime.datetime"
+
+    def docstring_text(self, **kwargs: Any) -> str:
+        return "datetime"
+
+    def get_declaration(self, value: datetime.datetime) -> str:
+        """Could be discussed, since technically I should return a datetime object,
+        but msrest will do fine.
+        """
+        return f'"{value}"'
+
+    def imports(self, **kwargs: Any) -> FileImport:
+        file_import = FileImport(self.code_model)
+        file_import.add_import("datetime", ImportType.STDLIB)
+        return file_import
+
+    @property
+    def default_template_representation_declaration(self) -> str:
+        return self.get_declaration(datetime.datetime(2020, 2, 20))
+
+    @property
+    def instance_check_template(self) -> str:
+        return "isinstance({}, datetime.time)"
+
+    def imports_for_sample(self) -> FileImport:
+        file_import = super().imports_for_sample()
+        file_import.add_import("datetime", ImportType.STDLIB)
+        return file_import
+
+    @staticmethod
+    def serialize_sample_value(value: Any) -> str:
+        return f"datetime.datetime.fromtimestamp({repr(value)}, datetime.timezone.utc)"
+
+
+class DateType(PrimitiveType):
+    @property
+    def serialization_type(self) -> str:
+        return "date"
+
+    def docstring_type(self, **kwargs: Any) -> str:
+        return "~" + self.type_annotation()
+
+    def type_annotation(self, **kwargs: Any) -> str:
+        return "datetime.date"
+
+    def docstring_text(self, **kwargs: Any) -> str:
+        return "date"
+
+    def get_declaration(self, value: datetime.date) -> str:
+        """Could be discussed, since technically I should return a datetime object,
+        but msrest will do fine.
+        """
+        return f'"{value}"'
+
+    def imports(self, **kwargs: Any) -> FileImport:
+        file_import = FileImport(self.code_model)
+        file_import.add_import("datetime", ImportType.STDLIB)
+        return file_import
+
+    @property
+    def default_template_representation_declaration(self) -> str:
+        return self.get_declaration(datetime.date(2020, 2, 20))
+
+    @property
+    def instance_check_template(self) -> str:
+        return "isinstance({}, datetime.date)"
+
+    def imports_for_sample(self) -> FileImport:
+        file_import = super().imports_for_sample()
+        file_import.add_import("isodate", ImportType.STDLIB)
+        return file_import
+
+    @staticmethod
+    def serialize_sample_value(value: Any) -> str:
+        return f"isodate.parse_date({repr(value)})"
+
+
+class DurationType(PrimitiveType):
+    @property
+    def serialization_type(self) -> str:
+        return "duration"
+
+    def docstring_type(self, **kwargs: Any) -> str:
+        return "~" + self.type_annotation()
+
+    def type_annotation(self, **kwargs: Any) -> str:
+        return "datetime.timedelta"
+
+    def docstring_text(self, **kwargs: Any) -> str:
+        return "timedelta"
+
+    def get_declaration(self, value: datetime.timedelta) -> str:
+        """Could be discussed, since technically I should return a datetime object,
+        but msrest will do fine.
+        """
+        return f'"{value}"'
+
+    def imports(self, **kwargs: Any) -> FileImport:
+        file_import = FileImport(self.code_model)
+        file_import.add_import("datetime", ImportType.STDLIB)
+        return file_import
+
+    @property
+    def default_template_representation_declaration(self) -> str:
+        return self.get_declaration(datetime.timedelta(1))
+
+    @property
+    def instance_check_template(self) -> str:
+        return "isinstance({}, datetime.timedelta)"
+
+    def imports_for_sample(self) -> FileImport:
+        file_import = super().imports_for_sample()
+        file_import.add_import("isodate", ImportType.STDLIB)
+        return file_import
+
+    @staticmethod
+    def serialize_sample_value(value: Any) -> str:
+        return f"isodate.parse_duration({repr(value)})"
+
+
+class ByteArraySchema(PrimitiveType):
+    def __init__(self, yaml_data: Dict[str, Any], code_model: "CodeModel") -> None:
+        super().__init__(yaml_data=yaml_data, code_model=code_model)
+        self.encode = yaml_data.get("encode", "base64")
+
+    @property
+    def serialization_type(self) -> str:
+        if self.encode == "base64url":
+            return "base64"
+        return "bytearray"
+
+    def docstring_type(self, **kwargs: Any) -> str:
+        return "bytes"
+
+    def get_declaration(self, value: str) -> str:
+        return f'bytes("{value}", encoding="utf-8")'
+
+    @property
+    def instance_check_template(self) -> str:
+        return "isinstance({}, bytes)"
+
+
+class SdkCoreType(PrimitiveType):
+    def __init__(self, yaml_data: Dict[str, Any], code_model: "CodeModel") -> None:
+        super().__init__(yaml_data=yaml_data, code_model=code_model)
+        self.name = yaml_data.get("name", "")
+
+    def docstring_type(self, **kwargs: Any) -> str:
+        return f"~{self.code_model.core_library}.{self.type_annotation(**kwargs)}"
+
+    def type_annotation(self, **kwargs: Any) -> str:
+        return self.name
+
+    def imports(self, **kwargs: Any) -> FileImport:
+        file_import = super().imports(**kwargs)
+        file_import.add_submodule_import("", self.name, ImportType.SDKCORE)
+        return file_import
+
+    @property
+    def instance_check_template(self) -> str:
+        return f"isinstance({{}}, {self.name})"
+
+    @property
+    def serialization_type(self) -> str:
+        return self.name