♻️ Simplify DB base class declaration (#117)

* Simplify DB base class declaration

* ♻️ Remove object inheritance

Co-authored-by: Sebastián Ramírez <tiangolo@gmail.com>
This commit is contained in:
Albert Iribarne
2020-04-17 09:39:25 +02:00
committed by GitHub
parent 8ff61e813e
commit 8a2252f654

View File

@@ -1,11 +1,9 @@
from sqlalchemy.ext.declarative import declarative_base, declared_attr
from sqlalchemy.ext.declarative import as_declarative, declared_attr
class CustomBase(object):
@as_declarative()
class Base:
# Generate __tablename__ automatically
@declared_attr
def __tablename__(cls):
return cls.__name__.lower()
Base = declarative_base(cls=CustomBase)