1#!/usr/bin/env python3
2"""
3Related keywords from Google "searches related to" using DataForSEO API
4Get up to 4,680 keyword ideas from Google's related searches
5 
6Usage: python3 scripts/related_keywords.py "AI agent" --depth 2 --limit 50
7"""
8import argparse
9from dataforseo_api import api_post, get_result, format_count
10 
11 
12def main():
13    parser = argparse.ArgumentParser(description="Related keywords from Google")
14    parser.add_argument("keyword", help="Seed keyword")
15    parser.add_argument("--location", "-loc", type=int, default=2840,
16                        help="Location code (default: 2840 = US)")
17    parser.add_argument("--depth", "-d", type=int, default=1,
18                        help="Search depth 1-3 (default: 1, max keywords: depth^3 * 10)")
19    parser.add_argument("--limit", "-l", type=int, default=50, 
20                        help="Max results to display (default: 50)")
21    args = parser.parse_args()
22 
23    # Validate depth
24    if args.depth < 1 or args.depth > 3:
25        print("Error: depth must be between 1 and 3")
26        return
27 
28    data = [{
29        "keyword": args.keyword,
30        "location_code": args.location,
31        "language_code": "en",
32        "depth": args.depth,
33        "limit": 1000  # API limit, we'll filter in display
34    }]
35    
36    response = api_post("dataforseo_labs/google/related_keywords/live", data)
37    results = get_result(response)
38    
39    print(f"keyword: {args.keyword}")
40    print(f"location: {args.location}")
41    print(f"depth: {args.depth}")
42    print()
43    
44    if results:
45        keywords = []
46        for result in results:
47            items = result.get("items", [])
48            for item in items:
49                kw_data = item.get("keyword_data", {})
50                keyword = kw_data.get("keyword", item.get("keyword", ""))
51                volume = kw_data.get("search_volume", item.get("search_volume", 0))
52                difficulty = kw_data.get("keyword_difficulty", item.get("keyword_difficulty", "N/A"))
53                
54                if keyword:
55                    keywords.append({
56                        "keyword": keyword,
57                        "volume": volume if volume is not None else 0,
58                        "difficulty": difficulty
59                    })
60        
61        # Sort by volume desc
62        keywords.sort(key=lambda x: x["volume"], reverse=True)
63        
64        # Display limited results
65        display_keywords = keywords[:args.limit]
66        print(f"related_keywords[{len(display_keywords)} of {len(keywords)}]{{keyword,volume,difficulty}}:")
67        for kw in display_keywords:
68            keyword = kw["keyword"]
69            volume = format_count(kw["volume"])
70            difficulty = kw["difficulty"]
71            print(f"  {keyword},{volume},{difficulty}")
72        
73        if len(keywords) > args.limit:
74            print(f"\n... and {len(keywords) - args.limit} more keywords (use --limit to show more)")
75    else:
76        print("No related keywords found")
77    
78    print()
79    print("Tip: Higher depth finds more keywords but costs more API credits")
80    print(f"  Depth 1: ~10 keywords, Depth 2: ~100, Depth 3: ~1,000+")
81 
82 
83if __name__ == "__main__":
84    main()
85