sync_categories.py

import os
import sys
import json

# Add the project root to the path to allow for module imports
sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))

# Load local.settings.json for environment variables
def load_local_settings():
    settings_path = os.path.join(os.path.dirname(__file__), 'local.settings.json')
    if not os.path.exists(settings_path):
        print(f"Warning: '{settings_path}' not found. Assuming environment is already configured.")
        return

    with open(settings_path, 'r') as f:
        settings = json.load(f)
    
    if 'Values' in settings:
        for key, value in settings['Values'].items():
            os.environ[key] = value

# Ensure environment variables are loaded before importing app modules
load_local_settings()

from uncap_app.core import SessionLocal
from uncap_app.models import ScanIQCategories, ScanIQCategories_Product

def sync_scaniq_categories():
    """Synchronizes ScanIQCategories from the default DB to the product DB."""
    session = SessionLocal()
    try:
        print("Synchronizing ScanIQCategories to product database...")
        source_categories = session.query(ScanIQCategories).all()
        for src_cat in source_categories:
            product_cat = session.query(ScanIQCategories_Product).filter_by(site=src_cat.site, category_name=src_cat.category_name).first()
            if product_cat:
                product_cat.search_term_1, product_cat.search_term_2, product_cat.search_term_3, product_cat.search_term_4, product_cat.search_term_5 = \
                    src_cat.search_term_1, src_cat.search_term_2, src_cat.search_term_3, src_cat.search_term_4, src_cat.search_term_5
            else:
                session.add(ScanIQCategories_Product(site=src_cat.site, category_name=src_cat.category_name,
                                                   search_term_1=src_cat.search_term_1, search_term_2=src_cat.search_term_2,
                                                   search_term_3=src_cat.search_term_3, search_term_4=src_cat.search_term_4,
                                                   search_term_5=src_cat.search_term_5))
        session.commit()
        print("Synchronization complete.")
    except Exception as e:
        print(f"An error occurred during category synchronization: {e}")
        session.rollback()
    finally:
        session.close()

if __name__ == "__main__":
    sync_scaniq_categories()