6

2026-09-01 20:51:46 +08:00
parent 70ece67910
commit b80161972a
18 changed files with 1878 additions and 359 deletions
@@ -0,0 +1,47 @@
+"""add_category_tree_and_text_content
+
+Revision ID: 4bd4c7f26818
+Revises: 4fcdb390ae18
+Create Date: 2026-09-01 18:02:12.469005
+"""
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision: str = '4bd4c7f26818'
+down_revision: Union[str, None] = '4fcdb390ae18'
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    # document_categories 新增字段
+    op.add_column('document_categories', sa.Column('parent_id', sa.String(length=32), nullable=True, comment='父分类 ID（NULL = 顶层）'))
+    op.add_column('document_categories', sa.Column('path', sa.Text(), server_default='/', nullable=False, comment='物化路径'))
+    op.add_column('document_categories', sa.Column('is_folder', sa.Boolean(), server_default=sa.text('1'), nullable=False, comment='True=文件夹'))
+    op.create_index(op.f('ix_document_categories_parent_id'), 'document_categories', ['parent_id'], unique=False)
+
+    # documents 新增字段
+    op.add_column('documents', sa.Column('content', sa.Text(), nullable=True, comment='直接输入的文本内容'))
+    op.add_column('documents', sa.Column('content_format', sa.String(length=16), server_default='markdown', nullable=False, comment='内容格式'))
+
+    # SQLite 不支持 ALTER COLUMN，用 batch mode 重建表
+    with op.batch_alter_table('documents', schema=None) as batch_op:
+        batch_op.alter_column('storage_path', existing_type=sa.String(1024), nullable=True)
+        batch_op.alter_column('sha256', existing_type=sa.String(64), nullable=True)
+
+
+def downgrade() -> None:
+    with op.batch_alter_table('documents', schema=None) as batch_op:
+        batch_op.alter_column('sha256', existing_type=sa.String(64), nullable=False)
+        batch_op.alter_column('storage_path', existing_type=sa.String(1024), nullable=False)
+
+    op.drop_column('documents', 'content_format')
+    op.drop_column('documents', 'content')
+    op.drop_index(op.f('ix_document_categories_parent_id'), table_name='document_categories')
+    op.drop_column('document_categories', 'is_folder')
+    op.drop_column('document_categories', 'path')
+    op.drop_column('document_categories', 'parent_id')