Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
56 changes: 24 additions & 32 deletions sqlspec/_typing.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,15 +6,7 @@

from collections.abc import Iterable, Mapping
from enum import Enum
from typing import (
Any,
ClassVar,
Final,
Protocol,
Union,
cast,
runtime_checkable,
)
from typing import Any, ClassVar, Final, Optional, Protocol, Union, cast, runtime_checkable

from typing_extensions import Literal, TypeVar, dataclass_transform

Expand Down Expand Up @@ -50,9 +42,9 @@ def model_dump(
self,
/,
*,
include: "Optional[Any]" = None,
exclude: "Optional[Any]" = None,
context: "Optional[Any]" = None,
include: Optional[Any] = None,
exclude: Optional[Any] = None,
context: Optional[Any] = None,
by_alias: bool = False,
exclude_unset: bool = False,
exclude_defaults: bool = False,
Expand All @@ -68,9 +60,9 @@ def model_dump_json(
self,
/,
*,
include: "Optional[Any]" = None,
exclude: "Optional[Any]" = None,
context: "Optional[Any]" = None,
include: Optional[Any] = None,
exclude: Optional[Any] = None,
context: Optional[Any] = None,
by_alias: bool = False,
exclude_unset: bool = False,
exclude_defaults: bool = False,
Expand All @@ -90,9 +82,9 @@ def __init__(
self,
type: Any, # noqa: A002
*,
config: "Optional[Any]" = None,
config: Optional[Any] = None,
_parent_depth: int = 2,
module: "Optional[str]" = None,
module: Optional[str] = None,
) -> None:
"""Init"""

Expand All @@ -101,10 +93,10 @@ def validate_python(
object: Any,
/,
*,
strict: "Optional[bool]" = None,
from_attributes: "Optional[bool]" = None,
context: "Optional[dict[str, Any]]" = None,
experimental_allow_partial: "Union[bool, Literal['off', 'on', 'trailing-strings']]" = False,
strict: Optional[bool] = None,
from_attributes: Optional[bool] = None,
context: Optional[dict[str, Any]] = None,
experimental_allow_partial: Union[bool, Literal["off", "on", "trailing-strings"]] = False,
) -> "T_co":
"""Stub"""
return cast("T_co", object)
Expand Down Expand Up @@ -136,16 +128,16 @@ class FailFast: # type: ignore[no-redef]
class Struct(Protocol): # type: ignore[no-redef]
"""Placeholder Implementation"""

__struct_fields__: "ClassVar[tuple[str, ...]]"
__struct_fields__: ClassVar[tuple[str, ...]]

def convert( # type: ignore[no-redef]
obj: Any,
type: "Union[Any, type[T]]", # noqa: A002
type: Union[Any, type[T]], # noqa: A002
*,
strict: bool = True,
from_attributes: bool = False,
dec_hook: "Optional[Callable[[type, Any], Any]]" = None,
builtin_types: "Union[Iterable[type], None]" = None,
dec_hook: Optional[Callable[[type, Any], Any]] = None,
builtin_types: Optional[Iterable[type]] = None,
str_keys: bool = False,
) -> "Union[T, Any]":
"""Placeholder implementation"""
Expand All @@ -172,10 +164,10 @@ class DTOData(Protocol[T]): # type: ignore[no-redef]
def __init__(self, backend: Any, data_as_builtins: Any) -> None:
"""Placeholder init"""

def create_instance(self, **kwargs: Any) -> "T":
def create_instance(self, **kwargs: Any) -> T:
return cast("T", kwargs)

def update_instance(self, instance: "T", **kwargs: Any) -> "T":
def update_instance(self, instance: T, **kwargs: Any) -> T:
"""Placeholder implementation"""
return cast("T", kwargs)

Expand Down Expand Up @@ -214,15 +206,15 @@ def to_string(self) -> str: ...
def from_arrays(
self,
arrays: list[Any],
names: "Optional[list[str]]" = None,
schema: "Optional[Any]" = None,
metadata: "Optional[Mapping[str, Any]]" = None,
names: Optional[list[str]] = None,
schema: Optional[Any] = None,
metadata: Optional[Mapping[str, Any]] = None,
) -> Any: ...
def from_pydict(
self,
mapping: dict[str, Any],
schema: "Optional[Any]" = None,
metadata: "Optional[Mapping[str, Any]]" = None,
schema: Optional[Any] = None,
metadata: Optional[Mapping[str, Any]] = None,
) -> Any: ...
def from_batches(self, batches: Iterable[Any], schema: Optional[Any] = None) -> Any: ...

Expand Down