blob: a471edf6c7e3997860faca348f7f8e35d214b860 (
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
|
from sqlalchemy import Column
from sqlalchemy import Integer
from sqlalchemy import String
from sqlalchemy.orm import declarative_base
from sqlalchemy.orm import registry
reg: registry = registry()
# TODO: also reg.as_declarative_base()
Base = declarative_base()
class HasUpdatedAt:
updated_at = Column(Integer)
@reg.mapped
class Foo(HasUpdatedAt):
__tablename__ = "foo"
id: int = Column(Integer(), primary_key=True)
name: str = Column(String)
class Bar(HasUpdatedAt, Base):
__tablename__ = "bar"
id = Column(Integer(), primary_key=True)
num = Column(Integer)
Foo.updated_at.in_([1, 2, 3])
Bar.updated_at.in_([1, 2, 3])
f1 = Foo(name="name", updated_at=5)
b1 = Bar(num=5, updated_at=6)
# test that we detected this as an unmapped mixin
# EXPECTED_MYPY: Unexpected keyword argument "updated_at" for "HasUpdatedAt"
HasUpdatedAt(updated_at=5)
|
