from qucanft import AstroDataFetcher, ZodiacCalculator, HousesCalculator, AspectsCalculator, VisualizationHelper
from datetime import datetime

# Initialize components
fetcher = AstroDataFetcher()
zodiac_calc = ZodiacCalculator()
houses_calc = HousesCalculator()
aspects_calc = AspectsCalculator()
viz = VisualizationHelper()

# Fetch planetary positions
date = "2024-01-01T12:00:00"
location = "New York, NY"
planets = ["Sun", "Moon", "Mercury", "Venus", "Mars", "Jupiter", "Saturn"]

planetary_data = fetcher.get_planet_positions(date=date, location=location, planets=planets)

# Calculate zodiac positions
zodiac_data = zodiac_calc.calculate_zodiac_positions(planetary_data)

# Calculate house positions
ascendant = 75.0  # Example ascendant in Gemini
house_cusps = houses_calc.calculate_houses(ascendant=ascendant, house_system='equal')
house_data = houses_calc.add_house_positions(zodiac_data, house_cusps)

# Calculate aspects
aspects_data = aspects_calc.calculate_all_aspects(house_data)

# Create visualizations
natal_chart = viz.create_natal_chart(house_data, house_cusps, aspects_data)
natal_chart.savefig('natal_chart.png')

from qucanft import AstroDataFetcher

fetcher = AstroDataFetcher()

# Fetch planetary positions for a specific date and location
planetary_data = fetcher.get_planet_positions(
    date="2024-01-01T12:00:00",
    location="New York, NY",
    planets=["Sun", "Moon", "Mercury", "Venus", "Mars"]
)

# Fetch ephemeris data over a date range
ephemeris_data = fetcher.get_ephemeris_range(
    start_date="2024-01-01",
    end_date="2024-01-31",
    step="1d",
    planet="Sun"
)

# Custom query for any celestial object
asteroid_data = fetcher.get_custom_query(
    target_id="433",  # Eros asteroid
    date="2024-01-01T12:00:00",
    quantities="1,2,3,4"
)

from qucanft import ZodiacCalculator

zodiac_calc = ZodiacCalculator()

# Calculate zodiac positions from planetary data
zodiac_data = zodiac_calc.calculate_zodiac_positions(planetary_data)

# Get zodiac sign information
aries_info = zodiac_calc.get_zodiac_sign_info('Aries')
print(f"Aries: {aries_info['element']}, {aries_info['quality']}, ruled by {aries_info['ruler']}")

# Calculate zodiac compatibility
compatibility = zodiac_calc.get_zodiac_compatibility('Leo', 'Aquarius')
print(f"Leo-Aquarius compatibility: {compatibility['overall_compatibility']}")

from qucanft import HousesCalculator

houses_calc = HousesCalculator()

# Calculate house cusps using different systems
equal_houses = houses_calc.calculate_houses(ascendant=75.0, house_system='equal')
placidus_houses = houses_calc.calculate_houses(
    ascendant=75.0, 
    midheaven=345.0, 
    latitude=40.7128, 
    house_system='placidus'
)

# Add house positions to planetary data
house_data = houses_calc.add_house_positions(zodiac_data, equal_houses)

# Get house information
house_info = houses_calc.get_house_info(1)
print(f"1st House: {house_info['name']} - {house_info['theme']}")

from qucanft import AspectsCalculator

aspects_calc = AspectsCalculator()

# Calculate all aspects between planets
aspects_data = aspects_calc.calculate_all_aspects(house_data)

# Get strongest aspects
strongest = aspects_calc.get_strongest_aspects(aspects_data, limit=5)

# Filter aspects by nature
harmonious = aspects_calc.get_aspects_by_nature(aspects_data, 'Harmonious')
challenging = aspects_calc.get_aspects_by_nature(aspects_data, 'Challenging')

# Get aspects for a specific planet
sun_aspects = aspects_calc.get_planet_aspects(aspects_data, 'Sun')

from qucanft import VisualizationHelper

viz = VisualizationHelper()

# Create a natal chart
natal_chart = viz.create_natal_chart(house_data, house_cusps, aspects_data)
natal_chart.savefig('natal_chart.png')

# Create planetary distribution chart
positions_chart = viz.create_planetary_positions_chart(house_data)
positions_chart.savefig('planetary_positions.png')

# Create aspects summary
aspects_chart = viz.create_aspects_summary_chart(aspects_data)
aspects_chart.savefig('aspects_summary.png')

# Export data summary
summary = viz.export_data_summary(house_data, aspects_data, house_cusps)

# Using coordinates
location = {
    'lat': 40.7128,
    'lon': -74.0060,
    'elevation': 10
}

# Using string (common locations are recognized)
location = "Tokyo, Japan"

# Using JPL Horizons location codes
location = "500"  # Geocentric

# Single date
date = "2024-01-01T12:00:00"

# Date range for ephemeris
ephemeris = fetcher.get_ephemeris_range(
    start_date="2024-01-01",
    end_date="2024-12-31",
    step="1d",
    planet="Mars"
)