Floke
198eaef2b2
This commit introduces a new unsubscribe feature to allow contacts to opt-out from marketing automation. Key changes include: - Database schema migration: Added (UUID) to the model. - Data population: Implemented a script to assign unique tokens to existing contacts. - API endpoint: Created a public GET endpoint to handle opt-out requests. - Automation: New contacts automatically receive an unsubscribe token upon creation. - Integration: The full unsubscribe link is now returned via the provisioning API for storage in SuperOffice UDFs (ProgID: SuperOffice:9). - Documentation: Updated and to reflect the new feature and its integration requirements. - Added for quick overview and next steps.
116 lines
4.4 KiB
Python
116 lines
4.4 KiB
Python
import sqlite3
|
|
import sys
|
|
import os
|
|
import logging
|
|
|
|
# Add parent path to import config
|
|
sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "../..")))
|
|
from backend.config import settings
|
|
|
|
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
|
|
logger = logging.getLogger(__name__)
|
|
|
|
# Extract DB file path from SQLAlchemy URL
|
|
DB_FILE = settings.DATABASE_URL.replace("sqlite:///", "")
|
|
|
|
def get_db_connection():
|
|
"""Establishes a connection to the SQLite database."""
|
|
return sqlite3.connect(DB_FILE)
|
|
|
|
def get_table_columns(cursor, table_name):
|
|
"""Returns a list of column names for a given table."""
|
|
cursor.execute(f"PRAGMA table_info({table_name})")
|
|
return [row[1] for row in cursor.fetchall()]
|
|
|
|
def migrate_tables():
|
|
"""
|
|
Adds new columns to existing tables to support v0.7.0 features.
|
|
"""
|
|
logger.info(f"Connecting to database at {DB_FILE} to run migrations...")
|
|
conn = get_db_connection()
|
|
cursor = conn.cursor()
|
|
|
|
try:
|
|
# 1. Update INDUSTRIES Table
|
|
logger.info("Checking 'industries' table schema...")
|
|
ind_columns = get_table_columns(cursor, "industries")
|
|
|
|
ind_migrations = {
|
|
"metric_type": "TEXT",
|
|
"scraper_search_term": "TEXT",
|
|
"standardization_logic": "TEXT",
|
|
"proxy_factor": "FLOAT",
|
|
"scraper_keywords": "TEXT",
|
|
"scraper_search_term": "TEXT"
|
|
}
|
|
|
|
for col, col_type in ind_migrations.items():
|
|
if col not in ind_columns:
|
|
logger.info(f"Adding column '{col}' to 'industries' table...")
|
|
cursor.execute(f"ALTER TABLE industries ADD COLUMN {col} {col_type}")
|
|
|
|
# 2. Update COMPANIES Table (New for v0.7.0)
|
|
logger.info("Checking 'companies' table schema...")
|
|
comp_columns = get_table_columns(cursor, "companies")
|
|
|
|
comp_migrations = {
|
|
"status": "TEXT", # Added to fix missing column error
|
|
"calculated_metric_name": "TEXT",
|
|
"calculated_metric_value": "FLOAT",
|
|
"calculated_metric_unit": "TEXT",
|
|
"standardized_metric_value": "FLOAT",
|
|
"standardized_metric_unit": "TEXT",
|
|
"metric_source": "TEXT",
|
|
"metric_source_url": "TEXT"
|
|
}
|
|
|
|
for col, col_type in comp_migrations.items():
|
|
if col not in comp_columns:
|
|
logger.info(f"Adding column '{col}' to 'companies' table...")
|
|
cursor.execute(f"ALTER TABLE companies ADD COLUMN {col} {col_type}")
|
|
|
|
# 3. Create REPORTED_MISTAKES Table
|
|
logger.info("Checking 'reported_mistakes' table schema...")
|
|
cursor.execute("""
|
|
CREATE TABLE IF NOT EXISTS reported_mistakes (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
company_id INTEGER NOT NULL,
|
|
field_name TEXT NOT NULL,
|
|
wrong_value TEXT,
|
|
corrected_value TEXT,
|
|
source_url TEXT,
|
|
quote TEXT,
|
|
user_comment TEXT,
|
|
status TEXT NOT NULL DEFAULT 'PENDING',
|
|
created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
|
|
updated_at DATETIME DEFAULT CURRENT_TIMESTAMP,
|
|
FOREIGN KEY (company_id) REFERENCES companies (id)
|
|
)
|
|
""")
|
|
logger.info("Table 'reported_mistakes' ensured to exist.")
|
|
|
|
# 4. Update CONTACTS Table (Two-step for SQLite compatibility)
|
|
logger.info("Checking 'contacts' table schema for unsubscribe_token...")
|
|
contacts_columns = get_table_columns(cursor, "contacts")
|
|
|
|
if 'unsubscribe_token' not in contacts_columns:
|
|
logger.info("Adding column 'unsubscribe_token' to 'contacts' table...")
|
|
cursor.execute("ALTER TABLE contacts ADD COLUMN unsubscribe_token TEXT")
|
|
|
|
logger.info("Creating UNIQUE index on 'unsubscribe_token' column...")
|
|
cursor.execute("CREATE UNIQUE INDEX IF NOT EXISTS idx_contacts_unsubscribe_token ON contacts (unsubscribe_token)")
|
|
|
|
conn.commit()
|
|
logger.info("All migrations completed successfully.")
|
|
|
|
except Exception as e:
|
|
logger.error(f"An error occurred during migration: {e}", exc_info=True)
|
|
conn.rollback()
|
|
finally:
|
|
conn.close()
|
|
|
|
if __name__ == "__main__":
|
|
if not os.path.exists(DB_FILE):
|
|
logger.error(f"Database file not found at {DB_FILE}.")
|
|
else:
|
|
migrate_tables() |