diff --git a/padelnomics/src/padelnomics/migrations/versions/0001_rename_ls_to_paddle.py b/padelnomics/src/padelnomics/migrations/versions/0001_rename_ls_to_paddle.py index 0b5d7be..28f676c 100644 --- a/padelnomics/src/padelnomics/migrations/versions/0001_rename_ls_to_paddle.py +++ b/padelnomics/src/padelnomics/migrations/versions/0001_rename_ls_to_paddle.py @@ -2,14 +2,17 @@ def up(conn): - conn.execute( - "ALTER TABLE subscriptions" - " RENAME COLUMN lemonsqueezy_customer_id TO paddle_customer_id" - ) - conn.execute( - "ALTER TABLE subscriptions" - " RENAME COLUMN lemonsqueezy_subscription_id TO paddle_subscription_id" - ) + cols = {r[1] for r in conn.execute("PRAGMA table_info(subscriptions)")} + if "lemonsqueezy_customer_id" in cols: + conn.execute( + "ALTER TABLE subscriptions" + " RENAME COLUMN lemonsqueezy_customer_id TO paddle_customer_id" + ) + conn.execute( + "ALTER TABLE subscriptions" + " RENAME COLUMN lemonsqueezy_subscription_id" + " TO paddle_subscription_id" + ) conn.execute("DROP INDEX IF EXISTS idx_subscriptions_provider") conn.execute( "CREATE INDEX IF NOT EXISTS idx_subscriptions_provider"