some refactors

This commit is contained in:
collerek
2020-09-17 15:21:11 +02:00
parent 1cfd061e17
commit 12487e1089
5 changed files with 10 additions and 22 deletions

View File

@ -1,5 +1,5 @@
import inspect
from typing import List, Optional, Set, TYPE_CHECKING, Type, TypeVar
from typing import List, Optional, Set, TYPE_CHECKING, Type, TypeVar, Union
import ormar
from ormar.exceptions import RelationshipInstanceError
@ -94,7 +94,9 @@ class ModelTableProxy:
return name
@staticmethod
def resolve_relation_field(item: "Model", related: "Model") -> Type[Field]:
def resolve_relation_field(
item: Union["Model", Type["Model"]], related: Union["Model", Type["Model"]]
) -> Type[Field]:
name = ModelTableProxy.resolve_relation_name(item, related)
to_field = item.Meta.model_fields.get(name)
if not to_field: # pragma no cover