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