liniting and applying black
This commit is contained in:
collerek
133
orm/fields.py
133
orm/fields.py
@ -1,14 +1,15 @@
|
||||
import datetime
|
||||
import decimal
|
||||
from typing import List, Optional, TYPE_CHECKING, Type, Any, Union
|
||||
|
||||
import sqlalchemy
|
||||
from pydantic import Json, BaseModel
|
||||
from pydantic.fields import ModelField
|
||||
from typing import Any, List, Optional, TYPE_CHECKING, Type, Union
|
||||
|
||||
import orm
|
||||
from orm.exceptions import ModelDefinitionError, RelationshipInstanceError
|
||||
|
||||
from pydantic import BaseModel, Json
|
||||
from pydantic.fields import ModelField
|
||||
|
||||
import sqlalchemy
|
||||
|
||||
if TYPE_CHECKING: # pragma no cover
|
||||
from orm.models import Model
|
||||
|
||||
@ -16,33 +17,39 @@ if TYPE_CHECKING: # pragma no cover
|
||||
class BaseField:
|
||||
__type__ = None
|
||||
|
||||
def __init__(self, *args, **kwargs) -> None:
|
||||
name = kwargs.pop('name', None)
|
||||
def __init__(self, *args: Any, **kwargs: Any) -> None:
|
||||
name = kwargs.pop("name", None)
|
||||
args = list(args)
|
||||
if args:
|
||||
if isinstance(args[0], str):
|
||||
if name is not None:
|
||||
raise ModelDefinitionError('Column name cannot be passed positionally and as a keyword.')
|
||||
raise ModelDefinitionError(
|
||||
"Column name cannot be passed positionally and as a keyword."
|
||||
)
|
||||
name = args.pop(0)
|
||||
|
||||
self.name = name
|
||||
self.primary_key = kwargs.pop('primary_key', False)
|
||||
self.autoincrement = kwargs.pop('autoincrement', self.primary_key and self.__type__ == int)
|
||||
self.primary_key = kwargs.pop("primary_key", False)
|
||||
self.autoincrement = kwargs.pop(
|
||||
"autoincrement", self.primary_key and self.__type__ == int
|
||||
)
|
||||
|
||||
self.nullable = kwargs.pop('nullable', not self.primary_key)
|
||||
self.default = kwargs.pop('default', None)
|
||||
self.server_default = kwargs.pop('server_default', None)
|
||||
self.nullable = kwargs.pop("nullable", not self.primary_key)
|
||||
self.default = kwargs.pop("default", None)
|
||||
self.server_default = kwargs.pop("server_default", None)
|
||||
|
||||
self.index = kwargs.pop('index', None)
|
||||
self.unique = kwargs.pop('unique', None)
|
||||
self.index = kwargs.pop("index", None)
|
||||
self.unique = kwargs.pop("unique", None)
|
||||
|
||||
self.pydantic_only = kwargs.pop('pydantic_only', False)
|
||||
self.pydantic_only = kwargs.pop("pydantic_only", False)
|
||||
if self.pydantic_only and self.primary_key:
|
||||
raise ModelDefinitionError('Primary key column cannot be pydantic only.')
|
||||
raise ModelDefinitionError("Primary key column cannot be pydantic only.")
|
||||
|
||||
@property
|
||||
def is_required(self) -> bool:
|
||||
return not self.nullable and not self.has_default and not self.is_auto_primary_key
|
||||
return (
|
||||
not self.nullable and not self.has_default and not self.is_auto_primary_key
|
||||
)
|
||||
|
||||
@property
|
||||
def default_value(self) -> Any:
|
||||
@ -81,16 +88,19 @@ class BaseField:
|
||||
def get_constraints(self) -> Optional[List]:
|
||||
return []
|
||||
|
||||
def expand_relationship(self, value, child) -> Any:
|
||||
def expand_relationship(self, value: Any, child: "Model") -> Any:
|
||||
return value
|
||||
|
||||
|
||||
class String(BaseField):
|
||||
__type__ = str
|
||||
|
||||
def __init__(self, *args, **kwargs):
|
||||
assert 'length' in kwargs, 'length is required'
|
||||
self.length = kwargs.pop('length')
|
||||
def __init__(self, *args: Any, **kwargs: Any) -> None:
|
||||
if "length" not in kwargs:
|
||||
raise ModelDefinitionError(
|
||||
"Param length is required for String model field."
|
||||
)
|
||||
self.length = kwargs.pop("length")
|
||||
super().__init__(*args, **kwargs)
|
||||
|
||||
def get_column_type(self) -> sqlalchemy.Column:
|
||||
@ -163,27 +173,41 @@ class BigInteger(BaseField):
|
||||
class Decimal(BaseField):
|
||||
__type__ = decimal.Decimal
|
||||
|
||||
def __init__(self, *args, **kwargs):
|
||||
assert 'precision' in kwargs, 'precision is required'
|
||||
assert 'length' in kwargs, 'length is required'
|
||||
self.length = kwargs.pop('length')
|
||||
self.precision = kwargs.pop('precision')
|
||||
def __init__(self, *args: Any, **kwargs: Any) -> None:
|
||||
if "length" not in kwargs or "precision" not in kwargs:
|
||||
raise ModelDefinitionError(
|
||||
"Params length and precision are required for Decimal model field."
|
||||
)
|
||||
self.length = kwargs.pop("length")
|
||||
self.precision = kwargs.pop("precision")
|
||||
super().__init__(*args, **kwargs)
|
||||
|
||||
def get_column_type(self) -> sqlalchemy.Column:
|
||||
return sqlalchemy.DECIMAL(self.length, self.precision)
|
||||
|
||||
|
||||
def create_dummy_instance(fk: Type['Model'], pk: int = None) -> 'Model':
|
||||
def create_dummy_instance(fk: Type["Model"], pk: int = None) -> "Model":
|
||||
init_dict = {fk.__pkname__: pk or -1}
|
||||
init_dict = {**init_dict, **{k: create_dummy_instance(v.to)
|
||||
for k, v in fk.__model_fields__.items()
|
||||
if isinstance(v, ForeignKey) and not v.nullable and not v.virtual}}
|
||||
init_dict = {
|
||||
**init_dict,
|
||||
**{
|
||||
k: create_dummy_instance(v.to)
|
||||
for k, v in fk.__model_fields__.items()
|
||||
if isinstance(v, ForeignKey) and not v.nullable and not v.virtual
|
||||
},
|
||||
}
|
||||
return fk(**init_dict)
|
||||
|
||||
|
||||
class ForeignKey(BaseField):
|
||||
def __init__(self, to, name: str = None, related_name: str = None, nullable: bool = True, virtual: bool = False):
|
||||
def __init__(
|
||||
self,
|
||||
to: Type["Model"],
|
||||
name: str = None,
|
||||
related_name: str = None,
|
||||
nullable: bool = True,
|
||||
virtual: bool = False,
|
||||
) -> None:
|
||||
super().__init__(nullable=nullable, name=name)
|
||||
self.virtual = virtual
|
||||
self.related_name = related_name
|
||||
@ -201,11 +225,16 @@ class ForeignKey(BaseField):
|
||||
to_column = self.to.__model_fields__[self.to.__pkname__]
|
||||
return to_column.get_column_type()
|
||||
|
||||
def expand_relationship(self, value, child) -> Union['Model', List['Model']]:
|
||||
def expand_relationship(
|
||||
self, value: Any, child: "Model"
|
||||
) -> Union["Model", List["Model"]]:
|
||||
if not isinstance(value, (self.to, dict, int, str, list)) or (
|
||||
isinstance(value, orm.models.Model) and not isinstance(value, self.to)):
|
||||
isinstance(value, orm.models.Model) and not isinstance(value, self.to)
|
||||
):
|
||||
raise RelationshipInstanceError(
|
||||
'Relationship model can be build only from orm.Model, dict and integer or string (pk).')
|
||||
"Relationship model can be build only from orm.Model, "
|
||||
"dict and integer or string (pk)."
|
||||
)
|
||||
if isinstance(value, list) and not isinstance(value, self.to):
|
||||
model = [self.expand_relationship(val, child) for val in value]
|
||||
return model
|
||||
@ -217,19 +246,27 @@ class ForeignKey(BaseField):
|
||||
else:
|
||||
model = create_dummy_instance(fk=self.to, pk=value)
|
||||
|
||||
child_model_name = self.related_name or child.__class__.__name__.lower() + 's'
|
||||
model._orm_relationship_manager.add_relation(model.__class__.__name__.lower(),
|
||||
child.__class__.__name__.lower(),
|
||||
model, child, virtual=self.virtual)
|
||||
child_model_name = self.related_name or child.__class__.__name__.lower() + "s"
|
||||
model._orm_relationship_manager.add_relation(
|
||||
model.__class__.__name__.lower(),
|
||||
child.__class__.__name__.lower(),
|
||||
model,
|
||||
child,
|
||||
virtual=self.virtual,
|
||||
)
|
||||
|
||||
if child_model_name not in model.__fields__ \
|
||||
and child.__class__.__name__.lower() not in model.__fields__:
|
||||
model.__fields__[child_model_name] = ModelField(name=child_model_name,
|
||||
type_=Optional[child.__pydantic_model__],
|
||||
model_config=child.__pydantic_model__.__config__,
|
||||
class_validators=child.__pydantic_model__.__validators__)
|
||||
model.__model_fields__[child_model_name] = ForeignKey(child.__class__,
|
||||
name=child_model_name,
|
||||
virtual=True)
|
||||
if (
|
||||
child_model_name not in model.__fields__
|
||||
and child.__class__.__name__.lower() not in model.__fields__
|
||||
):
|
||||
model.__fields__[child_model_name] = ModelField(
|
||||
name=child_model_name,
|
||||
type_=Optional[child.__pydantic_model__],
|
||||
model_config=child.__pydantic_model__.__config__,
|
||||
class_validators=child.__pydantic_model__.__validators__,
|
||||
)
|
||||
model.__model_fields__[child_model_name] = ForeignKey(
|
||||
child.__class__, name=child_model_name, virtual=True
|
||||
)
|
||||
|
||||
return model
|
||||
|
||||
Reference in New Issue
Block a user