1#!/usr/bin/env python3
2"""
3SERP analysis using DataForSEO API
4Usage: python3 scripts/serp_analysis.py "best seo tools" --depth 20
5"""
6import argparse
7from dataforseo_api import api_post, get_result, print_serp_list, format_count
8 
9 
10def main():
11    parser = argparse.ArgumentParser(description="SERP analysis")
12    parser.add_argument("keyword", help="Search keyword")
13    parser.add_argument("--location", "-loc", type=int, default=2840,
14                        help="Location code (default: 2840 = US)")
15    parser.add_argument("--depth", "-d", type=int, default=20, help="Search depth")
16    args = parser.parse_args()
17 
18    data = [{
19        "keyword": args.keyword,
20        "location_code": args.location,
21        "language_code": "en",
22        "depth": args.depth
23    }]
24    
25    response = api_post("serp/google/organic/live/advanced", data)
26    results = get_result(response)
27    
28    print(f"keyword: {args.keyword}")
29    print(f"location: {args.location}")
30    
31    if results:
32        result = results[0]
33        print(f"total_results: {format_count(result.get('se_results_count'))}")
34        items = result.get("items", [])
35        print_serp_list(items)
36    else:
37        print("No results found")
38 
39 
40if __name__ == "__main__":
41    main()
42