36 lines
1.2 KiB
Python
36 lines
1.2 KiB
Python
from django.core.management import call_command
|
|
from django.core.management.base import BaseCommand
|
|
|
|
from network.base.models import Antenna
|
|
from network.base.tests import DemodDataFactory, ObservationFactory, \
|
|
StationFactory, generate_payload, generate_payload_name
|
|
|
|
|
|
class Command(BaseCommand):
|
|
help = 'Create initial fixtures'
|
|
|
|
def handle(self, *args, **options):
|
|
# Migrate
|
|
self.stdout.write("Creating database...")
|
|
call_command('migrate')
|
|
|
|
# Initial data
|
|
call_command('loaddata', 'antennas')
|
|
call_command('fetch_data')
|
|
|
|
# Create random fixtures for remaining models
|
|
self.stdout.write("Creating fixtures...")
|
|
StationFactory.create_batch(40,
|
|
antennas=(Antenna.objects.all().values_list('id', flat=True)))
|
|
ObservationFactory.create_batch(200)
|
|
for _ in range(40):
|
|
DemodDataFactory.create(payload_demod__data=generate_payload(),
|
|
payload_demod__filename=generate_payload_name())
|
|
|
|
# Update TLEs
|
|
call_command('update_all_tle')
|
|
|
|
# Create superuser
|
|
self.stdout.write("Creating a superuser...")
|
|
call_command('createsuperuser')
|