mirror of
https://github.com/pacnpal/thrillwiki_django_no_react.git
synced 2025-12-20 18:31:09 -05:00
Add management command to seed comprehensive sample data for ThrillWiki application
- 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.
This commit is contained in:
pacnpal
317
parks/management/commands/create_sample_data.py
Normal file
317
parks/management/commands/create_sample_data.py
Normal file
@@ -0,0 +1,317 @@
|
||||
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
|
||||
}
|
||||
]
|
||||
Reference in New Issue
Block a user