Create astro_core.py

astro_core.py

@@ -0,0 +1,162 @@
+# astro_core.py
+"""
+Unified astrology calculation module supporting both Western and Vedic systems.
+"""
+
+from datetime import datetime
+from typing import Dict, Any, Literal
+import pytz
+from dataclasses import dataclass
+
+# Western astrology imports
+import flatlib
+from flatlib.datetime import Datetime
+from flatlib.geopos import GeoPos
+from flatlib.chart import Chart
+
+# Vedic astrology imports
+import jyotish
+from jyotish.core import Chart as VedicChart
+
+@dataclass
+class BirthInfo:
+    """Birth information container"""
+    datetime: datetime
+    latitude: float
+    longitude: float
+    timezone: str
+    system: Literal['western', 'vedic'] = 'western'
+
+class ChartCalculator:
+    """Unified chart calculator for Western and Vedic astrology"""
+    
+    def __init__(self):
+        self.western_planets = [
+            'Sun', 'Moon', 'Mercury', 'Venus', 'Mars', 
+            'Jupiter', 'Saturn', 'Uranus', 'Neptune', 'Pluto'
+        ]
+        self.vedic_planets = [
+            'Sun', 'Moon', 'Mercury', 'Venus', 'Mars',
+            'Jupiter', 'Saturn', 'Rahu', 'Ketu'
+        ]
+        
+    def calculate_chart(self, birth_info: BirthInfo) -> Dict[str, Any]:
+        """Calculate birth chart based on specified system"""
+        if birth_info.system == 'western':
+            return self._calculate_western_chart(birth_info)
+        else:
+            return self._calculate_vedic_chart(birth_info)
+    
+    def _calculate_western_chart(self, birth_info: BirthInfo) -> Dict[str, Any]:
+        """Calculate Western birth chart"""
+        date = Datetime(birth_info.datetime, birth_info.timezone)
+        pos = GeoPos(birth_info.latitude, birth_info.longitude)
+        
+        chart = Chart(date, pos)
+        
+        positions = {}
+        for planet in self.western_planets:
+            obj = chart.get(planet)
+            positions[planet] = {
+                'sign': obj.sign,
+                'degrees': obj.lon,
+                'house': obj.house,
+                'retrograde': obj.speed < 0
+            }
+        
+        houses = {}
+        for i in range(1, 13):
+            house = chart.houses.get(i)
+            houses[f'House_{i}'] = {
+                'sign': house.sign,
+                'degrees': house.lon
+            }
+        
+        return {
+            'system': 'western',
+            'planets': positions,
+            'houses': houses,
+            'aspects': self._calculate_western_aspects(chart)
+        }
+    
+    def _calculate_vedic_chart(self, birth_info: BirthInfo) -> Dict[str, Any]:
+        """Calculate Vedic birth chart"""
+        # Create Vedic chart using jyotish
+        chart = VedicChart(
+            datetime=birth_info.datetime,
+            latitude=birth_info.latitude,
+            longitude=birth_info.longitude,
+            timezone=birth_info.timezone
+        )
+        
+        positions = {}
+        for planet in self.vedic_planets:
+            graha = chart.get_graha(planet)
+            positions[planet] = {
+                'rashi': graha.rashi,  # Vedic sign
+                'degrees': graha.longitude,
+                'nakshatra': graha.nakshatra,
+                'pada': graha.pada,
+                'house': graha.house
+            }
+        
+        houses = {}
+        for i in range(1, 13):
+            bhava = chart.get_bhava(i)
+            houses[f'House_{i}'] = {
+                'rashi': bhava.rashi,
+                'degrees': bhava.longitude
+            }
+            
+        # Calculate Vedic specific details
+        dasha = chart.get_dasha()  # Current dasha period
+        
+        return {
+            'system': 'vedic',
+            'planets': positions,
+            'houses': houses,
+            'dasha': dasha,
+            'nakshatras': self._calculate_nakshatras(chart)
+        }
+    
+    def _calculate_western_aspects(self, chart: Chart) -> Dict[str, list]:
+        """Calculate major aspects between planets in Western astrology"""
+        aspects = {}
+        major_aspects = {
+            0: 'Conjunction',
+            60: 'Sextile',
+            90: 'Square',
+            120: 'Trine',
+            180: 'Opposition'
+        }
+        
+        for p1 in self.western_planets:
+            aspects[p1] = []
+            planet1 = chart.get(p1)
+            
+            for p2 in self.western_planets:
+                if p1 != p2:
+                    planet2 = chart.get(p2)
+                    angle = abs(planet1.lon - planet2.lon) % 360
+                    
+                    for degrees, aspect_type in major_aspects.items():
+                        if abs(angle - degrees) <= 8:  # 8 degree orb
+                            aspects[p1].append({
+                                'planet': p2,
+                                'type': aspect_type,
+                                'orb': abs(angle - degrees)
+                            })
+        
+        return aspects
+    
+    def _calculate_nakshatras(self, chart: VedicChart) -> Dict[str, Any]:
+        """Calculate Nakshatra positions for Vedic chart"""
+        nakshatra_data = {}
+        for planet in self.vedic_planets:
+            graha = chart.get_graha(planet)
+            nakshatra_data[planet] = {
+                'nakshatra': graha.nakshatra,
+                'pada': graha.pada,
+                'lord': graha.nakshatra_lord
+            }
+        return nakshatra_data