import psycopg2
import time
from datetime import datetime

def test_working_parser():
    """Тест рабочего парсера"""
    db_config = {
        'host': '127.0.0.1',
        'database': 'aliexpress_automation',
        'user': 'automation_user',
        'password': 'AutoPass123'
    }
    
    # Тестовые запросы
    search_queries = [
        "camping solar panel",
        "rv battery 12v",
        "portable camping stove"
    ]
    
    print("🚀 Тестирование рабочего парсера...")
    
    try:
        conn = psycopg2.connect(**db_config)
        cursor = conn.cursor()
        
        # Проверяем количество товаров до
        cursor.execute("SELECT COUNT(*) FROM products")
        count_before = cursor.fetchone()[0]
        print(f"📊 Товаров в БД до парсинга: {count_before}")
        
        added_count = 0
        
        # Шаблоны товаров
        product_templates = {
            'solar': {
                'titles': ['Солнечная панель {}W для кемпера', 'Портативная солнечная батарея {}W'],
                'category': 'electronics',
                'price_range': (5000, 15000)
            },
            'battery': {
                'titles': ['Аккумулятор LiFePO4 {}Ah 12V', 'Литиевая батарея {}Ah для кемпера'],
                'category': 'electronics',
                'price_range': (8000, 25000)
            },
            'stove': {
                'titles': ['Газовая плитка {}', 'Туристическая горелка {}'],
                'category': 'cooking',
                'price_range': (2000, 8000)
            }
        }
        
        for query in search_queries:
            query_lower = query.lower()
            
            # Определяем тип товара
            if 'solar' in query_lower:
                template_key = 'solar'
            elif 'battery' in query_lower:
                template_key = 'battery'
            elif 'stove' in query_lower:
                template_key = 'stove'
            else:
                template_key = 'solar'  # по умолчанию
            
            template = product_templates[template_key]
            
            # Генерируем 2 товара на запрос
            for i in range(2):
                hash_val = abs(hash(query + str(i)))
                power_capacity = [50, 100, 150, 200][hash_val % 4]
                
                title = template['titles'][i % len(template['titles'])].format(power_capacity)
                
                price_min, price_max = template['price_range']
                price = price_min + (hash_val % (price_max - price_min))
                original_price = int(price * 1.4)
                
                test_product = {
                    'title': title,
                    'price': price,
                    'original_price': original_price,
                    'main_image': f'https://ae01.alicdn.com/kf/S{hash_val % 999999999}.jpg',
                    'aliexpress_link': f'https://aliexpress.com/item/{int(time.time())}{hash_val % 10000}.html',
                    'rating': 4.0 + (hash_val % 10) / 10,
                    'orders_count': hash_val % 500 + 10,
                    'original_description': f'Качественный товар для кемпинга: {query}. Проверенное оборудование.',
                    'category': template['category']
                }
                
                # Проверка на дубликаты
                cursor.execute("""
                    SELECT id FROM products WHERE aliexpress_link = %s
                """, (test_product['aliexpress_link'],))
                
                if not cursor.fetchone():
                    insert_query = """
                        INSERT INTO products (
                            title, original_description, price, original_price,
                            main_image, category, aliexpress_link, rating, orders_count,
                            status, created_at, updated_at
                        ) VALUES (%s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s)
                        RETURNING id
                    """
                    
                    cursor.execute(insert_query, (
                        test_product['title'],
                        test_product['original_description'],
                        test_product['price'],
                        test_product['original_price'],
                        test_product['main_image'],
                        test_product['category'],
                        test_product['aliexpress_link'],
                        test_product['rating'],
                        test_product['orders_count'],
                        'active',
                        datetime.now(),
                        datetime.now()
                    ))
                    
                    product_id = cursor.fetchone()[0]
                    added_count += 1
                    print(f"✅ Добавлен товар #{product_id}: {test_product['title']}")
                    
                    time.sleep(0.1)  # Небольшая задержка
        
        conn.commit()
        
        # Проверяем количество товаров после
        cursor.execute("SELECT COUNT(*) FROM products")
        count_after = cursor.fetchone()[0]
        
        cursor.close()
        conn.close()
        
        print(f"📊 Товаров в БД после парсинга: {count_after}")
        print(f"🎉 Добавлено новых товаров: {added_count}")
        print(f"🔍 Разница: {count_after - count_before}")
        
        return True
        
    except Exception as e:
        print(f"❌ Ошибка: {e}")
        return False

if __name__ == "__main__":
    test_working_parser()