get_traffic
Check estimated search traffic for any website by domain or URL, with options for country-specific data and query modes.
Instructions
Check the estimated search traffic for any website.
Args:
domain_or_url (str): The domain or URL to query
country (str): The country to query, default is "None"
mode (["subdomains", "exact"]): The mode to use for the query
Returns:
Traffic data for the specified domain or URLInput Schema
TableJSON Schema
| Name | Required | Description | Default |
|---|---|---|---|
| domain_or_url | Yes | ||
| country | No | None | |
| mode | No | subdomains |
Implementation Reference
- src/seo_mcp/server.py:112-128 (handler)The get_traffic tool handler, registered via @mcp.tool() decorator. Handles captcha token acquisition and delegates to check_traffic for core logic.
@mcp.tool() def get_traffic(domain_or_url: str, country: str = "None", mode: Literal["subdomains", "exact"] = "subdomains") -> Optional[Dict[str, Any]]: """ Check the estimated search traffic for any website. Args: domain_or_url (str): The domain or URL to query country (str): The country to query, default is "None" mode (["subdomains", "exact"]): The mode to use for the query Returns: Traffic data for the specified domain or URL """ site_url = f"https://ahrefs.com/traffic-checker/?input={domain_or_url}&mode={mode}" token = get_capsolver_token(site_url) if not token: raise Exception(f"Failed to get verification token for domain: {domain_or_url}") return check_traffic(token, domain_or_url, mode, country) - src/seo_mcp/traffic.py:10-72 (helper)Core helper function implementing the traffic check by making authenticated request to Ahrefs API and parsing the response.
def check_traffic(token: str, domain_or_url: str, mode: Literal["subdomains", "exact"] = "subdomains", country: str = "None") -> Optional[Dict[str, Any]]: """ Check the estimated search traffic for any website. Args: domain_or_url (str): The domain or URL to query token (str): Verification token mode (str): Query mode, default is "subdomains" country (str): Country, default is "None" Returns: Optional[Dict[str, Any]]: Dictionary containing traffic data, returns None if request fails """ if not token: return None url = "https://ahrefs.com/v4/stGetFreeTrafficOverview" # 将参数转换为JSON字符串，然后作为单个input参数传递 params = { "input": json.dumps({ "captcha": token, "country": country, "protocol": "None", "mode": mode, "url": domain_or_url }) } headers = { "accept": "*/*", "content-type": "application/json", "referer": f"https://ahrefs.com/traffic-checker/?input={domain_or_url}&mode={mode}" } try: response = requests.get(url, params=params, headers=headers) if response.status_code != 200: return None data: Optional[List[Any]] = response.json() # 检查响应数据格式 if not isinstance(data, list) or len(data) < 2 or data[0] != "Ok": return None # 提取有效数据 traffic_data = data[1] # 格式化返回结果 result = { "traffic_history": traffic_data.get("traffic_history", []), "traffic": { "trafficMonthlyAvg": traffic_data.get("traffic", {}).get("trafficMonthlyAvg", 0), "costMontlyAvg": traffic_data.get("traffic", {}).get("costMontlyAvg", 0) }, "top_pages": traffic_data.get("top_pages", []), "top_countries": traffic_data.get("top_countries", []), "top_keywords": traffic_data.get("top_keywords", []) } return result except Exception as e: