refactor

2026-01-17 11:40:39 +03:00 · 2022-09-12 13:24:15 +02:00
parent 102771aee5
commit 9b1938c450
35 changed files with 50 additions and 50 deletions
--- a/app/models/shakespeare.py
+++ b/app/models/shakespeare.py
@@ -0,0 +1,134 @@
+from sqlalchemy import (
+    Column,
+    ForeignKey,
+    ForeignKeyConstraint,
+    Integer,
+    PrimaryKeyConstraint,
+    String,
+    Table,
+    Text,
+    UniqueConstraint,
+    select,
+)
+from sqlalchemy.ext.asyncio import AsyncSession
+from sqlalchemy.orm import relationship
+
+from app.models.base import Base
+
+metadata = Base.metadata
+
+
+class Character(Base):
+    __tablename__ = "character"
+    __table_args__ = (PrimaryKeyConstraint("id", name="character_pkey"), {"schema": "shakespeare"})
+
+    id = Column(String(32))
+    name = Column(String(64), nullable=False)
+    speech_count = Column(Integer, nullable=False)
+    abbrev = Column(String(32))
+    description = Column(String(2056))
+
+    work = relationship("Work", secondary="shakespeare.character_work", back_populates="character")
+    paragraph = relationship("Paragraph", back_populates="character")
+
+
+class Wordform(Base):
+    __tablename__ = "wordform"
+    __table_args__ = (PrimaryKeyConstraint("id", name="wordform_pkey"), {"schema": "shakespeare"})
+
+    id = Column(Integer)
+    plain_text = Column(String(64), nullable=False)
+    phonetic_text = Column(String(64), nullable=False)
+    stem_text = Column(String(64), nullable=False)
+    occurences = Column(Integer, nullable=False)
+
+
+class Work(Base):
+    __tablename__ = "work"
+    __table_args__ = (PrimaryKeyConstraint("id", name="work_pkey"), {"schema": "shakespeare"})
+
+    id = Column(String(32))
+    title = Column(String(32), nullable=False)
+    long_title = Column(String(64), nullable=False)
+    year = Column(Integer, nullable=False)
+    genre_type = Column(String(1), nullable=False)
+    source = Column(String(16), nullable=False)
+    total_words = Column(Integer, nullable=False)
+    total_paragraphs = Column(Integer, nullable=False)
+    notes = Column(Text)
+
+    character = relationship("Character", secondary="shakespeare.character_work", back_populates="work")
+    chapter = relationship("Chapter", back_populates="work")
+    paragraph = relationship("Paragraph", back_populates="work")
+
+
+class Chapter(Base):
+    __tablename__ = "chapter"
+    __table_args__ = (
+        ForeignKeyConstraint(["work_id"], ["shakespeare.work.id"], name="chapter_work_id_fkey"),
+        PrimaryKeyConstraint("id", name="chapter_pkey"),
+        UniqueConstraint(
+            "work_id", "section_number", "chapter_number", name="chapter_work_id_section_number_chapter_number_key"
+        ),
+        {"schema": "shakespeare"},
+    )
+
+    id = Column(Integer)
+    work_id = Column(ForeignKey("shakespeare.work.id"), nullable=False)
+    section_number = Column(Integer, nullable=False)
+    chapter_number = Column(Integer, nullable=False)
+    description = Column(String(256), nullable=False)
+
+    work = relationship("Work", back_populates="chapter")
+    paragraph = relationship("Paragraph", back_populates="chapter")
+
+
+t_character_work = Table(
+    "character_work",
+    metadata,
+    Column("character_id", ForeignKey("shakespeare.character.id"), nullable=False),
+    Column("work_id", ForeignKey("shakespeare.work.id"), nullable=False),
+    ForeignKeyConstraint(["character_id"], ["shakespeare.character.id"], name="character_work_character_id_fkey"),
+    ForeignKeyConstraint(["work_id"], ["shakespeare.work.id"], name="character_work_work_id_fkey"),
+    PrimaryKeyConstraint("character_id", "work_id", name="character_work_pkey"),
+    schema="shakespeare",
+)
+
+
+class Paragraph(Base):
+    __tablename__ = "paragraph"
+    __table_args__ = (
+        ForeignKeyConstraint(["character_id"], ["shakespeare.character.id"], name="paragraph_character_id_fkey"),
+        ForeignKeyConstraint(
+            ["work_id", "section_number", "chapter_number"],
+            ["shakespeare.chapter.work_id", "shakespeare.chapter.section_number", "shakespeare.chapter.chapter_number"],
+            name="paragraph_chapter_fkey",
+        ),
+        ForeignKeyConstraint(["work_id"], ["shakespeare.work.id"], name="paragraph_work_id_fkey"),
+        PrimaryKeyConstraint("id", name="paragraph_pkey"),
+        {"schema": "shakespeare"},
+    )
+
+    id = Column(Integer)
+    work_id = Column(ForeignKey("shakespeare.work.id"), nullable=False)
+    paragraph_num = Column(Integer, nullable=False)
+    character_id = Column(ForeignKey("shakespeare.character.id"), nullable=False)
+    plain_text = Column(Text, nullable=False)
+    phonetic_text = Column(Text, nullable=False)
+    stem_text = Column(Text, nullable=False)
+    paragraph_type = Column(String(1), nullable=False)
+    section_number = Column(Integer, nullable=False)
+    chapter_number = Column(Integer, nullable=False)
+    char_count = Column(Integer, nullable=False)
+    word_count = Column(Integer, nullable=False)
+
+    character = relationship("Character", back_populates="paragraph", lazy="selectin")
+    chapter = relationship("Chapter", back_populates="paragraph", lazy="selectin")
+    work = relationship("Work", back_populates="paragraph", lazy="selectin")
+
+    @classmethod
+    async def find(cls, db_session: AsyncSession, character: str):
+        stmt = select(cls).join(Character).join(Chapter).join(Work).where(Character.name == character)
+        result = await db_session.execute(stmt)
+        instance = result.scalars().all()
+        return instance