33 lines
1.0 KiB
Python
33 lines
1.0 KiB
Python
from sqlalchemy.orm import mapped_column, relationship
|
|
from sqlalchemy.orm.attributes import Mapped
|
|
from sqlalchemy.orm.properties import ForeignKey
|
|
from sqlalchemy.types import Integer, String
|
|
import app_db
|
|
|
|
|
|
class TeamIntegration(app_db.BaseModel):
|
|
__tablename__ = "team_integrations"
|
|
|
|
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
|
team_id: Mapped[int] = mapped_column(Integer, ForeignKey("teams.id"))
|
|
integration_type: Mapped[str]
|
|
|
|
team: Mapped["Team"] = relationship(back_populates="integrations")
|
|
|
|
__mapper_args__ = {
|
|
"polymorphic_identity": "team_integrations",
|
|
"polymorphic_on": "integration_type",
|
|
}
|
|
|
|
class TeamDiscordIntegration(TeamIntegration):
|
|
__tablename__ = "team_discord_integrations"
|
|
|
|
integration_id: Mapped[int] = mapped_column(ForeignKey("team_integrations.id"), primary_key=True)
|
|
webhook_url: Mapped[str] = mapped_column(String(255))
|
|
|
|
__mapper_args__ = {
|
|
"polymorphic_identity": "team_discord_integrations",
|
|
}
|
|
|
|
from models.team import Team
|