mirror of
https://github.com/pacnpal/thrillwiki_django_no_react.git
synced 2025-12-20 13:31:08 -05:00
78248aa892
- Implemented cleanup of existing sample data to avoid conflicts. - Created functions to generate companies, parks, rides, park areas, and reviews. - Ensured proper relationships between models during data creation. - Added logging for better tracking of data seeding process. - Included checks for required database tables before seeding.
318 lines
14 KiB
Python
318 lines
14 KiB
Python
from django.core.management.base import BaseCommand
|
|
from django.utils import timezone
|
|
from django.db import transaction
|
|
from datetime import date, timedelta
|
|
import random
|
|
from decimal import Decimal
|
|
|
|
# Import models from both apps
|
|
from parks.models import Company as ParkCompany, Park, ParkArea, ParkReview
|
|
from parks.models.location import ParkLocation
|
|
from rides.models import Company as RideCompany, Ride, RideModel, RideReview, RollerCoasterStats
|
|
from accounts.models import User
|
|
|
|
|
|
class Command(BaseCommand):
|
|
help = 'Creates comprehensive sample data for the ThrillWiki theme park application'
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
self.created_companies = {}
|
|
self.created_parks = {}
|
|
self.created_rides = {}
|
|
|
|
def handle(self, *args, **options):
|
|
self.stdout.write('Starting sample data creation...')
|
|
|
|
try:
|
|
with transaction.atomic():
|
|
self.create_companies()
|
|
self.create_parks()
|
|
self.create_ride_models()
|
|
self.create_rides()
|
|
self.create_park_areas()
|
|
self.create_reviews()
|
|
|
|
self.stdout.write(self.style.SUCCESS('Successfully created comprehensive sample data!'))
|
|
self.print_summary()
|
|
|
|
except Exception as e:
|
|
self.stdout.write(self.style.ERROR(f'Error creating sample data: {e}'))
|
|
raise
|
|
|
|
def create_companies(self):
|
|
"""Create companies with different roles following entity relationship rules"""
|
|
self.stdout.write('Creating companies...')
|
|
|
|
# Park operators and property owners (using parks.models.Company)
|
|
park_operators_data = [
|
|
{
|
|
'name': 'The Walt Disney Company',
|
|
'slug': 'walt-disney-company',
|
|
'roles': ['OPERATOR', 'PROPERTY_OWNER'],
|
|
'description': 'World\'s largest entertainment company and theme park operator.',
|
|
'website': 'https://www.disney.com/',
|
|
'founded_year': 1923,
|
|
},
|
|
{
|
|
'name': 'Universal Parks & Resorts',
|
|
'slug': 'universal-parks-resorts',
|
|
'roles': ['OPERATOR', 'PROPERTY_OWNER'],
|
|
'description': 'Division of Comcast NBCUniversal, operating major theme parks worldwide.',
|
|
'website': 'https://www.universalparks.com/',
|
|
'founded_year': 1964,
|
|
},
|
|
{
|
|
'name': 'Six Flags Entertainment Corporation',
|
|
'slug': 'six-flags-entertainment',
|
|
'roles': ['OPERATOR', 'PROPERTY_OWNER'],
|
|
'description': 'World\'s largest regional theme park company.',
|
|
'website': 'https://www.sixflags.com/',
|
|
'founded_year': 1961,
|
|
},
|
|
{
|
|
'name': 'Cedar Fair Entertainment Company',
|
|
'slug': 'cedar-fair-entertainment',
|
|
'roles': ['OPERATOR', 'PROPERTY_OWNER'],
|
|
'description': 'One of North America\'s largest operators of regional amusement parks.',
|
|
'website': 'https://www.cedarfair.com/',
|
|
'founded_year': 1983,
|
|
},
|
|
{
|
|
'name': 'Herschend Family Entertainment',
|
|
'slug': 'herschend-family-entertainment',
|
|
'roles': ['OPERATOR', 'PROPERTY_OWNER'],
|
|
'description': 'Largest family-owned themed attractions corporation in the United States.',
|
|
'website': 'https://www.hfecorp.com/',
|
|
'founded_year': 1950,
|
|
},
|
|
{
|
|
'name': 'SeaWorld Parks & Entertainment',
|
|
'slug': 'seaworld-parks-entertainment',
|
|
'roles': ['OPERATOR', 'PROPERTY_OWNER'],
|
|
'description': 'Theme park and entertainment company focusing on nature-based themes.',
|
|
'website': 'https://www.seaworldentertainment.com/',
|
|
'founded_year': 1959,
|
|
},
|
|
{
|
|
'name': 'Merlin Entertainments',
|
|
'slug': 'merlin-entertainments',
|
|
'roles': ['OPERATOR', 'PROPERTY_OWNER'],
|
|
'description': 'European theme park operator with LEGOLAND and Madame Tussauds brands.',
|
|
'website': 'https://www.merlinentertainments.com/',
|
|
'founded_year': 1998,
|
|
},
|
|
]
|
|
|
|
for company_data in park_operators_data:
|
|
company, created = ParkCompany.objects.get_or_create(
|
|
slug=company_data['slug'],
|
|
defaults=company_data
|
|
)
|
|
self.created_companies[company.slug] = company
|
|
self.stdout.write(f' {"Created" if created else "Found"} park company: {company.name}')
|
|
|
|
# Ride manufacturers and designers (using rides.models.Company)
|
|
ride_companies_data = [
|
|
{
|
|
'name': 'Bolliger & Mabillard',
|
|
'slug': 'bolliger-mabillard',
|
|
'roles': ['MANUFACTURER', 'DESIGNER'],
|
|
'description': 'Swiss roller coaster manufacturer known for inverted and diving coasters.',
|
|
'website': 'https://www.bolliger-mabillard.com/',
|
|
'founded_date': '1988-01-01',
|
|
},
|
|
{
|
|
'name': 'Intamin Amusement Rides',
|
|
'slug': 'intamin-amusement-rides',
|
|
'roles': ['MANUFACTURER', 'DESIGNER'],
|
|
'description': 'Liechtenstein-based manufacturer of roller coasters and thrill rides.',
|
|
'website': 'https://www.intamin.com/',
|
|
'founded_date': '1967-01-01',
|
|
},
|
|
{
|
|
'name': 'Arrow Dynamics',
|
|
'slug': 'arrow-dynamics',
|
|
'roles': ['MANUFACTURER', 'DESIGNER'],
|
|
'description': 'American manufacturer known for corkscrew coasters and mine trains.',
|
|
'website': 'https://en.wikipedia.org/wiki/Arrow_Dynamics',
|
|
'founded_date': '1946-01-01',
|
|
},
|
|
{
|
|
'name': 'Vekoma Rides Manufacturing',
|
|
'slug': 'vekoma-rides-manufacturing',
|
|
'roles': ['MANUFACTURER', 'DESIGNER'],
|
|
'description': 'Dutch manufacturer of roller coasters and family rides.',
|
|
'website': 'https://www.vekoma.com/',
|
|
'founded_date': '1926-01-01',
|
|
},
|
|
{
|
|
'name': 'Rocky Mountain Construction',
|
|
'slug': 'rocky-mountain-construction',
|
|
'roles': ['MANUFACTURER', 'DESIGNER'],
|
|
'description': 'American manufacturer specializing in I-Box track and Raptor track coasters.',
|
|
'website': 'https://www.rockymtnconstruction.com/',
|
|
'founded_date': '2001-01-01',
|
|
},
|
|
{
|
|
'name': 'Mack Rides',
|
|
'slug': 'mack-rides',
|
|
'roles': ['MANUFACTURER', 'DESIGNER'],
|
|
'description': 'German manufacturer known for water rides and powered coasters.',
|
|
'website': 'https://www.mack-rides.com/',
|
|
'founded_date': '1780-01-01',
|
|
},
|
|
{
|
|
'name': 'Chance Rides',
|
|
'slug': 'chance-rides',
|
|
'roles': ['MANUFACTURER'],
|
|
'description': 'American manufacturer of thrill rides and amusement park equipment.',
|
|
'website': 'https://www.chancerides.com/',
|
|
'founded_date': '1961-01-01',
|
|
},
|
|
{
|
|
'name': 'S&S Worldwide',
|
|
'slug': 's-s-worldwide',
|
|
'roles': ['MANUFACTURER', 'DESIGNER'],
|
|
'description': 'American manufacturer known for drop towers and 4D free-fly coasters.',
|
|
'website': 'https://www.s-s.com/',
|
|
'founded_date': '1990-01-01',
|
|
},
|
|
{
|
|
'name': 'Zierer Rides',
|
|
'slug': 'zierer-rides',
|
|
'roles': ['MANUFACTURER'],
|
|
'description': 'German manufacturer of kiddie rides and family coasters.',
|
|
'website': 'https://www.zierer.com/',
|
|
'founded_date': '1950-01-01',
|
|
},
|
|
{
|
|
'name': 'Gerstlauer',
|
|
'slug': 'gerstlauer',
|
|
'roles': ['MANUFACTURER', 'DESIGNER'],
|
|
'description': 'German manufacturer known for Euro-Fighter and spinning coasters.',
|
|
'website': 'https://www.gerstlauer-rides.de/',
|
|
'founded_date': '1982-01-01',
|
|
},
|
|
]
|
|
|
|
for company_data in ride_companies_data:
|
|
company, created = RideCompany.objects.get_or_create(
|
|
slug=company_data['slug'],
|
|
defaults=company_data
|
|
)
|
|
self.created_companies[company.slug] = company
|
|
self.stdout.write(f' {"Created" if created else "Found"} ride company: {company.name}')
|
|
|
|
def create_parks(self):
|
|
"""Create parks with proper operator relationships"""
|
|
self.stdout.write('Creating parks...')
|
|
|
|
parks_data = [
|
|
{
|
|
'name': 'Magic Kingdom',
|
|
'slug': 'magic-kingdom',
|
|
'operator_slug': 'walt-disney-company',
|
|
'property_owner_slug': 'walt-disney-company',
|
|
'description': 'The first theme park at Walt Disney World Resort in Florida, opened in 1971.',
|
|
'opening_date': '1971-10-01',
|
|
'size_acres': 142,
|
|
'website': 'https://disneyworld.disney.go.com/destinations/magic-kingdom/',
|
|
'location': {
|
|
'street_address': '1180 Seven Seas Dr',
|
|
'city': 'Lake Buena Vista',
|
|
'state_province': 'Florida',
|
|
'country': 'USA',
|
|
'postal_code': '32830',
|
|
'latitude': 28.4177,
|
|
'longitude': -81.5812
|
|
}
|
|
},
|
|
{
|
|
'name': 'Universal Studios Florida',
|
|
'slug': 'universal-studios-florida',
|
|
'operator_slug': 'universal-parks-resorts',
|
|
'property_owner_slug': 'universal-parks-resorts',
|
|
'description': 'Movie and television-based theme park in Orlando, Florida.',
|
|
'opening_date': '1990-06-07',
|
|
'size_acres': 108,
|
|
'website': 'https://www.universalorlando.com/web/en/us/theme-parks/universal-studios-florida',
|
|
'location': {
|
|
'street_address': '6000 Universal Blvd',
|
|
'city': 'Orlando',
|
|
'state_province': 'Florida',
|
|
'country': 'USA',
|
|
'postal_code': '32819',
|
|
'latitude': 28.4749,
|
|
'longitude': -81.4687
|
|
}
|
|
},
|
|
{
|
|
'name': 'Cedar Point',
|
|
'slug': 'cedar-point',
|
|
'operator_slug': 'cedar-fair-entertainment',
|
|
'property_owner_slug': 'cedar-fair-entertainment',
|
|
'description': 'Known as the "Roller Coaster Capital of the World".',
|
|
'opening_date': '1870-06-01',
|
|
'size_acres': 364,
|
|
'website': 'https://www.cedarpoint.com/',
|
|
'location': {
|
|
'street_address': '1 Cedar Point Dr',
|
|
'city': 'Sandusky',
|
|
'state_province': 'Ohio',
|
|
'country': 'USA',
|
|
'postal_code': '44870',
|
|
'latitude': 41.4822,
|
|
'longitude': -82.6835
|
|
}
|
|
},
|
|
{
|
|
'name': 'Six Flags Magic Mountain',
|
|
'slug': 'six-flags-magic-mountain',
|
|
'operator_slug': 'six-flags-entertainment',
|
|
'property_owner_slug': 'six-flags-entertainment',
|
|
'description': 'Known for its world-record 19 roller coasters.',
|
|
'opening_date': '1971-05-29',
|
|
'size_acres': 262,
|
|
'website': 'https://www.sixflags.com/magicmountain',
|
|
'location': {
|
|
'street_address': '26101 Magic Mountain Pkwy',
|
|
'city': 'Valencia',
|
|
'state_province': 'California',
|
|
'country': 'USA',
|
|
'postal_code': '91355',
|
|
'latitude': 34.4253,
|
|
'longitude': -118.5971
|
|
}
|
|
},
|
|
{
|
|
'name': 'Europa-Park',
|
|
'slug': 'europa-park',
|
|
'operator_slug': 'merlin-entertainments',
|
|
'property_owner_slug': 'merlin-entertainments',
|
|
'description': 'One of the most popular theme parks in Europe, located in Germany.',
|
|
'opening_date': '1975-07-12',
|
|
'size_acres': 234,
|
|
'website': 'https://www.europapark.de/',
|
|
'location': {
|
|
'street_address': 'Europa-Park-Straße 2',
|
|
'city': 'Rust',
|
|
'state_province': 'Baden-Württemberg',
|
|
'country': 'Germany',
|
|
'postal_code': '77977',
|
|
'latitude': 48.2667,
|
|
'longitude': 7.7167
|
|
}
|
|
},
|
|
{
|
|
'name': 'Alton Towers',
|
|
'slug': 'alton-towers',
|
|
'operator_slug': 'merlin-entertainments',
|
|
'property_owner_slug': 'merlin-entertainments',
|
|
'description': 'Major theme park and former country estate in Staffordshire, England.',
|
|
'opening_date': '1980-04-23',
|
|
'size_acres': 500,
|
|
# Add other fields as needed
|
|
}
|
|
]
|