sheety_api.py

import requests
import os
from dotenv import load_dotenv

# Load environment variables
load_dotenv()

class SheetyAPI:
    def __init__(self):
        """Initialize Sheety API with the base URL from environment variables."""
        self.base_url = os.getenv("SHEETY_URL")
        if not self.base_url:
            raise ValueError("Sheety URL not found. Check your .env file.")
    def post_flight(self, flight):
        """Send flight data to Sheety API."""
        # Ensure correct key names matching Google Sheet headers
        new_data = {
            "price": {  # ✅ This should match the sheet name in Sheety
                "origin": flight['origin'],
                "destination": flight['destination'],
                "departure": flight['departure_date'],
                "return": flight['return_date'],
                "price": flight['price']
            }
        }
        # Send data to Sheety
        response = requests.post(self.base_url, json=new_data)
        # Handle response
        if response.status_code == 200:
            print("✅ Flight added successfully!")
            return response.json()
        else:
            print(f"❌ Error {response.status_code}: {response.text}")
            return None