"""Add created_at/updated_at to workspace_users and workspace_widgets.

TenantMixin inherits from TimestampMixin which adds created_at and updated_at.
Migration 0072 only added assigned_at to workspace_users, not created_at/updated_at.

Revision ID: 0074
Revises: 0073
"""

from alembic import op
import sqlalchemy as sa

revision = "0074"
down_revision = "0073"
branch_labels = None
depends_on = None


def upgrade() -> None:
    # workspace_users: add created_at and updated_at
    op.execute("ALTER TABLE workspace_users ADD COLUMN IF NOT EXISTS created_at TIMESTAMPTZ NOT NULL DEFAULT NOW()")
    op.execute("ALTER TABLE workspace_users ADD COLUMN IF NOT EXISTS updated_at TIMESTAMPTZ NOT NULL DEFAULT NOW()")

    # workspace_widgets: already has created_at/updated_at from migration 0072
    # workspace_modules: already has created_at/updated_at from migration 0072
    # workspaces: already has created_at/updated_at from migration 0072


def downgrade() -> None:
    op.drop_column("workspace_users", "updated_at")
    op.drop_column("workspace_users", "created_at")