66 lines
2.1 KiB
Python
66 lines
2.1 KiB
Python
import os
|
|
import requests
|
|
import json
|
|
from dotenv import load_dotenv
|
|
|
|
load_dotenv(dotenv_path="/home/node/clawd/.env")
|
|
|
|
NOTION_TOKEN = os.getenv("NOTION_API_KEY")
|
|
HEADERS = {
|
|
"Authorization": f"Bearer {NOTION_TOKEN}",
|
|
"Content-Type": "application/json",
|
|
"Notion-Version": "2022-06-28"
|
|
}
|
|
|
|
def find_db_id(query_name):
|
|
url = "https://api.notion.com/v1/search"
|
|
payload = {"query": query_name, "filter": {"value": "database", "property": "object"}}
|
|
resp = requests.post(url, headers=HEADERS, json=payload)
|
|
if resp.status_code == 200:
|
|
results = resp.json().get("results", [])
|
|
if results:
|
|
return results[0]['id']
|
|
return None
|
|
|
|
def fetch_products():
|
|
# Find Product DB (it's likely named "Product Categories" or similar based on schema)
|
|
# Or search for "Products"
|
|
db_id = find_db_id("Product Categories")
|
|
if not db_id:
|
|
db_id = find_db_id("Products") # Fallback
|
|
|
|
if not db_id:
|
|
print("❌ Could not find Product Database.")
|
|
return
|
|
|
|
print(f"--- Fetching Products from DB {db_id} ---")
|
|
|
|
url = f"https://api.notion.com/v1/databases/{db_id}/query"
|
|
resp = requests.post(url, headers=HEADERS, json={})
|
|
|
|
products = {}
|
|
if resp.status_code == 200:
|
|
results = resp.json().get("results", [])
|
|
for page in results:
|
|
props = page['properties']
|
|
# Find Title
|
|
name = "Unknown"
|
|
if "Name" in props and props["Name"]["title"]:
|
|
name = props["Name"]["title"][0]["plain_text"]
|
|
elif "Product Name" in props and props["Product Name"]["title"]:
|
|
name = props["Product Name"]["title"][0]["plain_text"]
|
|
|
|
products[name] = page['id']
|
|
print(f"Found: {name} ({page['id']})")
|
|
|
|
# Save to file
|
|
os.makedirs("data", exist_ok=True)
|
|
with open("data/product_mapping.json", "w") as f:
|
|
json.dump(products, f, indent=4)
|
|
print("✅ Saved to data/product_mapping.json")
|
|
else:
|
|
print(f"Error: {resp.text}")
|
|
|
|
if __name__ == "__main__":
|
|
fetch_products()
|