migrate legacy source kind constraint

2026-09-04 15:51:59 +02:00
parent 0653e7366c
commit 1e4c3a1e15
2 changed files with 34 additions and 1 deletions
@@ -84,6 +84,28 @@ def connect(db_path: str) -> sqlite3.Connection:
        existing = {r[1] for r in db.execute(f"PRAGMA table_info({table})")}
        for col, definition in additions:
            if col not in existing: db.execute(f"ALTER TABLE {table} ADD COLUMN {col} {definition}")
+    db.execute("INSERT OR IGNORE INTO organizations (id,name) VALUES (?,?)", (ORGANIZATION_ID, "Demo organization"))
+    source_sql_row=db.execute("SELECT sql FROM sqlite_master WHERE type='table' AND name='sources'").fetchone()
+    source_sql=(source_sql_row[0] or '') if source_sql_row else ''
+    if "CHECK(kind IN ('csv','manual'))" in source_sql:
+        # SQLite cannot change foreign-key enforcement during a transaction.
+        db.commit(); db.execute("PRAGMA foreign_keys=OFF")
+        db.executescript("""
+        CREATE TABLE sources_rebuilt (
+          id INTEGER PRIMARY KEY AUTOINCREMENT, organization_id TEXT NOT NULL REFERENCES organizations(id),
+          name TEXT NOT NULL, kind TEXT NOT NULL CHECK(kind IN ('csv','manual','google_places','bing_local','approved_directory','public_website','permitted_social','ct_logs','dns','rdap')), source_code TEXT NOT NULL DEFAULT '', display_name TEXT NOT NULL DEFAULT '', enabled INTEGER NOT NULL DEFAULT 0, approved INTEGER NOT NULL DEFAULT 0,
+          config_json TEXT NOT NULL DEFAULT '{}', policy_json TEXT NOT NULL DEFAULT '{}', quota_json TEXT NOT NULL DEFAULT '{}', health_status TEXT NOT NULL DEFAULT 'unknown',
+          consecutive_failures INTEGER NOT NULL DEFAULT 0, circuit_open INTEGER NOT NULL DEFAULT 0,
+          last_success_at TEXT, last_failure_at TEXT, last_error TEXT, created_at TEXT NOT NULL DEFAULT CURRENT_TIMESTAMP,
+          updated_at TEXT NOT NULL DEFAULT CURRENT_TIMESTAMP, UNIQUE(organization_id,name)
+        );
+        INSERT INTO sources_rebuilt(id,organization_id,name,kind,source_code,display_name,enabled,approved,config_json,policy_json,quota_json,health_status,consecutive_failures,circuit_open,last_success_at,last_failure_at,last_error,created_at,updated_at)
+        SELECT id,organization_id,name,kind,source_code,display_name,enabled,approved,config_json,policy_json,quota_json,health_status,consecutive_failures,circuit_open,last_success_at,last_failure_at,last_error,created_at,updated_at FROM sources;
+        DROP TABLE sources;
+        ALTER TABLE sources_rebuilt RENAME TO sources;
+        CREATE INDEX IF NOT EXISTS idx_sources_org ON sources(organization_id,id);
+        """)
+        db.execute("PRAGMA foreign_keys=ON")
    db.execute("CREATE UNIQUE INDEX IF NOT EXISTS uq_pipeline_idempotency ON pipeline_entries(organization_id,idempotency_key) WHERE idempotency_key IS NOT NULL AND idempotency_key <> ''")
    db.execute("CREATE UNIQUE INDEX IF NOT EXISTS uq_interaction_idempotency ON interactions(organization_id,idempotency_key) WHERE idempotency_key IS NOT NULL AND idempotency_key <> ''")
    db.execute("INSERT OR IGNORE INTO organizations (id,name) VALUES (?,?)", (ORGANIZATION_ID, "Demo organization"))