Merge branch 'pr/remove_fetch_tle' into 'master'
Remove fetch_tle management command Closes #568 See merge request librespacefoundation/satnogs/satnogs-network!818merge-requests/818/merge
commit
0e44bb8389
|
@ -1,48 +0,0 @@
|
|||
"""SatNOGS Network django management command to fetch TLEs"""
|
||||
from django.core.management.base import BaseCommand, CommandError
|
||||
from satellite_tle import fetch_tle_from_celestrak
|
||||
|
||||
from network.base.models import Satellite
|
||||
|
||||
|
||||
class Command(BaseCommand):
|
||||
"""Django management command to update/insert TLEs for certain Satellites"""
|
||||
help = 'Updates/Inserts TLEs for certain Satellites'
|
||||
|
||||
def add_arguments(self, parser):
|
||||
parser.add_argument('norad_cat_id', nargs='+', type=int)
|
||||
parser.add_argument(
|
||||
'--delete',
|
||||
action='store_true',
|
||||
dest='delete',
|
||||
default=False,
|
||||
help='Delete Satellite',
|
||||
)
|
||||
|
||||
def handle(self, *args, **options):
|
||||
for item in options['norad_cat_id']:
|
||||
if options['delete']:
|
||||
try:
|
||||
Satellite.objects.get(norad_cat_id=item).delete()
|
||||
self.stdout.write('Satellite {}: deleted'.format(item))
|
||||
continue
|
||||
except Satellite.DoesNotExist:
|
||||
raise CommandError('Satellite with Identifier {} does not exist'.format(item))
|
||||
|
||||
try:
|
||||
tle = fetch_tle_from_celestrak(item)
|
||||
except LookupError:
|
||||
raise CommandError('Satellite with Identifier {} does not exist'.format(item))
|
||||
|
||||
try:
|
||||
obj = Satellite.objects.get(norad_cat_id=item)
|
||||
except Satellite.DoesNotExist:
|
||||
obj = Satellite(norad_cat_id=item)
|
||||
|
||||
obj.name = tle[0]
|
||||
obj.tle0 = tle[0]
|
||||
obj.tle1 = tle[1]
|
||||
obj.tle2 = tle[2]
|
||||
obj.save()
|
||||
|
||||
self.stdout.write('fetched data for {}: {}'.format(obj.norad_cat_id, obj.name))
|
Loading…
Reference in New Issue