price-tracker/main.py

#!/uhttps://www.atoz-catering.co.uksr/bin/env python3
"""
Price Tracker - Web Scraper for Product Price Monitoring
Tracks product prices across multiple e-commerce sites
"""

import asyncio
import logging
from datetime import datetime
from typing import List, Dict, Optional
import argparse

from src.scraper_manager import ScraperManager
from src.database import DatabaseManager
from src.config import Config
from src.notification import NotificationManager
from src.web_ui import create_app

# Configure logging
logging.basicConfig(
    level=logging.INFO,
    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
    handlers=[
        logging.FileHandler('price_tracker.log'),
        logging.StreamHandler()
    ]
)

logger = logging.getLogger(__name__)

async def run_scraper():
    """Run the price scraping process."""
    try:
        config = Config()
        db_manager = DatabaseManager(config.database_path)
        scraper_manager = ScraperManager(config)
        notification_manager = NotificationManager(config)
        
        logger.info("Starting price tracking session")
        
        # Load products from database
        products = db_manager.get_all_products()
        if not products:
            logger.warning("No products found in database. Add products first.")
            return
        
        # Scrape prices for all products
        results = await scraper_manager.scrape_all_products(products)
        
        # Process results and save to database
        price_alerts = []
        for product_id, site_prices in results.items():
            for site_name, price_data in site_prices.items():
                if price_data['success']:
                    # Save price to database
                    db_manager.save_price_history(
                        product_id=product_id,
                        site_name=site_name,
                        price=price_data['price'],
                        currency=price_data.get('currency', 'USD'),
                        availability=price_data.get('availability', True),
                        timestamp=datetime.now()
                    )
                    
                    # Check for price alerts
                    product = db_manager.get_product(product_id)
                    if product and price_data['price'] <= product['target_price']:
                        price_alerts.append({
                            'product': product,
                            'site': site_name,
                            'current_price': price_data['price'],
                            'target_price': product['target_price']
                        })
        
        # Send notifications for price alerts
        if price_alerts:
            await notification_manager.send_price_alerts(price_alerts)
        
        logger.info(f"Scraping completed. Found {len(price_alerts)} price alerts.")
        
    except Exception as e:
        logger.error(f"Error during scraping: {e}")
        raise


def run_shopping_lists():
    """Generate and optionally send daily shopping lists."""
    from src.config import Config
    from src.database import DatabaseManager
    from src.notification import NotificationManager
    from src.shopping_list import AutoShoppingListGenerator
    
    config = Config()
    db_manager = DatabaseManager(config.database_path)
    notification_manager = NotificationManager(config)
    shopping_generator = AutoShoppingListGenerator(db_manager, notification_manager)
    
    print("🛒 Generating automated shopping lists...")
    
    # Generate shopping lists
    shopping_lists = shopping_generator.generate_shopping_lists()
    summary = shopping_generator.get_summary_stats()
    
    # Display results
    print(f"\n📊 Summary:")
    print(f"  • {summary['total_products']} products tracked")
    print(f"  • £{summary['total_cost']:.2f} total cost")
    print(f"  • £{summary['total_savings']:.2f} total savings")
    print(f"  • {summary['store_count']} stores involved")
    
    for store_list in shopping_lists:
        print(f"\n🏪 {store_list.store_display_name}: {store_list.item_count} items (£{store_list.total_cost:.2f})")
    
    # Send daily list if email is configured
    email_config = config.notification_config.get('email', {})
    if email_config.get('enabled') and email_config.get('recipient_email'):
        print(f"\n📧 Sending daily shopping list...")
        success = shopping_generator.send_daily_shopping_list()
        if success:
            print(f"✅ Shopping list sent successfully!")
        else:
            print(f"❌ Failed to send shopping list")
    
    return shopping_lists


def run_web_ui():
    """Run the web UI for managing products and viewing price history."""
    import os
    
    # Use environment variables for configuration
    host = os.environ.get('FLASK_HOST', '0.0.0.0')
    port = int(os.environ.get('FLASK_PORT', 5000))
    debug = os.environ.get('FLASK_ENV', 'production').lower() != 'production'
    
    app = create_app()
    logger.info(f"Starting Price Tracker web server on {host}:{port}")
    app.run(host=host, port=port, debug=debug)


def main():
    parser = argparse.ArgumentParser(description='Price Tracker')
    parser.add_argument('--mode', choices=['scrape', 'web', 'shopping'], default='web',
                       help='Run mode: scrape prices, start web UI, or generate shopping lists')
    parser.add_argument('--config', help='Path to config file')
    
    args = parser.parse_args()
    
    if args.mode == 'scrape':
        asyncio.run(run_scraper())
    elif args.mode == 'shopping':
        run_shopping_lists()
    else:
        run_web_ui()


if __name__ == "__main__":
    main()