259 lines
14 KiB
Python
259 lines
14 KiB
Python
"""initial migration
|
|
|
|
Revision ID: 29c02cb50737
|
|
Revises:
|
|
Create Date: 2026-02-16 19:28:02.116622
|
|
|
|
"""
|
|
from typing import Sequence, Union
|
|
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
from sqlalchemy.dialects import postgresql
|
|
|
|
# revision identifiers, used by Alembic.
|
|
revision: str = '29c02cb50737'
|
|
down_revision: Union[str, None] = None
|
|
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('drive_acl',
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('resource_type', sa.String(length=10), nullable=False),
|
|
sa.Column('resource_id', sa.Integer(), nullable=False),
|
|
sa.Column('subject_type', sa.String(length=10), nullable=False),
|
|
sa.Column('subject_id', sa.Integer(), nullable=True),
|
|
sa.Column('role', sa.String(length=20), nullable=False),
|
|
sa.Column('inherited_from', sa.Integer(), nullable=True),
|
|
sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_table('users',
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('name', sa.String(length=255), nullable=False),
|
|
sa.Column('email', sa.String(length=255), nullable=False),
|
|
sa.Column('password_hash', sa.String(length=255), nullable=False),
|
|
sa.Column('subscription', sa.String(length=50), nullable=False),
|
|
sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_index(op.f('ix_users_email'), 'users', ['email'], unique=True)
|
|
op.create_table('drive_activities',
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('resource_type', sa.String(length=10), nullable=False),
|
|
sa.Column('resource_id', sa.Integer(), nullable=False),
|
|
sa.Column('actor_id', sa.Integer(), nullable=False),
|
|
sa.Column('activity_type', sa.String(length=50), nullable=False),
|
|
sa.Column('activity_metadata', sa.Text(), nullable=True),
|
|
sa.Column('ip_address', sa.String(length=64), nullable=True),
|
|
sa.Column('device_info', sa.Text(), nullable=True),
|
|
sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
|
|
sa.ForeignKeyConstraint(['actor_id'], ['users.id'], ondelete='CASCADE'),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_table('drive_folders',
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('owner_id', sa.Integer(), nullable=False),
|
|
sa.Column('parent_id', sa.Integer(), nullable=True),
|
|
sa.Column('name', sa.String(length=255), nullable=False),
|
|
sa.Column('color', sa.String(length=20), nullable=True),
|
|
sa.Column('is_starred', sa.Boolean(), nullable=False),
|
|
sa.Column('is_trashed', sa.Boolean(), nullable=False),
|
|
sa.Column('trashed_at', sa.DateTime(timezone=True), nullable=True),
|
|
sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
|
|
sa.Column('updated_at', sa.DateTime(timezone=True), nullable=False),
|
|
sa.ForeignKeyConstraint(['owner_id'], ['users.id'], ondelete='CASCADE'),
|
|
sa.ForeignKeyConstraint(['parent_id'], ['drive_folders.id'], ondelete='CASCADE'),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_table('drive_shares',
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('resource_type', sa.String(length=10), nullable=False),
|
|
sa.Column('resource_id', sa.Integer(), nullable=False),
|
|
sa.Column('invited_email', sa.String(length=255), nullable=True),
|
|
sa.Column('link_token', sa.String(length=255), nullable=True),
|
|
sa.Column('role', sa.String(length=20), nullable=False),
|
|
sa.Column('expires_at', sa.DateTime(timezone=True), nullable=True),
|
|
sa.Column('created_by_id', sa.Integer(), nullable=False),
|
|
sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
|
|
sa.ForeignKeyConstraint(['created_by_id'], ['users.id'], ondelete='CASCADE'),
|
|
sa.PrimaryKeyConstraint('id'),
|
|
sa.UniqueConstraint('link_token')
|
|
)
|
|
op.create_table('drive_stars',
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('resource_type', sa.String(length=10), nullable=False),
|
|
sa.Column('resource_id', sa.Integer(), nullable=False),
|
|
sa.Column('user_id', sa.Integer(), nullable=False),
|
|
sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
|
|
sa.ForeignKeyConstraint(['user_id'], ['users.id'], ondelete='CASCADE'),
|
|
sa.PrimaryKeyConstraint('id'),
|
|
sa.UniqueConstraint('resource_type', 'resource_id', 'user_id', name='uq_drive_star')
|
|
)
|
|
op.create_table('projects',
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('user_id', sa.Integer(), nullable=False),
|
|
sa.Column('name', sa.String(length=255), nullable=False),
|
|
sa.Column('status', sa.String(length=50), nullable=False),
|
|
sa.Column('file_type', sa.String(length=50), nullable=False),
|
|
sa.Column('output_format', sa.String(length=50), nullable=False),
|
|
sa.Column('progress', sa.Integer(), nullable=False),
|
|
sa.Column('current_step', sa.String(length=50), nullable=False),
|
|
sa.Column('session_id', sa.String(length=255), nullable=True),
|
|
sa.Column('html_content', sa.Text(), nullable=True),
|
|
sa.Column('markdown_content', sa.Text(), nullable=True),
|
|
sa.Column('xml_content', sa.Text(), nullable=True),
|
|
sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
|
|
sa.Column('updated_at', sa.DateTime(timezone=True), nullable=False),
|
|
sa.ForeignKeyConstraint(['user_id'], ['users.id'], ondelete='CASCADE'),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_table('user_files',
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('user_id', sa.Integer(), nullable=False),
|
|
sa.Column('file_name', sa.String(length=512), nullable=False),
|
|
sa.Column('s3_key', sa.String(length=1024), nullable=False),
|
|
sa.Column('file_size', sa.BigInteger(), nullable=False),
|
|
sa.Column('mime_type', sa.String(length=255), nullable=True),
|
|
sa.Column('is_deleted', sa.Boolean(), nullable=False),
|
|
sa.Column('deleted_at', sa.DateTime(timezone=True), nullable=True),
|
|
sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
|
|
sa.ForeignKeyConstraint(['user_id'], ['users.id'], ondelete='CASCADE'),
|
|
sa.PrimaryKeyConstraint('id'),
|
|
sa.UniqueConstraint('s3_key')
|
|
)
|
|
op.create_table('user_storage_usage',
|
|
sa.Column('user_id', sa.Integer(), nullable=False),
|
|
sa.Column('total_bytes', sa.BigInteger(), nullable=False),
|
|
sa.Column('file_count', sa.Integer(), nullable=False),
|
|
sa.Column('last_updated', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
|
|
sa.ForeignKeyConstraint(['user_id'], ['users.id'], ondelete='CASCADE'),
|
|
sa.PrimaryKeyConstraint('user_id')
|
|
)
|
|
op.create_table('drive_files',
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('folder_id', sa.Integer(), nullable=False),
|
|
sa.Column('owner_id', sa.Integer(), nullable=False),
|
|
sa.Column('name', sa.String(length=512), nullable=False),
|
|
sa.Column('mime_type', sa.String(length=255), nullable=True),
|
|
sa.Column('size', sa.BigInteger(), nullable=False),
|
|
sa.Column('hash', sa.String(length=128), nullable=True),
|
|
sa.Column('latest_version_id', sa.Integer(), nullable=True),
|
|
sa.Column('is_starred', sa.Boolean(), nullable=False),
|
|
sa.Column('is_trashed', sa.Boolean(), nullable=False),
|
|
sa.Column('trashed_at', sa.DateTime(timezone=True), nullable=True),
|
|
sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
|
|
sa.Column('updated_at', sa.DateTime(timezone=True), nullable=False),
|
|
sa.ForeignKeyConstraint(['folder_id'], ['drive_folders.id'], ondelete='CASCADE'),
|
|
sa.ForeignKeyConstraint(['owner_id'], ['users.id'], ondelete='CASCADE'),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_table('project_activities',
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('project_id', sa.Integer(), nullable=False),
|
|
sa.Column('user_id', sa.Integer(), nullable=False),
|
|
sa.Column('activity_type', sa.String(length=50), nullable=False),
|
|
sa.Column('description', sa.Text(), nullable=True),
|
|
sa.Column('activity_metadata', sa.Text(), nullable=True),
|
|
sa.Column('ip_address', sa.String(length=64), nullable=True),
|
|
sa.Column('device_info', sa.Text(), nullable=True),
|
|
sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
|
|
sa.ForeignKeyConstraint(['project_id'], ['projects.id'], ondelete='CASCADE'),
|
|
sa.ForeignKeyConstraint(['user_id'], ['users.id'], ondelete='CASCADE'),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_table('project_comments',
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('project_id', sa.Integer(), nullable=False),
|
|
sa.Column('author_id', sa.Integer(), nullable=False),
|
|
sa.Column('body', sa.Text(), nullable=False),
|
|
sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
|
|
sa.Column('resolved_at', sa.DateTime(timezone=True), nullable=True),
|
|
sa.Column('anchor', postgresql.JSONB(astext_type=sa.Text()), nullable=True),
|
|
sa.ForeignKeyConstraint(['author_id'], ['users.id'], ondelete='CASCADE'),
|
|
sa.ForeignKeyConstraint(['project_id'], ['projects.id'], ondelete='CASCADE'),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_table('project_shares',
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('project_id', sa.Integer(), nullable=False),
|
|
sa.Column('shared_with_user_id', sa.Integer(), nullable=False),
|
|
sa.Column('shared_by_user_id', sa.Integer(), nullable=False),
|
|
sa.Column('permission', sa.String(length=50), nullable=False),
|
|
sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
|
|
sa.Column('updated_at', sa.DateTime(timezone=True), nullable=False),
|
|
sa.ForeignKeyConstraint(['project_id'], ['projects.id'], ondelete='CASCADE'),
|
|
sa.ForeignKeyConstraint(['shared_by_user_id'], ['users.id'], ondelete='CASCADE'),
|
|
sa.ForeignKeyConstraint(['shared_with_user_id'], ['users.id'], ondelete='CASCADE'),
|
|
sa.PrimaryKeyConstraint('id'),
|
|
sa.UniqueConstraint('project_id', 'shared_with_user_id', name='unique_project_user_share')
|
|
)
|
|
op.create_table('project_versions',
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('project_id', sa.Integer(), nullable=False),
|
|
sa.Column('created_by_id', sa.Integer(), nullable=False),
|
|
sa.Column('version_number', sa.Integer(), nullable=False),
|
|
sa.Column('changes_description', sa.String(length=500), nullable=True),
|
|
sa.Column('markdown_content', sa.Text(), nullable=True),
|
|
sa.Column('xml_content', sa.Text(), nullable=True),
|
|
sa.Column('html_content', sa.Text(), nullable=True),
|
|
sa.Column('session_id', sa.String(length=255), nullable=True),
|
|
sa.Column('is_current', sa.Boolean(), nullable=False),
|
|
sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
|
|
sa.ForeignKeyConstraint(['created_by_id'], ['users.id'], ),
|
|
sa.ForeignKeyConstraint(['project_id'], ['projects.id'], ondelete='CASCADE'),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_table('drive_comments',
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('file_id', sa.Integer(), nullable=False),
|
|
sa.Column('author_id', sa.Integer(), nullable=False),
|
|
sa.Column('body', sa.Text(), nullable=False),
|
|
sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
|
|
sa.Column('resolved_at', sa.DateTime(timezone=True), nullable=True),
|
|
sa.Column('anchor', postgresql.JSONB(astext_type=sa.Text()), nullable=True),
|
|
sa.ForeignKeyConstraint(['author_id'], ['users.id'], ondelete='CASCADE'),
|
|
sa.ForeignKeyConstraint(['file_id'], ['drive_files.id'], ondelete='CASCADE'),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_table('drive_file_versions',
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('file_id', sa.Integer(), nullable=False),
|
|
sa.Column('version_number', sa.Integer(), nullable=False),
|
|
sa.Column('storage_key', sa.Text(), nullable=False),
|
|
sa.Column('mime_type', sa.String(length=255), nullable=True),
|
|
sa.Column('size', sa.BigInteger(), nullable=False),
|
|
sa.Column('checksum', sa.String(length=128), nullable=True),
|
|
sa.Column('created_by_id', sa.Integer(), nullable=False),
|
|
sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
|
|
sa.ForeignKeyConstraint(['created_by_id'], ['users.id'], ),
|
|
sa.ForeignKeyConstraint(['file_id'], ['drive_files.id'], ondelete='CASCADE'),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
# ### end Alembic commands ###
|
|
|
|
|
|
def downgrade() -> None:
|
|
# ### commands auto generated by Alembic - please adjust! ###
|
|
op.drop_table('drive_file_versions')
|
|
op.drop_table('drive_comments')
|
|
op.drop_table('project_versions')
|
|
op.drop_table('project_shares')
|
|
op.drop_table('project_comments')
|
|
op.drop_table('project_activities')
|
|
op.drop_table('drive_files')
|
|
op.drop_table('user_storage_usage')
|
|
op.drop_table('user_files')
|
|
op.drop_table('projects')
|
|
op.drop_table('drive_stars')
|
|
op.drop_table('drive_shares')
|
|
op.drop_table('drive_folders')
|
|
op.drop_table('drive_activities')
|
|
op.drop_index(op.f('ix_users_email'), table_name='users')
|
|
op.drop_table('users')
|
|
op.drop_table('drive_acl')
|
|
# ### end Alembic commands ###
|