finance_dev 0.9.0__py3-none-any.whl

finance_dev/utils/retries.py

@@ -0,0 +1,373 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+# @generated-id: 9cba85fd95c5
+
+import asyncio
+import random
+import time
+from datetime import datetime
+from email.utils import parsedate_to_datetime
+from typing import List, Optional
+
+import httpx
+
+
+class BackoffStrategy:
+    """Exponential backoff strategy configuration."""
+
+    initial_interval: int
+    max_interval: int
+    exponent: float
+    max_elapsed_time: int
+    jitter_ms: Optional[int]
+
+    def __init__(
+        self,
+        initial_interval: int,
+        max_interval: int,
+        exponent: float,
+        max_elapsed_time: int,
+        jitter_ms: Optional[int] = None,
+    ):
+        """Initialize a backoff strategy.
+
+        Args:
+            initial_interval: Initial retry interval in milliseconds.
+            max_interval: Maximum retry interval in milliseconds.
+            exponent: Base of the exponential backoff; the interval grows as
+                ``initial_interval * exponent ** retries``.
+            max_elapsed_time: Maximum total elapsed time in milliseconds.
+            jitter_ms: Additive jitter bound in milliseconds. When set, adds a random
+                value in ``[0, jitter_ms]`` to each computed backoff interval (default
+                ``+[0, 1s]``).
+
+        Note:
+            When a response carries a ``Retry-After`` or ``retry-after-ms`` header,
+            that delay is used as-is and the sleep-shaping parameters
+            (``initial_interval``, ``max_interval``, ``exponent``, ``jitter_ms``) are
+            ignored for that attempt.
+        """
+        if jitter_ms is not None and jitter_ms < 0:
+            raise ValueError("jitter_ms must be >= 0")
+        self.initial_interval = initial_interval
+        self.max_interval = max_interval
+        self.exponent = exponent
+        self.max_elapsed_time = max_elapsed_time
+        self.jitter_ms = jitter_ms
+
+
+class RetryConfig:
+    """Runtime retry configuration."""
+
+    strategy: str
+    backoff: BackoffStrategy
+    retry_connection_errors: bool
+    status_codes_override: Optional[List[str]]
+
+    def __init__(
+        self,
+        strategy: str,
+        backoff: BackoffStrategy,
+        retry_connection_errors: bool,
+        status_codes_override: Optional[List[str]] = None,
+    ):
+        """Initialize a retry configuration.
+
+        Args:
+            strategy: Retry strategy: ``"none"`` or ``"backoff"``.
+            backoff: Backoff parameters.
+            retry_connection_errors: Whether to also retry transport-level connection errors.
+            status_codes_override: Retryable HTTP status codes that take precedence over the
+                per-operation defaults when non-empty.
+        """
+        self.strategy = strategy
+        self.backoff = backoff
+        self.retry_connection_errors = retry_connection_errors
+        self.status_codes_override = status_codes_override
+
+
+class Retries:
+    config: RetryConfig
+    status_codes: List[str]
+
+    def __init__(self, config: RetryConfig, status_codes: List[str]):
+        self.config = config
+        self.status_codes = config.status_codes_override or status_codes
+
+
+class TemporaryError(Exception):
+    response: httpx.Response
+    retry_after: Optional[int]
+
+    def __init__(self, response: httpx.Response):
+        self.response = response
+        self.retry_after = _parse_retry_after_header(response)
+
+
+class PermanentError(Exception):
+    inner: Exception
+
+    def __init__(self, inner: Exception):
+        self.inner = inner
+
+
+_TRANSPORT_ERROR_NAMES = frozenset({"NetworkError", "TimeoutException"})
+_TRANSPORT_ERROR_BASES = frozenset({"TransportError", "RequestError", "HTTPError"})
+
+
+def _is_transport_error(exception: BaseException) -> bool:
+    """Report whether an exception is a connection or timeout failure."""
+    if isinstance(exception, (httpx.NetworkError, httpx.TimeoutException)):
+        return True
+    names = {base.__name__ for base in type(exception).__mro__}
+    return bool(names & _TRANSPORT_ERROR_NAMES) and _TRANSPORT_ERROR_BASES <= names
+
+
+def _parse_retry_after_header(response: httpx.Response) -> Optional[int]:
+    """Parse Retry-After header from response.
+
+    Returns:
+        Retry interval in milliseconds, or None if header is missing or invalid.
+    """
+    retry_after_header = response.headers.get("retry-after")
+    if not retry_after_header:
+        return None
+
+    try:
+        seconds = float(retry_after_header)
+        return round(seconds * 1000)
+    except ValueError:
+        pass
+
+    try:
+        retry_date = parsedate_to_datetime(retry_after_header)
+        delta = (retry_date - datetime.now(retry_date.tzinfo)).total_seconds()
+        return round(max(0, delta) * 1000)
+    except (ValueError, TypeError):
+        pass
+
+    return None
+
+
+def _parse_retry_after_ms_header(response: httpx.Response) -> Optional[int]:
+    retry_after_ms_header = response.headers.get("retry-after-ms")
+    if not retry_after_ms_header:
+        return None
+
+    try:
+        milliseconds = float(retry_after_ms_header)
+        if milliseconds >= 0:
+            return round(milliseconds)
+    except (OverflowError, ValueError):
+        pass
+
+    return None
+
+
+def _get_sleep_interval(
+    exception: Exception,
+    initial_interval: int,
+    max_interval: int,
+    exponent: float,
+    retries: int,
+    jitter_ms: Optional[int] = None,
+) -> float:
+    """Get sleep interval for retry with exponential backoff.
+
+    Args:
+        exception: The exception that triggered the retry.
+        initial_interval: Initial retry interval in milliseconds.
+        max_interval: Maximum retry interval in milliseconds.
+        exponent: Base for exponential backoff calculation.
+        retries: Current retry attempt count.
+        jitter_ms: Additive jitter bound in ms; see ``BackoffStrategy.jitter_ms``.
+
+    Returns:
+        Sleep interval in seconds.
+    """
+    if (
+        isinstance(exception, TemporaryError)
+        and exception.retry_after is not None
+        and exception.retry_after > 0
+    ):
+        return exception.retry_after / 1000
+
+    sleep = (initial_interval / 1000) * exponent**retries
+    if jitter_ms is not None:
+        sleep += random.uniform(0, jitter_ms / 1000)
+    else:
+        sleep += random.uniform(0, 1)
+    return min(sleep, max_interval / 1000)
+
+
+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 TemporaryError:
+                raise
+            except Exception as exception:
+                if (
+                    _is_transport_error(exception)
+                    and retries.config.retry_connection_errors
+                ):
+                    raise
+
+                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,
+            retries.config.backoff.jitter_ms,
+        )
+
+    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 TemporaryError:
+                raise
+            except Exception as exception:
+                if (
+                    _is_transport_error(exception)
+                    and retries.config.retry_connection_errors
+                ):
+                    raise
+
+                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,
+            retries.config.backoff.jitter_ms,
+        )
+
+    return await func()
+
+
+def retry_with_backoff(
+    func,
+    initial_interval=500,
+    max_interval=60000,
+    exponent=1.5,
+    max_elapsed_time=3600000,
+    jitter_ms=None,
+):
+    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
+
+            if isinstance(exception, TemporaryError):
+                retry_after_ms = _parse_retry_after_ms_header(exception.response)
+                if retry_after_ms is not None:
+                    exception.retry_after = retry_after_ms
+                exception.response.close()
+            sleep = _get_sleep_interval(
+                exception,
+                initial_interval,
+                max_interval,
+                exponent,
+                retries,
+                jitter_ms=jitter_ms,
+            )
+            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,
+    jitter_ms=None,
+):
+    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
+
+            if isinstance(exception, TemporaryError):
+                retry_after_ms = _parse_retry_after_ms_header(exception.response)
+                if retry_after_ms is not None:
+                    exception.retry_after = retry_after_ms
+                await exception.response.aclose()
+            sleep = _get_sleep_interval(
+                exception,
+                initial_interval,
+                max_interval,
+                exponent,
+                retries,
+                jitter_ms=jitter_ms,
+            )
+            await asyncio.sleep(sleep)
+            retries += 1

finance_dev/utils/security.py

@@ -0,0 +1,199 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+# @generated-id: 9311a2317fd6
+
+import base64
+from typing import (
+    Any,
+    Dict,
+    List,
+    Optional,
+    Tuple,
+)
+from pydantic import BaseModel
+from pydantic.fields import FieldInfo
+
+from .metadata import (
+    SecurityMetadata,
+    find_field_metadata,
+)
+
+
+def get_security(
+    security: Any, allowed_fields: Optional[List[str]] = None
+) -> 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
+    sec_field_names = (
+        list(sec_fields.keys()) if allowed_fields is None else allowed_fields
+    )
+
+    for name in sec_field_names:
+        if name not in sec_fields:
+            continue
+
+        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)
+
+            if not metadata.composite:
+                return headers, query_params
+
+    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
+
+        value = getattr(option, name)
+        if (
+            metadata.scheme_type == "http"
+            and metadata.sub_type == "basic"
+            and not isinstance(value, BaseModel)
+        ):
+            _parse_basic_auth_scheme(headers, option)
+            return
+
+        _parse_security_scheme(headers, query_params, metadata, name, value)
+
+
+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 == "basic":
+            headers[header_name] = 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()}"