bmc_hub/scripts/import_sample_data.py

"""
Data import script for BMC Hub
Imports customers and contacts from CSV files or creates sample data
"""

import psycopg2
from psycopg2.extras import RealDictCursor
import os
import logging

logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
logger = logging.getLogger(__name__)


def get_connection():
    """Get database connection"""
    database_url = os.getenv('DATABASE_URL', 'postgresql://bmc_hub:bmc_hub@postgres:5432/bmc_hub')
    return psycopg2.connect(database_url, cursor_factory=RealDictCursor)


def import_sample_data():
    """Import sample customers and contacts for testing"""
    conn = get_connection()
    cursor = conn.cursor()
    
    try:
        # Sample customers
        customers = [
            {
                'name': 'TDC A/S',
                'cvr_number': '14773908',
                'address': 'Teglholmsgade 1',
                'postal_code': '2450',
                'city': 'København SV',
                'email': 'info@tdc.dk',
                'phone': '+45 70 70 40 30',
                'website': 'https://tdc.dk',
                'is_active': True
            },
            {
                'name': 'Dansk Supermarked Group',
                'cvr_number': '16314439',
                'address': 'Roskildevej 65',
                'postal_code': '2620',
                'city': 'Albertslund',
                'email': 'info@dsg.dk',
                'phone': '+45 43 86 43 86',
                'website': 'https://dansksupermarked.dk',
                'is_active': True
            },
            {
                'name': 'Nets Denmark A/S',
                'cvr_number': '20016175',
                'address': 'Lautrupbjerg 10',
                'postal_code': '2750',
                'city': 'Ballerup',
                'email': 'info@nets.eu',
                'phone': '+45 44 68 44 68',
                'website': 'https://nets.eu',
                'is_active': True
            },
            {
                'name': 'Salling Group',
                'cvr_number': '30521736',
                'address': 'Skanderborgvej 121',
                'postal_code': '8260',
                'city': 'Viby J',
                'email': 'info@sallinggroup.com',
                'phone': '+45 87 93 35 00',
                'website': 'https://sallinggroup.com',
                'is_active': True
            },
            {
                'name': 'ISS A/S',
                'cvr_number': '28861341',
                'address': 'Buddingevej 197',
                'postal_code': '2860',
                'city': 'Søborg',
                'email': 'info@dk.issworld.com',
                'phone': '+45 38 17 00 00',
                'website': 'https://issworld.com',
                'is_active': True
            }
        ]
        
        logger.info("Importing customers...")
        customer_ids = {}
        
        for customer in customers:
            cursor.execute("""
                INSERT INTO customers (
                    name, cvr_number, address, postal_code, city, 
                    email, phone, website, is_active
                ) VALUES (%s, %s, %s, %s, %s, %s, %s, %s, %s)
                RETURNING id
            """, (
                customer['name'], customer['cvr_number'], customer['address'],
                customer['postal_code'], customer['city'], customer['email'],
                customer['phone'], customer['website'], customer['is_active']
            ))
            result = cursor.fetchone()
            customer_id = result['id']
            customer_ids[customer['name']] = customer_id
            logger.info(f"✅ Imported customer: {customer['name']} (ID: {customer_id})")
        
        # Sample contacts
        contacts = [
            {
                'first_name': 'Lars',
                'last_name': 'Jensen',
                'email': 'lars.jensen@tdc.dk',
                'phone': '+45 70 70 40 31',
                'mobile': '+45 20 12 34 56',
                'title': 'CTO',
                'department': 'IT',
                'companies': ['TDC A/S'],
                'is_primary': True,
                'role': 'Teknisk kontakt'
            },
            {
                'first_name': 'Mette',
                'last_name': 'Nielsen',
                'email': 'mette.nielsen@dsg.dk',
                'phone': '+45 43 86 43 87',
                'mobile': '+45 30 98 76 54',
                'title': 'IT Manager',
                'department': 'IT Operations',
                'companies': ['Dansk Supermarked Group'],
                'is_primary': True,
                'role': 'Primær kontakt'
            },
            {
                'first_name': 'Peter',
                'last_name': 'Hansen',
                'email': 'peter.hansen@nets.eu',
                'phone': '+45 44 68 44 69',
                'mobile': '+45 40 11 22 33',
                'title': 'Network Engineer',
                'department': 'Infrastructure',
                'companies': ['Nets Denmark A/S'],
                'is_primary': True,
                'role': 'Teknisk ansvarlig'
            },
            {
                'first_name': 'Anne',
                'last_name': 'Andersen',
                'email': 'anne.andersen@sallinggroup.com',
                'phone': '+45 87 93 35 01',
                'mobile': '+45 50 44 55 66',
                'title': 'IT Director',
                'department': 'IT',
                'companies': ['Salling Group'],
                'is_primary': True,
                'role': 'Beslutningsansvarlig'
            },
            {
                'first_name': 'Thomas',
                'last_name': 'Christensen',
                'email': 'thomas.christensen@issworld.com',
                'phone': '+45 38 17 00 01',
                'mobile': '+45 60 77 88 99',
                'title': 'Senior IT Consultant',
                'department': 'IT Services',
                'companies': ['ISS A/S', 'Nets Denmark A/S'],  # Multi-company contact
                'is_primary': False,
                'role': 'Konsulent'
            }
        ]
        
        logger.info("Importing contacts...")
        
        for contact in contacts:
            # Insert contact
            cursor.execute("""
                INSERT INTO contacts (
                    first_name, last_name, email, phone, mobile, 
                    title, department, is_active
                ) VALUES (%s, %s, %s, %s, %s, %s, %s, %s)
                RETURNING id
            """, (
                contact['first_name'], contact['last_name'], contact['email'],
                contact['phone'], contact['mobile'], contact['title'],
                contact['department'], True
            ))
            result = cursor.fetchone()
            contact_id = result['id']
            
            # Link to companies
            for company_name in contact['companies']:
                if company_name in customer_ids:
                    cursor.execute("""
                        INSERT INTO contact_companies (
                            contact_id, customer_id, is_primary, role
                        ) VALUES (%s, %s, %s, %s)
                    """, (
                        contact_id, customer_ids[company_name],
                        contact['is_primary'] if contact['companies'][0] == company_name else False,
                        contact['role']
                    ))
            
            logger.info(f"✅ Imported contact: {contact['first_name']} {contact['last_name']} (ID: {contact_id}, Companies: {len(contact['companies'])})")
        
        conn.commit()
        logger.info("🎉 Sample data import completed successfully!")
        
        # Print summary
        cursor.execute("SELECT COUNT(*) as count FROM customers")
        customer_count = cursor.fetchone()['count']
        
        cursor.execute("SELECT COUNT(*) as count FROM contacts")
        contact_count = cursor.fetchone()['count']
        
        cursor.execute("SELECT COUNT(*) as count FROM contact_companies")
        link_count = cursor.fetchone()['count']
        
        logger.info(f"\n📊 Summary:")
        logger.info(f"   Customers: {customer_count}")
        logger.info(f"   Contacts: {contact_count}")
        logger.info(f"   Company-Contact Links: {link_count}")
        
    except Exception as e:
        conn.rollback()
        logger.error(f"❌ Import failed: {e}")
        raise
    finally:
        cursor.close()
        conn.close()


if __name__ == "__main__":
    logger.info("🚀 Starting data import...")
    import_sample_data()