Skip to content

Commit 3c195c1

Browse files
committed
Simplify DB base class declaration fastapi#117
1 parent 85a6c25 commit 3c195c1

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed
Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,10 @@
1-
from sqlalchemy.ext.declarative import declarative_base, declared_attr
1+
from sqlalchemy.ext.declarative import as_declarative, declared_attr
22

33

4-
class CustomBase(object):
4+
@as_declarative()
5+
class Base(object):
56
# Generate __tablename__ automatically
67
@declared_attr
78
def __tablename__(cls):
89
return cls.__name__.lower()
910

10-
11-
Base = declarative_base(cls=CustomBase)

0 commit comments

Comments
 (0)