blob: d2653a9208695558013b985ba4f987a791ba97c5 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
|
import typing
from sqlalchemy import Integer
from sqlalchemy import Text
from sqlalchemy.orm import DeclarativeBase
from sqlalchemy.orm import declared_attr
from sqlalchemy.orm import Mapped
from sqlalchemy.orm import mapped_column
class Base(DeclarativeBase):
pass
class HasRelatedDataMixin:
@declared_attr
def related_data(cls) -> Mapped[str]:
return mapped_column(Text(), deferred=True)
class User(HasRelatedDataMixin, Base):
@declared_attr.directive
def __tablename__(cls) -> str:
return "user"
@declared_attr.directive
def __mapper_args__(cls) -> typing.Dict[str, typing.Any]:
return {}
id = mapped_column(Integer, primary_key=True)
class Foo(Base):
__tablename__ = "foo"
id = mapped_column(Integer, primary_key=True)
u1 = User()
if typing.TYPE_CHECKING:
# EXPECTED_TYPE: str
reveal_type(User.__tablename__)
# EXPECTED_TYPE: str
reveal_type(Foo.__tablename__)
# EXPECTED_TYPE: str
reveal_type(u1.related_data)
# EXPECTED_TYPE: InstrumentedAttribute[str]
reveal_type(User.related_data)
|
