From 3294839b2a9defef407e95b763703dddf0dda6c8 Mon Sep 17 00:00:00 2001 From: Jakub Miazek Date: Mon, 18 Sep 2023 08:20:58 +0200 Subject: [PATCH] use DeclarativeBase class for models Base --- app/models/base.py | 16 ++-------------- 1 file changed, 2 insertions(+), 14 deletions(-) diff --git a/app/models/base.py b/app/models/base.py index 133e117..53920e0 100644 --- a/app/models/base.py +++ b/app/models/base.py @@ -4,22 +4,10 @@ from asyncpg import UniqueViolationError from fastapi import HTTPException, status from sqlalchemy.exc import SQLAlchemyError, IntegrityError from sqlalchemy.ext.asyncio import AsyncSession -from sqlalchemy.orm import as_declarative, declared_attr +from sqlalchemy.orm import declared_attr, DeclarativeBase -@as_declarative() -class BaseReadOnly: - id: Any - __name__: str - # Generate __tablename__ automatically - - @declared_attr - def __tablename__(self) -> str: - return self.__name__.lower() - - -@as_declarative() -class Base: +class Base(DeclarativeBase): id: Any __name__: str # Generate __tablename__ automatically