Files
HabitForge/alembic/versions/e4f5g6h7i8j9_add_hc_data_source_columns.py
T
2026-07-30 23:25:20 -04:00

52 lines
1.6 KiB
Python

"""Add data_source columns to health_connect_daily_metrics
Revision ID: e4f5g6h7i8j9
Revises: b2c3d4e5f6g7
Create Date: 2026-01-07 15:40:00.000000
"""
from typing import Sequence, Union
from alembic import op
import sqlalchemy as sa
# revision identifiers, used by Alembic.
revision: str = 'e4f5g6h7i8j9'
down_revision: Union[str, None] = 'b2c3d4e5f6g7'
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None
def upgrade() -> None:
# Add data_source column to health_connect_daily_metrics
op.add_column(
'health_connect_daily_metrics',
sa.Column('data_source', sa.String(200), nullable=True)
)
# Add data_source_name column to health_connect_daily_metrics
op.add_column(
'health_connect_daily_metrics',
sa.Column('data_source_name', sa.String(100), nullable=True)
)
# Add data_source column to health_connect_exercise_sessions
op.add_column(
'health_connect_exercise_sessions',
sa.Column('data_source', sa.String(200), nullable=True)
)
# Add data_source_name column to health_connect_exercise_sessions
op.add_column(
'health_connect_exercise_sessions',
sa.Column('data_source_name', sa.String(100), nullable=True)
)
def downgrade() -> None:
op.drop_column('health_connect_daily_metrics', 'data_source')
op.drop_column('health_connect_daily_metrics', 'data_source_name')
op.drop_column('health_connect_exercise_sessions', 'data_source')
op.drop_column('health_connect_exercise_sessions', 'data_source_name')