Multi-Location Tracking

#!/usr/bin/env python3
"""
Multi-Location Rank Tracker
Compares keyword rankings across different geographic locations
"""

import os
import time
import requests
from collections import defaultdict

API_KEY = os.environ.get("SERPWATCH_API_KEY")
BASE_URL = "https://engine.v2.serpwatch.io"

# Configuration
TARGET_DOMAIN = "yourdomain.com"
KEYWORDS = ["pizza delivery", "best pizza near me", "pizza restaurant"]
LOCATIONS = [
    {"name": "New York, United States", "code": 1023191},
    {"name": "Los Angeles, United States", "code": 1013962},
    {"name": "Chicago, United States", "code": 1016367},
]

def submit_batch(keywords, locations):
    """Submit batch request for all keyword/location combinations."""
    batch = []
    for keyword in keywords:
        for location in locations:
            batch.append({
                "keyword": keyword,
                "location_name": location["name"],
                "location_code": location["code"],
                "device": "mobile",
                "depth": 20,
                "language_code": "en"
            })

    response = requests.post(
        f"{BASE_URL}/api/v2/serp/crawl/google/batch",
        headers={
            "Authorization": f"Bearer {API_KEY}",
            "Content-Type": "application/json"
        },
        json=batch
    )
    response.raise_for_status()
    return response.json()

def wait_for_results(tasks, timeout=300):
    """Wait for all tasks to complete."""
    results = {}
    pending = {t["id"] for t in tasks}
    start = time.time()

    while pending and (time.time() - start) < timeout:
        for task_id in list(pending):
            response = requests.get(
                f"{BASE_URL}/api/v2/serp/crawl/{task_id}",
                headers={"Authorization": f"Bearer {API_KEY}"}
            )
            data = response.json()

            if data["status"] in ["success", "completed", "error"]:
                results[task_id] = data
                pending.remove(task_id)

        if pending:
            print(f"  Waiting... {len(pending)} tasks remaining")
            time.sleep(5)

    return results

def find_rank(result, domain):
    """Find domain position in organic results."""
    for item in result.get("result", {}).get("organic", []):
        item_domain = item.get("domain", "").lower()
        if domain.lower() in item_domain:
            return item["position"]
    return None

def generate_report(tasks, results, domain):
    """Generate location comparison report."""
    # Organize by keyword -> location -> rank
    report = defaultdict(dict)

    for task in tasks:
        task_id = task["id"]
        keyword = task["keyword"]
        location = task["location_name"]
        result = results.get(task_id, {})

        if result.get("status") in ["success", "completed"]:
            rank = find_rank(result, domain)
            report[keyword][location] = rank
        else:
            report[keyword][location] = "Error"

    return report

def print_report(report, locations):
    """Print formatted comparison report."""
    # Header
    loc_names = [loc["name"].split(",")[0] for loc in locations]
    print("\n" + "=" * 80)
    print("MULTI-LOCATION RANKING REPORT")
    print("=" * 80)
    print(f"\nDomain: {TARGET_DOMAIN}\n")

    # Column headers
    header = f"{'Keyword':<30}"
    for name in loc_names:
        header += f"{name:>12}"
    print(header)
    print("-" * 80)

    # Data rows
    for keyword, ranks in report.items():
        row = f"{keyword[:28]:<30}"
        for loc in locations:
            rank = ranks.get(loc["name"])
            if rank is None:
                display = "Not found"
            elif rank == "Error":
                display = "Error"
            else:
                display = f"#{rank}"
            row += f"{display:>12}"
        print(row)

    print("-" * 80)

def main():
    total_queries = len(KEYWORDS) * len(LOCATIONS)
    print(f"Tracking {len(KEYWORDS)} keywords across {len(LOCATIONS)} locations")
    print(f"Total queries: {total_queries}")
    print("-" * 50)

    # Submit batch
    print("\nSubmitting batch request...")
    tasks = submit_batch(KEYWORDS, LOCATIONS)
    print(f"Submitted {len(tasks)} tasks")

    # Wait for results
    print("\nWaiting for results...")
    results = wait_for_results(tasks)
    print(f"Received {len(results)} results")

    # Generate report
    report = generate_report(tasks, results, TARGET_DOMAIN)
    print_report(report, LOCATIONS)

if __name__ == "__main__":
    main()