53 lines
2.4 KiB
Python
53 lines
2.4 KiB
Python
"""Add missing columns to schedules table for full scheduler support
|
|
|
|
Revision ID: 0002_add_schedule_columns
|
|
Revises: 0001_initial
|
|
Create Date: 2025-12-05
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
|
|
# revision identifiers, used by Alembic.
|
|
revision = "0002_add_schedule_columns"
|
|
down_revision = "0001_initial"
|
|
branch_labels = None
|
|
depends_on = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
# Ajouter les colonnes manquantes à la table schedules
|
|
op.add_column("schedules", sa.Column("description", sa.Text(), nullable=True))
|
|
op.add_column("schedules", sa.Column("target_type", sa.String(), nullable=True, server_default="group"))
|
|
op.add_column("schedules", sa.Column("extra_vars", sa.JSON(), nullable=True))
|
|
op.add_column("schedules", sa.Column("timezone", sa.String(), nullable=True, server_default="America/Montreal"))
|
|
op.add_column("schedules", sa.Column("start_at", sa.DateTime(timezone=True), nullable=True))
|
|
op.add_column("schedules", sa.Column("end_at", sa.DateTime(timezone=True), nullable=True))
|
|
op.add_column("schedules", sa.Column("last_status", sa.String(), nullable=True, server_default="never"))
|
|
op.add_column("schedules", sa.Column("retry_on_failure", sa.Integer(), nullable=True, server_default="0"))
|
|
op.add_column("schedules", sa.Column("timeout", sa.Integer(), nullable=True, server_default="3600"))
|
|
op.add_column("schedules", sa.Column("run_count", sa.Integer(), nullable=True, server_default="0"))
|
|
op.add_column("schedules", sa.Column("success_count", sa.Integer(), nullable=True, server_default="0"))
|
|
op.add_column("schedules", sa.Column("failure_count", sa.Integer(), nullable=True, server_default="0"))
|
|
|
|
# Ajouter hosts_impacted à schedule_runs
|
|
op.add_column("schedule_runs", sa.Column("hosts_impacted", sa.Integer(), nullable=True, server_default="0"))
|
|
|
|
|
|
def downgrade() -> None:
|
|
op.drop_column("schedule_runs", "hosts_impacted")
|
|
op.drop_column("schedules", "failure_count")
|
|
op.drop_column("schedules", "success_count")
|
|
op.drop_column("schedules", "run_count")
|
|
op.drop_column("schedules", "timeout")
|
|
op.drop_column("schedules", "retry_on_failure")
|
|
op.drop_column("schedules", "last_status")
|
|
op.drop_column("schedules", "end_at")
|
|
op.drop_column("schedules", "start_at")
|
|
op.drop_column("schedules", "timezone")
|
|
op.drop_column("schedules", "extra_vars")
|
|
op.drop_column("schedules", "target_type")
|
|
op.drop_column("schedules", "description")
|