homelab_automation/alembic/versions/0002_add_schedule_columns.py
Bruno Charest 984d06a223
Some checks failed
Tests / Backend Tests (Python) (3.10) (push) Has been cancelled
Tests / Backend Tests (Python) (3.11) (push) Has been cancelled
Tests / Backend Tests (Python) (3.12) (push) Has been cancelled
Tests / Frontend Tests (JS) (push) Has been cancelled
Tests / Integration Tests (push) Has been cancelled
Tests / All Tests Passed (push) Has been cancelled
feat: Implement comprehensive database schema with new models, CRUD operations, and documentation for host metrics, Docker management, and terminal sessions, while removing old test files.
2026-03-05 10:16:13 -05:00

53 lines
2.5 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(255), 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(255), 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(255), 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")