116 lines
3.9 KiB
Python
116 lines
3.9 KiB
Python
#!/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_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'], default='web',
|
|
help='Run mode: scrape prices or start web UI')
|
|
parser.add_argument('--config', help='Path to config file')
|
|
|
|
args = parser.parse_args()
|
|
|
|
if args.mode == 'scrape':
|
|
asyncio.run(run_scraper())
|
|
else:
|
|
run_web_ui()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|