Agent Zero
32 lines
1.1 KiB
Python
32 lines
1.1 KiB
Python
"""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")
|