180 lines
9.2 KiB
Python
180 lines
9.2 KiB
Python
"""baseline schema
|
|
|
|
Revision ID: bacd5e6a253d
|
|
Revises:
|
|
Create Date: 2026-02-02 16:37:34.122971
|
|
|
|
"""
|
|
from typing import Sequence, Union
|
|
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
import sqlmodel
|
|
|
|
|
|
# revision identifiers, used by Alembic.
|
|
revision: str = 'bacd5e6a253d'
|
|
down_revision: Union[str, Sequence[str], None] = None
|
|
branch_labels: Union[str, Sequence[str], None] = None
|
|
depends_on: Union[str, Sequence[str], None] = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
"""Upgrade schema."""
|
|
# ### commands auto generated by Alembic - please adjust! ###
|
|
op.create_table('departments',
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('name', sqlmodel.sql.sqltypes.AutoString(), nullable=False),
|
|
sa.Column('head_employee_id', sa.Integer(), nullable=True),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_index(op.f('ix_departments_name'), 'departments', ['name'], unique=True)
|
|
op.create_table('employees',
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('name', sqlmodel.sql.sqltypes.AutoString(), nullable=False),
|
|
sa.Column('employee_type', sqlmodel.sql.sqltypes.AutoString(), nullable=False),
|
|
sa.Column('department_id', sa.Integer(), nullable=True),
|
|
sa.Column('manager_id', sa.Integer(), nullable=True),
|
|
sa.Column('title', sqlmodel.sql.sqltypes.AutoString(), nullable=True),
|
|
sa.Column('status', sqlmodel.sql.sqltypes.AutoString(), nullable=False),
|
|
sa.Column('openclaw_session_key', sqlmodel.sql.sqltypes.AutoString(), nullable=True),
|
|
sa.Column('notify_enabled', sa.Boolean(), nullable=False),
|
|
sa.ForeignKeyConstraint(['department_id'], ['departments.id'], ),
|
|
sa.ForeignKeyConstraint(['manager_id'], ['employees.id'], ),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_foreign_key(None, 'departments', 'employees', ['head_employee_id'], ['id'])
|
|
op.create_table('projects',
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('name', sqlmodel.sql.sqltypes.AutoString(), nullable=False),
|
|
sa.Column('status', sqlmodel.sql.sqltypes.AutoString(), nullable=False),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_index(op.f('ix_projects_name'), 'projects', ['name'], unique=True)
|
|
op.create_table('activities',
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('actor_employee_id', sa.Integer(), nullable=True),
|
|
sa.Column('entity_type', sqlmodel.sql.sqltypes.AutoString(), nullable=False),
|
|
sa.Column('entity_id', sa.Integer(), nullable=True),
|
|
sa.Column('verb', sqlmodel.sql.sqltypes.AutoString(), nullable=False),
|
|
sa.Column('payload_json', sqlmodel.sql.sqltypes.AutoString(), nullable=True),
|
|
sa.Column('created_at', sa.DateTime(), nullable=False),
|
|
sa.ForeignKeyConstraint(['actor_employee_id'], ['employees.id'], ),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_table('agent_onboardings',
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('agent_name', sqlmodel.sql.sqltypes.AutoString(), nullable=False),
|
|
sa.Column('role_title', sqlmodel.sql.sqltypes.AutoString(), nullable=False),
|
|
sa.Column('prompt', sqlmodel.sql.sqltypes.AutoString(), nullable=False),
|
|
sa.Column('cron_interval_ms', sa.Integer(), nullable=True),
|
|
sa.Column('tools_json', sqlmodel.sql.sqltypes.AutoString(), nullable=True),
|
|
sa.Column('owner_hr_id', sa.Integer(), nullable=True),
|
|
sa.Column('employee_id', sa.Integer(), nullable=True),
|
|
sa.Column('status', sqlmodel.sql.sqltypes.AutoString(), nullable=False),
|
|
sa.Column('spawned_agent_id', sqlmodel.sql.sqltypes.AutoString(), nullable=True),
|
|
sa.Column('session_key', sqlmodel.sql.sqltypes.AutoString(), nullable=True),
|
|
sa.Column('notes', sqlmodel.sql.sqltypes.AutoString(), nullable=True),
|
|
sa.Column('created_at', sa.DateTime(), nullable=False),
|
|
sa.Column('updated_at', sa.DateTime(), nullable=False),
|
|
sa.ForeignKeyConstraint(['employee_id'], ['employees.id'], ),
|
|
sa.ForeignKeyConstraint(['owner_hr_id'], ['employees.id'], ),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_table('employment_actions',
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('employee_id', sa.Integer(), nullable=False),
|
|
sa.Column('issued_by_employee_id', sa.Integer(), nullable=False),
|
|
sa.Column('action_type', sqlmodel.sql.sqltypes.AutoString(), nullable=False),
|
|
sa.Column('notes', sqlmodel.sql.sqltypes.AutoString(), nullable=True),
|
|
sa.Column('idempotency_key', sqlmodel.sql.sqltypes.AutoString(), nullable=True),
|
|
sa.Column('created_at', sa.DateTime(), nullable=False),
|
|
sa.ForeignKeyConstraint(['employee_id'], ['employees.id'], ),
|
|
sa.ForeignKeyConstraint(['issued_by_employee_id'], ['employees.id'], ),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_index(op.f('ix_employment_actions_idempotency_key'), 'employment_actions', ['idempotency_key'], unique=True)
|
|
op.create_table('project_members',
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('project_id', sa.Integer(), nullable=False),
|
|
sa.Column('employee_id', sa.Integer(), nullable=False),
|
|
sa.Column('role', sqlmodel.sql.sqltypes.AutoString(), nullable=True),
|
|
sa.ForeignKeyConstraint(['employee_id'], ['employees.id'], ),
|
|
sa.ForeignKeyConstraint(['project_id'], ['projects.id'], ),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_table('tasks',
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('project_id', sa.Integer(), nullable=False),
|
|
sa.Column('title', sqlmodel.sql.sqltypes.AutoString(), nullable=False),
|
|
sa.Column('description', sqlmodel.sql.sqltypes.AutoString(), nullable=True),
|
|
sa.Column('status', sqlmodel.sql.sqltypes.AutoString(), nullable=False),
|
|
sa.Column('assignee_employee_id', sa.Integer(), nullable=True),
|
|
sa.Column('reviewer_employee_id', sa.Integer(), nullable=True),
|
|
sa.Column('created_by_employee_id', sa.Integer(), nullable=True),
|
|
sa.Column('created_at', sa.DateTime(), nullable=False),
|
|
sa.Column('updated_at', sa.DateTime(), nullable=False),
|
|
sa.ForeignKeyConstraint(['assignee_employee_id'], ['employees.id'], ),
|
|
sa.ForeignKeyConstraint(['created_by_employee_id'], ['employees.id'], ),
|
|
sa.ForeignKeyConstraint(['project_id'], ['projects.id'], ),
|
|
sa.ForeignKeyConstraint(['reviewer_employee_id'], ['employees.id'], ),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_index(op.f('ix_tasks_project_id'), 'tasks', ['project_id'], unique=False)
|
|
op.create_index(op.f('ix_tasks_status'), 'tasks', ['status'], unique=False)
|
|
op.create_table('headcount_requests',
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('department_id', sa.Integer(), nullable=False),
|
|
sa.Column('requested_by_manager_id', sa.Integer(), nullable=False),
|
|
sa.Column('role_title', sqlmodel.sql.sqltypes.AutoString(), nullable=False),
|
|
sa.Column('employee_type', sqlmodel.sql.sqltypes.AutoString(), nullable=False),
|
|
sa.Column('quantity', sa.Integer(), nullable=False),
|
|
sa.Column('justification', sqlmodel.sql.sqltypes.AutoString(), nullable=True),
|
|
sa.Column('status', sqlmodel.sql.sqltypes.AutoString(), nullable=False),
|
|
sa.Column('fulfilled_employee_id', sa.Integer(), nullable=True),
|
|
sa.Column('fulfilled_onboarding_id', sa.Integer(), nullable=True),
|
|
sa.Column('fulfilled_at', sa.DateTime(), nullable=True),
|
|
sa.Column('created_at', sa.DateTime(), nullable=False),
|
|
sa.ForeignKeyConstraint(['department_id'], ['departments.id'], ),
|
|
sa.ForeignKeyConstraint(['fulfilled_employee_id'], ['employees.id'], ),
|
|
sa.ForeignKeyConstraint(['fulfilled_onboarding_id'], ['agent_onboardings.id'], ),
|
|
sa.ForeignKeyConstraint(['requested_by_manager_id'], ['employees.id'], ),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_table('task_comments',
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('task_id', sa.Integer(), nullable=False),
|
|
sa.Column('author_employee_id', sa.Integer(), nullable=True),
|
|
sa.Column('reply_to_comment_id', sa.Integer(), nullable=True),
|
|
sa.Column('body', sqlmodel.sql.sqltypes.AutoString(), nullable=False),
|
|
sa.Column('created_at', sa.DateTime(), nullable=False),
|
|
sa.ForeignKeyConstraint(['author_employee_id'], ['employees.id'], ),
|
|
sa.ForeignKeyConstraint(['reply_to_comment_id'], ['task_comments.id'], ),
|
|
sa.ForeignKeyConstraint(['task_id'], ['tasks.id'], ),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_index(op.f('ix_task_comments_task_id'), 'task_comments', ['task_id'], unique=False)
|
|
# ### end Alembic commands ###
|
|
|
|
|
|
def downgrade() -> None:
|
|
"""Downgrade schema."""
|
|
# ### commands auto generated by Alembic - please adjust! ###
|
|
op.drop_index(op.f('ix_task_comments_task_id'), table_name='task_comments')
|
|
op.drop_table('task_comments')
|
|
op.drop_table('headcount_requests')
|
|
op.drop_index(op.f('ix_tasks_status'), table_name='tasks')
|
|
op.drop_index(op.f('ix_tasks_project_id'), table_name='tasks')
|
|
op.drop_table('tasks')
|
|
op.drop_table('project_members')
|
|
op.drop_index(op.f('ix_employment_actions_idempotency_key'), table_name='employment_actions')
|
|
op.drop_table('employment_actions')
|
|
op.drop_table('agent_onboardings')
|
|
op.drop_table('activities')
|
|
op.drop_index(op.f('ix_projects_name'), table_name='projects')
|
|
op.drop_table('projects')
|
|
op.drop_table('employees')
|
|
op.drop_index(op.f('ix_departments_name'), table_name='departments')
|
|
op.drop_table('departments')
|
|
# ### end Alembic commands ###
|