blob: 78b7a9b633480c1c0619924e2a748fd18b8a5533 (
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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
|
from typing import List
from typing import Optional
from sqlalchemy import Column
from sqlalchemy.orm import Mapped
from sqlalchemy.orm import registry
from sqlalchemy.orm import relationship
from sqlalchemy.orm.decl_api import declared_attr
from sqlalchemy.sql.schema import ForeignKey
from sqlalchemy.sql.sqltypes import Integer
from sqlalchemy.sql.sqltypes import String
reg: registry = registry()
@reg.mapped
class User:
__tablename__ = "user"
id = Column(Integer, primary_key=True)
name = Column(String(50))
name3 = Column(String(50))
addresses: List["Address"] = relationship("Address")
@reg.mapped
class SubUser(User):
__tablename__ = "subuser"
id: int = Column(ForeignKey("user.id"), primary_key=True)
@declared_attr
def name(cls) -> Column[String]:
return Column(String(50))
@declared_attr
def name2(cls) -> Mapped[Optional[str]]:
return Column(String(50))
@declared_attr
def name3(cls) -> Mapped[str]:
return Column(String(50))
subname = Column(String)
@reg.mapped
class Address:
__tablename__ = "address"
id = Column(Integer, primary_key=True)
user_id: int = Column(ForeignKey("user.id"))
email = Column(String(50))
user = relationship(User, uselist=False)
s1 = SubUser()
# EXPECTED_MYPY: Incompatible types in assignment (expression has type "Optional[str]", variable has type "str" # noqa
x1: str = s1.name
# EXPECTED_MYPY: Incompatible types in assignment (expression has type "Optional[str]", variable has type "str") # noqa
x2: str = s1.name2
x3: str = s1.name3
u1 = User()
# EXPECTED_MYPY: Incompatible types in assignment (expression has type "Optional[str]", variable has type "str") # noqa
x4: str = u1.name3
|
