Files
postbot/app/models/templates.py
2025-08-17 11:44:54 +09:00

33 lines
1.7 KiB
Python

from __future__ import annotations
from datetime import datetime
from typing import Optional
from sqlalchemy import DateTime, Enum, ForeignKey, String, UniqueConstraint, func, Text, JSON, Boolean
from sqlalchemy.orm import Mapped, mapped_column
from sqlalchemy.sql.sqltypes import Enum as EnumType # Явный импорт Enum для типизации
from app.db.session import Base
from app.models.post import PostType
from enum import Enum as PyEnum
class TemplateVisibility(str, PyEnum):
private = "private"
org = "org"
public = "public"
class Template(Base):
__tablename__ = "templates"
__table_args__ = (
UniqueConstraint("owner_id", "name", name="uq_template_owner_name"),
)
id: Mapped[int] = mapped_column(primary_key=True)
owner_id: Mapped[int] = mapped_column(ForeignKey("users.id", ondelete="CASCADE"), index=True)
name: Mapped[str] = mapped_column(String(64))
title: Mapped[Optional[str]] = mapped_column(String(128))
type: Mapped[PostType] = mapped_column(EnumType(PostType), default=PostType.text)
content: Mapped[str] = mapped_column(Text)
keyboard_tpl: Mapped[Optional[list[dict]]] = mapped_column(JSON, nullable=True)
parse_mode: Mapped[Optional[str]] = mapped_column(String(16), default="HTML")
visibility: Mapped[TemplateVisibility] = mapped_column(EnumType(TemplateVisibility), default=TemplateVisibility.private)
is_archived: Mapped[bool] = mapped_column(Boolean, default=False)
created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), server_default=func.now())
updated_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), server_default=func.now(), onupdate=func.now())