File size: 4,824 Bytes
80feb1b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 |
import sqlite3
import os
def fix_selection_offers():
db_path = "tabble_new.db"
if not os.path.exists(db_path):
print(f"Database file {db_path} not found!")
return False
print(f"Fixing selection_offers table in {db_path}...")
# Connect to the database
conn = sqlite3.connect(db_path)
cursor = conn.cursor()
try:
# Check if the selection_offers table exists
cursor.execute("SELECT name FROM sqlite_master WHERE type='table' AND name='selection_offers'")
if not cursor.fetchone():
print("selection_offers table does not exist! Creating it...")
# Create the selection_offers table with all required columns
cursor.execute("""
CREATE TABLE selection_offers (
id INTEGER PRIMARY KEY AUTOINCREMENT,
min_amount FLOAT NOT NULL,
discount_amount FLOAT NOT NULL,
is_active BOOLEAN DEFAULT 1,
description TEXT,
created_at TIMESTAMP,
updated_at TIMESTAMP
)
""")
# Add sample data
cursor.execute("""
INSERT INTO selection_offers (min_amount, discount_amount, is_active, description, created_at, updated_at)
VALUES
(50.0, 5.0, 1, 'Spend $50, get $5 off', CURRENT_TIMESTAMP, CURRENT_TIMESTAMP),
(100.0, 15.0, 1, 'Spend $100, get $15 off', CURRENT_TIMESTAMP, CURRENT_TIMESTAMP),
(150.0, 25.0, 1, 'Spend $150, get $25 off', CURRENT_TIMESTAMP, CURRENT_TIMESTAMP)
""")
conn.commit()
print("Created selection_offers table with sample data.")
else:
# Check if the discount_amount column exists
cursor.execute("PRAGMA table_info(selection_offers)")
columns = cursor.fetchall()
column_names = [column[1] for column in columns]
if "discount_amount" not in column_names:
print("Adding discount_amount column to selection_offers table...")
# Create a new table with the correct schema
cursor.execute("""
CREATE TABLE selection_offers_new (
id INTEGER PRIMARY KEY AUTOINCREMENT,
min_amount FLOAT NOT NULL,
discount_amount FLOAT NOT NULL DEFAULT 0,
is_active BOOLEAN DEFAULT 1,
description TEXT,
created_at TIMESTAMP,
updated_at TIMESTAMP
)
""")
# Copy data from the old table to the new one
cursor.execute("""
INSERT INTO selection_offers_new (id, min_amount, is_active, description, created_at, updated_at)
SELECT id, min_amount, is_active, description, created_at, updated_at FROM selection_offers
""")
# Update the discount_amount based on min_amount (as a simple rule)
cursor.execute("""
UPDATE selection_offers_new
SET discount_amount = CASE
WHEN min_amount <= 50 THEN 5.0
WHEN min_amount <= 100 THEN 15.0
ELSE 25.0
END
""")
# Drop the old table and rename the new one
cursor.execute("DROP TABLE selection_offers")
cursor.execute("ALTER TABLE selection_offers_new RENAME TO selection_offers")
conn.commit()
print("Added discount_amount column and populated with values.")
else:
print("discount_amount column already exists. No changes needed.")
# Verify the table structure
cursor.execute("PRAGMA table_info(selection_offers)")
columns = cursor.fetchall()
print("\nCurrent selection_offers table structure:")
for column in columns:
print(f" {column[1]} ({column[2]})")
# Show the data
cursor.execute("SELECT * FROM selection_offers")
rows = cursor.fetchall()
print("\nCurrent selection_offers data:")
for row in rows:
print(f" {row}")
print("\nFix completed successfully!")
return True
except sqlite3.Error as e:
print(f"SQLite error: {e}")
conn.rollback()
return False
finally:
conn.close()
if __name__ == "__main__":
fix_selection_offers()
|