import requests
from bs4 import BeautifulSoup
import json
import sys


def fetch_fitx_occupancy(studio_url):
    headers = {
        'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36'
    }

    response = requests.get(studio_url, headers=headers)

    soup = BeautifulSoup(response.content, 'html.parser')

    # The data is hidden in a section tag with class 'studio_graph'
    graph_section = soup.find('section', class_='studio_graph')

    if not graph_section:
        print("Could not find graph section. Check URL or website structure.")
        return

    # Extracting raw JSON strings from attributes
    visitor_data_raw = graph_section.get('data-visitordata')
    current_day_raw = graph_section.get('data-current-day-data')

    # Parsing JSON
    # visitor_data is usually 7 arrays (Mon-Sun), each containing 24 integers (0h-23h)
    forecast_data = json.loads(visitor_data_raw)

    # current_day_data is an array of integers for the hours passed so far today
    live_data = json.loads(current_day_raw)

    studio_name = soup.find('h1', class_='studio_hero__headline').get_text(strip=True)

    # Helper function to map values to time strings
    def map_to_times(data_list):
        return {f"{i:02d}:00": value for i, value in enumerate(data_list)}

    week_days = ["Monday", "Tuesday", "Wednesday", "Thursday", "Friday", "Saturday", "Sunday"]

    # Create the forecast dictionary with time keys
    forecast_with_times = {}
    for i, day in enumerate(week_days):
        forecast_with_times[day] = map_to_times(forecast_data[i])

    result = {
        "studio": studio_name,
        "live_utilization_today": map_to_times(live_data),
        "forecast_week_data": forecast_with_times
    }

    return result


if __name__ == "__main__":
    # URL defaults to Berlin Tierpark, but can be overridden via command line
    target_url = "https://www.fitx.de/fitnessstudios/berlin-tierpark"

    if len(sys.argv) > 1:
        target_url = sys.argv[1]

    data = fetch_fitx_occupancy(target_url)

    if data:
        print(json.dumps(data, indent=2, ensure_ascii=False))