Merge pull request #596 from CBoYXD/add_tablename_generation

Add tablename generation

Author: djeck1432

apps/dashboard_app/app/models/base.py

@@ -1,8 +1,15 @@
+from re import sub
 from sqlalchemy.dialects.postgresql import UUID as PG_UUID
 from uuid import uuid4, UUID
-from sqlalchemy.orm import DeclarativeBase, Mapped
+from sqlalchemy.orm import DeclarativeBase, Mapped, declared_attr
 from sqlalchemy import Column, MetaData
 
+
 class Base(DeclarativeBase):
     id: Mapped[UUID] = Column(PG_UUID(as_uuid=True), default=uuid4, primary_key=True)
     metadata = MetaData()
+
+    @classmethod
+    @declared_attr.directive
+    def __tablename__(cls) -> str:
+        return sub(r"(?<!^)(?=[A-Z])", "_", cls.__name__).lower()

apps/dashboard_app/app/models/watcher.py

@@ -27,7 +27,6 @@ class NotificationData(Base):
     )
 
 class TelegramLog(Base):
-    __tablename__ = "telegram_log"
 
     sent_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.now, nullable=False)
     notification_data_id: Mapped[str] = mapped_column(ForeignKey(NotificationData.id), nullable=False)