修改为东南天坐标系
This commit is contained in:
@@ -0,0 +1,185 @@
|
||||
"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
|
||||
|
||||
from typing import TYPE_CHECKING
|
||||
from importlib import import_module
|
||||
import builtins
|
||||
|
||||
if TYPE_CHECKING:
|
||||
from .annotations import get_discriminator
|
||||
from .datetimes import parse_datetime
|
||||
from .enums import OpenEnumMeta
|
||||
from .headers import get_headers, get_response_headers
|
||||
from .metadata import (
|
||||
FieldMetadata,
|
||||
find_metadata,
|
||||
FormMetadata,
|
||||
HeaderMetadata,
|
||||
MultipartFormMetadata,
|
||||
PathParamMetadata,
|
||||
QueryParamMetadata,
|
||||
RequestMetadata,
|
||||
SecurityMetadata,
|
||||
)
|
||||
from .queryparams import get_query_params
|
||||
from .retries import BackoffStrategy, Retries, retry, retry_async, RetryConfig
|
||||
from .requestbodies import serialize_request_body, SerializedRequestBody
|
||||
from .security import get_security
|
||||
from .serializers import (
|
||||
get_pydantic_model,
|
||||
marshal_json,
|
||||
unmarshal,
|
||||
unmarshal_json,
|
||||
serialize_decimal,
|
||||
serialize_float,
|
||||
serialize_int,
|
||||
stream_to_text,
|
||||
stream_to_text_async,
|
||||
stream_to_bytes,
|
||||
stream_to_bytes_async,
|
||||
validate_const,
|
||||
validate_decimal,
|
||||
validate_float,
|
||||
validate_int,
|
||||
validate_open_enum,
|
||||
)
|
||||
from .url import generate_url, template_url, remove_suffix
|
||||
from .values import (
|
||||
get_global_from_env,
|
||||
match_content_type,
|
||||
match_status_codes,
|
||||
match_response,
|
||||
cast_partial,
|
||||
)
|
||||
from .logger import Logger, get_body_content, get_default_logger
|
||||
|
||||
__all__ = [
|
||||
"BackoffStrategy",
|
||||
"FieldMetadata",
|
||||
"find_metadata",
|
||||
"FormMetadata",
|
||||
"generate_url",
|
||||
"get_body_content",
|
||||
"get_default_logger",
|
||||
"get_discriminator",
|
||||
"parse_datetime",
|
||||
"get_global_from_env",
|
||||
"get_headers",
|
||||
"get_pydantic_model",
|
||||
"get_query_params",
|
||||
"get_response_headers",
|
||||
"get_security",
|
||||
"HeaderMetadata",
|
||||
"Logger",
|
||||
"marshal_json",
|
||||
"match_content_type",
|
||||
"match_status_codes",
|
||||
"match_response",
|
||||
"MultipartFormMetadata",
|
||||
"OpenEnumMeta",
|
||||
"PathParamMetadata",
|
||||
"QueryParamMetadata",
|
||||
"remove_suffix",
|
||||
"Retries",
|
||||
"retry",
|
||||
"retry_async",
|
||||
"RetryConfig",
|
||||
"RequestMetadata",
|
||||
"SecurityMetadata",
|
||||
"serialize_decimal",
|
||||
"serialize_float",
|
||||
"serialize_int",
|
||||
"serialize_request_body",
|
||||
"SerializedRequestBody",
|
||||
"stream_to_text",
|
||||
"stream_to_text_async",
|
||||
"stream_to_bytes",
|
||||
"stream_to_bytes_async",
|
||||
"template_url",
|
||||
"unmarshal",
|
||||
"unmarshal_json",
|
||||
"validate_decimal",
|
||||
"validate_const",
|
||||
"validate_float",
|
||||
"validate_int",
|
||||
"validate_open_enum",
|
||||
"cast_partial",
|
||||
]
|
||||
|
||||
_dynamic_imports: dict[str, str] = {
|
||||
"BackoffStrategy": ".retries",
|
||||
"FieldMetadata": ".metadata",
|
||||
"find_metadata": ".metadata",
|
||||
"FormMetadata": ".metadata",
|
||||
"generate_url": ".url",
|
||||
"get_body_content": ".logger",
|
||||
"get_default_logger": ".logger",
|
||||
"get_discriminator": ".annotations",
|
||||
"parse_datetime": ".datetimes",
|
||||
"get_global_from_env": ".values",
|
||||
"get_headers": ".headers",
|
||||
"get_pydantic_model": ".serializers",
|
||||
"get_query_params": ".queryparams",
|
||||
"get_response_headers": ".headers",
|
||||
"get_security": ".security",
|
||||
"HeaderMetadata": ".metadata",
|
||||
"Logger": ".logger",
|
||||
"marshal_json": ".serializers",
|
||||
"match_content_type": ".values",
|
||||
"match_status_codes": ".values",
|
||||
"match_response": ".values",
|
||||
"MultipartFormMetadata": ".metadata",
|
||||
"OpenEnumMeta": ".enums",
|
||||
"PathParamMetadata": ".metadata",
|
||||
"QueryParamMetadata": ".metadata",
|
||||
"remove_suffix": ".url",
|
||||
"Retries": ".retries",
|
||||
"retry": ".retries",
|
||||
"retry_async": ".retries",
|
||||
"RetryConfig": ".retries",
|
||||
"RequestMetadata": ".metadata",
|
||||
"SecurityMetadata": ".metadata",
|
||||
"serialize_decimal": ".serializers",
|
||||
"serialize_float": ".serializers",
|
||||
"serialize_int": ".serializers",
|
||||
"serialize_request_body": ".requestbodies",
|
||||
"SerializedRequestBody": ".requestbodies",
|
||||
"stream_to_text": ".serializers",
|
||||
"stream_to_text_async": ".serializers",
|
||||
"stream_to_bytes": ".serializers",
|
||||
"stream_to_bytes_async": ".serializers",
|
||||
"template_url": ".url",
|
||||
"unmarshal": ".serializers",
|
||||
"unmarshal_json": ".serializers",
|
||||
"validate_decimal": ".serializers",
|
||||
"validate_const": ".serializers",
|
||||
"validate_float": ".serializers",
|
||||
"validate_int": ".serializers",
|
||||
"validate_open_enum": ".serializers",
|
||||
"cast_partial": ".values",
|
||||
}
|
||||
|
||||
|
||||
def __getattr__(attr_name: str) -> object:
|
||||
module_name = _dynamic_imports.get(attr_name)
|
||||
if module_name is None:
|
||||
raise AttributeError(
|
||||
f"no {attr_name} found in _dynamic_imports, module name -> {__name__} "
|
||||
)
|
||||
|
||||
try:
|
||||
module = import_module(module_name, __package__)
|
||||
result = getattr(module, attr_name)
|
||||
return result
|
||||
except ImportError as e:
|
||||
raise ImportError(
|
||||
f"Failed to import {attr_name} from {module_name}: {e}"
|
||||
) from e
|
||||
except AttributeError as e:
|
||||
raise AttributeError(
|
||||
f"Failed to get {attr_name} from {module_name}: {e}"
|
||||
) from e
|
||||
|
||||
|
||||
def __dir__():
|
||||
lazy_attrs = builtins.list(_dynamic_imports.keys())
|
||||
return builtins.sorted(lazy_attrs)
|
||||
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
@@ -0,0 +1,55 @@
|
||||
"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
|
||||
|
||||
from enum import Enum
|
||||
from typing import Any, Optional
|
||||
|
||||
def get_discriminator(model: Any, fieldname: str, key: str) -> str:
|
||||
"""
|
||||
Recursively search for the discriminator attribute in a model.
|
||||
|
||||
Args:
|
||||
model (Any): The model to search within.
|
||||
fieldname (str): The name of the field to search for.
|
||||
key (str): The key to search for in dictionaries.
|
||||
|
||||
Returns:
|
||||
str: The name of the discriminator attribute.
|
||||
|
||||
Raises:
|
||||
ValueError: If the discriminator attribute is not found.
|
||||
"""
|
||||
upper_fieldname = fieldname.upper()
|
||||
|
||||
def get_field_discriminator(field: Any) -> Optional[str]:
|
||||
"""Search for the discriminator attribute in a given field."""
|
||||
|
||||
if isinstance(field, dict):
|
||||
if key in field:
|
||||
return f'{field[key]}'
|
||||
|
||||
if hasattr(field, fieldname):
|
||||
attr = getattr(field, fieldname)
|
||||
if isinstance(attr, Enum):
|
||||
return f'{attr.value}'
|
||||
return f'{attr}'
|
||||
|
||||
if hasattr(field, upper_fieldname):
|
||||
attr = getattr(field, upper_fieldname)
|
||||
if isinstance(attr, Enum):
|
||||
return f'{attr.value}'
|
||||
return f'{attr}'
|
||||
|
||||
return None
|
||||
|
||||
|
||||
if isinstance(model, list):
|
||||
for field in model:
|
||||
discriminator = get_field_discriminator(field)
|
||||
if discriminator is not None:
|
||||
return discriminator
|
||||
|
||||
discriminator = get_field_discriminator(model)
|
||||
if discriminator is not None:
|
||||
return discriminator
|
||||
|
||||
raise ValueError(f'Could not find discriminator field {fieldname} in {model}')
|
||||
@@ -0,0 +1,23 @@
|
||||
"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
|
||||
|
||||
from datetime import datetime
|
||||
import sys
|
||||
|
||||
|
||||
def parse_datetime(datetime_string: str) -> datetime:
|
||||
"""
|
||||
Convert a RFC 3339 / ISO 8601 formatted string into a datetime object.
|
||||
Python versions 3.11 and later support parsing RFC 3339 directly with
|
||||
datetime.fromisoformat(), but for earlier versions, this function
|
||||
encapsulates the necessary extra logic.
|
||||
"""
|
||||
# Python 3.11 and later can parse RFC 3339 directly
|
||||
if sys.version_info >= (3, 11):
|
||||
return datetime.fromisoformat(datetime_string)
|
||||
|
||||
# For Python 3.10 and earlier, a common ValueError is trailing 'Z' suffix,
|
||||
# so fix that upfront.
|
||||
if datetime_string.endswith("Z"):
|
||||
datetime_string = datetime_string[:-1] + "+00:00"
|
||||
|
||||
return datetime.fromisoformat(datetime_string)
|
||||
@@ -0,0 +1,74 @@
|
||||
"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
|
||||
|
||||
import enum
|
||||
import sys
|
||||
|
||||
class OpenEnumMeta(enum.EnumMeta):
|
||||
# The __call__ method `boundary` kwarg was added in 3.11 and must be present
|
||||
# for pyright. Refer also: https://github.com/pylint-dev/pylint/issues/9622
|
||||
# pylint: disable=unexpected-keyword-arg
|
||||
# The __call__ method `values` varg must be named for pyright.
|
||||
# pylint: disable=keyword-arg-before-vararg
|
||||
|
||||
if sys.version_info >= (3, 11):
|
||||
def __call__(
|
||||
cls, value, names=None, *values, module=None, qualname=None, type=None, start=1, boundary=None
|
||||
):
|
||||
# The `type` kwarg also happens to be a built-in that pylint flags as
|
||||
# redeclared. Safe to ignore this lint rule with this scope.
|
||||
# pylint: disable=redefined-builtin
|
||||
|
||||
if names is not None:
|
||||
return super().__call__(
|
||||
value,
|
||||
names=names,
|
||||
*values,
|
||||
module=module,
|
||||
qualname=qualname,
|
||||
type=type,
|
||||
start=start,
|
||||
boundary=boundary,
|
||||
)
|
||||
|
||||
try:
|
||||
return super().__call__(
|
||||
value,
|
||||
names=names, # pyright: ignore[reportArgumentType]
|
||||
*values,
|
||||
module=module,
|
||||
qualname=qualname,
|
||||
type=type,
|
||||
start=start,
|
||||
boundary=boundary,
|
||||
)
|
||||
except ValueError:
|
||||
return value
|
||||
else:
|
||||
def __call__(
|
||||
cls, value, names=None, *, module=None, qualname=None, type=None, start=1
|
||||
):
|
||||
# The `type` kwarg also happens to be a built-in that pylint flags as
|
||||
# redeclared. Safe to ignore this lint rule with this scope.
|
||||
# pylint: disable=redefined-builtin
|
||||
|
||||
if names is not None:
|
||||
return super().__call__(
|
||||
value,
|
||||
names=names,
|
||||
module=module,
|
||||
qualname=qualname,
|
||||
type=type,
|
||||
start=start,
|
||||
)
|
||||
|
||||
try:
|
||||
return super().__call__(
|
||||
value,
|
||||
names=names, # pyright: ignore[reportArgumentType]
|
||||
module=module,
|
||||
qualname=qualname,
|
||||
type=type,
|
||||
start=start,
|
||||
)
|
||||
except ValueError:
|
||||
return value
|
||||
@@ -0,0 +1,238 @@
|
||||
"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
|
||||
|
||||
import re
|
||||
import json
|
||||
from typing import (
|
||||
Callable,
|
||||
Generic,
|
||||
TypeVar,
|
||||
Optional,
|
||||
Generator,
|
||||
AsyncGenerator,
|
||||
Tuple,
|
||||
)
|
||||
import httpx
|
||||
|
||||
T = TypeVar("T")
|
||||
|
||||
|
||||
class EventStream(Generic[T]):
|
||||
response: httpx.Response
|
||||
generator: Generator[T, None, None]
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
response: httpx.Response,
|
||||
decoder: Callable[[str], T],
|
||||
sentinel: Optional[str] = None,
|
||||
):
|
||||
self.response = response
|
||||
self.generator = stream_events(response, decoder, sentinel)
|
||||
|
||||
def __iter__(self):
|
||||
return self
|
||||
|
||||
def __next__(self):
|
||||
return next(self.generator)
|
||||
|
||||
def __enter__(self):
|
||||
return self
|
||||
|
||||
def __exit__(self, exc_type, exc_val, exc_tb):
|
||||
self.response.close()
|
||||
|
||||
|
||||
class EventStreamAsync(Generic[T]):
|
||||
response: httpx.Response
|
||||
generator: AsyncGenerator[T, None]
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
response: httpx.Response,
|
||||
decoder: Callable[[str], T],
|
||||
sentinel: Optional[str] = None,
|
||||
):
|
||||
self.response = response
|
||||
self.generator = stream_events_async(response, decoder, sentinel)
|
||||
|
||||
def __aiter__(self):
|
||||
return self
|
||||
|
||||
async def __anext__(self):
|
||||
return await self.generator.__anext__()
|
||||
|
||||
async def __aenter__(self):
|
||||
return self
|
||||
|
||||
async def __aexit__(self, exc_type, exc_val, exc_tb):
|
||||
await self.response.aclose()
|
||||
|
||||
|
||||
class ServerEvent:
|
||||
id: Optional[str] = None
|
||||
event: Optional[str] = None
|
||||
data: Optional[str] = None
|
||||
retry: Optional[int] = None
|
||||
|
||||
|
||||
MESSAGE_BOUNDARIES = [
|
||||
b"\r\n\r\n",
|
||||
b"\n\n",
|
||||
b"\r\r",
|
||||
]
|
||||
|
||||
|
||||
async def stream_events_async(
|
||||
response: httpx.Response,
|
||||
decoder: Callable[[str], T],
|
||||
sentinel: Optional[str] = None,
|
||||
) -> AsyncGenerator[T, None]:
|
||||
buffer = bytearray()
|
||||
position = 0
|
||||
discard = False
|
||||
async for chunk in response.aiter_bytes():
|
||||
# We've encountered the sentinel value and should no longer process
|
||||
# incoming data. Instead we throw new data away until the server closes
|
||||
# the connection.
|
||||
if discard:
|
||||
continue
|
||||
|
||||
buffer += chunk
|
||||
for i in range(position, len(buffer)):
|
||||
char = buffer[i : i + 1]
|
||||
seq: Optional[bytes] = None
|
||||
if char in [b"\r", b"\n"]:
|
||||
for boundary in MESSAGE_BOUNDARIES:
|
||||
seq = _peek_sequence(i, buffer, boundary)
|
||||
if seq is not None:
|
||||
break
|
||||
if seq is None:
|
||||
continue
|
||||
|
||||
block = buffer[position:i]
|
||||
position = i + len(seq)
|
||||
event, discard = _parse_event(block, decoder, sentinel)
|
||||
if event is not None:
|
||||
yield event
|
||||
|
||||
if position > 0:
|
||||
buffer = buffer[position:]
|
||||
position = 0
|
||||
|
||||
event, discard = _parse_event(buffer, decoder, sentinel)
|
||||
if event is not None:
|
||||
yield event
|
||||
|
||||
|
||||
def stream_events(
|
||||
response: httpx.Response,
|
||||
decoder: Callable[[str], T],
|
||||
sentinel: Optional[str] = None,
|
||||
) -> Generator[T, None, None]:
|
||||
buffer = bytearray()
|
||||
position = 0
|
||||
discard = False
|
||||
for chunk in response.iter_bytes():
|
||||
# We've encountered the sentinel value and should no longer process
|
||||
# incoming data. Instead we throw new data away until the server closes
|
||||
# the connection.
|
||||
if discard:
|
||||
continue
|
||||
|
||||
buffer += chunk
|
||||
for i in range(position, len(buffer)):
|
||||
char = buffer[i : i + 1]
|
||||
seq: Optional[bytes] = None
|
||||
if char in [b"\r", b"\n"]:
|
||||
for boundary in MESSAGE_BOUNDARIES:
|
||||
seq = _peek_sequence(i, buffer, boundary)
|
||||
if seq is not None:
|
||||
break
|
||||
if seq is None:
|
||||
continue
|
||||
|
||||
block = buffer[position:i]
|
||||
position = i + len(seq)
|
||||
event, discard = _parse_event(block, decoder, sentinel)
|
||||
if event is not None:
|
||||
yield event
|
||||
|
||||
if position > 0:
|
||||
buffer = buffer[position:]
|
||||
position = 0
|
||||
|
||||
event, discard = _parse_event(buffer, decoder, sentinel)
|
||||
if event is not None:
|
||||
yield event
|
||||
|
||||
|
||||
def _parse_event(
|
||||
raw: bytearray, decoder: Callable[[str], T], sentinel: Optional[str] = None
|
||||
) -> Tuple[Optional[T], bool]:
|
||||
block = raw.decode()
|
||||
lines = re.split(r"\r?\n|\r", block)
|
||||
publish = False
|
||||
event = ServerEvent()
|
||||
data = ""
|
||||
for line in lines:
|
||||
if not line:
|
||||
continue
|
||||
|
||||
delim = line.find(":")
|
||||
if delim <= 0:
|
||||
continue
|
||||
|
||||
field = line[0:delim]
|
||||
value = line[delim + 1 :] if delim < len(line) - 1 else ""
|
||||
if len(value) and value[0] == " ":
|
||||
value = value[1:]
|
||||
|
||||
if field == "event":
|
||||
event.event = value
|
||||
publish = True
|
||||
elif field == "data":
|
||||
data += value + "\n"
|
||||
publish = True
|
||||
elif field == "id":
|
||||
event.id = value
|
||||
publish = True
|
||||
elif field == "retry":
|
||||
event.retry = int(value) if value.isdigit() else None
|
||||
publish = True
|
||||
|
||||
if sentinel and data == f"{sentinel}\n":
|
||||
return None, True
|
||||
|
||||
if data:
|
||||
data = data[:-1]
|
||||
event.data = data
|
||||
|
||||
data_is_primitive = (
|
||||
data.isnumeric() or data == "true" or data == "false" or data == "null"
|
||||
)
|
||||
data_is_json = (
|
||||
data.startswith("{") or data.startswith("[") or data.startswith('"')
|
||||
)
|
||||
|
||||
if data_is_primitive or data_is_json:
|
||||
try:
|
||||
event.data = json.loads(data)
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
out = None
|
||||
if publish:
|
||||
out = decoder(json.dumps(event.__dict__))
|
||||
|
||||
return out, False
|
||||
|
||||
|
||||
def _peek_sequence(position: int, buffer: bytearray, sequence: bytes):
|
||||
if len(sequence) > (len(buffer) - position):
|
||||
return None
|
||||
|
||||
for i, seq in enumerate(sequence):
|
||||
if buffer[position + i] != seq:
|
||||
return None
|
||||
|
||||
return sequence
|
||||
@@ -0,0 +1,223 @@
|
||||
"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
|
||||
|
||||
from typing import (
|
||||
Any,
|
||||
Dict,
|
||||
get_type_hints,
|
||||
List,
|
||||
Tuple,
|
||||
)
|
||||
from pydantic import BaseModel
|
||||
from pydantic.fields import FieldInfo
|
||||
|
||||
from .serializers import marshal_json
|
||||
|
||||
from .metadata import (
|
||||
FormMetadata,
|
||||
MultipartFormMetadata,
|
||||
find_field_metadata,
|
||||
)
|
||||
from .values import _is_set, _val_to_string
|
||||
|
||||
|
||||
def _populate_form(
|
||||
field_name: str,
|
||||
explode: bool,
|
||||
obj: Any,
|
||||
delimiter: str,
|
||||
form: Dict[str, List[str]],
|
||||
):
|
||||
if not _is_set(obj):
|
||||
return form
|
||||
|
||||
if isinstance(obj, BaseModel):
|
||||
items = []
|
||||
|
||||
obj_fields: Dict[str, FieldInfo] = obj.__class__.model_fields
|
||||
for name in obj_fields:
|
||||
obj_field = obj_fields[name]
|
||||
obj_field_name = obj_field.alias if obj_field.alias is not None else name
|
||||
if obj_field_name == "":
|
||||
continue
|
||||
|
||||
val = getattr(obj, name)
|
||||
if not _is_set(val):
|
||||
continue
|
||||
|
||||
if explode:
|
||||
form[obj_field_name] = [_val_to_string(val)]
|
||||
else:
|
||||
items.append(f"{obj_field_name}{delimiter}{_val_to_string(val)}")
|
||||
|
||||
if len(items) > 0:
|
||||
form[field_name] = [delimiter.join(items)]
|
||||
elif isinstance(obj, Dict):
|
||||
items = []
|
||||
for key, value in obj.items():
|
||||
if not _is_set(value):
|
||||
continue
|
||||
|
||||
if explode:
|
||||
form[key] = [_val_to_string(value)]
|
||||
else:
|
||||
items.append(f"{key}{delimiter}{_val_to_string(value)}")
|
||||
|
||||
if len(items) > 0:
|
||||
form[field_name] = [delimiter.join(items)]
|
||||
elif isinstance(obj, List):
|
||||
items = []
|
||||
|
||||
for value in obj:
|
||||
if not _is_set(value):
|
||||
continue
|
||||
|
||||
if explode:
|
||||
if not field_name in form:
|
||||
form[field_name] = []
|
||||
form[field_name].append(_val_to_string(value))
|
||||
else:
|
||||
items.append(_val_to_string(value))
|
||||
|
||||
if len(items) > 0:
|
||||
form[field_name] = [delimiter.join([str(item) for item in items])]
|
||||
else:
|
||||
form[field_name] = [_val_to_string(obj)]
|
||||
|
||||
return form
|
||||
|
||||
|
||||
def _extract_file_properties(file_obj: Any) -> Tuple[str, Any, Any]:
|
||||
"""Extract file name, content, and content type from a file object."""
|
||||
file_fields: Dict[str, FieldInfo] = file_obj.__class__.model_fields
|
||||
|
||||
file_name = ""
|
||||
content = None
|
||||
content_type = None
|
||||
|
||||
for file_field_name in file_fields:
|
||||
file_field = file_fields[file_field_name]
|
||||
|
||||
file_metadata = find_field_metadata(file_field, MultipartFormMetadata)
|
||||
if file_metadata is None:
|
||||
continue
|
||||
|
||||
if file_metadata.content:
|
||||
content = getattr(file_obj, file_field_name, None)
|
||||
elif file_field_name == "content_type":
|
||||
content_type = getattr(file_obj, file_field_name, None)
|
||||
else:
|
||||
file_name = getattr(file_obj, file_field_name)
|
||||
|
||||
if file_name == "" or content is None:
|
||||
raise ValueError("invalid multipart/form-data file")
|
||||
|
||||
return file_name, content, content_type
|
||||
|
||||
|
||||
def serialize_multipart_form(
|
||||
media_type: str, request: Any
|
||||
) -> Tuple[str, Dict[str, Any], List[Tuple[str, Any]]]:
|
||||
form: Dict[str, Any] = {}
|
||||
files: List[Tuple[str, Any]] = []
|
||||
|
||||
if not isinstance(request, BaseModel):
|
||||
raise TypeError("invalid request body type")
|
||||
|
||||
request_fields: Dict[str, FieldInfo] = request.__class__.model_fields
|
||||
request_field_types = get_type_hints(request.__class__)
|
||||
|
||||
for name in request_fields:
|
||||
field = request_fields[name]
|
||||
|
||||
val = getattr(request, name)
|
||||
if not _is_set(val):
|
||||
continue
|
||||
|
||||
field_metadata = find_field_metadata(field, MultipartFormMetadata)
|
||||
if not field_metadata:
|
||||
continue
|
||||
|
||||
f_name = field.alias if field.alias else name
|
||||
|
||||
if field_metadata.file:
|
||||
if isinstance(val, List):
|
||||
# Handle array of files
|
||||
for file_obj in val:
|
||||
if not _is_set(file_obj):
|
||||
continue
|
||||
|
||||
file_name, content, content_type = _extract_file_properties(file_obj)
|
||||
|
||||
if content_type is not None:
|
||||
files.append((f_name + "[]", (file_name, content, content_type)))
|
||||
else:
|
||||
files.append((f_name + "[]", (file_name, content)))
|
||||
else:
|
||||
# Handle single file
|
||||
file_name, content, content_type = _extract_file_properties(val)
|
||||
|
||||
if content_type is not None:
|
||||
files.append((f_name, (file_name, content, content_type)))
|
||||
else:
|
||||
files.append((f_name, (file_name, content)))
|
||||
elif field_metadata.json:
|
||||
files.append((f_name, (
|
||||
None,
|
||||
marshal_json(val, request_field_types[name]),
|
||||
"application/json",
|
||||
)))
|
||||
else:
|
||||
if isinstance(val, List):
|
||||
values = []
|
||||
|
||||
for value in val:
|
||||
if not _is_set(value):
|
||||
continue
|
||||
values.append(_val_to_string(value))
|
||||
|
||||
form[f_name + "[]"] = values
|
||||
else:
|
||||
form[f_name] = _val_to_string(val)
|
||||
return media_type, form, files
|
||||
|
||||
|
||||
def serialize_form_data(data: Any) -> Dict[str, Any]:
|
||||
form: Dict[str, List[str]] = {}
|
||||
|
||||
if isinstance(data, BaseModel):
|
||||
data_fields: Dict[str, FieldInfo] = data.__class__.model_fields
|
||||
data_field_types = get_type_hints(data.__class__)
|
||||
for name in data_fields:
|
||||
field = data_fields[name]
|
||||
|
||||
val = getattr(data, name)
|
||||
if not _is_set(val):
|
||||
continue
|
||||
|
||||
metadata = find_field_metadata(field, FormMetadata)
|
||||
if metadata is None:
|
||||
continue
|
||||
|
||||
f_name = field.alias if field.alias is not None else name
|
||||
|
||||
if metadata.json:
|
||||
form[f_name] = [marshal_json(val, data_field_types[name])]
|
||||
else:
|
||||
if metadata.style == "form":
|
||||
_populate_form(
|
||||
f_name,
|
||||
metadata.explode,
|
||||
val,
|
||||
",",
|
||||
form,
|
||||
)
|
||||
else:
|
||||
raise ValueError(f"Invalid form style for field {name}")
|
||||
elif isinstance(data, Dict):
|
||||
for key, value in data.items():
|
||||
if _is_set(value):
|
||||
form[key] = [_val_to_string(value)]
|
||||
else:
|
||||
raise TypeError(f"Invalid request body type {type(data)} for form data")
|
||||
|
||||
return form
|
||||
@@ -0,0 +1,136 @@
|
||||
"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
|
||||
|
||||
from typing import (
|
||||
Any,
|
||||
Dict,
|
||||
List,
|
||||
Optional,
|
||||
)
|
||||
from httpx import Headers
|
||||
from pydantic import BaseModel
|
||||
from pydantic.fields import FieldInfo
|
||||
|
||||
from .metadata import (
|
||||
HeaderMetadata,
|
||||
find_field_metadata,
|
||||
)
|
||||
|
||||
from .values import _is_set, _populate_from_globals, _val_to_string
|
||||
|
||||
|
||||
def get_headers(headers_params: Any, gbls: Optional[Any] = None) -> Dict[str, str]:
|
||||
headers: Dict[str, str] = {}
|
||||
|
||||
globals_already_populated = []
|
||||
if _is_set(headers_params):
|
||||
globals_already_populated = _populate_headers(headers_params, gbls, headers, [])
|
||||
if _is_set(gbls):
|
||||
_populate_headers(gbls, None, headers, globals_already_populated)
|
||||
|
||||
return headers
|
||||
|
||||
|
||||
def _populate_headers(
|
||||
headers_params: Any,
|
||||
gbls: Any,
|
||||
header_values: Dict[str, str],
|
||||
skip_fields: List[str],
|
||||
) -> List[str]:
|
||||
globals_already_populated: List[str] = []
|
||||
|
||||
if not isinstance(headers_params, BaseModel):
|
||||
return globals_already_populated
|
||||
|
||||
param_fields: Dict[str, FieldInfo] = headers_params.__class__.model_fields
|
||||
for name in param_fields:
|
||||
if name in skip_fields:
|
||||
continue
|
||||
|
||||
field = param_fields[name]
|
||||
f_name = field.alias if field.alias is not None else name
|
||||
|
||||
metadata = find_field_metadata(field, HeaderMetadata)
|
||||
if metadata is None:
|
||||
continue
|
||||
|
||||
value, global_found = _populate_from_globals(
|
||||
name, getattr(headers_params, name), HeaderMetadata, gbls
|
||||
)
|
||||
if global_found:
|
||||
globals_already_populated.append(name)
|
||||
value = _serialize_header(metadata.explode, value)
|
||||
|
||||
if value != "":
|
||||
header_values[f_name] = value
|
||||
|
||||
return globals_already_populated
|
||||
|
||||
|
||||
def _serialize_header(explode: bool, obj: Any) -> str:
|
||||
if not _is_set(obj):
|
||||
return ""
|
||||
|
||||
if isinstance(obj, BaseModel):
|
||||
items = []
|
||||
obj_fields: Dict[str, FieldInfo] = obj.__class__.model_fields
|
||||
for name in obj_fields:
|
||||
obj_field = obj_fields[name]
|
||||
obj_param_metadata = find_field_metadata(obj_field, HeaderMetadata)
|
||||
|
||||
if not obj_param_metadata:
|
||||
continue
|
||||
|
||||
f_name = obj_field.alias if obj_field.alias is not None else name
|
||||
|
||||
val = getattr(obj, name)
|
||||
if not _is_set(val):
|
||||
continue
|
||||
|
||||
if explode:
|
||||
items.append(f"{f_name}={_val_to_string(val)}")
|
||||
else:
|
||||
items.append(f_name)
|
||||
items.append(_val_to_string(val))
|
||||
|
||||
if len(items) > 0:
|
||||
return ",".join(items)
|
||||
elif isinstance(obj, Dict):
|
||||
items = []
|
||||
|
||||
for key, value in obj.items():
|
||||
if not _is_set(value):
|
||||
continue
|
||||
|
||||
if explode:
|
||||
items.append(f"{key}={_val_to_string(value)}")
|
||||
else:
|
||||
items.append(key)
|
||||
items.append(_val_to_string(value))
|
||||
|
||||
if len(items) > 0:
|
||||
return ",".join([str(item) for item in items])
|
||||
elif isinstance(obj, List):
|
||||
items = []
|
||||
|
||||
for value in obj:
|
||||
if not _is_set(value):
|
||||
continue
|
||||
|
||||
items.append(_val_to_string(value))
|
||||
|
||||
if len(items) > 0:
|
||||
return ",".join(items)
|
||||
elif _is_set(obj):
|
||||
return f"{_val_to_string(obj)}"
|
||||
|
||||
return ""
|
||||
|
||||
|
||||
def get_response_headers(headers: Headers) -> Dict[str, List[str]]:
|
||||
res: Dict[str, List[str]] = {}
|
||||
for k, v in headers.items():
|
||||
if not k in res:
|
||||
res[k] = []
|
||||
|
||||
res[k].append(v)
|
||||
return res
|
||||
@@ -0,0 +1,22 @@
|
||||
"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
|
||||
|
||||
import httpx
|
||||
from typing import Any, Protocol
|
||||
|
||||
|
||||
class Logger(Protocol):
|
||||
def debug(self, msg: str, *args: Any, **kwargs: Any) -> None:
|
||||
pass
|
||||
|
||||
|
||||
class NoOpLogger:
|
||||
def debug(self, msg: str, *args: Any, **kwargs: Any) -> None:
|
||||
pass
|
||||
|
||||
|
||||
def get_body_content(req: httpx.Request) -> str:
|
||||
return "<streaming body>" if not hasattr(req, "_content") else str(req.content)
|
||||
|
||||
|
||||
def get_default_logger() -> Logger:
|
||||
return NoOpLogger()
|
||||
@@ -0,0 +1,118 @@
|
||||
"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
|
||||
|
||||
from typing import Optional, Type, TypeVar, Union
|
||||
from dataclasses import dataclass
|
||||
from pydantic.fields import FieldInfo
|
||||
|
||||
|
||||
T = TypeVar("T")
|
||||
|
||||
|
||||
@dataclass
|
||||
class SecurityMetadata:
|
||||
option: bool = False
|
||||
scheme: bool = False
|
||||
scheme_type: Optional[str] = None
|
||||
sub_type: Optional[str] = None
|
||||
field_name: Optional[str] = None
|
||||
|
||||
def get_field_name(self, default: str) -> str:
|
||||
return self.field_name or default
|
||||
|
||||
|
||||
@dataclass
|
||||
class ParamMetadata:
|
||||
serialization: Optional[str] = None
|
||||
style: str = "simple"
|
||||
explode: bool = False
|
||||
|
||||
|
||||
@dataclass
|
||||
class PathParamMetadata(ParamMetadata):
|
||||
pass
|
||||
|
||||
|
||||
@dataclass
|
||||
class QueryParamMetadata(ParamMetadata):
|
||||
style: str = "form"
|
||||
explode: bool = True
|
||||
|
||||
|
||||
@dataclass
|
||||
class HeaderMetadata(ParamMetadata):
|
||||
pass
|
||||
|
||||
|
||||
@dataclass
|
||||
class RequestMetadata:
|
||||
media_type: str = "application/octet-stream"
|
||||
|
||||
|
||||
@dataclass
|
||||
class MultipartFormMetadata:
|
||||
file: bool = False
|
||||
content: bool = False
|
||||
json: bool = False
|
||||
|
||||
|
||||
@dataclass
|
||||
class FormMetadata:
|
||||
json: bool = False
|
||||
style: str = "form"
|
||||
explode: bool = True
|
||||
|
||||
|
||||
class FieldMetadata:
|
||||
security: Optional[SecurityMetadata] = None
|
||||
path: Optional[PathParamMetadata] = None
|
||||
query: Optional[QueryParamMetadata] = None
|
||||
header: Optional[HeaderMetadata] = None
|
||||
request: Optional[RequestMetadata] = None
|
||||
form: Optional[FormMetadata] = None
|
||||
multipart: Optional[MultipartFormMetadata] = None
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
security: Optional[SecurityMetadata] = None,
|
||||
path: Optional[Union[PathParamMetadata, bool]] = None,
|
||||
query: Optional[Union[QueryParamMetadata, bool]] = None,
|
||||
header: Optional[Union[HeaderMetadata, bool]] = None,
|
||||
request: Optional[Union[RequestMetadata, bool]] = None,
|
||||
form: Optional[Union[FormMetadata, bool]] = None,
|
||||
multipart: Optional[Union[MultipartFormMetadata, bool]] = None,
|
||||
):
|
||||
self.security = security
|
||||
self.path = PathParamMetadata() if isinstance(path, bool) else path
|
||||
self.query = QueryParamMetadata() if isinstance(query, bool) else query
|
||||
self.header = HeaderMetadata() if isinstance(header, bool) else header
|
||||
self.request = RequestMetadata() if isinstance(request, bool) else request
|
||||
self.form = FormMetadata() if isinstance(form, bool) else form
|
||||
self.multipart = (
|
||||
MultipartFormMetadata() if isinstance(multipart, bool) else multipart
|
||||
)
|
||||
|
||||
|
||||
def find_field_metadata(field_info: FieldInfo, metadata_type: Type[T]) -> Optional[T]:
|
||||
metadata = find_metadata(field_info, FieldMetadata)
|
||||
if not metadata:
|
||||
return None
|
||||
|
||||
fields = metadata.__dict__
|
||||
|
||||
for field in fields:
|
||||
if isinstance(fields[field], metadata_type):
|
||||
return fields[field]
|
||||
|
||||
return None
|
||||
|
||||
|
||||
def find_metadata(field_info: FieldInfo, metadata_type: Type[T]) -> Optional[T]:
|
||||
metadata = field_info.metadata
|
||||
if not metadata:
|
||||
return None
|
||||
|
||||
for md in metadata:
|
||||
if isinstance(md, metadata_type):
|
||||
return md
|
||||
|
||||
return None
|
||||
@@ -0,0 +1,205 @@
|
||||
"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
|
||||
|
||||
from typing import (
|
||||
Any,
|
||||
Dict,
|
||||
get_type_hints,
|
||||
List,
|
||||
Optional,
|
||||
)
|
||||
|
||||
from pydantic import BaseModel
|
||||
from pydantic.fields import FieldInfo
|
||||
|
||||
from .metadata import (
|
||||
QueryParamMetadata,
|
||||
find_field_metadata,
|
||||
)
|
||||
from .values import (
|
||||
_get_serialized_params,
|
||||
_is_set,
|
||||
_populate_from_globals,
|
||||
_val_to_string,
|
||||
)
|
||||
from .forms import _populate_form
|
||||
|
||||
|
||||
def get_query_params(
|
||||
query_params: Any,
|
||||
gbls: Optional[Any] = None,
|
||||
) -> Dict[str, List[str]]:
|
||||
params: Dict[str, List[str]] = {}
|
||||
|
||||
globals_already_populated = _populate_query_params(query_params, gbls, params, [])
|
||||
if _is_set(gbls):
|
||||
_populate_query_params(gbls, None, params, globals_already_populated)
|
||||
|
||||
return params
|
||||
|
||||
|
||||
def _populate_query_params(
|
||||
query_params: Any,
|
||||
gbls: Any,
|
||||
query_param_values: Dict[str, List[str]],
|
||||
skip_fields: List[str],
|
||||
) -> List[str]:
|
||||
globals_already_populated: List[str] = []
|
||||
|
||||
if not isinstance(query_params, BaseModel):
|
||||
return globals_already_populated
|
||||
|
||||
param_fields: Dict[str, FieldInfo] = query_params.__class__.model_fields
|
||||
param_field_types = get_type_hints(query_params.__class__)
|
||||
for name in param_fields:
|
||||
if name in skip_fields:
|
||||
continue
|
||||
|
||||
field = param_fields[name]
|
||||
|
||||
metadata = find_field_metadata(field, QueryParamMetadata)
|
||||
if not metadata:
|
||||
continue
|
||||
|
||||
value = getattr(query_params, name) if _is_set(query_params) else None
|
||||
|
||||
value, global_found = _populate_from_globals(
|
||||
name, value, QueryParamMetadata, gbls
|
||||
)
|
||||
if global_found:
|
||||
globals_already_populated.append(name)
|
||||
|
||||
f_name = field.alias if field.alias is not None else name
|
||||
serialization = metadata.serialization
|
||||
if serialization is not None:
|
||||
serialized_parms = _get_serialized_params(
|
||||
metadata, f_name, value, param_field_types[name]
|
||||
)
|
||||
for key, value in serialized_parms.items():
|
||||
if key in query_param_values:
|
||||
query_param_values[key].extend(value)
|
||||
else:
|
||||
query_param_values[key] = [value]
|
||||
else:
|
||||
style = metadata.style
|
||||
if style == "deepObject":
|
||||
_populate_deep_object_query_params(f_name, value, query_param_values)
|
||||
elif style == "form":
|
||||
_populate_delimited_query_params(
|
||||
metadata, f_name, value, ",", query_param_values
|
||||
)
|
||||
elif style == "pipeDelimited":
|
||||
_populate_delimited_query_params(
|
||||
metadata, f_name, value, "|", query_param_values
|
||||
)
|
||||
else:
|
||||
raise NotImplementedError(
|
||||
f"query param style {style} not yet supported"
|
||||
)
|
||||
|
||||
return globals_already_populated
|
||||
|
||||
|
||||
def _populate_deep_object_query_params(
|
||||
field_name: str,
|
||||
obj: Any,
|
||||
params: Dict[str, List[str]],
|
||||
):
|
||||
if not _is_set(obj):
|
||||
return
|
||||
|
||||
if isinstance(obj, BaseModel):
|
||||
_populate_deep_object_query_params_basemodel(field_name, obj, params)
|
||||
elif isinstance(obj, Dict):
|
||||
_populate_deep_object_query_params_dict(field_name, obj, params)
|
||||
|
||||
|
||||
def _populate_deep_object_query_params_basemodel(
|
||||
prior_params_key: str,
|
||||
obj: Any,
|
||||
params: Dict[str, List[str]],
|
||||
):
|
||||
if not _is_set(obj) or not isinstance(obj, BaseModel):
|
||||
return
|
||||
|
||||
obj_fields: Dict[str, FieldInfo] = obj.__class__.model_fields
|
||||
for name in obj_fields:
|
||||
obj_field = obj_fields[name]
|
||||
|
||||
f_name = obj_field.alias if obj_field.alias is not None else name
|
||||
|
||||
params_key = f"{prior_params_key}[{f_name}]"
|
||||
|
||||
obj_param_metadata = find_field_metadata(obj_field, QueryParamMetadata)
|
||||
if not _is_set(obj_param_metadata):
|
||||
continue
|
||||
|
||||
obj_val = getattr(obj, name)
|
||||
if not _is_set(obj_val):
|
||||
continue
|
||||
|
||||
if isinstance(obj_val, BaseModel):
|
||||
_populate_deep_object_query_params_basemodel(params_key, obj_val, params)
|
||||
elif isinstance(obj_val, Dict):
|
||||
_populate_deep_object_query_params_dict(params_key, obj_val, params)
|
||||
elif isinstance(obj_val, List):
|
||||
_populate_deep_object_query_params_list(params_key, obj_val, params)
|
||||
else:
|
||||
params[params_key] = [_val_to_string(obj_val)]
|
||||
|
||||
|
||||
def _populate_deep_object_query_params_dict(
|
||||
prior_params_key: str,
|
||||
value: Dict,
|
||||
params: Dict[str, List[str]],
|
||||
):
|
||||
if not _is_set(value):
|
||||
return
|
||||
|
||||
for key, val in value.items():
|
||||
if not _is_set(val):
|
||||
continue
|
||||
|
||||
params_key = f"{prior_params_key}[{key}]"
|
||||
|
||||
if isinstance(val, BaseModel):
|
||||
_populate_deep_object_query_params_basemodel(params_key, val, params)
|
||||
elif isinstance(val, Dict):
|
||||
_populate_deep_object_query_params_dict(params_key, val, params)
|
||||
elif isinstance(val, List):
|
||||
_populate_deep_object_query_params_list(params_key, val, params)
|
||||
else:
|
||||
params[params_key] = [_val_to_string(val)]
|
||||
|
||||
|
||||
def _populate_deep_object_query_params_list(
|
||||
params_key: str,
|
||||
value: List,
|
||||
params: Dict[str, List[str]],
|
||||
):
|
||||
if not _is_set(value):
|
||||
return
|
||||
|
||||
for val in value:
|
||||
if not _is_set(val):
|
||||
continue
|
||||
|
||||
if params.get(params_key) is None:
|
||||
params[params_key] = []
|
||||
|
||||
params[params_key].append(_val_to_string(val))
|
||||
|
||||
|
||||
def _populate_delimited_query_params(
|
||||
metadata: QueryParamMetadata,
|
||||
field_name: str,
|
||||
obj: Any,
|
||||
delimiter: str,
|
||||
query_param_values: Dict[str, List[str]],
|
||||
):
|
||||
_populate_form(
|
||||
field_name,
|
||||
metadata.explode,
|
||||
obj,
|
||||
delimiter,
|
||||
query_param_values,
|
||||
)
|
||||
@@ -0,0 +1,66 @@
|
||||
"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
|
||||
|
||||
import io
|
||||
from dataclasses import dataclass
|
||||
import re
|
||||
from typing import (
|
||||
Any,
|
||||
Optional,
|
||||
)
|
||||
|
||||
from .forms import serialize_form_data, serialize_multipart_form
|
||||
|
||||
from .serializers import marshal_json
|
||||
|
||||
SERIALIZATION_METHOD_TO_CONTENT_TYPE = {
|
||||
"json": "application/json",
|
||||
"form": "application/x-www-form-urlencoded",
|
||||
"multipart": "multipart/form-data",
|
||||
"raw": "application/octet-stream",
|
||||
"string": "text/plain",
|
||||
}
|
||||
|
||||
|
||||
@dataclass
|
||||
class SerializedRequestBody:
|
||||
media_type: Optional[str] = None
|
||||
content: Optional[Any] = None
|
||||
data: Optional[Any] = None
|
||||
files: Optional[Any] = None
|
||||
|
||||
|
||||
def serialize_request_body(
|
||||
request_body: Any,
|
||||
nullable: bool,
|
||||
optional: bool,
|
||||
serialization_method: str,
|
||||
request_body_type,
|
||||
) -> Optional[SerializedRequestBody]:
|
||||
if request_body is None:
|
||||
if not nullable and optional:
|
||||
return None
|
||||
|
||||
media_type = SERIALIZATION_METHOD_TO_CONTENT_TYPE[serialization_method]
|
||||
|
||||
serialized_request_body = SerializedRequestBody(media_type)
|
||||
|
||||
if re.match(r"(application|text)\/.*?\+*json.*", media_type) is not None:
|
||||
serialized_request_body.content = marshal_json(request_body, request_body_type)
|
||||
elif re.match(r"multipart\/.*", media_type) is not None:
|
||||
(
|
||||
serialized_request_body.media_type,
|
||||
serialized_request_body.data,
|
||||
serialized_request_body.files,
|
||||
) = serialize_multipart_form(media_type, request_body)
|
||||
elif re.match(r"application\/x-www-form-urlencoded.*", media_type) is not None:
|
||||
serialized_request_body.data = serialize_form_data(request_body)
|
||||
elif isinstance(request_body, (bytes, bytearray, io.BytesIO, io.BufferedReader)):
|
||||
serialized_request_body.content = request_body
|
||||
elif isinstance(request_body, str):
|
||||
serialized_request_body.content = request_body
|
||||
else:
|
||||
raise TypeError(
|
||||
f"invalid request body type {type(request_body)} for mediaType {media_type}"
|
||||
)
|
||||
|
||||
return serialized_request_body
|
||||
@@ -0,0 +1,217 @@
|
||||
"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
|
||||
|
||||
import asyncio
|
||||
import random
|
||||
import time
|
||||
from typing import List
|
||||
|
||||
import httpx
|
||||
|
||||
|
||||
class BackoffStrategy:
|
||||
initial_interval: int
|
||||
max_interval: int
|
||||
exponent: float
|
||||
max_elapsed_time: int
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
initial_interval: int,
|
||||
max_interval: int,
|
||||
exponent: float,
|
||||
max_elapsed_time: int,
|
||||
):
|
||||
self.initial_interval = initial_interval
|
||||
self.max_interval = max_interval
|
||||
self.exponent = exponent
|
||||
self.max_elapsed_time = max_elapsed_time
|
||||
|
||||
|
||||
class RetryConfig:
|
||||
strategy: str
|
||||
backoff: BackoffStrategy
|
||||
retry_connection_errors: bool
|
||||
|
||||
def __init__(
|
||||
self, strategy: str, backoff: BackoffStrategy, retry_connection_errors: bool
|
||||
):
|
||||
self.strategy = strategy
|
||||
self.backoff = backoff
|
||||
self.retry_connection_errors = retry_connection_errors
|
||||
|
||||
|
||||
class Retries:
|
||||
config: RetryConfig
|
||||
status_codes: List[str]
|
||||
|
||||
def __init__(self, config: RetryConfig, status_codes: List[str]):
|
||||
self.config = config
|
||||
self.status_codes = status_codes
|
||||
|
||||
|
||||
class TemporaryError(Exception):
|
||||
response: httpx.Response
|
||||
|
||||
def __init__(self, response: httpx.Response):
|
||||
self.response = response
|
||||
|
||||
|
||||
class PermanentError(Exception):
|
||||
inner: Exception
|
||||
|
||||
def __init__(self, inner: Exception):
|
||||
self.inner = inner
|
||||
|
||||
|
||||
def retry(func, retries: Retries):
|
||||
if retries.config.strategy == "backoff":
|
||||
|
||||
def do_request() -> httpx.Response:
|
||||
res: httpx.Response
|
||||
try:
|
||||
res = func()
|
||||
|
||||
for code in retries.status_codes:
|
||||
if "X" in code.upper():
|
||||
code_range = int(code[0])
|
||||
|
||||
status_major = res.status_code / 100
|
||||
|
||||
if code_range <= status_major < code_range + 1:
|
||||
raise TemporaryError(res)
|
||||
else:
|
||||
parsed_code = int(code)
|
||||
|
||||
if res.status_code == parsed_code:
|
||||
raise TemporaryError(res)
|
||||
except httpx.ConnectError as exception:
|
||||
if retries.config.retry_connection_errors:
|
||||
raise
|
||||
|
||||
raise PermanentError(exception) from exception
|
||||
except httpx.TimeoutException as exception:
|
||||
if retries.config.retry_connection_errors:
|
||||
raise
|
||||
|
||||
raise PermanentError(exception) from exception
|
||||
except TemporaryError:
|
||||
raise
|
||||
except Exception as exception:
|
||||
raise PermanentError(exception) from exception
|
||||
|
||||
return res
|
||||
|
||||
return retry_with_backoff(
|
||||
do_request,
|
||||
retries.config.backoff.initial_interval,
|
||||
retries.config.backoff.max_interval,
|
||||
retries.config.backoff.exponent,
|
||||
retries.config.backoff.max_elapsed_time,
|
||||
)
|
||||
|
||||
return func()
|
||||
|
||||
|
||||
async def retry_async(func, retries: Retries):
|
||||
if retries.config.strategy == "backoff":
|
||||
|
||||
async def do_request() -> httpx.Response:
|
||||
res: httpx.Response
|
||||
try:
|
||||
res = await func()
|
||||
|
||||
for code in retries.status_codes:
|
||||
if "X" in code.upper():
|
||||
code_range = int(code[0])
|
||||
|
||||
status_major = res.status_code / 100
|
||||
|
||||
if code_range <= status_major < code_range + 1:
|
||||
raise TemporaryError(res)
|
||||
else:
|
||||
parsed_code = int(code)
|
||||
|
||||
if res.status_code == parsed_code:
|
||||
raise TemporaryError(res)
|
||||
except httpx.ConnectError as exception:
|
||||
if retries.config.retry_connection_errors:
|
||||
raise
|
||||
|
||||
raise PermanentError(exception) from exception
|
||||
except httpx.TimeoutException as exception:
|
||||
if retries.config.retry_connection_errors:
|
||||
raise
|
||||
|
||||
raise PermanentError(exception) from exception
|
||||
except TemporaryError:
|
||||
raise
|
||||
except Exception as exception:
|
||||
raise PermanentError(exception) from exception
|
||||
|
||||
return res
|
||||
|
||||
return await retry_with_backoff_async(
|
||||
do_request,
|
||||
retries.config.backoff.initial_interval,
|
||||
retries.config.backoff.max_interval,
|
||||
retries.config.backoff.exponent,
|
||||
retries.config.backoff.max_elapsed_time,
|
||||
)
|
||||
|
||||
return await func()
|
||||
|
||||
|
||||
def retry_with_backoff(
|
||||
func,
|
||||
initial_interval=500,
|
||||
max_interval=60000,
|
||||
exponent=1.5,
|
||||
max_elapsed_time=3600000,
|
||||
):
|
||||
start = round(time.time() * 1000)
|
||||
retries = 0
|
||||
|
||||
while True:
|
||||
try:
|
||||
return func()
|
||||
except PermanentError as exception:
|
||||
raise exception.inner
|
||||
except Exception as exception: # pylint: disable=broad-exception-caught
|
||||
now = round(time.time() * 1000)
|
||||
if now - start > max_elapsed_time:
|
||||
if isinstance(exception, TemporaryError):
|
||||
return exception.response
|
||||
|
||||
raise
|
||||
sleep = (initial_interval / 1000) * exponent**retries + random.uniform(0, 1)
|
||||
sleep = min(sleep, max_interval / 1000)
|
||||
time.sleep(sleep)
|
||||
retries += 1
|
||||
|
||||
|
||||
async def retry_with_backoff_async(
|
||||
func,
|
||||
initial_interval=500,
|
||||
max_interval=60000,
|
||||
exponent=1.5,
|
||||
max_elapsed_time=3600000,
|
||||
):
|
||||
start = round(time.time() * 1000)
|
||||
retries = 0
|
||||
|
||||
while True:
|
||||
try:
|
||||
return await func()
|
||||
except PermanentError as exception:
|
||||
raise exception.inner
|
||||
except Exception as exception: # pylint: disable=broad-exception-caught
|
||||
now = round(time.time() * 1000)
|
||||
if now - start > max_elapsed_time:
|
||||
if isinstance(exception, TemporaryError):
|
||||
return exception.response
|
||||
|
||||
raise
|
||||
sleep = (initial_interval / 1000) * exponent**retries + random.uniform(0, 1)
|
||||
sleep = min(sleep, max_interval / 1000)
|
||||
await asyncio.sleep(sleep)
|
||||
retries += 1
|
||||
@@ -0,0 +1,174 @@
|
||||
"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
|
||||
|
||||
import base64
|
||||
from typing import (
|
||||
Any,
|
||||
Dict,
|
||||
List,
|
||||
Tuple,
|
||||
)
|
||||
from pydantic import BaseModel
|
||||
from pydantic.fields import FieldInfo
|
||||
|
||||
from .metadata import (
|
||||
SecurityMetadata,
|
||||
find_field_metadata,
|
||||
)
|
||||
|
||||
|
||||
def get_security(security: Any) -> Tuple[Dict[str, str], Dict[str, List[str]]]:
|
||||
headers: Dict[str, str] = {}
|
||||
query_params: Dict[str, List[str]] = {}
|
||||
|
||||
if security is None:
|
||||
return headers, query_params
|
||||
|
||||
if not isinstance(security, BaseModel):
|
||||
raise TypeError("security must be a pydantic model")
|
||||
|
||||
sec_fields: Dict[str, FieldInfo] = security.__class__.model_fields
|
||||
for name in sec_fields:
|
||||
sec_field = sec_fields[name]
|
||||
|
||||
value = getattr(security, name)
|
||||
if value is None:
|
||||
continue
|
||||
|
||||
metadata = find_field_metadata(sec_field, SecurityMetadata)
|
||||
if metadata is None:
|
||||
continue
|
||||
if metadata.option:
|
||||
_parse_security_option(headers, query_params, value)
|
||||
return headers, query_params
|
||||
if metadata.scheme:
|
||||
# Special case for basic auth or custom auth which could be a flattened model
|
||||
if metadata.sub_type in ["basic", "custom"] and not isinstance(
|
||||
value, BaseModel
|
||||
):
|
||||
_parse_security_scheme(headers, query_params, metadata, name, security)
|
||||
else:
|
||||
_parse_security_scheme(headers, query_params, metadata, name, value)
|
||||
|
||||
return headers, query_params
|
||||
|
||||
|
||||
def _parse_security_option(
|
||||
headers: Dict[str, str], query_params: Dict[str, List[str]], option: Any
|
||||
):
|
||||
if not isinstance(option, BaseModel):
|
||||
raise TypeError("security option must be a pydantic model")
|
||||
|
||||
opt_fields: Dict[str, FieldInfo] = option.__class__.model_fields
|
||||
for name in opt_fields:
|
||||
opt_field = opt_fields[name]
|
||||
|
||||
metadata = find_field_metadata(opt_field, SecurityMetadata)
|
||||
if metadata is None or not metadata.scheme:
|
||||
continue
|
||||
_parse_security_scheme(
|
||||
headers, query_params, metadata, name, getattr(option, name)
|
||||
)
|
||||
|
||||
|
||||
def _parse_security_scheme(
|
||||
headers: Dict[str, str],
|
||||
query_params: Dict[str, List[str]],
|
||||
scheme_metadata: SecurityMetadata,
|
||||
field_name: str,
|
||||
scheme: Any,
|
||||
):
|
||||
scheme_type = scheme_metadata.scheme_type
|
||||
sub_type = scheme_metadata.sub_type
|
||||
|
||||
if isinstance(scheme, BaseModel):
|
||||
if scheme_type == "http":
|
||||
if sub_type == "basic":
|
||||
_parse_basic_auth_scheme(headers, scheme)
|
||||
return
|
||||
if sub_type == "custom":
|
||||
return
|
||||
|
||||
scheme_fields: Dict[str, FieldInfo] = scheme.__class__.model_fields
|
||||
for name in scheme_fields:
|
||||
scheme_field = scheme_fields[name]
|
||||
|
||||
metadata = find_field_metadata(scheme_field, SecurityMetadata)
|
||||
if metadata is None or metadata.field_name is None:
|
||||
continue
|
||||
|
||||
value = getattr(scheme, name)
|
||||
|
||||
_parse_security_scheme_value(
|
||||
headers, query_params, scheme_metadata, metadata, name, value
|
||||
)
|
||||
else:
|
||||
_parse_security_scheme_value(
|
||||
headers, query_params, scheme_metadata, scheme_metadata, field_name, scheme
|
||||
)
|
||||
|
||||
|
||||
def _parse_security_scheme_value(
|
||||
headers: Dict[str, str],
|
||||
query_params: Dict[str, List[str]],
|
||||
scheme_metadata: SecurityMetadata,
|
||||
security_metadata: SecurityMetadata,
|
||||
field_name: str,
|
||||
value: Any,
|
||||
):
|
||||
scheme_type = scheme_metadata.scheme_type
|
||||
sub_type = scheme_metadata.sub_type
|
||||
|
||||
header_name = security_metadata.get_field_name(field_name)
|
||||
|
||||
if scheme_type == "apiKey":
|
||||
if sub_type == "header":
|
||||
headers[header_name] = value
|
||||
elif sub_type == "query":
|
||||
query_params[header_name] = [value]
|
||||
else:
|
||||
raise ValueError("sub type {sub_type} not supported")
|
||||
elif scheme_type == "openIdConnect":
|
||||
headers[header_name] = _apply_bearer(value)
|
||||
elif scheme_type == "oauth2":
|
||||
if sub_type != "client_credentials":
|
||||
headers[header_name] = _apply_bearer(value)
|
||||
elif scheme_type == "http":
|
||||
if sub_type == "bearer":
|
||||
headers[header_name] = _apply_bearer(value)
|
||||
elif sub_type == "custom":
|
||||
return
|
||||
else:
|
||||
raise ValueError("sub type {sub_type} not supported")
|
||||
else:
|
||||
raise ValueError("scheme type {scheme_type} not supported")
|
||||
|
||||
|
||||
def _apply_bearer(token: str) -> str:
|
||||
return token.lower().startswith("bearer ") and token or f"Bearer {token}"
|
||||
|
||||
|
||||
def _parse_basic_auth_scheme(headers: Dict[str, str], scheme: Any):
|
||||
username = ""
|
||||
password = ""
|
||||
|
||||
if not isinstance(scheme, BaseModel):
|
||||
raise TypeError("basic auth scheme must be a pydantic model")
|
||||
|
||||
scheme_fields: Dict[str, FieldInfo] = scheme.__class__.model_fields
|
||||
for name in scheme_fields:
|
||||
scheme_field = scheme_fields[name]
|
||||
|
||||
metadata = find_field_metadata(scheme_field, SecurityMetadata)
|
||||
if metadata is None or metadata.field_name is None:
|
||||
continue
|
||||
|
||||
field_name = metadata.field_name
|
||||
value = getattr(scheme, name)
|
||||
|
||||
if field_name == "username":
|
||||
username = value
|
||||
if field_name == "password":
|
||||
password = value
|
||||
|
||||
data = f"{username}:{password}".encode()
|
||||
headers["Authorization"] = f"Basic {base64.b64encode(data).decode()}"
|
||||
@@ -0,0 +1,249 @@
|
||||
"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
|
||||
|
||||
from decimal import Decimal
|
||||
import functools
|
||||
import json
|
||||
import typing
|
||||
from typing import Any, Dict, List, Tuple, Union, get_args
|
||||
import typing_extensions
|
||||
from typing_extensions import get_origin
|
||||
|
||||
import httpx
|
||||
from pydantic import ConfigDict, create_model
|
||||
from pydantic_core import from_json
|
||||
|
||||
from ..types.basemodel import BaseModel, Nullable, OptionalNullable, Unset
|
||||
|
||||
|
||||
def serialize_decimal(as_str: bool):
|
||||
def serialize(d):
|
||||
# Optional[T] is a Union[T, None]
|
||||
if is_union(type(d)) and type(None) in get_args(type(d)) and d is None:
|
||||
return None
|
||||
if isinstance(d, Unset):
|
||||
return d
|
||||
|
||||
if not isinstance(d, Decimal):
|
||||
raise ValueError("Expected Decimal object")
|
||||
|
||||
return str(d) if as_str else float(d)
|
||||
|
||||
return serialize
|
||||
|
||||
|
||||
def validate_decimal(d):
|
||||
if d is None:
|
||||
return None
|
||||
|
||||
if isinstance(d, (Decimal, Unset)):
|
||||
return d
|
||||
|
||||
if not isinstance(d, (str, int, float)):
|
||||
raise ValueError("Expected string, int or float")
|
||||
|
||||
return Decimal(str(d))
|
||||
|
||||
|
||||
def serialize_float(as_str: bool):
|
||||
def serialize(f):
|
||||
# Optional[T] is a Union[T, None]
|
||||
if is_union(type(f)) and type(None) in get_args(type(f)) and f is None:
|
||||
return None
|
||||
if isinstance(f, Unset):
|
||||
return f
|
||||
|
||||
if not isinstance(f, float):
|
||||
raise ValueError("Expected float")
|
||||
|
||||
return str(f) if as_str else f
|
||||
|
||||
return serialize
|
||||
|
||||
|
||||
def validate_float(f):
|
||||
if f is None:
|
||||
return None
|
||||
|
||||
if isinstance(f, (float, Unset)):
|
||||
return f
|
||||
|
||||
if not isinstance(f, str):
|
||||
raise ValueError("Expected string")
|
||||
|
||||
return float(f)
|
||||
|
||||
|
||||
def serialize_int(as_str: bool):
|
||||
def serialize(i):
|
||||
# Optional[T] is a Union[T, None]
|
||||
if is_union(type(i)) and type(None) in get_args(type(i)) and i is None:
|
||||
return None
|
||||
if isinstance(i, Unset):
|
||||
return i
|
||||
|
||||
if not isinstance(i, int):
|
||||
raise ValueError("Expected int")
|
||||
|
||||
return str(i) if as_str else i
|
||||
|
||||
return serialize
|
||||
|
||||
|
||||
def validate_int(b):
|
||||
if b is None:
|
||||
return None
|
||||
|
||||
if isinstance(b, (int, Unset)):
|
||||
return b
|
||||
|
||||
if not isinstance(b, str):
|
||||
raise ValueError("Expected string")
|
||||
|
||||
return int(b)
|
||||
|
||||
|
||||
def validate_open_enum(is_int: bool):
|
||||
def validate(e):
|
||||
if e is None:
|
||||
return None
|
||||
|
||||
if isinstance(e, Unset):
|
||||
return e
|
||||
|
||||
if is_int:
|
||||
if not isinstance(e, int):
|
||||
raise ValueError("Expected int")
|
||||
else:
|
||||
if not isinstance(e, str):
|
||||
raise ValueError("Expected string")
|
||||
|
||||
return e
|
||||
|
||||
return validate
|
||||
|
||||
|
||||
def validate_const(v):
|
||||
def validate(c):
|
||||
# Optional[T] is a Union[T, None]
|
||||
if is_union(type(c)) and type(None) in get_args(type(c)) and c is None:
|
||||
return None
|
||||
|
||||
if v != c:
|
||||
raise ValueError(f"Expected {v}")
|
||||
|
||||
return c
|
||||
|
||||
return validate
|
||||
|
||||
|
||||
def unmarshal_json(raw, typ: Any) -> Any:
|
||||
return unmarshal(from_json(raw), typ)
|
||||
|
||||
|
||||
def unmarshal(val, typ: Any) -> Any:
|
||||
unmarshaller = create_model(
|
||||
"Unmarshaller",
|
||||
body=(typ, ...),
|
||||
__config__=ConfigDict(populate_by_name=True, arbitrary_types_allowed=True),
|
||||
)
|
||||
|
||||
m = unmarshaller(body=val)
|
||||
|
||||
# pyright: ignore[reportAttributeAccessIssue]
|
||||
return m.body # type: ignore
|
||||
|
||||
|
||||
def marshal_json(val, typ):
|
||||
if is_nullable(typ) and val is None:
|
||||
return "null"
|
||||
|
||||
marshaller = create_model(
|
||||
"Marshaller",
|
||||
body=(typ, ...),
|
||||
__config__=ConfigDict(populate_by_name=True, arbitrary_types_allowed=True),
|
||||
)
|
||||
|
||||
m = marshaller(body=val)
|
||||
|
||||
d = m.model_dump(by_alias=True, mode="json", exclude_none=True)
|
||||
|
||||
if len(d) == 0:
|
||||
return ""
|
||||
|
||||
return json.dumps(d[next(iter(d))], separators=(",", ":"))
|
||||
|
||||
|
||||
def is_nullable(field):
|
||||
origin = get_origin(field)
|
||||
if origin is Nullable or origin is OptionalNullable:
|
||||
return True
|
||||
|
||||
if not origin is Union or type(None) not in get_args(field):
|
||||
return False
|
||||
|
||||
for arg in get_args(field):
|
||||
if get_origin(arg) is Nullable or get_origin(arg) is OptionalNullable:
|
||||
return True
|
||||
|
||||
return False
|
||||
|
||||
|
||||
def is_union(obj: object) -> bool:
|
||||
"""
|
||||
Returns True if the given object is a typing.Union or typing_extensions.Union.
|
||||
"""
|
||||
return any(
|
||||
obj is typing_obj for typing_obj in _get_typing_objects_by_name_of("Union")
|
||||
)
|
||||
|
||||
|
||||
def stream_to_text(stream: httpx.Response) -> str:
|
||||
return "".join(stream.iter_text())
|
||||
|
||||
|
||||
async def stream_to_text_async(stream: httpx.Response) -> str:
|
||||
return "".join([chunk async for chunk in stream.aiter_text()])
|
||||
|
||||
|
||||
def stream_to_bytes(stream: httpx.Response) -> bytes:
|
||||
return stream.content
|
||||
|
||||
|
||||
async def stream_to_bytes_async(stream: httpx.Response) -> bytes:
|
||||
return await stream.aread()
|
||||
|
||||
|
||||
def get_pydantic_model(data: Any, typ: Any) -> Any:
|
||||
if not _contains_pydantic_model(data):
|
||||
return unmarshal(data, typ)
|
||||
|
||||
return data
|
||||
|
||||
|
||||
def _contains_pydantic_model(data: Any) -> bool:
|
||||
if isinstance(data, BaseModel):
|
||||
return True
|
||||
if isinstance(data, List):
|
||||
return any(_contains_pydantic_model(item) for item in data)
|
||||
if isinstance(data, Dict):
|
||||
return any(_contains_pydantic_model(value) for value in data.values())
|
||||
|
||||
return False
|
||||
|
||||
|
||||
@functools.cache
|
||||
def _get_typing_objects_by_name_of(name: str) -> Tuple[Any, ...]:
|
||||
"""
|
||||
Get typing objects by name from typing and typing_extensions.
|
||||
Reference: https://typing-extensions.readthedocs.io/en/latest/#runtime-use-of-types
|
||||
"""
|
||||
result = tuple(
|
||||
getattr(module, name)
|
||||
for module in (typing, typing_extensions)
|
||||
if hasattr(module, name)
|
||||
)
|
||||
if not result:
|
||||
raise ValueError(
|
||||
f"Neither typing nor typing_extensions has an object called {name!r}"
|
||||
)
|
||||
return result
|
||||
@@ -0,0 +1,24 @@
|
||||
"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
|
||||
|
||||
from typing import Any, Optional
|
||||
|
||||
import httpx
|
||||
|
||||
from .serializers import unmarshal_json
|
||||
from unstructured_client.models import errors
|
||||
|
||||
|
||||
def unmarshal_json_response(
|
||||
typ: Any, http_res: httpx.Response, body: Optional[str] = None
|
||||
) -> Any:
|
||||
if body is None:
|
||||
body = http_res.text
|
||||
try:
|
||||
return unmarshal_json(body, typ)
|
||||
except Exception as e:
|
||||
raise errors.ResponseValidationError(
|
||||
"Response validation failed",
|
||||
http_res,
|
||||
e,
|
||||
body,
|
||||
) from e
|
||||
@@ -0,0 +1,155 @@
|
||||
"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
|
||||
|
||||
from decimal import Decimal
|
||||
from typing import (
|
||||
Any,
|
||||
Dict,
|
||||
get_type_hints,
|
||||
List,
|
||||
Optional,
|
||||
Union,
|
||||
get_args,
|
||||
get_origin,
|
||||
)
|
||||
from pydantic import BaseModel
|
||||
from pydantic.fields import FieldInfo
|
||||
|
||||
from .metadata import (
|
||||
PathParamMetadata,
|
||||
find_field_metadata,
|
||||
)
|
||||
from .values import (
|
||||
_get_serialized_params,
|
||||
_is_set,
|
||||
_populate_from_globals,
|
||||
_val_to_string,
|
||||
)
|
||||
|
||||
|
||||
def generate_url(
|
||||
server_url: str,
|
||||
path: str,
|
||||
path_params: Any,
|
||||
gbls: Optional[Any] = None,
|
||||
) -> str:
|
||||
path_param_values: Dict[str, str] = {}
|
||||
|
||||
globals_already_populated = _populate_path_params(
|
||||
path_params, gbls, path_param_values, []
|
||||
)
|
||||
if _is_set(gbls):
|
||||
_populate_path_params(gbls, None, path_param_values, globals_already_populated)
|
||||
|
||||
for key, value in path_param_values.items():
|
||||
path = path.replace("{" + key + "}", value, 1)
|
||||
|
||||
return remove_suffix(server_url, "/") + path
|
||||
|
||||
|
||||
def _populate_path_params(
|
||||
path_params: Any,
|
||||
gbls: Any,
|
||||
path_param_values: Dict[str, str],
|
||||
skip_fields: List[str],
|
||||
) -> List[str]:
|
||||
globals_already_populated: List[str] = []
|
||||
|
||||
if not isinstance(path_params, BaseModel):
|
||||
return globals_already_populated
|
||||
|
||||
path_param_fields: Dict[str, FieldInfo] = path_params.__class__.model_fields
|
||||
path_param_field_types = get_type_hints(path_params.__class__)
|
||||
for name in path_param_fields:
|
||||
if name in skip_fields:
|
||||
continue
|
||||
|
||||
field = path_param_fields[name]
|
||||
|
||||
param_metadata = find_field_metadata(field, PathParamMetadata)
|
||||
if param_metadata is None:
|
||||
continue
|
||||
|
||||
param = getattr(path_params, name) if _is_set(path_params) else None
|
||||
param, global_found = _populate_from_globals(
|
||||
name, param, PathParamMetadata, gbls
|
||||
)
|
||||
if global_found:
|
||||
globals_already_populated.append(name)
|
||||
|
||||
if not _is_set(param):
|
||||
continue
|
||||
|
||||
f_name = field.alias if field.alias is not None else name
|
||||
serialization = param_metadata.serialization
|
||||
if serialization is not None:
|
||||
serialized_params = _get_serialized_params(
|
||||
param_metadata, f_name, param, path_param_field_types[name]
|
||||
)
|
||||
for key, value in serialized_params.items():
|
||||
path_param_values[key] = value
|
||||
else:
|
||||
pp_vals: List[str] = []
|
||||
if param_metadata.style == "simple":
|
||||
if isinstance(param, List):
|
||||
for pp_val in param:
|
||||
if not _is_set(pp_val):
|
||||
continue
|
||||
pp_vals.append(_val_to_string(pp_val))
|
||||
path_param_values[f_name] = ",".join(pp_vals)
|
||||
elif isinstance(param, Dict):
|
||||
for pp_key in param:
|
||||
if not _is_set(param[pp_key]):
|
||||
continue
|
||||
if param_metadata.explode:
|
||||
pp_vals.append(f"{pp_key}={_val_to_string(param[pp_key])}")
|
||||
else:
|
||||
pp_vals.append(f"{pp_key},{_val_to_string(param[pp_key])}")
|
||||
path_param_values[f_name] = ",".join(pp_vals)
|
||||
elif not isinstance(param, (str, int, float, complex, bool, Decimal)):
|
||||
param_fields: Dict[str, FieldInfo] = param.__class__.model_fields
|
||||
for name in param_fields:
|
||||
param_field = param_fields[name]
|
||||
|
||||
param_value_metadata = find_field_metadata(
|
||||
param_field, PathParamMetadata
|
||||
)
|
||||
if param_value_metadata is None:
|
||||
continue
|
||||
|
||||
param_name = (
|
||||
param_field.alias if param_field.alias is not None else name
|
||||
)
|
||||
|
||||
param_field_val = getattr(param, name)
|
||||
if not _is_set(param_field_val):
|
||||
continue
|
||||
if param_metadata.explode:
|
||||
pp_vals.append(
|
||||
f"{param_name}={_val_to_string(param_field_val)}"
|
||||
)
|
||||
else:
|
||||
pp_vals.append(
|
||||
f"{param_name},{_val_to_string(param_field_val)}"
|
||||
)
|
||||
path_param_values[f_name] = ",".join(pp_vals)
|
||||
elif _is_set(param):
|
||||
path_param_values[f_name] = _val_to_string(param)
|
||||
|
||||
return globals_already_populated
|
||||
|
||||
|
||||
def is_optional(field):
|
||||
return get_origin(field) is Union and type(None) in get_args(field)
|
||||
|
||||
|
||||
def template_url(url_with_params: str, params: Dict[str, str]) -> str:
|
||||
for key, value in params.items():
|
||||
url_with_params = url_with_params.replace("{" + key + "}", value)
|
||||
|
||||
return url_with_params
|
||||
|
||||
|
||||
def remove_suffix(input_string, suffix):
|
||||
if suffix and input_string.endswith(suffix):
|
||||
return input_string[: -len(suffix)]
|
||||
return input_string
|
||||
@@ -0,0 +1,137 @@
|
||||
"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
|
||||
|
||||
from datetime import datetime
|
||||
from enum import Enum
|
||||
from email.message import Message
|
||||
from functools import partial
|
||||
import os
|
||||
from typing import Any, Callable, Dict, List, Optional, Tuple, TypeVar, Union, cast
|
||||
|
||||
from httpx import Response
|
||||
from pydantic import BaseModel
|
||||
from pydantic.fields import FieldInfo
|
||||
|
||||
from ..types.basemodel import Unset
|
||||
|
||||
from .serializers import marshal_json
|
||||
|
||||
from .metadata import ParamMetadata, find_field_metadata
|
||||
|
||||
|
||||
def match_content_type(content_type: str, pattern: str) -> bool:
|
||||
if pattern in (content_type, "*", "*/*"):
|
||||
return True
|
||||
|
||||
msg = Message()
|
||||
msg["content-type"] = content_type
|
||||
media_type = msg.get_content_type()
|
||||
|
||||
if media_type == pattern:
|
||||
return True
|
||||
|
||||
parts = media_type.split("/")
|
||||
if len(parts) == 2:
|
||||
if pattern in (f"{parts[0]}/*", f"*/{parts[1]}"):
|
||||
return True
|
||||
|
||||
return False
|
||||
|
||||
|
||||
def match_status_codes(status_codes: List[str], status_code: int) -> bool:
|
||||
if "default" in status_codes:
|
||||
return True
|
||||
|
||||
for code in status_codes:
|
||||
if code == str(status_code):
|
||||
return True
|
||||
|
||||
if code.endswith("XX") and code.startswith(str(status_code)[:1]):
|
||||
return True
|
||||
return False
|
||||
|
||||
|
||||
T = TypeVar("T")
|
||||
|
||||
def cast_partial(typ):
|
||||
return partial(cast, typ)
|
||||
|
||||
def get_global_from_env(
|
||||
value: Optional[T], env_key: str, type_cast: Callable[[str], T]
|
||||
) -> Optional[T]:
|
||||
if value is not None:
|
||||
return value
|
||||
env_value = os.getenv(env_key)
|
||||
if env_value is not None:
|
||||
try:
|
||||
return type_cast(env_value)
|
||||
except ValueError:
|
||||
pass
|
||||
return None
|
||||
|
||||
|
||||
def match_response(
|
||||
response: Response, code: Union[str, List[str]], content_type: str
|
||||
) -> bool:
|
||||
codes = code if isinstance(code, list) else [code]
|
||||
return match_status_codes(codes, response.status_code) and match_content_type(
|
||||
response.headers.get("content-type", "application/octet-stream"), content_type
|
||||
)
|
||||
|
||||
|
||||
def _populate_from_globals(
|
||||
param_name: str, value: Any, param_metadata_type: type, gbls: Any
|
||||
) -> Tuple[Any, bool]:
|
||||
if gbls is None:
|
||||
return value, False
|
||||
|
||||
if not isinstance(gbls, BaseModel):
|
||||
raise TypeError("globals must be a pydantic model")
|
||||
|
||||
global_fields: Dict[str, FieldInfo] = gbls.__class__.model_fields
|
||||
found = False
|
||||
for name in global_fields:
|
||||
field = global_fields[name]
|
||||
if name is not param_name:
|
||||
continue
|
||||
|
||||
found = True
|
||||
|
||||
if value is not None:
|
||||
return value, True
|
||||
|
||||
global_value = getattr(gbls, name)
|
||||
|
||||
param_metadata = find_field_metadata(field, param_metadata_type)
|
||||
if param_metadata is None:
|
||||
return value, True
|
||||
|
||||
return global_value, True
|
||||
|
||||
return value, found
|
||||
|
||||
|
||||
def _val_to_string(val) -> str:
|
||||
if isinstance(val, bool):
|
||||
return str(val).lower()
|
||||
if isinstance(val, datetime):
|
||||
return str(val.isoformat().replace("+00:00", "Z"))
|
||||
if isinstance(val, Enum):
|
||||
return str(val.value)
|
||||
|
||||
return str(val)
|
||||
|
||||
|
||||
def _get_serialized_params(
|
||||
metadata: ParamMetadata, field_name: str, obj: Any, typ: type
|
||||
) -> Dict[str, str]:
|
||||
params: Dict[str, str] = {}
|
||||
|
||||
serialization = metadata.serialization
|
||||
if serialization == "json":
|
||||
params[field_name] = marshal_json(obj, typ)
|
||||
|
||||
return params
|
||||
|
||||
|
||||
def _is_set(value: Any) -> bool:
|
||||
return value is not None and not isinstance(value, Unset)
|
||||
Reference in New Issue
Block a user