mirror of
https://github.com/All-Hands-AI/OpenHands.git
synced 2026-01-08 22:38:05 -05:00
105 lines
3.6 KiB
Python
105 lines
3.6 KiB
Python
"""Create openhands_prs table
|
|
|
|
Revision ID: 050
|
|
Revises: 049
|
|
Create Date: 2025-06-18
|
|
|
|
"""
|
|
|
|
from typing import Sequence, Union
|
|
|
|
import sqlalchemy as sa
|
|
from alembic import op
|
|
|
|
# revision identifiers, used by Alembic.
|
|
revision: str = '050'
|
|
down_revision: Union[str, None] = '049'
|
|
branch_labels: Union[str, Sequence[str], None] = None
|
|
depends_on: Union[str, Sequence[str], None] = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
# Create enum types
|
|
op.create_table(
|
|
'openhands_prs',
|
|
sa.Column('id', sa.Integer(), sa.Identity(), nullable=False, primary_key=True),
|
|
sa.Column('repo_id', sa.String(), nullable=False),
|
|
sa.Column('repo_name', sa.String(), nullable=False),
|
|
sa.Column('pr_number', sa.Integer(), nullable=False),
|
|
sa.Column('provider', sa.String(), nullable=False),
|
|
sa.Column('installation_id', sa.String(), nullable=True),
|
|
sa.Column('private', sa.Boolean(), nullable=True),
|
|
sa.Column(
|
|
'status', sa.Enum('MERGED', 'CLOSED', name='prstatus'), nullable=False
|
|
),
|
|
sa.Column(
|
|
'processed', sa.Boolean(), nullable=False, server_default=sa.text('FALSE')
|
|
),
|
|
sa.Column(
|
|
'process_attempts',
|
|
sa.Integer(),
|
|
nullable=False,
|
|
server_default=sa.text('0'),
|
|
),
|
|
sa.Column(
|
|
'updated_at',
|
|
sa.DateTime(),
|
|
server_default=sa.text('CURRENT_TIMESTAMP'),
|
|
nullable=False,
|
|
),
|
|
sa.Column(
|
|
'created_at',
|
|
sa.DateTime(),
|
|
nullable=False,
|
|
),
|
|
sa.Column(
|
|
'closed_at',
|
|
sa.DateTime(),
|
|
nullable=False,
|
|
),
|
|
# PR metrics columns (optional fields)
|
|
sa.Column('num_reviewers', sa.Integer(), nullable=True),
|
|
sa.Column('num_commits', sa.Integer(), nullable=True),
|
|
sa.Column('num_review_comments', sa.Integer(), nullable=True),
|
|
sa.Column('num_general_comments', sa.Integer(), nullable=True),
|
|
sa.Column('num_changed_files', sa.Integer(), nullable=True),
|
|
sa.Column('num_additions', sa.Integer(), nullable=True),
|
|
sa.Column('num_deletions', sa.Integer(), nullable=True),
|
|
sa.Column('merged', sa.Boolean(), nullable=True),
|
|
sa.Column('openhands_helped_author', sa.Boolean(), nullable=True),
|
|
sa.Column('num_openhands_commits', sa.Integer(), nullable=True),
|
|
sa.Column('num_openhands_review_comments', sa.Integer(), nullable=True),
|
|
sa.Column('num_openhands_general_comments', sa.Integer(), nullable=True),
|
|
)
|
|
|
|
# Create indexes for efficient querying
|
|
op.create_index(
|
|
'ix_openhands_prs_repo_id', 'openhands_prs', ['repo_id'], unique=False
|
|
)
|
|
op.create_index(
|
|
'ix_openhands_prs_pr_number', 'openhands_prs', ['pr_number'], unique=False
|
|
)
|
|
op.create_index(
|
|
'ix_openhands_prs_status', 'openhands_prs', ['status'], unique=False
|
|
)
|
|
|
|
# Create unique constraint on repo_id + pr_number combination
|
|
op.create_index(
|
|
'ix_openhands_prs_repo_pr_unique',
|
|
'openhands_prs',
|
|
['repo_id', 'pr_number'],
|
|
unique=True,
|
|
)
|
|
|
|
|
|
def downgrade() -> None:
|
|
op.drop_index('ix_openhands_prs_repo_id', table_name='openhands_prs')
|
|
op.drop_index('ix_openhands_prs_pr_number', table_name='openhands_prs')
|
|
op.drop_index('ix_openhands_prs_status', table_name='openhands_prs')
|
|
op.drop_index('ix_openhands_prs_repo_pr_unique', table_name='openhands_prs')
|
|
op.drop_table('openhands_prs')
|
|
|
|
# Drop enum types
|
|
op.execute('DROP TYPE IF EXISTS prstatus')
|
|
op.execute('DROP TYPE IF EXISTS providertype')
|