welearn-database 1.1.1.dev0__tar.gz → 1.2.0.dev0__tar.gz

{welearn_database-1.1.1.dev0 → welearn_database-1.2.0.dev0}/PKG-INFO

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

{welearn_database-1.1.1.dev0 → welearn_database-1.2.0.dev0}/pyproject.toml

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

welearn_database-1.2.0.dev0/welearn_database/alembic/versions/068312e7800c_add_referrer_origin_column_to_user_and_.py

@@ -0,0 +1,45 @@
+"""add referrer origin column to user and session
+
+Revision ID: 068312e7800c
+Revises: 4f5a188dd614
+Create Date: 2026-01-12 11:45:40.359740
+
+"""
+
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision: str = "068312e7800c"
+down_revision: Union[str, None] = "4f5a188dd614"
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    op.add_column(
+        "session",
+        sa.Column("origin_referrer", sa.String(), nullable=True),
+        schema="user_related",
+    )
+    op.add_column(
+        "inferred_user",
+        sa.Column("origin_referrer", sa.String(), nullable=True),
+        schema="user_related",
+    )
+
+
+def downgrade() -> None:
+    op.drop_column(
+        "inferred_user",
+        "origin_referrer",
+        schema="user_related",
+    )
+    op.drop_column(
+        "session",
+        "origin_referrer",
+        schema="user_related",
+    )

{welearn_database-1.1.1.dev0 → welearn_database-1.2.0.dev0}/welearn_database/data/models/document_related.py

@@ -475,7 +475,7 @@ class QtyDocumentInQdrantPerCorpus(Base):
     __read_only__ = True
 
     source_name: Mapped[str] = mapped_column(primary_key=True)
-    count: Mapped[int] = mapped_column(primary_key=True)
+    count: Mapped[int]
 
 
 class QtyDocumentPerCorpus(Base):
@@ -484,4 +484,4 @@ class QtyDocumentPerCorpus(Base):
     __read_only__ = True
 
     source_name: Mapped[str] = mapped_column(primary_key=True)
-    count: Mapped[int] = mapped_column(primary_key=True)
+    count: Mapped[int]

{welearn_database-1.1.1.dev0 → welearn_database-1.2.0.dev0}/welearn_database/data/models/grafana.py

@@ -21,6 +21,7 @@ class Corpus(Base):
     is_fix: Mapped[bool | None] = mapped_column(nullable=True)
     binary_treshold: Mapped[float | None] = mapped_column(nullable=True)
 
+
 class DocumentLatestState(Base):
     __tablename__ = "document_latest_state"
     __table_args__ = {"schema": schema_name}
@@ -93,6 +94,7 @@ class ProcessState(Base):
     created_at: Mapped[datetime] = mapped_column()
     operation_order: Mapped[int] = mapped_column()
 
+
 class QtyEndpointsPerUser(Base):
     __tablename__ = "qty_endpoints_per_user"
     __table_args__ = {"schema": schema_name}
@@ -106,7 +108,9 @@ class QtySessionEndpointPerUser(Base):
     __table_args__ = {"schema": schema_name}
     __read_only__ = True
 
-    inferred_user_id: Mapped[uuid.UUID] = mapped_column(UUID(as_uuid=True), primary_key=True)
+    inferred_user_id: Mapped[uuid.UUID] = mapped_column(
+        UUID(as_uuid=True), primary_key=True
+    )
     host: Mapped[str] = mapped_column(primary_key=True)
     count_sessions: Mapped[int] = mapped_column()
     count_endpoints: Mapped[int] = mapped_column()

{welearn_database-1.1.1.dev0 → welearn_database-1.2.0.dev0}/welearn_database/data/models/user_related.py

@@ -169,6 +169,7 @@ class Session(Base):
         ForeignKey("user_related.inferred_user.id"),
         nullable=False,
     )
+    origin_referrer: Mapped[str | None]
     created_at: Mapped[datetime] = mapped_column(
         TIMESTAMP(timezone=False),
         nullable=False,
@@ -186,6 +187,7 @@ class InferredUser(Base):
     id: Mapped[UUID] = mapped_column(
         types.Uuid, primary_key=True, nullable=False, server_default="gen_random_uuid()"
     )
+    origin_referrer: Mapped[str | None]
     created_at: Mapped[datetime] = mapped_column(
         TIMESTAMP(timezone=False),
         nullable=False,