feat: add more models to match workflows

backend/alembic/versions/04e02312343d_add_user_question_models.py

@@ -0,0 +1,66 @@
+"""Add user question models
+
+
+Revision ID: 04e02312343d
+Revises: d369cd69a23b
+Create Date: 2024-09-22 21:35:48.778303
+
+"""
+
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision: str = "04e02312343d"
+down_revision: Union[str, None] = "d369cd69a23b"
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.create_table(
+        "user_question",
+        sa.Column("id", sa.Integer(), nullable=False),
+        sa.Column("question", sa.String(), nullable=False),
+        sa.Column("user_id", sa.Integer(), nullable=False),
+        sa.ForeignKeyConstraint(
+            ["user_id"],
+            ["user.id"],
+        ),
+        sa.PrimaryKeyConstraint("id"),
+    )
+    op.create_table(
+        "answer",
+        sa.Column("id", sa.Integer(), nullable=False),
+        sa.Column("user_question_id", sa.Integer(), nullable=False),
+        sa.ForeignKeyConstraint(
+            ["user_question_id"],
+            ["user_question.id"],
+        ),
+        sa.PrimaryKeyConstraint("id"),
+    )
+    op.create_table(
+        "point",
+        sa.Column("id", sa.Integer(), nullable=False),
+        sa.Column("title", sa.String(), nullable=False),
+        sa.Column("body", sa.String(), nullable=False),
+        sa.Column("answer_id", sa.Integer(), nullable=False),
+        sa.ForeignKeyConstraint(
+            ["answer_id"],
+            ["answer.id"],
+        ),
+        sa.PrimaryKeyConstraint("id"),
+    )
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_table("point")
+    op.drop_table("answer")
+    op.drop_table("user_question")
+    # ### end Alembic commands ###

backend/alembic/versions/680d9bea2168_add_note_models.py

@@ -0,0 +1,40 @@
+"""Add note models
+
+Revision ID: 680d9bea2168
+Revises: 04e02312343d
+Create Date: 2024-09-22 22:07:11.979617
+
+"""
+
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision: str = "680d9bea2168"
+down_revision: Union[str, None] = "04e02312343d"
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.create_table(
+        "note",
+        sa.Column("id", sa.Integer(), nullable=False),
+        sa.Column("content", sa.String(), nullable=False),
+        sa.Column("start_index", sa.Integer(), nullable=False),
+        sa.Column("end_index", sa.Integer(), nullable=False),
+        sa.Column("parent_id", sa.Integer(), nullable=False),
+        sa.Column("parent_type", sa.String(), nullable=False),
+        sa.PrimaryKeyConstraint("id"),
+    )
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_table("note")
+    # ### end Alembic commands ###

backend/alembic/versions/8912e0e896bf_add_analysis_table.py

@@ -0,0 +1,58 @@
+"""Add Analysis table
+
+Revision ID: 8912e0e896bf
+Revises: ff5af4d7ce04
+Create Date: 2024-09-22 18:29:26.096419
+
+"""
+
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision: str = "8912e0e896bf"
+down_revision: Union[str, None] = "ff5af4d7ce04"
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.create_table(
+        "analysis",
+        sa.Column("event_id", sa.Integer(), nullable=False),
+        sa.Column("category_id", sa.Integer(), nullable=False),
+        sa.Column("content", sa.String(), nullable=False),
+        sa.ForeignKeyConstraint(
+            ["category_id"],
+            ["category.id"],
+        ),
+        sa.ForeignKeyConstraint(
+            ["event_id"],
+            ["event.id"],
+        ),
+        sa.PrimaryKeyConstraint("event_id", "category_id"),
+    )
+    op.drop_table("event_category")
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.create_table(
+        "event_category",
+        sa.Column("event_id", sa.INTEGER(), autoincrement=False, nullable=False),
+        sa.Column("category_id", sa.INTEGER(), autoincrement=False, nullable=False),
+        sa.ForeignKeyConstraint(
+            ["category_id"], ["category.id"], name="event_category_category_id_fkey"
+        ),
+        sa.ForeignKeyConstraint(
+            ["event_id"], ["event.id"], name="event_category_event_id_fkey"
+        ),
+        sa.PrimaryKeyConstraint("event_id", "category_id", name="event_category_pkey"),
+    )
+    op.drop_table("analysis")
+    # ### end Alembic commands ###

backend/alembic/versions/8c458d0adb18_add_gp_questions_tables.py

@@ -0,0 +1,57 @@
+"""Add GP Questions tables
+
+Revision ID: 8c458d0adb18
+Revises: 8912e0e896bf
+Create Date: 2024-09-22 18:49:41.011371
+
+"""
+
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision: str = "8c458d0adb18"
+down_revision: Union[str, None] = "8912e0e896bf"
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.create_table(
+        "gp_question",
+        sa.Column("id", sa.Integer(), nullable=False),
+        sa.Column("question", sa.String(), nullable=False),
+        sa.Column("is_llm_generated", sa.Boolean(), nullable=False),
+        sa.Column("event_id", sa.Integer(), nullable=True),
+        sa.ForeignKeyConstraint(
+            ["event_id"],
+            ["event.id"],
+        ),
+        sa.PrimaryKeyConstraint("id"),
+    )
+    op.create_table(
+        "gp_question_categories",
+        sa.Column("gp_question_id", sa.Integer(), nullable=False),
+        sa.Column("category_id", sa.Integer(), nullable=False),
+        sa.ForeignKeyConstraint(
+            ["category_id"],
+            ["category.id"],
+        ),
+        sa.ForeignKeyConstraint(
+            ["gp_question_id"],
+            ["gp_question.id"],
+        ),
+        sa.PrimaryKeyConstraint("gp_question_id", "category_id"),
+    )
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_table("gp_question_categories")
+    op.drop_table("gp_question")
+    # ### end Alembic commands ###

backend/alembic/versions/b8f3f95c78b6_add_image_url_to_article.py

@@ -0,0 +1,31 @@
+"""Add image url to article
+
+Revision ID: b8f3f95c78b6
+Revises: 8c458d0adb18
+Create Date: 2024-09-22 18:56:47.024304
+
+"""
+
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision: str = "b8f3f95c78b6"
+down_revision: Union[str, None] = "8c458d0adb18"
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.add_column("article", sa.Column("image_url", sa.String(), nullable=False))
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_column("article", "image_url")
+    # ### end Alembic commands ###

backend/alembic/versions/bc5fcbe47713_remove_analysis_field_from_event.py

@@ -0,0 +1,34 @@
+"""Remove analysis field from event
+
+Revision ID: bc5fcbe47713
+Revises: f624d58d3f42
+Create Date: 2024-09-22 22:33:38.823000
+
+"""
+
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision: str = "bc5fcbe47713"
+down_revision: Union[str, None] = "f624d58d3f42"
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_column("event", "analysis")
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.add_column(
+        "event",
+        sa.Column("analysis", sa.VARCHAR(), autoincrement=False, nullable=False),
+    )
+    # ### end Alembic commands ###

backend/alembic/versions/d369cd69a23b_add_rating_to_event.py

@@ -0,0 +1,31 @@
+"""Add rating to event
+
+Revision ID: d369cd69a23b
+Revises: b8f3f95c78b6
+Create Date: 2024-09-22 18:59:11.283743
+
+"""
+
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision: str = "d369cd69a23b"
+down_revision: Union[str, None] = "b8f3f95c78b6"
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.add_column("event", sa.Column("rating", sa.Integer(), nullable=False))
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_column("event", "rating")
+    # ### end Alembic commands ###

backend/alembic/versions/f624d58d3f42_add_user_id_to_note.py

@@ -0,0 +1,33 @@
+"""Add user id to note
+
+Revision ID: f624d58d3f42
+Revises: 680d9bea2168
+Create Date: 2024-09-22 22:29:21.220643
+
+"""
+
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision: str = "f624d58d3f42"
+down_revision: Union[str, None] = "680d9bea2168"
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.add_column("note", sa.Column("user_id", sa.Integer(), nullable=False))
+    op.create_foreign_key(None, "note", "user", ["user_id"], ["id"])
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_constraint(None, "note", type_="foreignkey")
+    op.drop_column("note", "user_id")
+    # ### end Alembic commands ###