From 8a2252f654c3efcaf0f20ed3bde40e6479b75fd4 Mon Sep 17 00:00:00 2001 From: Albert Iribarne Date: Fri, 17 Apr 2020 09:39:25 +0200 Subject: [PATCH] :recycle: Simplify DB base class declaration (#117) MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit * Simplify DB base class declaration * :recycle: Remove object inheritance Co-authored-by: Sebastián Ramírez --- .../backend/app/app/db/base_class.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/{{cookiecutter.project_slug}}/backend/app/app/db/base_class.py b/{{cookiecutter.project_slug}}/backend/app/app/db/base_class.py index 2228a37..3dadcb3 100644 --- a/{{cookiecutter.project_slug}}/backend/app/app/db/base_class.py +++ b/{{cookiecutter.project_slug}}/backend/app/app/db/base_class.py @@ -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)