1#!/usr/bin/env python3
2"""
3DataForSEO API wrapper
4"""
5import urllib.request
6import urllib.parse
7import json
8import base64
9import sys
10from credential import get_dataforseo_credentials
11 
12API_BASE = "https://api.dataforseo.com/v3"
13 
14 
15def api_post(endpoint: str, data: list) -> dict:
16    """Make POST request to DataForSEO API"""
17    login, password = get_dataforseo_credentials()
18    if not login or not password:
19        print("error: DATAFORSEO_LOGIN and DATAFORSEO_PASSWORD not set", file=sys.stderr)
20        print("Run: export DATAFORSEO_LOGIN=your_login", file=sys.stderr)
21        print("     export DATAFORSEO_PASSWORD=your_password", file=sys.stderr)
22        sys.exit(1)
23    
24    url = f"{API_BASE}/{endpoint}"
25    auth = base64.b64encode(f"{login}:{password}".encode()).decode()
26    headers = {
27        "Authorization": f"Basic {auth}",
28        "Content-Type": "application/json"
29    }
30    
31    req = urllib.request.Request(
32        url,
33        data=json.dumps(data).encode(),
34        headers=headers,
35        method="POST"
36    )
37    
38    try:
39        with urllib.request.urlopen(req, timeout=60) as resp:
40            return json.loads(resp.read().decode())
41    except urllib.error.HTTPError as e:
42        error_body = e.read().decode()
43        print(f"error: HTTP {e.code} - {error_body}", file=sys.stderr)
44        sys.exit(1)
45    except Exception as e:
46        print(f"error: {e}", file=sys.stderr)
47        sys.exit(1)
48 
49 
50def format_count(n) -> str:
51    """Format numbers (1234567 -> 1.2M)"""
52    if n is None:
53        return "0"
54    n = int(n)
55    if n >= 1_000_000_000:
56        return f"{n/1_000_000_000:.1f}B"
57    if n >= 1_000_000:
58        return f"{n/1_000_000:.1f}M"
59    if n >= 1_000:
60        return f"{n/1_000:.1f}K"
61    return str(n)
62 
63 
64def get_result(response: dict) -> list:
65    """Extract result from API response"""
66    if not response.get("tasks"):
67        return []
68    task = response["tasks"][0]
69    if task.get("status_code") != 20000:
70        print(f"error: {task.get('status_message', 'Unknown error')}", file=sys.stderr)
71        return []
72    return task.get("result", [])
73 
74 
75def print_keywords_list(keywords: list):
76    """Print list of keywords"""
77    print(f"keywords[{len(keywords)}]{{keyword,volume,difficulty}}:")
78    for kw in keywords:
79        keyword = kw.get("keyword", "N/A")
80        volume = format_count(kw.get("search_volume"))
81        difficulty = kw.get("keyword_difficulty", "N/A")
82        print(f"  {keyword},{volume},{difficulty}")
83 
84 
85def print_serp_list(items: list):
86    """Print SERP results"""
87    organic = [i for i in items if i.get("type") == "organic"]
88    print(f"serp[{len(organic)}]{{position,title,domain}}:")
89    for item in organic[:20]:
90        pos = item.get("rank_absolute", "N/A")
91        title = (item.get("title") or "")[:50]
92        domain = item.get("domain", "N/A")
93        print(f"  {pos},{title},{domain}")
94 
95 
96def print_backlinks_list(items: list):
97    """Print backlinks"""
98    print(f"backlinks[{len(items)}]{{from,to,rank,dofollow}}:")
99    for item in items:
100        url_from = (item.get("url_from") or "")[:50]
101        url_to = (item.get("url_to") or "")[:30]
102        rank = item.get("rank", "N/A")
103        dofollow = item.get("dofollow", False)
104        print(f"  {url_from},{url_to},{rank},{dofollow}")
105