wehub-resource-sync
111 lines
4.1 KiB
Python
111 lines
4.1 KiB
Python
import os
|
|
import time
|
|
import requests
|
|
import json
|
|
import sys
|
|
import datetime
|
|
import io
|
|
|
|
# Force UTF-8 encoding for standard output and error streams
|
|
sys.stdout = io.TextIOWrapper(sys.stdout.buffer, encoding='utf-8')
|
|
sys.stderr = io.TextIOWrapper(sys.stderr.buffer, encoding='utf-8')
|
|
|
|
# API Configuration
|
|
TEMPLATE_ID = "77577579210625331"
|
|
API_BASE_URL = "https://api.browseract.com/v2/workflow"
|
|
|
|
def run_google_maps_task(api_key, keywords, language="en", country="us"):
|
|
headers = {"Authorization": f"Bearer {api_key}"}
|
|
payload = {
|
|
"workflow_template_id": TEMPLATE_ID,
|
|
"input_parameters": [
|
|
{"name": "keywords", "value": keywords},
|
|
{"name": "language", "value": language},
|
|
{"name": "country", "value": country}
|
|
]
|
|
}
|
|
|
|
# 1. Start Task
|
|
print(f"Start Task", flush=True)
|
|
try:
|
|
res = requests.post(f"{API_BASE_URL}/run-task-by-template", json=payload, headers=headers, timeout=30).json()
|
|
except Exception as e:
|
|
print(f"Error: Connection to API failed - {e}", flush=True)
|
|
return None
|
|
|
|
if "id" not in res:
|
|
# Check for authorization error
|
|
if "Invalid authorization" in str(res):
|
|
print(f"Error: Invalid authorization. Please check your API key.", flush=True)
|
|
else:
|
|
print(f"Error: Could not start task. Response: {res}", flush=True)
|
|
return None
|
|
|
|
task_id = res["id"]
|
|
print(f"Task started. ID: {task_id}", flush=True)
|
|
|
|
# 2. Poll for Completion
|
|
max_poll_time = 300
|
|
poll_start = time.time()
|
|
while time.time() - poll_start < max_poll_time:
|
|
try:
|
|
status_res = requests.get(f"{API_BASE_URL}/get-task-status?task_id={task_id}", headers=headers, timeout=30).json()
|
|
status = status_res.get("status")
|
|
|
|
timestamp = datetime.datetime.now().strftime("%H:%M:%S")
|
|
print(f"[{timestamp}] Task Status: {status}", flush=True)
|
|
|
|
if status == "finished":
|
|
print(f"[{timestamp}] Task finished successfully.", flush=True)
|
|
break
|
|
elif status in ["failed", "canceled"]:
|
|
print(f"Error: Task {status}. Please check your BrowserAct dashboard.", flush=True)
|
|
return None
|
|
except Exception as e:
|
|
timestamp = datetime.datetime.now().strftime("%H:%M:%S")
|
|
print(f"[{timestamp}] Polling error: {e}. Retrying...", flush=True)
|
|
|
|
time.sleep(10)
|
|
print(f"Error: Task polling timed out after {max_poll_time} seconds.", flush=True)
|
|
return None
|
|
|
|
# 3. Get Results
|
|
try:
|
|
task_info = requests.get(f"{API_BASE_URL}/get-task?task_id={task_id}", headers=headers, timeout=30).json()
|
|
|
|
# The result is usually in output['string'] or within the raw JSON response
|
|
output = task_info.get("output", {})
|
|
result_string = output.get("string")
|
|
|
|
if result_string:
|
|
return result_string
|
|
else:
|
|
return json.dumps(task_info, ensure_ascii=False)
|
|
except Exception as e:
|
|
print(f"Error: Failed to retrieve results - {e}", flush=True)
|
|
return None
|
|
|
|
if __name__ == "__main__":
|
|
# Get API key from environment variable
|
|
api_key = os.getenv("BROWSERACT_API_KEY")
|
|
|
|
if len(sys.argv) < 2:
|
|
print("Usage: python google_maps_api.py <keywords> [language] [country]", flush=True)
|
|
sys.exit(1)
|
|
|
|
if not api_key:
|
|
print("\n[!] ERROR: BrowserAct API Key is missing.", flush=True)
|
|
print("Please follow these steps:", flush=True)
|
|
print("1. Go to: https://www.browseract.com/reception/integrations", flush=True)
|
|
print("2. Copy your API Key.", flush=True)
|
|
print("3. Provide it to me or set it as an environment variable (BROWSERACT_API_KEY).", flush=True)
|
|
sys.exit(1)
|
|
|
|
keywords = sys.argv[1]
|
|
language = sys.argv[2] if len(sys.argv) > 2 else "en"
|
|
country = sys.argv[3] if len(sys.argv) > 3 else "us"
|
|
|
|
result = run_google_maps_task(api_key, keywords, language, country)
|
|
if result:
|
|
print(result, flush=True)
|