sheety_data_manager.py

import os
import requests
from requests.auth import HTTPBasicAuth
from dotenv import load_dotenv

# Load environment variables
load_dotenv()

# Sheety API endpoint for flight deals
SHEETY_PRICES_ENDPOINT = "https://api.sheety.co/8f7dac75c174db2fc7a1b07db76b96fd/flightDeals/prices"

class SheetyDataManager:
    def __init__(self):
        # Initialize authentication credentials from environment variables
        self._user = os.environ["USER_NAME"]  # Username for authentication
        self._password = os.environ["PASSWORD"]  # Password for authentication
        self._authorization = HTTPBasicAuth(self._user, self._password)  # HTTP Basic Authentication
        self.display_data = {}  # Dictionary to store fetched data

    def get_sheet_data(self):
        """
        Fetches all data from the Sheety API and stores it in `self.display_data`.
        Returns the fetched data.
        """
        response = requests.get(url=SHEETY_PRICES_ENDPOINT, auth=self._authorization)

        if response.status_code == 200:
            data = response.json()
            self.display_data = data["prices"]  # Extract "prices" data from the API response
            return self.display_data
        else:
            print(f"Error fetching data: {response.status_code}, {response.text}")
            return None

    def update_sheet_data(self):
        """
        Updates the IATA codes for all cities in the Sheety database.
        Uses the `iataCode` field for each city to send PUT requests.
        """
        for city in self.display_data:
            new_data = {
                "price": {
                    "iataCode": city["iataCode"]  # New IATA code to update
                }
            }

            # Send a PUT request to update the data for the current city
            response = requests.put(
                url=f"{SHEETY_PRICES_ENDPOINT}/{city['id']}",  # Target URL with city ID
                json=new_data,  # JSON data for the update
                auth=self._authorization
            )

            # Log the response status and details
            print(f"Response for {city['city']}: {response.status_code}, {response.text}")

    def delete_sheet_data(self):
        """
        Resets the IATA codes for all cities in the Sheety database to an empty string.
        """
        for city in self.display_data:
            update = {
                "price": {
                    "iataCode": ""  # Reset the IATA code
                }
            }

            # Send a PUT request to reset the IATA code for the current city
            response = requests.put(
                url=f"{SHEETY_PRICES_ENDPOINT}/{city['id']}",
                json=update,
                auth=self._authorization
            )

            # Log the response details
            print(f"Reset response for {city['city']}: {response.text}")