70 lines
1.8 KiB
Python
70 lines
1.8 KiB
Python
from typing import Any, List, Optional, TYPE_CHECKING, Union
|
|
|
|
import sqlalchemy
|
|
from pydantic import Field
|
|
|
|
from ormar import ModelDefinitionError # noqa I101
|
|
|
|
if TYPE_CHECKING: # pragma no cover
|
|
from ormar.models import Model
|
|
from ormar.models import NewBaseModel
|
|
|
|
|
|
class BaseField:
|
|
__type__ = None
|
|
|
|
column_type: sqlalchemy.Column
|
|
constraints: List = []
|
|
|
|
primary_key: bool
|
|
autoincrement: bool
|
|
nullable: bool
|
|
index: bool
|
|
unique: bool
|
|
pydantic_only: bool
|
|
|
|
default: Any
|
|
server_default: Any
|
|
|
|
@classmethod
|
|
def default_value(cls) -> Optional[Field]:
|
|
if cls.is_auto_primary_key():
|
|
return Field(default=None)
|
|
if cls.has_default():
|
|
default = cls.default if cls.default is not None else cls.server_default
|
|
if callable(default):
|
|
return Field(default_factory=default)
|
|
else:
|
|
return Field(default=default)
|
|
return None
|
|
|
|
@classmethod
|
|
def has_default(cls) -> bool:
|
|
return cls.default is not None or cls.server_default is not None
|
|
|
|
@classmethod
|
|
def is_auto_primary_key(cls) -> bool:
|
|
if cls.primary_key:
|
|
return cls.autoincrement
|
|
return False
|
|
|
|
@classmethod
|
|
def get_column(cls, name: str) -> sqlalchemy.Column:
|
|
return sqlalchemy.Column(
|
|
name,
|
|
cls.column_type,
|
|
*cls.constraints,
|
|
primary_key=cls.primary_key,
|
|
nullable=cls.nullable and not cls.primary_key,
|
|
index=cls.index,
|
|
unique=cls.unique,
|
|
default=cls.default,
|
|
server_default=cls.server_default,
|
|
)
|
|
|
|
@classmethod
|
|
def expand_relationship(
|
|
cls, value: Any, child: Union["Model", "NewBaseModel"], to_register: bool = True
|
|
) -> Any:
|
|
return value
|