seed_data.py

import os
import django
os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'geo_bucket_system.settings')
django.setup()
from property_search.models import Property, GeoBucket
from property_search.services import LocationService
from django.contrib.gis.geos import Point


"""
Seed script for ExpertListing Geo-Bucket System
Loads the required test data.
"""

def clear_database():
    """Clear existing data."""
    print("Clearing existing data...")
    Property.objects.all().delete()
    GeoBucket.objects.all().delete()
    print("✓ Database cleared")


def seed_required_test_cases():

    print("\n📋 Creating required test data...")
    
    test_data = [
        {
            'title': '3 Bedroom Flat - Sangotedo',
            'location_name': 'Sangotedo',
            'lat': 6.4698,
            'lng': 3.6285,
            'price': 45000000,
            'bedrooms': 3,
            'bathrooms': 2
        },
        {
            'title': '4 Bedroom Duplex - Sangotedo, Ajah',
            'location_name': 'Sangotedo, Ajah',
            'lat': 6.4720,
            'lng': 3.6301,
            'price': 65000000,
            'bedrooms': 4,
            'bathrooms': 3
        },
        {
            'title': '2 Bedroom Apartment - sangotedo lagos',
            'location_name': 'sangotedo lagos',
            'lat': 6.4705,
            'lng': 3.6290,
            'price': 38000000,
            'bedrooms': 2,
            'bathrooms': 2
        },
         # Different location - Lekki
        {
            'title': 'Luxury Villa - Lekki Phase 1',
            'location_name': 'Lekki Phase 1',
            'lat': 6.4474,
            'lng': 3.4738,
            'price': 150000000,
            'bedrooms': 5,
            'bathrooms': 5
        },
        {
            'title': 'Penthouse - lekki',
            'location_name': 'lekki',
            'lat': 6.4480,
            'lng': 3.4745,
            'price': 200000000,
            'bedrooms': 4,
            'bathrooms': 4
        },
        # Different location - Victoria Island
        {
            'title': 'Modern Apartment - Victoria Island',
            'location_name': 'Victoria Island',
            'lat': 6.4281,
            'lng': 3.4219,
            'price': 180000000,
            'bedrooms': 3,
            'bathrooms': 3
        },
        {
            'title': 'Office Space - VI',
            'location_name': 'VI',
            'lat': 6.4285,
            'lng': 3.4225,
            'price': 95000000,
            'bedrooms': 2,
            'bathrooms': 2
        },
        # Ikoyi
        {
            'title': 'Waterfront Property - Ikoyi',
            'location_name': 'Ikoyi',
            'lat': 6.4541,
            'lng': 3.4316,
            'price': 250000000,
            'bedrooms': 6,
            'bathrooms': 6
        },
        # Ajah (close to Sangotedo)
        {
            'title': 'Family Home - Ajah',
            'location_name': 'Ajah',
            'lat': 6.4655,
            'lng': 3.5737,
            'price': 42000000,
            'bedrooms': 3,
            'bathrooms': 2
        },
    ]
    
    created_properties = []
    
    for idx, data in enumerate(test_data, 1):
        lat = data.pop('lat')
        lng = data.pop('lng')
        location_name = data['location_name']
        
        # Find or create bucket
        bucket = LocationService.find_or_create_bucket(lat, lng, location_name)
        
        # Create property
        prop = Property.objects.create(
            location_point=Point(lng, lat, srid=4326),
            geo_bucket=bucket,
            **data
        )
        
        created_properties.append(prop)
        print(f"  ✓ Property {idx}: '{prop.title}' → Bucket '{bucket.canonical_name}' (ID: {bucket.id})")
    
    return created_properties



def main():
    """Main seed function."""
    
    # Clear existing data
    clear_database()
    
    # Seed required test cases
    seed_required_test_cases()
    

    
    print("\n✅ Seeding completed successfully!")
    


if __name__ == '__main__':
    main()