db: comment alembic migration (pinterest#1252)

pinterest#1252

package.json

@@ -1,6 +1,6 @@
 {
     "name": "querybook",
-    "version": "3.23.2",
+    "version": "3.24.0",
     "description": "A Big Data Webapp",
     "private": true,
     "scripts": {

querybook/migrations/versions/3d9f98ca7ec1_add_comment_feature.py

@@ -0,0 +1,74 @@
+"""add comment feature
+
+Revision ID: 3d9f98ca7ec1
+Revises: 7f6cdb3621f7
+Create Date: 2023-05-22 22:06:30.972865
+
+"""
+from alembic import op
+import sqlalchemy as sa
+from sqlalchemy.dialects import mysql
+
+# revision identifiers, used by Alembic.
+revision = "3d9f98ca7ec1"
+down_revision = "7f6cdb3621f7"
+branch_labels = None
+depends_on = None
+
+
+def upgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.create_table(
+        "comment",
+        sa.Column("id", sa.Integer(), nullable=False),
+        sa.Column("created_at", sa.DateTime(), nullable=True),
+        sa.Column("updated_at", sa.DateTime(), nullable=True),
+        sa.Column("created_by", sa.Integer(), nullable=True),
+        sa.Column("text", sa.Text(length=16777215), nullable=True),
+        sa.Column("parent_comment_id", sa.Integer(), nullable=True),
+        sa.ForeignKeyConstraint(["created_by"], ["user.id"], ondelete="SET NULL"),
+        sa.ForeignKeyConstraint(
+            ["parent_comment_id"], ["comment.id"], ondelete="CASCADE"
+        ),
+        sa.PrimaryKeyConstraint("id"),
+    )
+    op.create_table(
+        "comment_reaction",
+        sa.Column("id", sa.Integer(), autoincrement=True, nullable=False),
+        sa.Column("comment_id", sa.Integer(), nullable=False),
+        sa.Column("reaction", sa.String(length=255), nullable=False),
+        sa.Column("created_by", sa.Integer(), nullable=True),
+        sa.ForeignKeyConstraint(["comment_id"], ["comment.id"], ondelete="CASCADE"),
+        sa.ForeignKeyConstraint(["created_by"], ["user.id"], ondelete="SET NULL"),
+        sa.PrimaryKeyConstraint("id"),
+    )
+    op.create_table(
+        "data_cell_comment",
+        sa.Column("id", sa.Integer(), autoincrement=True, nullable=False),
+        sa.Column("data_cell_id", sa.Integer(), nullable=False),
+        sa.Column("comment_id", sa.Integer(), nullable=False),
+        sa.ForeignKeyConstraint(["comment_id"], ["comment.id"], ondelete="CASCADE"),
+        sa.ForeignKeyConstraint(["data_cell_id"], ["data_cell.id"], ondelete="CASCADE"),
+        sa.PrimaryKeyConstraint("id"),
+    )
+    op.create_table(
+        "data_table_comment",
+        sa.Column("id", sa.Integer(), autoincrement=True, nullable=False),
+        sa.Column("data_table_id", sa.Integer(), nullable=False),
+        sa.Column("comment_id", sa.Integer(), nullable=False),
+        sa.ForeignKeyConstraint(["comment_id"], ["comment.id"], ondelete="CASCADE"),
+        sa.ForeignKeyConstraint(
+            ["data_table_id"], ["data_table.id"], ondelete="CASCADE"
+        ),
+        sa.PrimaryKeyConstraint("id"),
+    )
+    # ### end Alembic commands ###
+
+
+def downgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_table("data_table_comment")
+    op.drop_table("data_cell_comment")
+    op.drop_table("comment_reaction")
+    op.drop_table("comment")
+    # ### end Alembic commands ###

querybook/server/models/__init__.py

@@ -13,3 +13,4 @@
 from .tag import *
 from .event_log import *
 from .data_element import *
+from .comment import *