改用mysql

2026-09-02 11:19:49 +08:00
parent 410bd30b2e
commit 73c4d83f42
24 changed files with 831 additions and 258 deletions
@@ -0,0 +1,125 @@
+"""add_user_role
+
+Revision ID: 4e40432cab9f
+Revises: 4bd4c7f26818
+Create Date: 2026-09-02 10:29:41.369850
+"""
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+from sqlalchemy.dialects import mysql
+
+# revision identifiers, used by Alembic.
+revision: str = '4e40432cab9f'
+down_revision: Union[str, None] = '4bd4c7f26818'
+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.alter_column('document_categories', 'path',
+               existing_type=mysql.VARCHAR(collation='utf8mb4_unicode_ci', length=500),
+               comment='物化路径，如 /01_公司层/04_岗位AI角色/',
+               existing_comment='物化路径',
+               existing_nullable=False,
+               existing_server_default=sa.text("'/'"))
+    op.alter_column('document_categories', 'is_folder',
+               existing_type=mysql.TINYINT(display_width=1),
+               comment='True=文件夹（可含子项），False=叶子分类',
+               existing_comment='True=文件夹',
+               existing_nullable=False,
+               existing_server_default=sa.text("'1'"))
+    op.create_foreign_key(None, 'document_categories', 'document_categories', ['parent_id'], ['id'])
+    op.alter_column('documents', 'original_filename',
+               existing_type=mysql.VARCHAR(collation='utf8mb4_unicode_ci', length=512),
+               comment='原始文件名或文本标题 (仅展示)',
+               existing_comment='原始文件名 (仅展示)',
+               existing_nullable=False)
+    op.alter_column('documents', 'storage_path',
+               existing_type=mysql.VARCHAR(collation='utf8mb4_unicode_ci', length=1024),
+               comment='相对于 data/ 的物理存储路径（文本内容文档为 NULL）',
+               existing_nullable=True)
+    op.alter_column('documents', 'content',
+               existing_type=mysql.TEXT(collation='utf8mb4_unicode_ci'),
+               comment='直接输入的文本内容（非文件上传时使用）',
+               existing_comment='直接输入的文本内容',
+               existing_nullable=True)
+    op.alter_column('documents', 'content_format',
+               existing_type=mysql.VARCHAR(collation='utf8mb4_unicode_ci', length=16),
+               comment='内容格式：markdown / text',
+               existing_comment='内容格式',
+               existing_nullable=False,
+               existing_server_default=sa.text("'markdown'"))
+    op.alter_column('documents', 'file_size',
+               existing_type=mysql.INTEGER(),
+               comment='文件大小 (字节)，文本内容文档为内容长度',
+               existing_comment='文件大小 (字节)',
+               existing_nullable=False)
+    op.alter_column('documents', 'file_ext',
+               existing_type=mysql.VARCHAR(collation='utf8mb4_unicode_ci', length=16),
+               comment='文件扩展名 (.docx/.pdf/.md)',
+               existing_comment='文件扩展名 (.docx/.pdf)',
+               existing_nullable=False)
+    op.alter_column('documents', 'sha256',
+               existing_type=mysql.VARCHAR(collation='utf8mb4_unicode_ci', length=64),
+               comment='文件 SHA-256 哈希（文本内容文档为 NULL）',
+               existing_nullable=True)
+    op.add_column('users', sa.Column('role', sa.String(length=16), nullable=False, comment='角色 (internal=内部员工 / customer=客户)'))
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_column('users', 'role')
+    op.alter_column('documents', 'sha256',
+               existing_type=mysql.VARCHAR(collation='utf8mb4_unicode_ci', length=64),
+               comment=None,
+               existing_comment='文件 SHA-256 哈希（文本内容文档为 NULL）',
+               existing_nullable=True)
+    op.alter_column('documents', 'file_ext',
+               existing_type=mysql.VARCHAR(collation='utf8mb4_unicode_ci', length=16),
+               comment='文件扩展名 (.docx/.pdf)',
+               existing_comment='文件扩展名 (.docx/.pdf/.md)',
+               existing_nullable=False)
+    op.alter_column('documents', 'file_size',
+               existing_type=mysql.INTEGER(),
+               comment='文件大小 (字节)',
+               existing_comment='文件大小 (字节)，文本内容文档为内容长度',
+               existing_nullable=False)
+    op.alter_column('documents', 'content_format',
+               existing_type=mysql.VARCHAR(collation='utf8mb4_unicode_ci', length=16),
+               comment='内容格式',
+               existing_comment='内容格式：markdown / text',
+               existing_nullable=False,
+               existing_server_default=sa.text("'markdown'"))
+    op.alter_column('documents', 'content',
+               existing_type=mysql.TEXT(collation='utf8mb4_unicode_ci'),
+               comment='直接输入的文本内容',
+               existing_comment='直接输入的文本内容（非文件上传时使用）',
+               existing_nullable=True)
+    op.alter_column('documents', 'storage_path',
+               existing_type=mysql.VARCHAR(collation='utf8mb4_unicode_ci', length=1024),
+               comment=None,
+               existing_comment='相对于 data/ 的物理存储路径（文本内容文档为 NULL）',
+               existing_nullable=True)
+    op.alter_column('documents', 'original_filename',
+               existing_type=mysql.VARCHAR(collation='utf8mb4_unicode_ci', length=512),
+               comment='原始文件名 (仅展示)',
+               existing_comment='原始文件名或文本标题 (仅展示)',
+               existing_nullable=False)
+    op.drop_constraint(None, 'document_categories', type_='foreignkey')
+    op.alter_column('document_categories', 'is_folder',
+               existing_type=mysql.TINYINT(display_width=1),
+               comment='True=文件夹',
+               existing_comment='True=文件夹（可含子项），False=叶子分类',
+               existing_nullable=False,
+               existing_server_default=sa.text("'1'"))
+    op.alter_column('document_categories', 'path',
+               existing_type=mysql.VARCHAR(collation='utf8mb4_unicode_ci', length=500),
+               comment='物化路径',
+               existing_comment='物化路径，如 /01_公司层/04_岗位AI角色/',
+               existing_nullable=False,
+               existing_server_default=sa.text("'/'"))
+    # ### end Alembic commands ###