test_supabase.py

#!/usr/bin/env python3
"""
Simple Supabase connection test script.
This script tests both the Supabase Python client and direct REST API access.
"""

import os

import requests
from dotenv import load_dotenv
from supabase import create_client

# Load environment variables
load_dotenv()

# Get Supabase credentials from environment and strip any whitespace
SUPABASE_URL = os.getenv("SUPABASE_URL", "").strip()
SUPABASE_ANON_KEY = os.getenv("SUPABASE_KEY", "").strip()
SUPABASE_SERVICE_KEY = os.getenv("SUPABASE_SERVICE_ROLE_KEY", "").strip()

print("=== Supabase Connection Test ===")
print(f"URL: {SUPABASE_URL}")
print(f"Anon Key: {SUPABASE_ANON_KEY[:10]}...{SUPABASE_ANON_KEY[-5:]}")
print(
    f"Service Key: {SUPABASE_SERVICE_KEY[:10]}...{SUPABASE_SERVICE_KEY[-5:]}"
)  # Line break
print("=" * 30)

# Test 1: Try with Python client and anon key
print("\n1. Testing with Python client and anon key:")
try:
    client = create_client(SUPABASE_URL, SUPABASE_ANON_KEY)
    print("✅ Client initialized successfully")

    # Try a simple query
    try:
        response = client.table("_dummy_table_").select("*").limit(1).execute()
        print("✅ Query executed successfully")
    except Exception as e:
        print(f"❌ Query failed (expected for non-existent table): {e}")
        print("✅ But client connection was established")
except Exception as e:
    print(f"❌ Failed to initialize client: {e}")

# Test 2: Try with Python client and service role key
print("\n2. Testing with Python client and service role key:")
try:
    client = create_client(SUPABASE_URL, SUPABASE_SERVICE_KEY)
    print("✅ Client initialized successfully")

    # Try a simple query
    try:
        response = client.table("_dummy_table_").select("*").limit(1).execute()
        print("✅ Query executed successfully")
    except Exception as e:
        print(f"❌ Query failed (expected for non-existent table): {e}")
        print("✅ But client connection was established")
except Exception as e:
    print(f"❌ Failed to initialize client: {e}")

# Test 3: Direct REST API test with anon key
print("\n3. Testing direct REST API with anon key:")
headers = {
    "apikey": SUPABASE_ANON_KEY,
    "Authorization": f"Bearer {SUPABASE_ANON_KEY}",
    "Content-Type": "application/json",
}
try:
    response = requests.get(
        f"{SUPABASE_URL}/rest/v1/_dummy_table_?limit=1", headers=headers
    )
    print(f"Status code: {response.status_code}")
    if response.status_code == 200:
        print("✅ API connection successful")
    elif response.status_code == 404:
        print("✅ API connection successful (404 expected for non-existent table)")
    else:
        print(f"❌ API connection failed: {response.text}")
except Exception as e:
    print(f"❌ API request failed: {e}")

# Test 4: Direct REST API test with service role key
print("\n4. Testing direct REST API with service role key:")
headers = {
    "apikey": SUPABASE_SERVICE_KEY,
    "Authorization": f"Bearer {SUPABASE_SERVICE_KEY}",
    "Content-Type": "application/json",
}
try:
    response = requests.get(
        f"{SUPABASE_URL}/rest/v1/_dummy_table_?limit=1", headers=headers
    )
    print(f"Status code: {response.status_code}")
    if response.status_code == 200:
        print("✅ API connection successful")
    elif response.status_code == 404:
        print("✅ API connection successful (404 expected for non-existent table)")
    else:
        print(f"❌ API connection failed: {response.text}")
except Exception as e:
    print(f"❌ API request failed: {e}")

print("\n=== Test Complete ===")