welearn-database 1.5.2__tar.gz → 1.5.3__tar.gz

{welearn_database-1.5.2 → welearn_database-1.5.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: welearn-database
-Version: 1.5.2
+Version: 1.5.3
 Summary: All stuff related to relationnal database from the WeLearn project
 License: cc-by-sa-nc
 Author: Théo

{welearn_database-1.5.2 → welearn_database-1.5.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "welearn-database"
-version = "1.5.2"
+version = "1.5.3"
 description = "All stuff related to relationnal database from the WeLearn project"
 authors = [
     {name = "Théo",email = "theo.nardin@cri-paris.org"}

welearn_database-1.5.3/welearn_database/alembic/versions/5d46d7920342_support_sub_corpus.py

@@ -0,0 +1,75 @@
+"""support_sub_corpus
+
+Revision ID: 5d46d7920342
+Revises: 95ffe7afa64c
+Create Date: 2026-09-21 14:47:28.116941
+
+"""
+
+from typing import Sequence, Union
+
+import sqlalchemy as sa
+from alembic import op
+
+# revision identifiers, used by Alembic.
+revision: str = "5d46d7920342"
+down_revision: Union[str, None] = "95ffe7afa64c"
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    op.add_column(
+        "corpus",
+        sa.Column("parent_corpus_id", sa.Uuid(), nullable=True),
+        schema="corpus_related",
+    )
+    op.create_foreign_key(
+        "corpus_parent_corpus_id_fkey",
+        "corpus",
+        "corpus",
+        ["parent_corpus_id"],
+        ["id"],
+        source_schema="corpus_related",
+        referent_schema="corpus_related",
+    )
+    op.execute("DROP VIEW grafana.corpus")
+    op.drop_column("corpus", "binary_treshold", schema="corpus_related")
+    op.execute("""
+            CREATE OR REPLACE VIEW grafana.corpus
+            AS SELECT corpus.id,
+            corpus.parent_corpus_id,
+            corpus.source_name,
+            corpus.main_url,
+            corpus.is_fix,
+            corpus.is_active,
+            corpus.category_id
+           FROM corpus_related.corpus;
+    """)
+
+
+def downgrade() -> None:
+    op.drop_constraint(
+        "corpus_parent_corpus_id_fkey", "corpus", schema="corpus_related"
+    )
+    op.execute("DROP VIEW grafana.corpus")
+    op.drop_column("corpus", "parent_corpus_id", schema="corpus_related")
+    op.add_column(
+        "corpus",
+        sa.Column(
+            "binary_treshold",
+            sa.Float(),
+            nullable=False,
+            default=0.5,
+            server_default="0.5",
+        ),
+        schema="corpus_related",
+    )
+    op.execute("""
+            CREATE OR REPLACE VIEW grafana.corpus
+            AS SELECT corpus.id,
+            corpus.source_name,
+            corpus.is_fix,
+            corpus.binary_treshold
+           FROM corpus_related.corpus;
+    """)

{welearn_database-1.5.2 → welearn_database-1.5.3}/welearn_database/data/models/corpus_related.py

@@ -19,10 +19,13 @@ class Corpus(Base):
     id: Mapped[UUID] = mapped_column(
         types.Uuid, primary_key=True, nullable=False, server_default="gen_random_uuid()"
     )
+    parent_corpus_id: Mapped[UUID | None] = mapped_column(
+        types.Uuid,
+        ForeignKey(f"{schema_name}.corpus.id"),
+    )
     source_name: Mapped[str]
     main_url: Mapped[str | None] = mapped_column(nullable=True)
     is_fix: Mapped[bool]
-    binary_treshold: Mapped[float] = mapped_column(nullable=False, default=0.5)
     is_active: Mapped[bool]
     category_id: Mapped[UUID] = mapped_column(
         types.Uuid,

{welearn_database-1.5.2 → welearn_database-1.5.3}/welearn_database/data/models/grafana.py

@@ -16,10 +16,13 @@ class Corpus(Base):
     __table_args__ = {"schema": schema_name}
     __read_only__ = True
 
-    id: Mapped[uuid.UUID] = mapped_column(UUID(as_uuid=True), primary_key=True)
-    source_name: Mapped[str] = mapped_column()
-    is_fix: Mapped[bool | None] = mapped_column(nullable=True)
-    binary_treshold: Mapped[float | None] = mapped_column(nullable=True)
+    id: Mapped[UUID] = mapped_column(UUID(as_uuid=True), primary_key=True)
+    parent_corpus_id: Mapped[UUID | None] = mapped_column(UUID(as_uuid=True))
+    source_name: Mapped[str]
+    main_url: Mapped[str | None] = mapped_column(nullable=True)
+    is_fix: Mapped[bool]
+    is_active: Mapped[bool]
+    category_id: Mapped[UUID | None] = mapped_column(UUID(as_uuid=True))
 
 
 class DocumentLatestState(Base):