import sys
import warnings
from enum import Enum
from typing import (
TYPE_CHECKING,
Any,
Callable,
Dict,
Generic,
List,
Optional,
Tuple,
Type,
TypeVar,
Union,
overload,
)
from pypika.terms import Term
from tortoise.exceptions import ConfigurationError, ValidationError
from tortoise.validators import Validator
if TYPE_CHECKING: # pragma: nocoverage
from tortoise.models import Model
if sys.version_info >= (3, 11):
from enum import StrEnum
else: # pragma: no cover
[docs] class StrEnum(str, Enum):
__str__ = str.__str__
VALUE = TypeVar("VALUE")
[docs]class OnDelete(StrEnum):
CASCADE = "CASCADE"
RESTRICT = "RESTRICT"
SET_NULL = "SET NULL"
SET_DEFAULT = "SET DEFAULT"
NO_ACTION = "NO ACTION"
CASCADE = OnDelete.CASCADE
RESTRICT = OnDelete.RESTRICT
SET_NULL = OnDelete.SET_NULL
SET_DEFAULT = OnDelete.SET_DEFAULT
NO_ACTION = OnDelete.NO_ACTION
class _FieldMeta(type):
# TODO: Require functions to return field instances instead of this hack
def __new__(mcs, name: str, bases: Tuple[Type, ...], attrs: dict):
if len(bases) > 1 and bases[0] is Field:
# Instantiate class with only the 1st base class (should be Field)
cls = type.__new__(mcs, name, (bases[0],), attrs)
# All other base classes are our meta types, we store them in class attributes
cls.field_type = bases[1] if len(bases) == 2 else Union[bases[1:]] # type: ignore
return cls
return type.__new__(mcs, name, bases, attrs)
[docs]class Field(Generic[VALUE], metaclass=_FieldMeta):
"""
Base Field type.
:param source_field: Provide a source_field name if the DB column name needs to be
something specific instead of generated off the field name.
:param generated: Is this field DB-generated?
:param primary_key: Is this field a Primary Key? Can only have a single such field on the Model,
and if none is specified it will autogenerate a default primary key called ``id``.
:param null: Is this field nullable?
:param default: A default value for the field if not specified on Model creation.
This can also be a callable for dynamic defaults in which case we will call it.
The default value will not be part of the schema.
:param unique: Is this field unique?
:param db_index: Should this field be indexed by itself?
:param description: Field description. Will also appear in ``Tortoise.describe_model()``
and as DB comments in the generated DDL.
:param validators: Validators for this field.
**Class Attributes:**
These attributes needs to be defined when defining an actual field type.
.. attribute:: field_type
:annotation: Type[Any]
The Python type the field is.
If adding a type as a mixin, _FieldMeta will automatically set this to that.
.. attribute:: indexable
:annotation: bool = True
Is the field indexable? Set to False if this field can't be indexed reliably.
.. attribute:: has_db_field
:annotation: bool = True
Does this field have a direct corresponding DB column? Or is the field virtualized?
.. attribute:: skip_to_python_if_native
:annotation: bool = False
If the DB driver natively supports this Python type, should we skip it?
This is for optimization purposes only, where we don't need to force type conversion
between Python and the DB.
.. attribute:: allows_generated
:annotation: bool = False
Is this field able to be DB-generated?
.. attribute:: function_cast
:annotation: Optional[pypika.Term] = None
A casting term that we need to apply in case the DB needs emulation help.
.. attribute:: SQL_TYPE
:annotation: str
The SQL type as a string that the DB will use.
.. attribute:: GENERATED_SQL
:annotation: str
The SQL that instructs the DB to auto-generate this field.
Required if ``allows_generated`` is ``True``.
**Per-DB overrides:**
One can specify per-DB overrides of any of the class attributes,
or the ``to_db_value`` or ``to_python_value`` methods.
To do so, specify a inner class in the form of :samp:`class _db__{SQL_DIALECT}:` like so:
.. code-block:: py3
class _db_sqlite:
SQL_TYPE = "VARCHAR(40)"
skip_to_python_if_native = False
def function_cast(self, term: Term) -> Term:
return functions.Cast(term, SqlTypes.NUMERIC)
Tortoise will then use the overridden attributes/functions for that dialect.
If you need a dynamic attribute, you can use a property.
"""
# Field_type is a readonly property for the instance, it is set by _FieldMeta
field_type: Type[Any] = None # type: ignore
indexable: bool = True
has_db_field: bool = True
skip_to_python_if_native: bool = False
allows_generated: bool = False
function_cast: Optional[Callable[[Term], Term]] = None
SQL_TYPE: str = None # type: ignore
GENERATED_SQL: str = None # type: ignore
# These methods are just to make IDE/Linters happy:
if TYPE_CHECKING:
def __new__(cls, *args: Any, **kwargs: Any) -> "Field[VALUE]":
return super().__new__(cls)
@overload
def __get__(self, instance: None, owner: Type["Model"]) -> "Field[VALUE]": ...
@overload
def __get__(self, instance: "Model", owner: Type["Model"]) -> VALUE: ...
def __get__(
self, instance: Optional["Model"], owner: Type["Model"]
) -> "Field[VALUE] | VALUE": ...
def __set__(self, instance: "Model", value: VALUE) -> None: ...
def __init__(
self,
source_field: Optional[str] = None,
generated: bool = False,
primary_key: Optional[bool] = None,
null: bool = False,
default: Any = None,
unique: bool = False,
db_index: Optional[bool] = None,
description: Optional[str] = None,
model: "Optional[Model]" = None,
validators: Optional[List[Union[Validator, Callable]]] = None,
**kwargs: Any,
) -> None:
if (index := kwargs.pop("index", None)) is not None:
if db_index is None:
warnings.warn(
"`index` is deprecated, please use `db_index` instead",
DeprecationWarning,
stacklevel=2,
)
db_index = index
elif db_index != index:
raise ConfigurationError(
f"{self.__class__.__name__} can't set both db_index and index"
)
if not self.indexable and (unique or db_index):
raise ConfigurationError(f"{self.__class__.__name__} can't be indexed")
if (pk := kwargs.pop("pk", None)) is not None:
if primary_key is None:
warnings.warn(
"`pk` is deprecated, please use `primary_key` instead",
DeprecationWarning,
stacklevel=2,
)
primary_key = pk
elif primary_key != pk:
raise ConfigurationError(
f"{self.__class__.__name__} can't set both primary_key and pk"
)
if null:
if pk:
raise ConfigurationError(
f"{self.__class__.__name__} can't be both null=True and pk=True"
)
if primary_key:
raise ConfigurationError(
f"{self.__class__.__name__} can't be both null=True and primary_key=True"
)
if primary_key:
db_index = True
unique = True
self.source_field = source_field
self.generated = generated
self.pk = bool(primary_key)
self.default = default
self.null = null
self.unique = unique
self.index = bool(db_index)
self.model_field_name = ""
self.description = description
self.docstring: Optional[str] = None
self.validators: List[Union[Validator, Callable]] = validators or []
# TODO: consider making this not be set from constructor
self.model: Type["Model"] = model # type: ignore
self.reference: "Optional[Field]" = None
[docs] def to_db_value(self, value: Any, instance: "Union[Type[Model], Model]") -> Any:
"""
Converts from the Python type to the DB type.
:param value: Current python value in model.
:param instance: Model class or Model instance provided to look up.
Due to metacoding, to determine if this is an instance reliably, please do a:
.. code-block:: py3
if hasattr(instance, "_saved_in_db"):
"""
if value is not None and not isinstance(value, self.field_type):
value = self.field_type(value) # pylint: disable=E1102
self.validate(value)
return value
[docs] def to_python_value(self, value: Any) -> Any:
"""
Converts from the DB type to the Python type.
:param value: Value from DB
"""
if value is not None and not isinstance(value, self.field_type):
value = self.field_type(value) # pylint: disable=E1102
self.validate(value)
return value
[docs] def validate(self, value: Any):
"""
Validate whether given value is valid
:param value: Value to be validation
:raises ValidationError: If validator check is not passed
"""
for v in self.validators:
if self.null and value is None:
continue
try:
if isinstance(value, Enum):
v(value.value)
else:
v(value)
except ValidationError as exc:
raise ValidationError(f"{self.model_field_name}: {exc}")
@property
def required(self) -> bool:
"""
Returns ``True`` if the field is required to be provided.
It needs to be non-nullable and not have a default or be DB-generated to be required.
"""
return self.default is None and not self.null and not self.generated
@property
def constraints(self) -> dict:
"""
Returns a dict with constraints defined in the Pydantic/JSONSchema format.
"""
return {}
def _get_dialects(self) -> Dict[str, dict]:
ret = {}
for dialect in [key for key in dir(self) if key.startswith("_db_")]:
item = {}
cls = getattr(self, dialect)
try:
cls = cls(self)
except TypeError:
pass
for key, val in cls.__dict__.items():
if not key.startswith("_"):
item[key] = val
ret[dialect[4:]] = item
return ret
[docs] def get_db_field_types(self) -> Optional[Dict[str, str]]:
"""
Returns the DB types for this field.
:return: A dictionary that is keyed by dialect.
A blank dialect `""` means it is the default DB field type.
"""
if not self.has_db_field: # pragma: nocoverage
return None
return {
"": getattr(self, "SQL_TYPE"),
**{
dialect: _db["SQL_TYPE"]
for dialect, _db in self._get_dialects().items()
if "SQL_TYPE" in _db
},
}
[docs] def get_for_dialect(self, dialect: str, key: str) -> Any:
"""
Returns a field by dialect override.
:param dialect: The requested SQL Dialect.
:param key: The attribute/method name.
"""
dialect_data = self._get_dialects().get(dialect, {})
return dialect_data.get(key, getattr(self, key, None))
[docs] def describe(self, serializable: bool) -> dict:
"""
Describes the field.
:param serializable:
``False`` if you want raw python objects,
``True`` for JSON-serializable data. (Defaults to ``True``)
:return:
A dictionary containing the field description.
(This assumes ``serializable=True``, which is the default):
.. code-block:: python3
{
"name": str # Field name
"field_type": str # Field type
"db_column": str # Name of DB column
# Optional: Only for pk/data fields
"raw_field": str # Name of raw field of the Foreign Key
# Optional: Only for Foreign Keys
"db_field_types": dict # DB Field types for default and DB overrides
"python_type": str # Python type
"generated": bool # Is the field generated by the DB?
"nullable": bool # Is the column nullable?
"unique": bool # Is the field unique?
"indexed": bool # Is the field indexed?
"default": ... # The default value (coerced to int/float/str/bool/null)
"description": str # Description of the field (nullable)
"docstring": str # Field docstring (nullable)
}
When ``serializable=False`` is specified some fields are not coerced to valid
JSON types. The changes are:
.. code-block:: python3
{
"field_type": Field # The Field class used
"python_type": Type # The actual Python type
"default": ... # The default value as native type OR a callable
}
"""
def _type_name(typ: Type) -> str:
if typ.__module__ == "builtins":
return typ.__name__
if typ.__module__ == "typing":
return str(typ).replace("typing.", "")
return f"{typ.__module__}.{typ.__name__}"
def type_name(typ: Any) -> Union[str, List[str]]:
try:
return typ._meta.full_name
except (AttributeError, TypeError):
pass
try:
return _type_name(typ)
except AttributeError:
try:
return [_type_name(_typ) for _typ in typ] # pragma: nobranch
except TypeError:
return str(typ)
def default_name(default: Any) -> Optional[Union[int, float, str, bool]]:
if isinstance(default, (int, float, str, bool, type(None))):
return default
if callable(default):
return f"<function {default.__module__}.{default.__name__}>"
return str(default)
field_type = getattr(self, "related_model", self.field_type)
desc = {
"name": self.model_field_name,
"field_type": self.__class__.__name__ if serializable else self.__class__,
"db_column": self.source_field or self.model_field_name,
"python_type": type_name(field_type) if serializable else field_type,
"generated": self.generated,
"nullable": self.null,
"unique": self.unique,
"indexed": self.index or self.unique,
"default": default_name(self.default) if serializable else self.default,
"description": self.description,
"docstring": self.docstring,
"constraints": self.constraints,
}
if self.has_db_field:
desc["db_field_types"] = self.get_db_field_types()
return desc