37 lines
1009 B
Python
37 lines
1009 B
Python
|
"""Proactive actions table for Argus layer
|
||
|
|
|
||
|
|
Revision ID: 003
|
||
|
|
Revises: 002
|
||
|
|
Create Date: 2026-03-28
|
||
|
|
"""
|
||
|
|
|
||
|
|
from alembic import op
|
||
|
|
|
||
|
|
revision = "003"
|
||
|
|
down_revision = "002"
|
||
|
|
branch_labels = None
|
||
|
|
depends_on = None
|
||
|
|
|
||
|
|
|
||
|
|
def upgrade():
|
||
|
|
op.execute("""
|
||
|
|
CREATE TABLE IF NOT EXISTS public.proactive_actions (
|
||
|
|
id UUID PRIMARY KEY DEFAULT gen_random_uuid(),
|
||
|
|
user_id UUID NOT NULL REFERENCES public.users(id) ON DELETE CASCADE,
|
||
|
|
session_id UUID REFERENCES public.sessions(id) ON DELETE SET NULL,
|
||
|
|
friction_type TEXT NOT NULL,
|
||
|
|
proposed_action TEXT NOT NULL,
|
||
|
|
user_choice TEXT,
|
||
|
|
chosen_action TEXT,
|
||
|
|
executed BOOLEAN DEFAULT false,
|
||
|
|
detected_at TIMESTAMPTZ DEFAULT now(),
|
||
|
|
responded_at TIMESTAMPTZ
|
||
|
|
);
|
||
|
|
|
||
|
|
CREATE INDEX IF NOT EXISTS idx_proactive_user ON proactive_actions(user_id, friction_type);
|
||
|
|
""")
|
||
|
|
|
||
|
|
|
||
|
|
def downgrade():
|
||
|
|
op.execute("DROP TABLE IF EXISTS public.proactive_actions CASCADE;")
|