65 lines
2.3 KiB
Python
65 lines
2.3 KiB
Python
|
import uuid
|
||
|
|
from datetime import datetime, timezone
|
||
|
|
|
||
|
|
from sqlalchemy import Boolean, ForeignKey, String, Text, UniqueConstraint, DateTime
|
||
|
|
from sqlalchemy.orm import Mapped, mapped_column
|
||
|
|
from sqlalchemy.dialects.postgresql import UUID
|
||
|
|
|
||
|
|
from ..database import Base
|
||
|
|
|
||
|
|
|
||
|
|
class UserLanguagePairEntity(Base):
|
||
|
|
__tablename__ = "user_language_pair"
|
||
|
|
__table_args__ = (UniqueConstraint("user_id", "source_lang", "target_lang"),)
|
||
|
|
|
||
|
|
id: Mapped[uuid.UUID] = mapped_column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
|
||
|
|
user_id: Mapped[uuid.UUID] = mapped_column(
|
||
|
|
UUID(as_uuid=True),
|
||
|
|
ForeignKey("users.id", ondelete="CASCADE"),
|
||
|
|
nullable=False,
|
||
|
|
index=True,
|
||
|
|
)
|
||
|
|
source_lang: Mapped[str] = mapped_column(String(2), nullable=False)
|
||
|
|
target_lang: Mapped[str] = mapped_column(String(2), nullable=False)
|
||
|
|
|
||
|
|
|
||
|
|
class LearnableWordBankEntryEntity(Base):
|
||
|
|
__tablename__ = "learnable_word_bank_entry"
|
||
|
|
|
||
|
|
id: Mapped[uuid.UUID] = mapped_column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
|
||
|
|
user_id: Mapped[uuid.UUID] = mapped_column(
|
||
|
|
UUID(as_uuid=True),
|
||
|
|
ForeignKey("users.id", ondelete="CASCADE"),
|
||
|
|
nullable=False,
|
||
|
|
index=True,
|
||
|
|
)
|
||
|
|
language_pair_id: Mapped[uuid.UUID] = mapped_column(
|
||
|
|
UUID(as_uuid=True),
|
||
|
|
ForeignKey("user_language_pair.id", ondelete="CASCADE"),
|
||
|
|
nullable=False,
|
||
|
|
index=True,
|
||
|
|
)
|
||
|
|
sense_id: Mapped[uuid.UUID | None] = mapped_column(
|
||
|
|
UUID(as_uuid=True),
|
||
|
|
ForeignKey("dictionary_sense.id", ondelete="SET NULL"),
|
||
|
|
nullable=True,
|
||
|
|
index=True,
|
||
|
|
)
|
||
|
|
wordform_id: Mapped[uuid.UUID | None] = mapped_column(
|
||
|
|
UUID(as_uuid=True),
|
||
|
|
ForeignKey("dictionary_wordform.id", ondelete="SET NULL"),
|
||
|
|
nullable=True,
|
||
|
|
)
|
||
|
|
surface_text: Mapped[str] = mapped_column(Text, nullable=False)
|
||
|
|
is_phrase: Mapped[bool] = mapped_column(Boolean, nullable=False, default=False)
|
||
|
|
entry_pathway: Mapped[str] = mapped_column(Text, nullable=False)
|
||
|
|
source_article_id: Mapped[uuid.UUID | None] = mapped_column(
|
||
|
|
UUID(as_uuid=True), nullable=True
|
||
|
|
)
|
||
|
|
disambiguation_status: Mapped[str] = mapped_column(Text, nullable=False, default="pending")
|
||
|
|
created_at: Mapped[datetime] = mapped_column(
|
||
|
|
DateTime(timezone=True),
|
||
|
|
nullable=False,
|
||
|
|
default=lambda: datetime.now(timezone.utc),
|
||
|
|
)
|